OpenList/utils/check.go

package utils

import (
	log "github.com/sirupsen/logrus"
	"net/url"
	"strconv"
	"strings"
)

// get code from url
func GetCode(rawUrl string) string {
	u, err := url.Parse(rawUrl)
	if err != nil {
		log.Errorf("解析url出错:%s", err.Error())
		return ""
	}
	code := u.Query().Get("code")
	return code
}

// determine whether to include
func ContainsString(array []string, val string) (index int) {
	index = -1
	for i := 0; i < len(array); i++ {
		if array[i] == val {
			index = i
			return
		}
	}
	return
}

// compare version
func VersionCompare(version1, version2 string) int {
	a := strings.Split(version1, ".")
	b := strings.Split(version2, ".")
	flag := 1
	if len(a) > len(b) {
		a, b = b, a
		flag = -1
	}
	for i := range a {
		x, _ := strconv.Atoi(a[i])
		y, _ := strconv.Atoi(b[i])
		if x < y {
			return -1 * flag
		} else if x > y {
			return 1 * flag
		}
	}
	for _, v := range b[len(a):] {
		y, _ := strconv.Atoi(v)
		if y > 0 {
			return -1 * flag
		}
	}
	return 0
}