瀏覽代碼

check for host limit on free tier

Abhishek Kondur 2 年之前
父節點
當前提交
a43fea35d8
共有 1 個文件被更改,包括 8 次插入1 次删除
  1. 8 1
      logic/hosts.go

+ 8 - 1
logic/hosts.go

@@ -86,7 +86,14 @@ func GetHost(hostid string) (*models.Host, error) {
 
 // CreateHost - creates a host if not exist
 func CreateHost(h *models.Host) error {
-	_, err := GetHost(h.ID.String())
+	hosts, err := GetAllHosts()
+	if err != nil && !database.IsEmptyRecord(err) {
+		return err
+	}
+	if len(hosts) >= Hosts_Limit {
+		return errors.New("free tier limits exceeded on hosts")
+	}
+	_, err = GetHost(h.ID.String())
 	if (err != nil && !database.IsEmptyRecord(err)) || (err == nil) {
 		return ErrHostExists
 	}