瀏覽代碼

Use switch instead of repeated else if

gabrielseibel1 2 年之前
父節點
當前提交
e570bfb3d1
共有 1 個文件被更改,包括 4 次插入3 次删除
  1. 4 3
      controllers/limits.go

+ 4 - 3
controllers/limits.go

@@ -23,20 +23,21 @@ func checkFreeTierLimits(limitChoice int, next http.Handler) http.HandlerFunc {
 		}
 
 		if logic.Free_Tier { // check that free tier limits not exceeded
-			if limitChoice == networksLimit {
+			switch limitChoice {
+			case networksLimit:
 				currentNetworks, err := logic.GetNetworks()
 				if (err != nil && !database.IsEmptyRecord(err)) || len(currentNetworks) >= logic.Networks_Limit {
 					logic.ReturnErrorResponse(w, r, errorResponse)
 					return
 				}
-			} else if limitChoice == usersLimit {
+			case usersLimit:
 				users, err := logic.GetUsers()
 				if (err != nil && !database.IsEmptyRecord(err)) || len(users) >= logic.Users_Limit {
 					errorResponse.Message = "free tier limits exceeded on users"
 					logic.ReturnErrorResponse(w, r, errorResponse)
 					return
 				}
-			} else if limitChoice == clientsLimit {
+			case clientsLimit:
 				clients, err := logic.GetAllExtClients()
 				if (err != nil && !database.IsEmptyRecord(err)) || len(clients) >= logic.Clients_Limit {
 					errorResponse.Message = "free tier limits exceeded on external clients"