mirror of
https://github.com/fatedier/frp.git
synced 2026-10-15 12:15:55 +08:00
fix(server): control work connection refill scheduling
This commit is contained in:
+5
-2
@@ -121,8 +121,9 @@ func (ctl *Control) SetInWorkConnCallback(cb func(*v1.ProxyBaseConfig, net.Conn,
|
||||
ctl.pm.SetInWorkConnCallback(cb)
|
||||
}
|
||||
|
||||
func (ctl *Control) handleReqWorkConn(_ msg.Message) {
|
||||
func (ctl *Control) handleReqWorkConn(raw msg.Message) {
|
||||
xl := ctl.xl
|
||||
req := raw.(*msg.ReqWorkConn)
|
||||
workConn, err := ctl.connectServer()
|
||||
if err != nil {
|
||||
xl.Warnf("start new connection to server error: %v", err)
|
||||
@@ -130,7 +131,9 @@ func (ctl *Control) handleReqWorkConn(_ msg.Message) {
|
||||
}
|
||||
|
||||
m := &msg.NewWorkConn{
|
||||
RunID: ctl.sessionCtx.RunID,
|
||||
RunID: ctl.sessionCtx.RunID,
|
||||
WorkConnType: req.WorkConnType,
|
||||
ControlID: req.ControlID,
|
||||
}
|
||||
if err = ctl.sessionCtx.Auth.Setter.SetNewWorkConn(m); err != nil {
|
||||
xl.Warnf("error during NewWorkConn authentication: %v", err)
|
||||
|
||||
Reference in New Issue
Block a user