Mxmilu666/frp
1
1
Fork 1
mirror of https://github.com/fatedier/frp.git synced 2026-03-19 08:19:14 +08:00
Code Issues Packages Projects Releases Wiki Activity

update for metric

Browse Source
This commit is contained in:
fatedier
2017-03-27 01:39:05 +08:00
parent 634e048d0c
commit 8b2cde3a30
5 changed files with 79 additions and 116 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
server/proxy.go
View File

@@ -306,7 +306,7 @@ func (pxy *UdpProxy) Run() (err error) {
if errRet := msg.ReadMsgInto(conn, &udpMsg); errRet != nil {
pxy.Warn("read from workConn for udp error: %v", errRet)
conn.Close()
// notity proxy to start a new work connection
// notify proxy to start a new work connection
errors.PanicToError(func() {
pxy.checkCloseCh <- 1
})
@@ -314,6 +314,7 @@ func (pxy *UdpProxy) Run() (err error) {
}
if errRet := errors.PanicToError(func() {
pxy.readCh <- &udpMsg
StatsAddTrafficOut(pxy.GetName(), int64(len(udpMsg.Content)))
}); errRet != nil {
pxy.Info("reader goroutine for udp work connection closed")
return
@@ -332,6 +333,7 @@ func (pxy *UdpProxy) Run() (err error) {
pxy.Info("sender goroutine for udp work connection closed: %v", errRet)
return
} else {
StatsAddTrafficIn(pxy.GetName(), int64(len(udpMsg.Content)))
continue
}
case <-ctx.Done():
@@ -420,7 +422,7 @@ func HandleUserTcpConnection(pxy Proxy, userConn frpNet.Conn) {
StatsOpenConnection(pxy.GetName())
inCount, outCount := tcp.Join(local, userConn)
StatsCloseConnection(pxy.GetName())
StatsAddFlowIn(pxy.GetName(), inCount)
StatsAddFlowOut(pxy.GetName(), outCount)
StatsAddTrafficIn(pxy.GetName(), inCount)
StatsAddTrafficOut(pxy.GetName(), outCount)
pxy.Debug("join connections closed")
}