EdgeCommon/pkg/nodeconfigs/toa_config.go

package nodeconfigs

import (
	"fmt"
	"github.com/iwind/TeaGo/rands"
	"net"
	"runtime"
)

// DefaultTOAConfig 默认的TOA配置
func DefaultTOAConfig() *TOAConfig {
	return &TOAConfig{
		IsOn:       false,
		Debug:      false,
		OptionType: 0xfe,
		MinQueueId: 100,
		AutoSetup:  true,
	}
}

// TOAConfig TOA相关配置
type TOAConfig struct {
	IsOn         bool     `yaml:"isOn" json:"isOn"` // 是否启用
	Debug        bool     `yaml:"debug" json:"debug"`
	OptionType   uint8    `yaml:"optionType" json:"optionType"`
	MinQueueId   uint8    `yaml:"minQueueId" json:"minQueueId"`
	AutoSetup    bool     `yaml:"autoSetup" json:"autoSetup"`
	MinLocalPort uint16   `yaml:"minLocalPort" json:"minLocalPort"` // 本地可使用的最小端口 TODO
	MaxLocalPort uint16   `yaml:"maxLocalPort" json:"maxLocalPort"` // 本地可使用的最大端口 TODO
	SockPath     string   `yaml:"sockPath" json:"sockPath"`         // Sock文件路径 TODO
	ByPassPorts  []uint16 `yaml:"byPassPorts" json:"byPassPorts"`   // 忽略的端口 TODO

	minLocalPort int
	maxLocalPort int
}

func (this *TOAConfig) Init() error {
	// LocalPort
	minPort := this.MinLocalPort
	maxPort := this.MaxLocalPort
	if minPort == 0 {
		minPort = 1025
	}
	if maxPort == 0 {
		maxPort = 65534
	}
	if minPort > maxPort {
		minPort, maxPort = maxPort, minPort
	}
	this.minLocalPort = int(minPort)
	this.maxLocalPort = int(maxPort)

	return nil
}

// SockFile Sock路径
func (this *TOAConfig) SockFile() string {
	if len(this.SockPath) == 0 {
		return "/tmp/edge-toa.sock"
	}
	return this.SockPath
}

// RandLocalPort 获取随机端口
func (this *TOAConfig) RandLocalPort() uint16 {
	listener, err := net.Listen("tcp", ":0")
	if err != nil {
		return uint16(rands.Int(this.minLocalPort, this.maxLocalPort))
	}
	_ = listener.Close()
	return uint16(listener.Addr().(*net.TCPAddr).Port)
}

// AsArgs 转换为参数的形式
func (this *TOAConfig) AsArgs() (args []string) {
	args = append(args, "run")
	args = append(args, "-option-type="+fmt.Sprintf("%d", this.OptionType))
	args = append(args, "-min-queue-id="+fmt.Sprintf("%d", this.MinQueueId))
	args = append(args, "-max-queue-id="+fmt.Sprintf("%d", this.MaxQueueId()))
	if this.AutoSetup {
		args = append(args, "-auto-setup")
	}
	if this.Debug {
		args = append(args, "-debug")
	}
	return
}

// MaxQueueId 获取队列ID最大值
func (this *TOAConfig) MaxQueueId() uint8 {
	var maxQueueId = int(this.MinQueueId) + runtime.NumCPU() - 1
	if maxQueueId > 255 {
		maxQueueId = 255
	}
	return uint8(maxQueueId)
}