golang实现PHP的version_compare函数

alonli

　　PHP的version_compare是对比两个「PHP 规范化」的版本数字字符串。

mixed version_compare ( string $version1 , string $version2 [, string $operator ] )
　　使用golang实现其中一部分常用的功能。

package util
import (
"strings"
"strconv"
)
// 该函数比较两个版本号是否相等，是否大于或小于的关系
// 返回值：0表示v1与v2相等；1表示v1大于v2；2表示v1小于v2
func Compare(v1, v2 string) int {
// 替换一些常见的版本符号
replaceMap := map[string]string{"V":"","v": "", "-": ".",}
//keywords := {"alpha,beta,rc,p"}
for k, v := range replaceMap {
if strings.Contains(v1, k) {
strings.Replace(v1, k, v, -1)
}
if strings.Contains(v2, k) {
strings.Replace(v2, k, v, -1)
}
}
verStr1 := strings.Split(v1, ".")
verStr2 := strings.Split(v2, ".")
ver1 := strSlice2IntSlice(verStr1)
ver2 := strSlice2IntSlice(verStr2)
// 找出v1和v2哪一个最短
var shorter int
if len(ver1) > len(ver2) {
shorter = len(ver2)
} else {
shorter = len(ver1)
}
// 循环比较
for i := 0; i < shorter; i++ {
if ver1 == ver2 {
if shorter-1 == i {
if len(ver1) == len(ver2) {
return 0
} else {
if len(ver1) > len(ver2) {
return 1
} else {
return 2
}
}
}
} else if ver1 > ver2 {
return 1
} else {
return 2
}
}
return -1
}
func strSlice2IntSlice(strs []string) []int64 {
if len(strs) == 0 {
return []int64{}
}
retInt := make([]int64,0,len(strs))
for _,str:=range strs {
i,err := strconv.ParseInt(str,10,64)
if err == nil {
retInt = append(retInt,i)
}
}
return  retInt
}
func VersionCompare(v1, v2 , operator string) bool {
com := Compare(v1,v2)
switch operator {
case "==":
if com == 0 {
return true
}
case "":
if com == 1 {
return true
}
case "=":
if com == 0 || com == 1{
return true
}
}
return false
}
　　测试单元：

package version
import (
"testing"
version "github.com/mcuadros/go-version"
)
var (
v1 = "2.0.1"
v1_1 = "2.0.1.1"
v2 = "2.1.1"
v3 = "2.10.22.1"
v4 = "2.09.22.2"
v5 = "2.9.22.2"
)
// go test -test.bench=".*"
func Benchmark_MyVersion(b *testing.B) {
for i:=0;i