mirrors/frp
1
0
Fork 0
mirror of https://github.com/fatedier/frp.git synced 2025-07-19 20:30:08 +07:00
Code Issues Packages Projects Releases Wiki Activity

fix new http no traffic stats, fix #590

Browse Source
This commit is contained in:
fatedier
2018-01-17 23:17:15 +08:00
parent 5b08201e5d
commit 4f7dfcdb31
3 changed files with 45 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
server/proxy.go
View File

@ -287,9 +287,18 @@ func (pxy *HttpProxy) GetRealConn() (workConn frpNet.Conn, err error) {
rwc = frpIo.WithCompression(rwc)
}
workConn = frpNet.WrapReadWriteCloserToConn(rwc, tmpConn)
workConn = frpNet.WrapStatsConn(workConn, pxy.updateStatsAfterClosedConn)
StatsOpenConnection(pxy.GetName())
return
}
func (pxy *HttpProxy) updateStatsAfterClosedConn(totalRead, totalWrite int64) {
name := pxy.GetName()
StatsCloseConnection(name)
StatsAddTrafficIn(name, totalWrite)
StatsAddTrafficOut(name, totalRead)
}
func (pxy *HttpProxy) Close() {
pxy.BaseProxy.Close()
for _, closeFn := range pxy.closeFuncs {