mirror of
https://github.com/emirpasic/gods.git
synced 2026-06-14 16:06:42 +03:00
Paul Chesnais
14f714261f
* Generics migration This attempts to migrate this library in the least invasive way by preserving as much of the original API as possible. It does not change the tests in a meaningful way nor does it attempt to upgrade any logic that can be simplified or improved with generics. This is purely an API migration, and still requires a lot of additional work to be fully ready. * Fix a few broken tests around serialization * Add v2 suffix * Temporarily change mod name for testing * Rename module to /v2
43 lines
774 B
Go
43 lines
774 B
Go
// Copyright (c) 2015, Emir Pasic. All rights reserved.
|
|
// Use of this source code is governed by a BSD-style
|
|
// license that can be found in the LICENSE file.
|
|
|
|
package main
|
|
|
|
import (
|
|
"fmt"
|
|
|
|
"github.com/emirpasic/gods/v2/sets/treeset"
|
|
)
|
|
|
|
// User model (id and name)
|
|
type User struct {
|
|
id int
|
|
name string
|
|
}
|
|
|
|
// Comparator function (sort by IDs)
|
|
func byID(a, b User) int {
|
|
|
|
switch {
|
|
case a.id > b.id:
|
|
return 1
|
|
case a.id < b.id:
|
|
return -1
|
|
default:
|
|
return 0
|
|
}
|
|
}
|
|
|
|
// CustomComparatorExample to demonstrate basic usage of CustomComparator
|
|
func main() {
|
|
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}
|
|
}
|