1 files changed, 162 insertions, 0 deletions
diff --git a/src/dcm/api/quotaHandler.go b/src/dcm/api/quotaHandler.go
new file mode 100644
index 00000000..bca5206a
--- /dev/null
+++ b/src/dcm/api/quotaHandler.go
@@ -0,0 +1,162 @@
+/*
+* Copyright 2020 Intel Corporation, Inc
+*
+* Licensed under the Apache License, Version 2.0 (the "License");
+* you may not use this file except in compliance with the License.
+* You may obtain a copy of the License at
+*
+*     http://www.apache.org/licenses/LICENSE-2.0
+*
+* Unless required by applicable law or agreed to in writing, software
+* distributed under the License is distributed on an "AS IS" BASIS,
+* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or
+* implied.
+* See the License for the specific language governing permissions
+* and
+* limitations under the License.
+*/
+
+package api
+
+import (
+    "encoding/json"
+    "net/http"
+    "io"
+    "github.com/onap/multicloud-k8s/src/dcm/pkg/module"
+
+    "github.com/gorilla/mux"
+)
+
+
+// quotaHandler is used to store backend implementations objects
+type quotaHandler struct {
+    client module.QuotaManager
+}
+
+// CreateHandler handles creation of the quota entry in the database
+
+func (h quotaHandler) createHandler(w http.ResponseWriter, r *http.Request) {
+    vars := mux.Vars(r)
+    project := vars["project-name"]
+    logicalCloud := vars["logical-cloud-name"]
+    var v module.Quota
+
+    err := json.NewDecoder(r.Body).Decode(&v)
+    switch {
+    case err == io.EOF:
+        http.Error(w, "Empty body", http.StatusBadRequest)
+        return
+    case err != nil:
+        http.Error(w, err.Error(), http.StatusUnprocessableEntity)
+        return
+    }
+
+    // Quota Name is required.
+    if v.MetaData.QuotaName == "" {
+        http.Error(w, "Missing name in POST request", http.StatusBadRequest)
+        return
+    }
+
+    ret, err := h.client.CreateQuota(project, logicalCloud, v)
+    if err != nil {
+        http.Error(w, err.Error(), http.StatusInternalServerError)
+        return
+    }
+
+    w.Header().Set("Content-Type", "application/json")
+    w.WriteHeader(http.StatusCreated)
+    err = json.NewEncoder(w).Encode(ret)
+    if err != nil {
+        http.Error(w, err.Error(), http.StatusInternalServerError)
+        return
+    }
+}
+
+// getHandler handle GET operations on a particular name
+// Returns a quota
+func (h quotaHandler) getHandler(w http.ResponseWriter, r *http.Request) {
+    vars := mux.Vars(r)
+    project := vars["project-name"]
+    logicalCloud := vars["logical-cloud-name"]
+    name := vars["quota-name"]
+    var ret interface{}
+    var err error
+
+    if len(name) == 0 {
+        ret, err = h.client.GetAllQuotas(project, logicalCloud)
+        if err != nil {
+            http.Error(w, err.Error(), http.StatusInternalServerError)
+            return
+        }
+    } else {
+        ret, err = h.client.GetQuota(project, logicalCloud, name)
+        if err != nil {
+            http.Error(w, err.Error(), http.StatusInternalServerError)
+            return
+        }
+    }
+
+    w.Header().Set("Content-Type", "application/json")
+    w.WriteHeader(http.StatusOK)
+    err = json.NewEncoder(w).Encode(ret)
+    if err != nil {
+        http.Error(w, err.Error(), http.StatusInternalServerError)
+        return
+    }
+}
+
+// UpdateHandler handles Update operations on a particular quota
+func (h quotaHandler) updateHandler(w http.ResponseWriter, r *http.Request) {
+    var v module.Quota
+    vars := mux.Vars(r)
+    project := vars["project-name"]
+    logicalCloud := vars["logical-cloud-name"]
+    name := vars["quota-name"]
+
+    err := json.NewDecoder(r.Body).Decode(&v)
+    switch {
+        case err == io.EOF:
+            http.Error(w, "Empty body", http.StatusBadRequest)
+            return
+        case err != nil:
+            http.Error(w, err.Error(), http.StatusUnprocessableEntity)
+            return
+    }
+
+    // Name is required.
+    if v.MetaData.QuotaName == "" {
+        http.Error(w, "Missing name in PUT request", http.StatusBadRequest)
+        return
+    }
+
+    ret, err := h.client.UpdateQuota(project, logicalCloud, name, v)
+    if err != nil {
+        http.Error(w, err.Error(),
+        http.StatusInternalServerError)
+        return
+    }
+    w.Header().Set("Content-Type", "application/json")
+    w.WriteHeader(http.StatusCreated)
+    err = json.NewEncoder(w).Encode(ret)
+    if err != nil {
+        http.Error(w, err.Error(),
+        http.StatusInternalServerError)
+        return
+    }
+}
+
+//deleteHandler handles DELETE operations on a particular record
+func (h quotaHandler) deleteHandler(w http.ResponseWriter, r *http.Request) {
+    vars := mux.Vars(r)
+    project := vars["project-name"]
+    logicalCloud := vars["logical-cloud-name"]
+    name := vars["quota-name"]
+
+    err := h.client.DeleteQuota(project, logicalCloud, name)
+    if err != nil {
+        http.Error(w, err.Error(), http.StatusInternalServerError)
+        return
+    }
+
+    w.WriteHeader(http.StatusNoContent)
+}