mirror of
https://github.com/TeaOSLab/EdgeCommon.git
synced 2025-11-04 05:00:24 +08:00
94 lines
2.4 KiB
Go
94 lines
2.4 KiB
Go
package nodeconfigs
|
|
|
|
import (
|
|
"fmt"
|
|
"github.com/iwind/TeaGo/rands"
|
|
"net"
|
|
)
|
|
|
|
// 默认的TOA配置
|
|
func DefaultTOAConfig() *TOAConfig {
|
|
return &TOAConfig{
|
|
IsOn: false,
|
|
Debug: false,
|
|
OptionType: 0xfe,
|
|
MinQueueId: 100,
|
|
MaxQueueId: 109,
|
|
AutoSetup: true,
|
|
}
|
|
}
|
|
|
|
// 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"`
|
|
MaxQueueId uint8 `yaml:"maxQueueId" json:"maxQueueId"`
|
|
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)
|
|
|
|
// QueueId
|
|
if this.MinQueueId > this.MaxQueueId {
|
|
this.MinQueueId, this.MaxQueueId = this.MaxQueueId, this.MinQueueId
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
// Sock路径
|
|
func (this *TOAConfig) SockFile() string {
|
|
if len(this.SockPath) == 0 {
|
|
return "/tmp/edge-toa.sock"
|
|
}
|
|
return this.SockPath
|
|
}
|
|
|
|
// 获取随机端口
|
|
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)
|
|
}
|
|
|
|
// 转换为参数的形式
|
|
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
|
|
}
|