mirrors/hysteria
1
0
Fork 0
mirror of https://github.com/apernet/hysteria.git synced 2025-04-03 04:27:39 +03:00
Code Issues Projects Releases Packages Wiki Activity

refactor(online): track count instead of raddr

Browse source
This commit is contained in:
Haruue 2024-05-11 11:05:32 +08:00
parent 88eef7617f
commit ba9b3cdebb
No known key found for this signature in database
GPG key ID: F6083B28CBCBC148
3 changed files with 17 additions and 63 deletions
Download patch file Download diff file Expand all files Collapse all files

3
core/server/config.go
View file

@ -196,6 +196,5 @@ type EventLogger interface {
// The implementation of this interface must be thread-safe.
type TrafficLogger interface {
Log(id string, tx, rx uint64) (ok bool)
LogOnline(id string, addr net.Addr)
LogOffline(id string, addr net.Addr)
LogOnlineStateChanged(id string, online bool)
}

4
core/server/server.go
View file

@ -83,7 +83,7 @@ func (s *serverImpl) handleClient(conn quic.Connection) {
err := h3s.ServeQUICConn(conn)
// If the client is authenticated, we need to log the disconnect event
if handler.authenticated && s.config.EventLogger != nil {
s.config.TrafficLogger.LogOffline(handler.authID, conn.RemoteAddr())
s.config.TrafficLogger.LogOnlineStateChanged(handler.authID, false)
s.config.EventLogger.Disconnect(conn.RemoteAddr(), handler.authID, err)
}
_ = conn.CloseWithError(closeErrCodeOK, "")
@ -155,7 +155,7 @@ func (h *h3sHandler) ServeHTTP(w http.ResponseWriter, r *http.Request) {
w.WriteHeader(protocol.StatusAuthOK)
// Call event logger
if h.config.EventLogger != nil {
h.config.TrafficLogger.LogOnline(id, h.conn.RemoteAddr())
h.config.TrafficLogger.LogOnlineStateChanged(id, true)
h.config.EventLogger.Connect(h.conn.RemoteAddr(), id, actualTx)
}
// Initialize UDP session manager (if UDP is enabled)