mirror of
https://gitee.com/rainbond/Rainbond.git
synced 2024-11-29 18:27:58 +08:00
f1fd46a8eb
* feat: support arm amd multi-architecture cluster * feat: builder runner dual architecture * fix: build rainbond arm use failure (#1674) * perf: ComponentK8sAttributeGet add note --------- Co-authored-by: yangkaa <30614084+yangkaa@users.noreply.github.com>
122 lines
4.3 KiB
Go
122 lines
4.3 KiB
Go
// RAINBOND, Application Management Platform
|
|
// Copyright (C) 2022-2022 Goodrain Co., Ltd.
|
|
|
|
// This program is free software: you can redistribute it and/or modify
|
|
// it under the terms of the GNU General Public License as published by
|
|
// the Free Software Foundation, either version 3 of the License, or
|
|
// (at your option) any later version. For any non-GPL usage of Rainbond,
|
|
// one or multiple Commercial Licenses authorized by Goodrain Co., Ltd.
|
|
// must be obtained first.
|
|
|
|
// This program is distributed in the hope that it will be useful,
|
|
// but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
// GNU General Public License for more details.
|
|
|
|
// You should have received a copy of the GNU General Public License
|
|
// along with this program. If not, see <http://www.gnu.org/licenses/>.
|
|
|
|
package controller
|
|
|
|
import (
|
|
"fmt"
|
|
"github.com/goodrain/rainbond/api/handler"
|
|
api_model "github.com/goodrain/rainbond/api/model"
|
|
ctxutil "github.com/goodrain/rainbond/api/util/ctx"
|
|
dbmodel "github.com/goodrain/rainbond/db/model"
|
|
httputil "github.com/goodrain/rainbond/util/http"
|
|
"net/http"
|
|
)
|
|
|
|
// K8sAttributeController -
|
|
type K8sAttributeController struct{}
|
|
|
|
// K8sAttributes -
|
|
func (k *K8sAttributeController) K8sAttributes(w http.ResponseWriter, r *http.Request) {
|
|
switch r.Method {
|
|
case "POST":
|
|
k.createK8sAttributes(w, r)
|
|
case "PUT":
|
|
k.updateK8sAttributes(w, r)
|
|
case "DELETE":
|
|
k.deleteK8sAttributes(w, r)
|
|
case "GET":
|
|
k.getK8sAttributes(w, r)
|
|
}
|
|
}
|
|
|
|
func (k *K8sAttributeController) getK8sAttributes(w http.ResponseWriter, r *http.Request) {
|
|
componentID := r.Context().Value(ctxutil.ContextKey("service_id")).(string)
|
|
var k8sAttrGet api_model.ComponentK8sAttributeGet
|
|
if ok := httputil.ValidatorRequestStructAndErrorResponse(r, w, &k8sAttrGet, nil); !ok {
|
|
httputil.ReturnBcodeError(r, w, fmt.Errorf("k8s attributes is not valid"))
|
|
return
|
|
}
|
|
atts, err := handler.GetServiceManager().GetK8sAttribute(componentID, k8sAttrGet.Name)
|
|
if err != nil {
|
|
httputil.ReturnBcodeError(r, w, err)
|
|
return
|
|
}
|
|
if k8sAttrGet.Arch != "" {
|
|
var attsValue string
|
|
if atts != nil {
|
|
attsValue, err = handler.GetServiceManager().ReviseAttributeAffinityByArch(atts.AttributeValue, k8sAttrGet.Arch)
|
|
atts.AttributeValue = attsValue
|
|
} else {
|
|
attsValue, err = handler.GetServiceManager().ReviseAttributeAffinityByArch("", k8sAttrGet.Arch)
|
|
atts = &dbmodel.ComponentK8sAttributes{
|
|
AttributeValue: attsValue,
|
|
}
|
|
}
|
|
if err != nil {
|
|
httputil.ReturnBcodeError(r, w, err)
|
|
return
|
|
}
|
|
}
|
|
|
|
httputil.ReturnSuccess(r, w, atts)
|
|
}
|
|
|
|
func (k *K8sAttributeController) createK8sAttributes(w http.ResponseWriter, r *http.Request) {
|
|
tenantID := r.Context().Value(ctxutil.ContextKey("tenant_id")).(string)
|
|
componentID := r.Context().Value(ctxutil.ContextKey("service_id")).(string)
|
|
var k8sAttr api_model.ComponentK8sAttribute
|
|
if ok := httputil.ValidatorRequestStructAndErrorResponse(r, w, &k8sAttr, nil); !ok {
|
|
httputil.ReturnBcodeError(r, w, fmt.Errorf("k8s attributes is not valid"))
|
|
return
|
|
}
|
|
if err := handler.GetServiceManager().CreateK8sAttribute(tenantID, componentID, &k8sAttr); err != nil {
|
|
httputil.ReturnBcodeError(r, w, err)
|
|
return
|
|
}
|
|
w.WriteHeader(http.StatusOK)
|
|
}
|
|
|
|
func (k *K8sAttributeController) updateK8sAttributes(w http.ResponseWriter, r *http.Request) {
|
|
componentID := r.Context().Value(ctxutil.ContextKey("service_id")).(string)
|
|
var k8sAttr api_model.ComponentK8sAttribute
|
|
if ok := httputil.ValidatorRequestStructAndErrorResponse(r, w, &k8sAttr, nil); !ok {
|
|
httputil.ReturnBcodeError(r, w, fmt.Errorf("k8s attributes is not valid"))
|
|
return
|
|
}
|
|
if err := handler.GetServiceManager().UpdateK8sAttribute(componentID, &k8sAttr); err != nil {
|
|
httputil.ReturnBcodeError(r, w, err)
|
|
return
|
|
}
|
|
w.WriteHeader(http.StatusOK)
|
|
}
|
|
|
|
func (k *K8sAttributeController) deleteK8sAttributes(w http.ResponseWriter, r *http.Request) {
|
|
componentID := r.Context().Value(ctxutil.ContextKey("service_id")).(string)
|
|
var req api_model.DeleteK8sAttributeReq
|
|
if ok := httputil.ValidatorRequestStructAndErrorResponse(r, w, &req, nil); !ok {
|
|
httputil.ReturnBcodeError(r, w, fmt.Errorf("k8s attributes is not valid"))
|
|
return
|
|
}
|
|
if err := handler.GetServiceManager().DeleteK8sAttribute(componentID, req.Name); err != nil {
|
|
httputil.ReturnBcodeError(r, w, err)
|
|
return
|
|
}
|
|
w.WriteHeader(http.StatusOK)
|
|
}
|