Max Ma 1 год назад
Родитель
Сommit
0a68ddc7b4
3 измененных файлов с 13 добавлено и 0 удалено
  1. 1 0
      config/config.go
  2. 1 0
      models/structs.go
  3. 11 0
      servercfg/serverconf.go

+ 1 - 0
config/config.go

@@ -100,6 +100,7 @@ type ServerConfig struct {
 	SmtpHost                   string        `json:"smtp_host"`
 	SmtpPort                   int           `json:"smtp_port"`
 	MetricInterval             string        `yaml:"metric_interval"`
+	ManageDNS                  bool          `yaml:"manage_dns"`
 }
 
 // SQLConfig - Generic SQL Config

+ 1 - 0
models/structs.go

@@ -256,6 +256,7 @@ type ServerConfig struct {
 	IsPro          bool   `yaml:"isee" json:"Is_EE"`
 	TrafficKey     []byte `yaml:"traffickey"`
 	MetricInterval string `yaml:"metric_interval"`
+	ManageDNS      bool   `yaml:"manage_dns"`
 }
 
 // User.NameInCharset - returns if name is in charset below or not

+ 11 - 0
servercfg/serverconf.go

@@ -92,6 +92,7 @@ func GetServerConfig() config.ServerConfig {
 	cfg.JwtValidityDuration = GetJwtValidityDuration()
 	cfg.RacAutoDisable = GetRacAutoDisable()
 	cfg.MetricInterval = GetMetricInterval()
+	cfg.ManageDNS = GetManageDNS()
 	return cfg
 }
 
@@ -136,6 +137,7 @@ func GetServerInfo() models.ServerConfig {
 	cfg.Version = GetVersion()
 	cfg.IsPro = IsPro
 	cfg.MetricInterval = GetMetricInterval()
+	cfg.ManageDNS = GetManageDNS()
 	return cfg
 }
 
@@ -650,6 +652,15 @@ func GetMetricInterval() string {
 	return mi
 }
 
+// GetManageDNS - if manage DNS enabled or not
+func GetManageDNS() bool {
+	enabled := true
+	if os.Getenv("MANAGE_DNS") != "" {
+		enabled = os.Getenv("MANAGE_DNS") == "true"
+	}
+	return enabled
+}
+
 // GetBatchPeerUpdate - if batch peer update
 func GetBatchPeerUpdate() bool {
 	enabled := true