KJZH001/nezhahq-agent
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

make edit and service flag subcommands (#77)

Browse Source 
* make edit and service argument subcommands

* generate uuid if non-exist, use default report_delay or ip_report_period value if not specified
This commit is contained in:
UUBulb
2024-10-23 22:34:59 +08:00
committed by GitHub
parent fbf099e437
commit 0a890a2021
7 changed files with 193 additions and 114 deletions
Download Patch File Download Diff File Expand all files Collapse all files
model/config.go
+29 -20
View File
@@ -8,28 +8,28 @@ import (
)
type AgentConfig struct {
Debug bool `mapstructure:"debug"`
Debug bool `mapstructure:"debug" json:"debug"`
Server string `mapstructure:"server"` // 服务器地址
ClientSecret string `mapstructure:"client_secret"` // 客户端密钥
UUID string `mapstructure:"uuid"`
Server string `mapstructure:"server" json:"server"` // 服务器地址
ClientSecret string `mapstructure:"client_secret" json:"client_secret"` // 客户端密钥
UUID string `mapstructure:"uuid" json:"uuid"`
HardDrivePartitionAllowlist []string `mapstructure:"hard_drive_partition_allowlist"`
NICAllowlist map[string]bool `mapstructure:"nic_allowlist"`
DNS []string `mapstructure:"dns"`
GPU bool `mapstructure:"gpu"` // 是否检查GPU
Temperature bool `mapstructure:"temperature"` // 是否检查温度
SkipConnectionCount bool `mapstructure:"skip_connection_count"` // 跳过连接数检查
SkipProcsCount bool `mapstructure:"skip_procs_count"` // 跳过进程数量检查
DisableAutoUpdate bool `mapstructure:"disable_auto_update"` // 关闭自动更新
DisableForceUpdate bool `mapstructure:"disable_force_update"` // 关闭强制更新
DisableCommandExecute bool `mapstructure:"disable_command_execute"` // 关闭命令执行
ReportDelay int `mapstructure:"report_delay"` // 报告间隔
TLS bool `mapstructure:"tls"` // 是否使用TLS加密传输至服务端
InsecureTLS bool `mapstructure:"insecure_tls"` // 是否禁用证书检查
UseIPv6CountryCode bool `mapstructure:"use_i_pv_6_country_code"` // 默认优先展示IPv6旗帜
UseGiteeToUpgrade bool `mapstructure:"use_gitee_to_upgrade"` // 强制从Gitee获取更新
IPReportPeriod uint32 `mapstructure:"ip_report_period"` // IP上报周期
HardDrivePartitionAllowlist []string `mapstructure:"hard_drive_partition_allowlist" json:"hard_drive_partition_allowlist"`
NICAllowlist map[string]bool `mapstructure:"nic_allowlist" json:"nic_allowlist"`
DNS []string `mapstructure:"dns" json:"dns"`
GPU bool `mapstructure:"gpu" json:"gpu"` // 是否检查GPU
Temperature bool `mapstructure:"temperature" json:"temperature"` // 是否检查温度
SkipConnectionCount bool `mapstructure:"skip_connection_count" json:"skip_connection_count"` // 跳过连接数检查
SkipProcsCount bool `mapstructure:"skip_procs_count" json:"skip_procs_count"` // 跳过进程数量检查
DisableAutoUpdate bool `mapstructure:"disable_auto_update" json:"disable_auto_update"` // 关闭自动更新
DisableForceUpdate bool `mapstructure:"disable_force_update" json:"disable_force_update"` // 关闭强制更新
DisableCommandExecute bool `mapstructure:"disable_command_execute" json:"disable_command_execute"` // 关闭命令执行
ReportDelay int `mapstructure:"report_delay" json:"report_delay"` // 报告间隔
TLS bool `mapstructure:"tls" json:"tls"` // 是否使用TLS加密传输至服务端
InsecureTLS bool `mapstructure:"insecure_tls" json:"insecure_tls"` // 是否禁用证书检查
UseIPv6CountryCode bool `mapstructure:"use_ipv6_country_code" json:"use_ipv6_country_code"` // 默认优先展示IPv6旗帜
UseGiteeToUpgrade bool `mapstructure:"use_gitee_to_upgrade" json:"use_gitee_to_upgrade"` // 强制从Gitee获取更新
IPReportPeriod uint32 `mapstructure:"ip_report_period" json:"ip_report_period"` // IP上报周期
v *viper.Viper
}
@@ -46,6 +46,15 @@ func (c *AgentConfig) Read(path string) error {
if err != nil {
return err
}
if c.ReportDelay == 0 {
c.ReportDelay = 1
}
if c.IPReportPeriod == 0 {
c.IPReportPeriod = 1800
}
return nil
}