options.go

package hydra

import (
	"fmt"
	"os"
	"path/filepath"

	"github.com/micro-plat/hydra/global"
	"github.com/micro-plat/lib4go/types"
)

//Option 配置选项
type Option func()

//WithRegistry 设置注册中心地址
func WithRegistry(addr string) Option {
	return func() {
		global.Def.RegistryAddr = addr
	}
}

//WithPlatName 设置平台名称
func WithPlatName(platName string, platCNName ...string) Option {
	return func() {
		global.Def.PlatName = platName
		global.Def.PlatCNName = types.GetStringByIndex(platCNName, 0, platName)
	}
}

//WithSystemName 设置系统名称
func WithSystemName(sysName string, sysCNName ...string) Option {
	return func() {
		global.Def.SysName = sysName
		global.Def.SysCNName = types.GetStringByIndex(sysCNName, 0, sysName)
	}
}

//WithServerTypes 设置系统类型
func WithServerTypes(serverType ...string) Option {
	return func() {
		for _, s := range serverType {
			if !types.StringContains(global.Def.ServerTypes, s) {
				global.Def.ServerTypes = append(global.Def.ServerTypes, s)
			}
		}
	}
}

//WithClusterName 设置集群名称
func WithClusterName(clusterName string) Option {
	return func() {
		global.Def.ClusterName = clusterName
	}
}

//WithDebug 设置dubug模式
func WithDebug() Option {
	return func() {
		global.IsDebug = true
	}
}

//WithVersion 设置当前软件版本号
func WithVersion(v string) Option {
	return func() {
		global.Version = v
	}
}

//WithUsage 设置使用说明
func WithUsage(usage string) Option {
	return func() {
		global.Usage = fmt.Sprintf("%s(%s)", filepath.Base(os.Args[0]), usage)
	}
}

//WithRunFlag 添加run命令扩展参数
func WithRunFlag(flags ...FlagOption) Option {
	return func() {
		global.RunCli.AddFlags(flags...)
	}
}

//WithConfFlag 添加conf命令扩展参数
func WithConfFlag(flags ...FlagOption) Option {
	return func() {
		global.ConfCli.AddFlags(flags...)
	}
}

//WithDBFlag 添加db命令扩展参数
func WithDBFlag(flags ...FlagOption) Option {
	return func() {
		global.DBCli.AddFlags(flags...)
	}
}

//WithInstallFlag 添加install命令扩展参数
func WithInstallFlag(flags ...FlagOption) Option {
	return func() {
		global.InstallCli.AddFlags(flags...)
	}
}

//WithIPMask 设置获取本地IP的掩码
func WithIPMask(mask string) Option {
	return func() {
		global.Def.IPMask = mask
	}
}

//WithTrace 用于生成pprof的性能分析数据,支持的模式有:cpu,mem,block,mutex,web
func WithTrace(trace string) Option {
	return func() {
		global.Def.Trace = trace
	}
}

//WithTracePort 性能跟踪端口，当Trace为web时候可用，指定pprof的端口
func WithTracePort(port string) Option {
	return func() {
		global.Def.TracePort = port
	}
}