mirror of
https://github.com/fankes/komari-agent.git
synced 2025-10-18 18:49:23 +08:00
feat: 添加任务执行功能
fix: 版本解析逻辑
This commit is contained in:
@@ -4,6 +4,7 @@ import (
|
||||
"fmt"
|
||||
"log"
|
||||
"os"
|
||||
"strings"
|
||||
"time"
|
||||
|
||||
"github.com/blang/semver"
|
||||
@@ -15,6 +16,19 @@ var (
|
||||
Repo string = "komari-monitor/komari-agent"
|
||||
)
|
||||
|
||||
// parseVersion 解析可能带有 v/V 前缀,以及预发布或构建元数据的版本字符串
|
||||
func parseVersion(ver string) (semver.Version, error) {
|
||||
ver = strings.TrimPrefix(ver, "v")
|
||||
ver = strings.TrimPrefix(ver, "V")
|
||||
return semver.ParseTolerant(ver)
|
||||
}
|
||||
|
||||
// needUpdate 判断是否需要更新
|
||||
func needUpdate(current, latest semver.Version) bool {
|
||||
// 返回最新版本大于当前版本时需要更新
|
||||
return latest.Compare(current) > 0
|
||||
}
|
||||
|
||||
func DoUpdateWorks() {
|
||||
ticker_ := time.NewTicker(time.Duration(6) * time.Hour)
|
||||
for range ticker_.C {
|
||||
@@ -26,7 +40,7 @@ func DoUpdateWorks() {
|
||||
func CheckAndUpdate() error {
|
||||
log.Println("Checking update...")
|
||||
// Parse current version
|
||||
currentSemVer, err := semver.Parse(CurrentVersion)
|
||||
currentSemVer, err := parseVersion(CurrentVersion)
|
||||
if err != nil {
|
||||
return fmt.Errorf("failed to parse current version: %v", err)
|
||||
}
|
||||
|
72
update/update_test.go
Normal file
72
update/update_test.go
Normal file
@@ -0,0 +1,72 @@
|
||||
package update
|
||||
|
||||
import (
|
||||
"strings"
|
||||
"testing"
|
||||
)
|
||||
|
||||
// TestParseVersion 验证 parseVersion 能够解析各种版本号格式,包括带 v/V 前缀、预发布和构建元数据
|
||||
func TestParseVersion(t *testing.T) {
|
||||
tests := []struct {
|
||||
input string
|
||||
want string
|
||||
}{
|
||||
{"v1.2.3", "1.2.3"},
|
||||
{"V1.2.3", "1.2.3"},
|
||||
{"1.2.3-beta.1", "1.2.3-beta.1"},
|
||||
{"v1.2.3+meta", "1.2.3+meta"},
|
||||
{"1.2.3-pre.1+build.123", "1.2.3-pre.1+build.123"},
|
||||
{" v2.0.0 ", "2.0.0"},
|
||||
{"invalid", ""},
|
||||
}
|
||||
|
||||
for _, tt := range tests {
|
||||
got, err := parseVersion(strings.TrimSpace(tt.input))
|
||||
if tt.want == "" {
|
||||
if err == nil {
|
||||
t.Errorf("parseVersion(%q) expected error, got %v", tt.input, got)
|
||||
}
|
||||
} else {
|
||||
if err != nil {
|
||||
t.Errorf("parseVersion(%q) unexpected error: %v", tt.input, err)
|
||||
continue
|
||||
}
|
||||
if got.String() != tt.want {
|
||||
t.Errorf("parseVersion(%q) = %q, want %q", tt.input, got.String(), tt.want)
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
// TestNeedUpdate 验证 needUpdate 在不同版本组合下的判断
|
||||
func TestNeedUpdate(t *testing.T) {
|
||||
tests := []struct {
|
||||
current string
|
||||
latest string
|
||||
want bool
|
||||
}{
|
||||
{"1.0.0", "1.0.1", true},
|
||||
{"v1.0.0", "1.1.0", true},
|
||||
{"1.2.3", "1.2.3", false},
|
||||
{"1.2.4", "1.2.3", false},
|
||||
{"1.2.3-beta", "1.2.3", true},
|
||||
{"1.2.3", "1.2.3-beta", false},
|
||||
{"0.0.5", "0.0.6+build.1", true},
|
||||
{"0.0.6", "v0.0.6+build.1", false},
|
||||
}
|
||||
|
||||
for _, tt := range tests {
|
||||
cur, err := parseVersion(strings.TrimSpace(tt.current))
|
||||
if err != nil {
|
||||
t.Fatalf("parseVersion(%q) error: %v", tt.current, err)
|
||||
}
|
||||
lat, err := parseVersion(strings.TrimSpace(tt.latest))
|
||||
if err != nil {
|
||||
t.Fatalf("parseVersion(%q) error: %v", tt.latest, err)
|
||||
}
|
||||
got := needUpdate(cur, lat)
|
||||
if got != tt.want {
|
||||
t.Errorf("needUpdate(%q, %q) = %v, want %v", tt.current, tt.latest, got, tt.want)
|
||||
}
|
||||
}
|
||||
}
|
Reference in New Issue
Block a user