fumiama/go-registry
1
0
Fork 0
mirror of https://github.com/fumiama/go-registry.git synced 2026-06-27 14:20:26 +08:00
Code Activity

fix: time sequence

Browse Source
This commit is contained in:
源文雨
2022-04-17 21:13:50 +08:00
parent 2c62a8ecf3
commit f93d4d7437
2 changed files with 2 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
reg.go
View File

@@ -129,18 +129,17 @@ func (r *Regedit) Set(key, value string) error {
p := NewCmdPacket(CMDSET, StringToBytes(key), r.ts) p := NewCmdPacket(CMDSET, StringToBytes(key), r.ts)
defer p.Put() defer p.Put()
r.Lock() r.Lock()
defer r.Unlock()
r.conn.Write(p.Encrypt(r.seq)) r.conn.Write(p.Encrypt(r.seq))
r.seq++ r.seq++
ack := NewCmdPacket(CMDACK, nil, &r.tp) ack := NewCmdPacket(CMDACK, nil, &r.tp)
defer ack.Put() defer ack.Put()
err := r.ack(ack) err := r.ack(ack)
if err != nil { if err != nil {
r.Unlock()
return err return err
} }
err = ack.Decrypt(r.seq) err = ack.Decrypt(r.seq)
r.seq++ r.seq++
r.Unlock()
if err != nil { if err != nil {
return ErrDecAck return ErrDecAck
} }
@@ -152,17 +151,14 @@ func (r *Regedit) Set(key, value string) error {
return ErrUnknownAck return ErrUnknownAck
} }
p.Refresh(CMDDAT, StringToBytes(value), r.ts) p.Refresh(CMDDAT, StringToBytes(value), r.ts)
r.Lock()
r.conn.Write(p.Encrypt(r.seq)) r.conn.Write(p.Encrypt(r.seq))
r.seq++ r.seq++
err = r.ack(ack) err = r.ack(ack)
if err != nil { if err != nil {
r.Unlock()
return err return err
} }
err = ack.Decrypt(r.seq) err = ack.Decrypt(r.seq)
r.seq++ r.seq++
r.Unlock()
if err != nil { if err != nil {
return ErrDecAck return ErrDecAck
} }

2
reg_test.go
View File

@@ -45,7 +45,7 @@ func TestPush(t *testing.T) {
if err != nil { if err != nil {
t.Fatal(err) t.Fatal(err)
} }
err = r.Set("ZeroBot-Plugin/kanban", "QQ群:1048452984, 开发群:705749886. 禁止用于商业用途.\nv1.2.2及以下版本公告服已下线.") err = r.Set("ZeroBot-Plugin/kanban", "由于tx更改登录鉴权, 近期gocqzbp闪退增多,\n请及时处理以免对md5服务器产生过量请求, 导致IP被封.")
if err != nil { if err != nil {
t.Fatal(err) t.Fatal(err)
} }