guoswork/frp
1
0
Fork 0
mirror of https://gitee.com/IrisVega/frp.git synced 2024-11-01 22:31:29 +08:00
Code Issues Packages Projects Releases Wiki Activity

(1)新增心跳检测发送间隔和超时时间的配置

Browse Source
This commit is contained in:
Hurricanezwf 2016-02-05 14:36:04 +08:00
parent 5d6f37aa82
commit 04c26d1c31
6 changed files with 26 additions and 21 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
cmd/frpc/config.go
View File

@ -16,11 +16,11 @@ var (
LogFile string = "./frpc.log"
LogLevel string = "warn"
LogWay string = "file"
HeartBeatInterval int64 = 5
)
var ProxyClients map[string]*models.ProxyClient = make(map[string]*models.ProxyClient)
func LoadConf(confFile string) (err error) {
var tmpStr string
var ok bool
@ -56,6 +56,11 @@ func LoadConf(confFile string) (err error) {
LogWay = tmpStr
}
tmpStr, ok = conf.Get("common", "heartbeat_interval")
if ok {
HeartBeatInterval, _ = strconv.ParseInt(tmpStr, 10, 64)
}
// servers
for name, section := range conf {
if name != "common" {

6
cmd/frpc/control.go
View File

@ -11,10 +11,6 @@ import (
"frp/pkg/utils/log"
)
const (
heartbeatDuration = 2 //心跳检测时间间隔，单位秒
)
var isHeartBeatContinue bool = true
func ControlProcess(cli *models.ProxyClient, wait *sync.WaitGroup) {
@ -114,7 +110,7 @@ func loginToServer(cli *models.ProxyClient) (connection *conn.Conn) {
func startHeartBeat(con *conn.Conn) {
isHeartBeatContinue = true
for {
time.Sleep(heartbeatDuration * time.Second)
time.Sleep(time.Duration(HeartBeatInterval) * time.Second)
if isHeartBeatContinue { // 把isHeartBeatContinue放在这里是为了防止SIGPIPE
err := con.Write("\r\n")
//log.Debug("send heart beat to server!")

2
cmd/frps/config.go
View File

@ -16,11 +16,11 @@ var (
LogFile string = "./frps.log"
LogLevel string = "warn"
LogWay string = "file"
HeartBeatTimeout int64 = 30
)
var ProxyServers map[string]*models.ProxyServer = make(map[string]*models.ProxyServer)
func LoadConf(confFile string) (err error) {
var tmpStr string
var ok bool

4
cmd/frps/control.go
View File

@ -151,7 +151,7 @@ func readControlMsgFromClient(server *models.ProxyServer, c *conn.Conn) {
isContinueRead = false
server.StopWaitUserConn()
}
timer := time.AfterFunc(10*time.Second, f)
timer := time.AfterFunc(time.Duration(HeartBeatTimeout)*time.Second, f)
defer timer.Stop()
for isContinueRead {
@ -169,7 +169,7 @@ func readControlMsgFromClient(server *models.ProxyServer, c *conn.Conn) {
if content == "\r\n" {
log.Debug("receive hearbeat:%s", content)
timer.Reset(10 * time.Second)
timer.Reset(time.Duration(HeartBeatTimeout) * time.Second)
}
}
}

4
conf/frpc.ini
View File

@ -6,7 +6,9 @@ log_file = ./frpc.log
# debug, info, warn, error
log_level = info
# file, console
log_way = file
log_way = console
# 心跳检测时间间隔,单位秒,默认为2
heartbeat_interval = 2
# test1即为name
[test1]

4
conf/frps.ini
View File

@ -6,7 +6,9 @@ log_file = ./frps.log
# debug, info, warn, error
log_level = info
# file, console
log_way = file
log_way = console
# 心跳检测超时时间,单位秒,默认为30
heartbeat_timeout = 30
# test1即为name
[test1]