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 getOnline feature

Browse source
This commit is contained in:
HynoR 2024-05-09 10:56:19 +08:00
parent 2366882bd6
commit 88eef7617f
3 changed files with 64 additions and 14 deletions
Download patch file Download diff file Expand all files Collapse all files

2
core/server/config.go
View file

@ -196,4 +196,6 @@ 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)
}

2
core/server/server.go
View file

@ -83,6 +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.EventLogger.Disconnect(conn.RemoteAddr(), handler.authID, err)
}
_ = conn.CloseWithError(closeErrCodeOK, "")
@ -154,6 +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.EventLogger.Connect(h.conn.RemoteAddr(), id, actualTx)
}
// Initialize UDP session manager (if UDP is enabled)