shkong/go-cqhttp
1
0
Fork 0
mirror of https://github.com/Mrs4s/go-cqhttp.git synced 2025-05-05 03:23:49 +08:00
Code Issues Projects Releases Wiki Activity

fix #298.

Browse Source
This commit is contained in:
Mrs4s 2020-09-20 00:29:50 +08:00
parent b5b6117487
commit c31169b99b
1 changed files with 27 additions and 13 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
main.go
View File

@ -239,7 +239,14 @@ func main() {
cli.OnServerUpdated(func(bot *client.QQClient, e *client.ServerUpdatedEvent) { cli.OnServerUpdated(func(bot *client.QQClient, e *client.ServerUpdatedEvent) {
log.Infof("收到服务器地址更新通知, 将在下一次重连时应用. 地址信息已储存到 servers.bin 文件") log.Infof("收到服务器地址更新通知, 将在下一次重连时应用. 地址信息已储存到 servers.bin 文件")
_ = ioutil.WriteFile("servers.bin", binary.NewWriterF(func(w *binary.Writer) { _ = ioutil.WriteFile("servers.bin", binary.NewWriterF(func(w *binary.Writer) {
w.WriteUInt16(uint16(len(e.Servers))) w.WriteUInt16(func() (c uint16) {
for _, s := range e.Servers {
if !strings.Contains(s.Server, "com") {
c++
}
}
return
}())
for _, s := range e.Servers { for _, s := range e.Servers {
if !strings.Contains(s.Server, "com") { if !strings.Contains(s.Server, "com") {
w.WriteString(s.Server) w.WriteString(s.Server)
@ -250,6 +257,12 @@ func main() {
}) })
if global.PathExists("servers.bin") { if global.PathExists("servers.bin") {
if data, err := ioutil.ReadFile("servers.bin"); err == nil { if data, err := ioutil.ReadFile("servers.bin"); err == nil {
func() {
defer func() {
if pan := recover(); pan != nil {
log.Error("读取服务器地址时出现错误: %v", pan)
}
}()
r := binary.NewReader(data) r := binary.NewReader(data)
var addr []*net.TCPAddr var addr []*net.TCPAddr
l := r.ReadUInt16() l := r.ReadUInt16()
@ -262,6 +275,7 @@ func main() {
if len(addr) > 0 { if len(addr) > 0 {
cli.SetCustomServer(addr) cli.SetCustomServer(addr)
} }
}()
} }
} }
rsp, err := cli.Login() rsp, err := cli.Login()