diff --git a/internal/controllers/daemon.go b/internal/controllers/daemon.go index 747c853..7a0f3a8 100644 --- a/internal/controllers/daemon.go +++ b/internal/controllers/daemon.go @@ -3,7 +3,6 @@ package controllers import ( "fmt" "github.com/Mmx233/BitSrunLoginGo/internal/global" - "github.com/Mmx233/tool" "github.com/howeyc/fsnotify" "os" "time" @@ -22,12 +21,12 @@ var Daemon = daemon{ // MarkDaemon 写入后台标记文件 func (a *daemon) MarkDaemon() error { - return tool.File.WriteAll(a.Path, []byte(a.Mark)) + return os.WriteFile(a.Path, []byte(a.Mark), 0600) } // CheckDaemon 检查后台标记文件 func (a *daemon) CheckDaemon() bool { - if data, err := tool.File.ReadAll(a.Path); err != nil { + if data, err := os.ReadFile(a.Path); err != nil { return false } else { return string(data) == a.Mark diff --git a/internal/pkg/dns/cloudflare/cloudflare.go b/internal/pkg/dns/cloudflare/cloudflare.go index c134cdd..4d0d408 100644 --- a/internal/pkg/dns/cloudflare/cloudflare.go +++ b/internal/pkg/dns/cloudflare/cloudflare.go @@ -60,9 +60,10 @@ func (a DnsProvider) SetDomainRecord(domain, ip string) error { if record.Content == ip { return nil } - return a.Api.UpdateDNSRecord(context.Background(), a.ZoneResource, cloudflare.UpdateDNSRecordParams{ + _, e = a.Api.UpdateDNSRecord(context.Background(), a.ZoneResource, cloudflare.UpdateDNSRecordParams{ ID: record.ID, Content: ip, }) + return e } }