修复待升级节点版本号对比错误

This commit is contained in:
GoEdgeLab
2021-01-18 12:35:39 +08:00
parent 6978d3d945
commit f59d54faae
3 changed files with 39 additions and 0 deletions

19
internal/utils/ip.go Normal file
View File

@@ -0,0 +1,19 @@
package utils
import (
"encoding/binary"
"net"
)
// 将IP转换为整型
func IP2Long(ip string) uint32 {
s := net.ParseIP(ip)
if s == nil {
return 0
}
if len(s) == 16 {
return binary.BigEndian.Uint32(s[12:16])
}
return binary.BigEndian.Uint32(s)
}

18
internal/utils/version.go Normal file
View File

@@ -0,0 +1,18 @@
package utils
import (
"strings"
)
// 计算版本代号
func VersionToLong(version string) uint32 {
countDots := strings.Count(version, ".")
if countDots == 2 {
version += ".0"
} else if countDots == 1 {
version += ".0.0"
} else if countDots == 0 {
version += ".0.0.0"
}
return IP2Long(version)
}