From e92a8727380c18c82c5aaaee4732967e9d7c1702 Mon Sep 17 00:00:00 2001 From: yuzuki999 Date: Tue, 6 Sep 2022 00:32:52 +0800 Subject: [PATCH] rename some method --- core/user.go | 6 +++--- node/node.go | 12 ++++++------ 2 files changed, 9 insertions(+), 9 deletions(-) diff --git a/core/user.go b/core/user.go index ab73d66..14b1c4a 100644 --- a/core/user.go +++ b/core/user.go @@ -74,14 +74,14 @@ func (p *Core) GetUserTraffic(email string) (up int64, down int64) { return up, down } -func (p *Core) GetOnlineIps(tag string) ([]dispatcher.UserIp, error) { +func (p *Core) ListOnlineIp(tag string) ([]dispatcher.UserIp, error) { return p.dispatcher.Limiter.GetOnlineUserIp(tag) } -func (p *Core) UpdateOnlineIps(tag string, ips []dispatcher.UserIp) { +func (p *Core) UpdateOnlineIp(tag string, ips []dispatcher.UserIp) { p.dispatcher.Limiter.UpdateOnlineUserIP(tag, ips) } -func (p *Core) ClearOnlineIps(tag string) { +func (p *Core) ClearOnlineIp(tag string) { p.dispatcher.Limiter.ClearOnlineUserIP(tag) } diff --git a/node/node.go b/node/node.go index d0bc88c..aec6bdb 100644 --- a/node/node.go +++ b/node/node.go @@ -382,14 +382,14 @@ func (c *Node) userInfoMonitor() (err error) { } userTraffic = nil if !c.config.EnableIpRecorder { - c.server.ClearOnlineIps(c.Tag) + c.server.ClearOnlineIp(c.Tag) } runtime.GC() return nil } func (c *Node) onlineIpReport() (err error) { - onlineIp, err := c.server.GetOnlineIps(c.Tag) + onlineIp, err := c.server.ListOnlineIp(c.Tag) if err != nil { log.Print(err) return nil @@ -402,7 +402,7 @@ func (c *Node) onlineIpReport() (err error) { c.config.IpRecorderConfig.Token) if err != nil { log.Print(err) - c.server.ClearOnlineIps(c.Tag) + c.server.ClearOnlineIp(c.Tag) return nil } log.Printf("[Node: %d] Report %d online ip", c.nodeInfo.NodeId, len(onlineIp)) @@ -411,13 +411,13 @@ func (c *Node) onlineIpReport() (err error) { err := json.Unmarshal(rsp.Body(), &onlineIp) if err != nil { log.Print(err) - c.server.ClearOnlineIps(c.Tag) + c.server.ClearOnlineIp(c.Tag) return nil } - c.server.UpdateOnlineIps(c.Tag, onlineIp) + c.server.UpdateOnlineIp(c.Tag, onlineIp) log.Printf("[Node: %d] Updated %d online ip", c.nodeInfo.NodeId, len(onlineIp)) } else { - c.server.ClearOnlineIps(c.Tag) + c.server.ClearOnlineIp(c.Tag) } return nil }