chinhwajie/BitSrunLoginGo
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix: 两处兼容错误

Browse Source
This commit is contained in:
Mmx233
2023-04-07 10:52:50 +08:00
parent fb3192e555
commit 16a5da155c
2 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
internal/controllers/daemon.go
View File

@@ -3,7 +3,6 @@ package controllers
import ( import (
"fmt" "fmt"
"github.com/Mmx233/BitSrunLoginGo/internal/global" "github.com/Mmx233/BitSrunLoginGo/internal/global"
"github.com/Mmx233/tool"
"github.com/howeyc/fsnotify" "github.com/howeyc/fsnotify"
"os" "os"
"time" "time"
@@ -22,12 +21,12 @@ var Daemon = daemon{
// MarkDaemon 写入后台标记文件 // MarkDaemon 写入后台标记文件
func (a *daemon) MarkDaemon() error { func (a *daemon) MarkDaemon() error {
return tool.File.WriteAll(a.Path, []byte(a.Mark)) return os.WriteFile(a.Path, []byte(a.Mark), 0600)
} }
// CheckDaemon 检查后台标记文件 // CheckDaemon 检查后台标记文件
func (a *daemon) CheckDaemon() bool { 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 return false
} else { } else {
return string(data) == a.Mark return string(data) == a.Mark

3
internal/pkg/dns/cloudflare/cloudflare.go
View File

@@ -60,9 +60,10 @@ func (a DnsProvider) SetDomainRecord(domain, ip string) error {
if record.Content == ip { if record.Content == ip {
return nil 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, ID: record.ID,
Content: ip, Content: ip,
}) })
return e
} }
} }