onyx_online/q3rcon-proxy
1
0
Fork 0
mirror of https://github.com/onyx-and-iris/q3rcon-proxy.git synced 2025-01-18 00:30:54 +00:00
Code Issues Packages Projects Releases Wiki Activity

no need to export sessionCache methods

Browse Source 
rename upsert to insert since we only call it if there isn't a session.
This commit is contained in:
onyx-and-iris 2024-10-23 17:22:03 +01:00
parent abc1ea9d3f
commit bfe31c28c8
2 changed files with 9 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
pkg/udpproxy/sessioncache.go
View File

@ -15,8 +15,8 @@ func newSessionCache() sessionCache {
}
}
// Read returns the associated session for an addr
func (sc *sessionCache) Read(addr string) (*session, bool) {
// read returns the associated session for an addr
func (sc *sessionCache) read(addr string) (*session, bool) {
sc.mu.RLock()
defer sc.mu.RUnlock()
@ -24,16 +24,16 @@ func (sc *sessionCache) Read(addr string) (*session, bool) {
return v, ok
}
// Upsert overrides the session for a given addr.
func (sc *sessionCache) Upsert(addr string, session *session) {
// insert adds a session for a given addr.
func (sc *sessionCache) insert(addr string, session *session) {
sc.mu.Lock()
defer sc.mu.Unlock()
sc.data[addr] = session
}
// Delete removes the session for the given addr.
func (sc *sessionCache) Delete(addr string) {
// delete removes the session for the given addr.
func (sc *sessionCache) delete(addr string) {
sc.mu.Lock()
defer sc.mu.Unlock()

6
pkg/udpproxy/udpproxy.go
View File

@ -73,7 +73,7 @@ func (c *Client) ListenAndServe() error {
log.Error(err)
}
session, ok := c.sessionCache.Read(caddr.String())
session, ok := c.sessionCache.read(caddr.String())
if !ok {
session, err = newSession(caddr, c.raddr, c.proxyConn)
if err != nil {
@ -81,7 +81,7 @@ func (c *Client) ListenAndServe() error {
continue
}
c.sessionCache.Upsert(caddr.String(), session)
c.sessionCache.insert(caddr.String(), session)
}
go session.proxyTo(buf[:n])
@ -94,7 +94,7 @@ func (c *Client) pruneSessions() {
for range ticker.C {
for _, session := range c.sessionCache.data {
if time.Since(session.updateTime) > c.sessionTimeout {
c.sessionCache.Delete(session.caddr.String())
c.sessionCache.delete(session.caddr.String())
log.Tracef("session for %s deleted", session.caddr)
}
}