mirror of
https://gitee.com/rainbond/Rainbond.git
synced 2024-12-05 21:27:47 +08:00
148 lines
4.7 KiB
Go
148 lines
4.7 KiB
Go
// Copyright (C) 2014-2018 Goodrain Co., Ltd.
|
|
// RAINBOND, Application Management Platform
|
|
|
|
// 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 (
|
|
"encoding/json"
|
|
"net/http"
|
|
|
|
"github.com/goodrain/rainbond/api/handler"
|
|
"github.com/goodrain/rainbond/api/model"
|
|
ctxutil "github.com/goodrain/rainbond/api/util/ctx"
|
|
"github.com/goodrain/rainbond/db"
|
|
"github.com/goodrain/rainbond/db/errors"
|
|
validation "github.com/goodrain/rainbond/util/endpoint"
|
|
httputil "github.com/goodrain/rainbond/util/http"
|
|
"github.com/sirupsen/logrus"
|
|
)
|
|
|
|
// ThirdPartyServiceController implements ThirdPartyServicer
|
|
type ThirdPartyServiceController struct{}
|
|
|
|
// Endpoints POST->add endpoints, PUT->update endpoints, DELETE->delete endpoints
|
|
func (t *ThirdPartyServiceController) Endpoints(w http.ResponseWriter, r *http.Request) {
|
|
switch r.Method {
|
|
case "POST":
|
|
t.addEndpoints(w, r)
|
|
case "PUT":
|
|
t.updEndpoints(w, r)
|
|
case "DELETE":
|
|
t.delEndpoints(w, r)
|
|
case "GET":
|
|
t.listEndpoints(w, r)
|
|
}
|
|
}
|
|
|
|
func (t *ThirdPartyServiceController) addEndpoints(w http.ResponseWriter, r *http.Request) {
|
|
var data model.AddEndpiontsReq
|
|
if !httputil.ValidatorRequestStructAndErrorResponse(r, w, &data, nil) {
|
|
return
|
|
}
|
|
// if address is not ip, and then it is domain
|
|
address := validation.SplitEndpointAddress(data.Address)
|
|
sid := r.Context().Value(ctxutil.ContextKey("service_id")).(string)
|
|
if validation.IsDomainNotIP(address) {
|
|
// handle domain, check can add new endpoint or not
|
|
if !canAddDomainEndpoint(sid, true) {
|
|
logrus.Warningf("new endpoint addres[%s] is domian", address)
|
|
httputil.ReturnError(r, w, 400, "do not support multi domain endpoints")
|
|
return
|
|
}
|
|
}
|
|
if !canAddDomainEndpoint(sid, false) {
|
|
// handle ip, check can add new endpoint or not
|
|
logrus.Warningf("new endpoint address[%s] is ip, but already has domain endpoint", address)
|
|
httputil.ReturnError(r, w, 400, "do not support multi domain endpoints")
|
|
return
|
|
}
|
|
|
|
if err := handler.Get3rdPartySvcHandler().AddEndpoints(sid, &data); err != nil {
|
|
if err == errors.ErrRecordAlreadyExist {
|
|
httputil.ReturnError(r, w, 400, err.Error())
|
|
return
|
|
}
|
|
httputil.ReturnError(r, w, 500, err.Error())
|
|
return
|
|
}
|
|
httputil.ReturnSuccess(r, w, "success")
|
|
}
|
|
|
|
func canAddDomainEndpoint(sid string, isDomain bool) bool {
|
|
endpoints, err := db.GetManager().EndpointsDao().List(sid)
|
|
if err != nil {
|
|
logrus.Errorf("find endpoints by sid[%s], error: %s", sid, err.Error())
|
|
return false
|
|
}
|
|
|
|
if len(endpoints) > 0 && isDomain {
|
|
return false
|
|
}
|
|
if !isDomain {
|
|
for _, ep := range endpoints {
|
|
address := validation.SplitEndpointAddress(ep.IP)
|
|
if validation.IsDomainNotIP(address) {
|
|
return false
|
|
}
|
|
}
|
|
}
|
|
return true
|
|
}
|
|
|
|
func (t *ThirdPartyServiceController) updEndpoints(w http.ResponseWriter, r *http.Request) {
|
|
var data model.UpdEndpiontsReq
|
|
if !httputil.ValidatorRequestStructAndErrorResponse(r, w, &data, nil) {
|
|
return
|
|
}
|
|
|
|
if err := handler.Get3rdPartySvcHandler().UpdEndpoints(&data); err != nil {
|
|
httputil.ReturnError(r, w, 500, err.Error())
|
|
return
|
|
}
|
|
httputil.ReturnSuccess(r, w, "success")
|
|
}
|
|
|
|
func (t *ThirdPartyServiceController) delEndpoints(w http.ResponseWriter, r *http.Request) {
|
|
var data model.DelEndpiontsReq
|
|
if !httputil.ValidatorRequestStructAndErrorResponse(r, w, &data, nil) {
|
|
return
|
|
}
|
|
sid := r.Context().Value(ctxutil.ContextKey("service_id")).(string)
|
|
if err := handler.Get3rdPartySvcHandler().DelEndpoints(data.EpID, sid); err != nil {
|
|
httputil.ReturnError(r, w, 500, err.Error())
|
|
return
|
|
}
|
|
httputil.ReturnSuccess(r, w, "success")
|
|
}
|
|
|
|
func (t *ThirdPartyServiceController) listEndpoints(w http.ResponseWriter, r *http.Request) {
|
|
sid := r.Context().Value(ctxutil.ContextKey("service_id")).(string)
|
|
res, err := handler.Get3rdPartySvcHandler().ListEndpoints(sid)
|
|
if err != nil {
|
|
httputil.ReturnError(r, w, 500, err.Error())
|
|
return
|
|
}
|
|
b, _ := json.Marshal(res)
|
|
logrus.Debugf("response endpoints: %s", string(b))
|
|
if res == nil || len(res) == 0 {
|
|
httputil.ReturnSuccess(r, w, []*model.EndpointResp{})
|
|
return
|
|
}
|
|
httputil.ReturnSuccess(r, w, res)
|
|
}
|