golang实现PHP的version_compare函数

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[i] == ver2[i] {
            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[i] > ver2[i] {
            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 == 2 {
            return true
        }
    case ">":
        if com == 1 {
            return true
        }
    case "<=":
        if com == 0 || com == 2 {
            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<b.N ;i++  {
        VersionCompare("2.3.1","2.1.3.4",">=")
    }
}

func Benchmark_GoVersion(b *testing.B) {
    for i:=0;i<b.N ;i++  {
        version.Compare("2.3.1","2.1.3.4",">=")
    }
}

// 该函数比较两个版本号是否相等，是否大于或小于的关系
// 返回值：0表示v1与v2相等；1表示v1大于v2；2表示v1小于v2
func TestCompare(t *testing.T) {
    if 0 != Compare(v1,v1) {
        t.Errorf("v1 %s == v1 %s ",v1,v1)
    }

    if 1 != Compare(v2,v1) {
        t.Errorf("v2 %s > v1 %s",v2,v1)
    }

    if 2 != Compare(v1,v2) {
        t.Errorf("v1 %s < v2 %s",v1,v2)
    }

}

func TestVersionCompare(t *testing.T)  {
    if !VersionCompare(v1,v1_1,"<") {
        t.Errorf("v1 %s < v1_1 %s ",v1,v1_1)
    }

    if !VersionCompare(v2,v1_1,">") {
        t.Errorf("v2 %s < v1_1 %s ",v2,v1_1)
    }

    if !VersionCompare(v2,v1,">=") {
        t.Errorf("v2 %s >= v1 %s ",v2,v1)
    }

    if !VersionCompare(v1,v2,"<=") {
        t.Errorf("v1 %s >= v2 %s ",v1,v2)
    }

    if !VersionCompare(v1,v1,"==") {
        t.Errorf("v1 %s == v1 %s ",v1,v1)
    }
    if !VersionCompare(v3,v4,">") {
        t.Errorf("v3 %s > v4 %s ",v3,v4)
    }
    if !VersionCompare(v3,v5,">") {
        t.Errorf("v3 %s > v5 %s ",v3,v5)
    }

    /*if !VersionCompare(v3,v5,"<=") {
        t.Errorf("v3 %s > v5 %s ",v3,v5)
    }*/

    if !VersionCompare(v4,v5,"==") {
        t.Errorf("v4 %s == v5 %s ",v4,v5)
    }

}

该代码只是实现其中的『<,>,<=,>=,==』。