mirror of https://github.com/emirpasic/gods
You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
75 lines
2.2 KiB
Go
75 lines
2.2 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 utils
|
|
|
|
// Comparator will make type assertion (see IntComparator for example),
|
|
// which will panic if a or b are not of the asserted type.
|
|
//
|
|
// Should return a number:
|
|
// negative , if a < b
|
|
// zero , if a == b
|
|
// positive , if a > b
|
|
type Comparator func(a, b interface{}) int
|
|
|
|
// IntComparator provides a basic comparison on ints
|
|
func IntComparator(a, b interface{}) int {
|
|
aInt := a.(int)
|
|
bInt := b.(int)
|
|
switch {
|
|
case aInt > bInt:
|
|
return 1
|
|
case aInt < bInt:
|
|
return -1
|
|
default:
|
|
return 0
|
|
}
|
|
}
|
|
|
|
// StringComparator provides a fast comparison on strings
|
|
func StringComparator(a, b interface{}) int {
|
|
s1 := a.(string)
|
|
s2 := b.(string)
|
|
min := len(s2)
|
|
if len(s1) < len(s2) {
|
|
min = len(s1)
|
|
}
|
|
diff := 0
|
|
for i := 0; i < min && diff == 0; i++ {
|
|
diff = int(s1[i]) - int(s2[i])
|
|
}
|
|
if diff == 0 {
|
|
diff = len(s1) - len(s2)
|
|
}
|
|
if diff < 0 {
|
|
return -1
|
|
}
|
|
if diff > 0 {
|
|
return 1
|
|
}
|
|
return 0
|
|
}
|