mirror of
https://github.com/emirpasic/gods
synced 2024-11-08 13:10:26 +00:00
66 lines
1.9 KiB
Go
66 lines
1.9 KiB
Go
/*
|
|
Copyright (c) 2015, Emir Pasic
|
|
All rights reserved.
|
|
|
|
Redistribution and use in source and binary forms, with or without
|
|
modification, are permitted provided that the following conditions are met:
|
|
|
|
* Redistributions of source code must retain the above copyright notice, this
|
|
list of conditions and the following disclaimer.
|
|
|
|
* Redistributions in binary form must reproduce the above copyright notice,
|
|
this list of conditions and the following disclaimer in the documentation
|
|
and/or other materials provided with the distribution.
|
|
|
|
THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
|
|
AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
|
|
IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
|
|
DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE LIABLE
|
|
FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
|
|
DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR
|
|
SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER
|
|
CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY,
|
|
OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
|
|
OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
|
|
*/
|
|
|
|
package examples
|
|
|
|
import (
|
|
"fmt"
|
|
"github.com/emirpasic/gods/sets/treeset"
|
|
)
|
|
|
|
type User struct {
|
|
id int
|
|
name string
|
|
}
|
|
|
|
// Comparator function (sort by IDs)
|
|
func byID(a, b interface{}) int {
|
|
|
|
// Type assertion, program will panic if this is not respected
|
|
c1 := a.(User)
|
|
c2 := b.(User)
|
|
|
|
switch {
|
|
case c1.id > c2.id:
|
|
return 1
|
|
case c1.id < c2.id:
|
|
return -1
|
|
default:
|
|
return 0
|
|
}
|
|
}
|
|
|
|
func CustomComparatorExample() {
|
|
set := treeset.NewWith(byID)
|
|
|
|
set.Add(User{2, "Second"})
|
|
set.Add(User{3, "Third"})
|
|
set.Add(User{1, "First"})
|
|
set.Add(User{4, "Fourth"})
|
|
|
|
fmt.Println(set) // {1 First}, {2 Second}, {3 Third}, {4 Fourth}
|
|
}
|