Buriburizaemon/nezha_domains
1
0
Fork 0
mirror of https://github.com/Buriburizaem0n/nezha_domains.git synced 2026-02-07 14:09:35 +00:00
Code Issues Packages Projects Releases Wiki Activity

♻️ 优化取网卡流量及硬盘容量

Browse Source 
Co-authored-by: CHN-STUDENT <21209416+CHN-STUDENT@users.noreply.github.com>
This commit is contained in:
naiba
2021-04-25 23:46:15 +08:00
parent b18227d64b
commit aeb48e50ac
2 changed files with 69 additions and 26 deletions
Download Patch File Download Diff File Expand all files Collapse all files

34
cmd/agent/main.go
View File

@@ -55,17 +55,17 @@ func doSelfUpdate() {
updateCh <- struct{}{}
}()
v := semver.MustParse(version)
log.Println("Check update", v)
println("Check update", v)
latest, err := selfupdate.UpdateSelf(v, "naiba/nezha")
if err != nil {
log.Println("Binary update failed:", err)
println("Binary update failed:", err)
return
}
if latest.Version.Equals(v) {
// latest version is the same as current version. It means current binary is up to date.
log.Println("Current binary is the latest version", version)
println("Current binary is the latest version", version)
} else {
log.Println("Successfully updated to version", latest.Version)
println("Successfully updated to version", latest.Version)
os.Exit(1)
}
}
@@ -120,18 +120,18 @@ func run() {
var conn *grpc.ClientConn
retry := func() {
log.Println("Error to close connection ...")
println("Error to close connection ...")
if conn != nil {
conn.Close()
}
time.Sleep(delayWhenError)
log.Println("Try to reconnect ...")
println("Try to reconnect ...")
}
for {
conn, err = grpc.Dial(server, grpc.WithInsecure(), grpc.WithPerRPCCredentials(&auth))
if err != nil {
log.Printf("grpc.Dial err: %v", err)
println("grpc.Dial err: ", err)
retry()
continue
}
@@ -139,26 +139,26 @@ func run() {
// 第一步注册
_, err = client.ReportSystemInfo(ctx, monitor.GetHost().PB())
if err != nil {
log.Printf("client.ReportSystemInfo err: %v", err)
println("client.ReportSystemInfo err: ", err)
retry()
continue
}
// 执行 Task
tasks, err := client.RequestTask(ctx, monitor.GetHost().PB())
if err != nil {
log.Printf("client.RequestTask err: %v", err)
println("client.RequestTask err: ", err)
retry()
continue
}
err = receiveTasks(tasks)
log.Printf("receiveTasks exit to main: %v", err)
println("receiveTasks exit to main: ", err)
retry()
}
}
func receiveTasks(tasks pb.NezhaService_RequestTaskClient) error {
var err error
defer log.Printf("receiveTasks exit %v => %v", time.Now(), err)
defer println("receiveTasks exit", time.Now(), "=>", err)
for {
var task *pb.Task
task, err = tasks.Recv()
@@ -264,7 +264,7 @@ func doTask(task *pb.Task) {
}
result.Delay = float32(time.Since(startedAt).Seconds())
default:
log.Printf("Unknown action: %v", task)
println("Unknown action: ", task)
}
client.ReportTask(ctx, &result)
}
@@ -272,13 +272,13 @@ func doTask(task *pb.Task) {
func reportState() {
var lastReportHostInfo time.Time
var err error
defer log.Printf("reportState exit %v => %v", time.Now(), err)
defer println("reportState exit", time.Now(), "=>", err)
for {
if client != nil {
monitor.TrackNetworkSpeed()
_, err = client.ReportSystemState(ctx, monitor.GetState(dao.ReportDelay).PB())
if err != nil {
log.Printf("reportState error %v", err)
println("reportState error", err)
time.Sleep(delayWhenError)
}
if lastReportHostInfo.Before(time.Now().Add(-10 * time.Minute)) {
@@ -288,3 +288,9 @@ func reportState() {
}
}
}
func println(v ...interface{}) {
if dao.Conf.Debug {
log.Println(v...)
}
}