feat: 添加任务执行功能

fix: 版本解析逻辑
This commit is contained in:
Akizon77
2025-06-04 17:20:39 +08:00
parent ccb8c06dac
commit b247a6ebee
4 changed files with 179 additions and 4 deletions

View File

@@ -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
View 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)
}
}
}