ソースを参照

add host limit to config

Abhishek Kondur 2 年 前
コミット
b00094afe8
3 ファイル変更14 行追加0 行削除
  1. 1 0
      config/config.go
  2. 3 0
      logic/serverconf.go
  3. 10 0
      servercfg/serverconf.go

+ 1 - 0
config/config.go

@@ -76,6 +76,7 @@ type ServerConfig struct {
 	UserLimit            int    `yaml:"user_limit"`
 	ClientLimit          int    `yaml:"client_limit"`
 	NetworkLimit         int    `yaml:"network_limit"`
+	HostLimit            int    `yaml:"host_limit"`
 	DeployedByOperator   bool   `yaml:"deployed_by_operator"`
 	StunList             string `yaml:"stun_list"`
 }

+ 3 - 0
logic/serverconf.go

@@ -16,6 +16,8 @@ var (
 	Users_Limit = 1000000000
 	// Clients_Limit - dummy var for community
 	Clients_Limit = 1000000000
+	// Hosts_Limit - dummy var for community
+	Hosts_Limit = 1000000000
 	// Free_Tier - specifies if free tier
 	Free_Tier = false
 )
@@ -92,4 +94,5 @@ func SetFreeTierLimits() {
 	Users_Limit = servercfg.GetUserLimit()
 	Clients_Limit = servercfg.GetClientLimit()
 	Networks_Limit = servercfg.GetNetworkLimit()
+	Hosts_Limit = servercfg.GetHostLimit()
 }

+ 10 - 0
servercfg/serverconf.go

@@ -666,6 +666,16 @@ func GetClientLimit() int {
 	return clientLimit
 }
 
+func GetHostLimit() int {
+	var hostLimit int
+	if os.Getenv("HOST_LIMIT") != "" {
+		hostLimit, _ = strconv.Atoi(os.Getenv("HOST_LIMIT"))
+	} else {
+		hostLimit = config.Config.Server.HostLimit
+	}
+	return hostLimit
+}
+
 func DeployedByOperator() bool {
 	if os.Getenv("DEPLOYED_BY_OPERATOR") != "" {
 		return os.Getenv("DEPLOYED_BY_OPERATOR") == "true"