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

⚡️ refactor: 小幅提升性能

Browse Source
This commit is contained in:
naiba
2021-01-17 22:05:59 +08:00
parent 67df0852f9
commit d059835877
5 changed files with 27 additions and 22 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
service/rpc/nezha.go
View File

@@ -69,10 +69,10 @@ func (s *NezhaHandler) RequestTask(h *pb.Host, stream pb.NezhaService_RequestTas
return err
}
closeCh := make(chan error)
dao.ServerLock.Lock()
dao.ServerLock.RLock()
dao.ServerList[clientID].TaskStream = stream
dao.ServerList[clientID].TaskClose = closeCh
dao.ServerLock.Unlock()
dao.ServerLock.RUnlock()
select {
case err = <-closeCh:
return err
@@ -88,7 +88,8 @@ func (s *NezhaHandler) ReportSystemState(c context.Context, r *pb.State) (*pb.Re
state := model.PB2State(r)
dao.ServerLock.RLock()
defer dao.ServerLock.RUnlock()
dao.ServerList[clientID].LastActive = time.Now()
now := time.Now()
dao.ServerList[clientID].LastActive = &now
dao.ServerList[clientID].State = &state
return &pb.Receipt{Proced: true}, nil
}