Function HandleRoundrobin

controller/roundrobin.go:17–57

HandleRoundrobin 顺序轮转目标，失败时回退到 boost 模式。

(conn net.Conn, rule *config.Rule)

// HandleRoundrobin 顺序轮转目标，失败时回退到 boost 模式。
func HandleRoundrobin(conn net.Conn, rule *config.Rule) {
	defer conn.Close()

	index := atomic.AddUint64(&tcpCounter, 1) % uint64(len(rule.Targets))
	if tcpCounter >= 100*uint64(len(rule.Targets)) {
		atomic.StoreUint64(&tcpCounter, 1)
	}

	v := rule.Targets[index]

	roundrobinBegin := time.Now()
	target, err := outboundDial(v.Address)
	if err != nil {
		utils.Logger.Error("无法建立连接，切换到 boost 模式",
			zap.String("ruleName", rule.Name),
			zap.String("remoteAddr", conn.RemoteAddr().String()),
			zap.String("targetAddr", v.Address),
			zap.Int64("failedTime(ms)", time.Since(roundrobinBegin).Milliseconds()))
		HandleBoost(conn, rule)
		return
	}
	if tc, ok := target.(*net.TCPConn); ok {
		_ = tc.SetNoDelay(true)
		_ = tc.SetKeepAlive(true)
		_ = tc.SetKeepAlivePeriod(30 * time.Second)
	}
	utils.Logger.Debug("建立连接",
		zap.String("ruleName", rule.Name),
		zap.String("remoteAddr", conn.RemoteAddr().String()),
		zap.String("targetAddr", target.RemoteAddr().String()),
		zap.Int64("roundrobinTime(ms)", time.Since(roundrobinBegin).Milliseconds()))

	defer target.Close()

	go func() {
		io.Copy(conn, target)
		conn.Close()
		target.Close()
	}()
	io.Copy(target, conn)
}