Rainbond/pkg/api/controller/resources.go

1730 lines
46 KiB
Go
Raw Normal View History

2017-11-07 11:40:44 +08:00
// RAINBOND, Application Management Platform
// Copyright (C) 2014-2017 Goodrain Co., Ltd.
2017-11-07 11:40:44 +08:00
// 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.
2017-11-07 11:40:44 +08:00
// 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.
2017-11-07 11:40:44 +08:00
// 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"
"io/ioutil"
"net/http"
"os"
"strconv"
"strings"
"github.com/goodrain/rainbond/pkg/api/middleware"
api_model "github.com/goodrain/rainbond/pkg/api/model"
dbmodel "github.com/goodrain/rainbond/pkg/db/model"
2017-11-07 11:40:44 +08:00
"github.com/pquerna/ffjson/ffjson"
"github.com/go-chi/chi"
"github.com/jinzhu/gorm"
validator "github.com/thedevsaddam/govalidator"
"github.com/goodrain/rainbond/pkg/api/handler"
httputil "github.com/goodrain/rainbond/pkg/util/http"
"github.com/Sirupsen/logrus"
"sort"
2017-12-06 17:18:26 +08:00
"github.com/renstorm/fuzzysearch/fuzzy"
2017-11-07 11:40:44 +08:00
)
//V2Routes v2Routes
type V2Routes struct {
TenantStruct
AcpNodeStruct
EntranceStruct
EventLogStruct
OpentsdbStruct
}
//Show test
func (v2 *V2Routes) Show(w http.ResponseWriter, r *http.Request) {
// swagger:operation GET /v2/show v2 getApiVersion
//
// 显示当前的api version 信息
//
// show api version
//
// ---
// produces:
// - application/json
// - application/xml
// responses:
// default:
// schema:
// "$ref": "#/responses/commandResponse"
// description: 统一返回格式
w.Write([]byte("v2 urls"))
}
//TenantStruct tenant struct
type TenantStruct struct{}
//AllTenantResources GetResources
func (t *TenantStruct) AllTenantResources(w http.ResponseWriter, r *http.Request) {
tenants, err := handler.GetTenantManager().GetTenants()
if err != nil {
msg := httputil.ResponseBody{
Msg: fmt.Sprintf("get tenant error, %v", err),
}
httputil.Return(r, w, 500, msg)
}
ts := &api_model.TotalStatsInfo{}
for _, tenant := range tenants {
services, err := handler.GetServiceManager().GetService(tenant.UUID)
if err != nil {
msg := httputil.ResponseBody{
Msg: fmt.Sprintf("get service error, %v", err),
}
httputil.Return(r, w, 500, msg)
}
statsInfo, _ := handler.GetTenantManager().StatsMemCPU(services)
statsInfo.UUID = tenant.UUID
logrus.Debugf("stats info is %v", statsInfo)
ts.Data = append(ts.Data, statsInfo)
}
httputil.ReturnSuccess(r, w, ts.Data)
}
//TenantResources TenantResources
func (t *TenantStruct) TenantResources(w http.ResponseWriter, r *http.Request) {
// swagger:operation POST /v2/resources/tenants v2 tenantResources
//
2017-11-27 18:01:04 +08:00
// 租户资源使用情况
2017-11-07 11:40:44 +08:00
//
// get tenant resources
//
// ---
// produces:
// - application/json
// - application/xml
//
// responses:
// default:
// schema:
// "$ref": "#/responses/commandResponse"
// description: 统一返回格式
var tr api_model.TenantResources
ok := httputil.ValidatorRequestStructAndErrorResponse(r, w, &tr.Body, nil)
if !ok {
return
}
rep, err := handler.GetTenantManager().GetTenantsResources(&tr)
if err != nil {
httputil.ReturnError(r, w, 500, fmt.Sprintf("get resources error, %v", err))
return
}
httputil.ReturnSuccess(r, w, rep)
return
}
2017-12-06 17:09:49 +08:00
//TenantsQuery TenantsQuery
func (t *TenantStruct) TenantsQuery(w http.ResponseWriter, r *http.Request) {
// swagger:operation GET /v2/tenants/query/{tenant_name} v2 tenants
//
// 租户带资源列表
//
// get tenant resources
//
// ---
// produces:
// - application/json
// - application/xml
//
2017-12-07 11:07:51 +08:00
// parameters:
// - name: tenant_name
// in: path
// description: '123'
// required: true
// type: string
// format: string
//
2017-12-06 17:09:49 +08:00
// responses:
// default:
// schema:
// "$ref": "#/responses/commandResponse"
// description: 统一返回格式
2017-12-07 11:07:51 +08:00
rules := validator.MapData{
"tenant_name": []string{"required"},
}
data, ok := httputil.ValidatorRequestMapAndErrorResponse(r, w, rules, nil)
if !ok {
return
}
tenantName := data["tenant_name"].(string)
2017-12-06 17:09:49 +08:00
rep, err := handler.GetTenantManager().GetTenantsName()
if err != nil {
httputil.ReturnError(r, w, 500, fmt.Sprintf("get tenants names error, %v", err))
return
}
result:=fuzzy.Find(tenantName, rep) // [cartwheel wheel]
httputil.ReturnSuccess(r, w, result)
return
}
2017-12-07 11:44:56 +08:00
//TenantsGetByName TenantsGetByName
func (t *TenantStruct) TenantsGetByName(w http.ResponseWriter, r *http.Request) {
// swagger:operation GET /v2/tenants/{tenant_name}/res v2 tenants
//
// 租户带资源单个
//
// get tenant resources
//
// ---
// produces:
// - application/json
// - application/xml
//
// parameters:
// - name: tenant_name
// in: path
// description: '123'
// required: true
// type: string
// format: string
//
// responses:
// default:
// schema:
// "$ref": "#/responses/commandResponse"
// description: 统一返回格式
2017-12-07 12:14:01 +08:00
tenantName := chi.URLParam(r, "tenant_name")
2017-12-07 11:44:56 +08:00
v, err := handler.GetTenantManager().GetTenantsByName(tenantName)
if err != nil {
httputil.ReturnError(r, w, 500, fmt.Sprintf("get tenants names error, %v", err))
return
}
2017-12-07 13:42:45 +08:00
logrus.Infof("query tenant from db by name %s ,got %v",tenantName,v)
2017-12-07 13:47:25 +08:00
var res =&api_model.TenantResource{}
2017-12-07 11:44:56 +08:00
services, err := handler.GetServiceManager().GetService(v.UUID)
if err != nil {
httputil.ReturnError(r, w, 500, fmt.Sprintf("get services by tenantID %s error, %v",v.UUID, err))
return
}
totalResInfo, _ := handler.GetTenantManager().TotalMemCPU(services)
usedResInfo, _ := handler.GetTenantManager().StatsMemCPU(services)
res.UUID=v.UUID
res.Name=v.Name
res.EID=v.EID
res.AllocatedCPU=totalResInfo.CPU
res.AllocatedMEM=totalResInfo.MEM
res.UsedCPU=usedResInfo.CPU
res.UsedMEM=usedResInfo.MEM
httputil.ReturnSuccess(r, w, res)
return
}
2017-12-06 15:26:21 +08:00
//TenantsWithResource TenantsWithResource
func (t *TenantStruct) TenantsWithResource(w http.ResponseWriter, r *http.Request) {
2017-12-07 11:07:51 +08:00
// swagger:operation GET /v2/resources/tenants/res/page/{curPage}/size/{pageLen} v2 PagedTenantResList
2017-12-06 15:26:21 +08:00
//
// 租户带资源列表
//
2017-12-07 11:07:51 +08:00
// get paged tenant resources
2017-12-06 15:26:21 +08:00
//
// ---
// produces:
// - application/json
// - application/xml
//
2017-12-07 11:07:51 +08:00
// parameters:
// - name: curPage
// in: path
// description: '123'
// required: true
// type: string
// format: string
// - name: pageLen
// in: path
// description: '25'
// required: true
// type: string
// format: string
//
2017-12-06 15:26:21 +08:00
// responses:
// default:
// schema:
// "$ref": "#/responses/commandResponse"
// description: 统一返回格式
2017-12-06 17:51:33 +08:00
2017-12-07 11:07:51 +08:00
rules := validator.MapData{
"curPage": []string{"required"},
"pageLen": []string{"required"},
}
data, ok := httputil.ValidatorRequestMapAndErrorResponse(r, w, rules, nil)
if !ok {
return
}
2017-12-06 17:51:33 +08:00
2017-12-07 11:07:51 +08:00
pageLenStr := data["pageLen"].(string)
curPageStr := data["curPage"].(string)
2017-12-06 17:51:33 +08:00
2017-12-06 23:51:34 +08:00
pageLen,err:=strconv.Atoi(pageLenStr)
if err != nil {
httputil.ReturnError(r, w, 400, fmt.Sprintf("bad request, %v", err))
return
}
curPage,err:=strconv.Atoi(curPageStr)
if err != nil {
httputil.ReturnError(r, w, 400, fmt.Sprintf("bad request, %v", err))
return
}
2017-12-06 15:26:21 +08:00
rep, err := handler.GetTenantManager().GetTenants()
if err != nil {
httputil.ReturnError(r, w, 500, fmt.Sprintf("get tenants error, %v", err))
return
}
var result []*api_model.TenantResource
for _,v:=range rep{
services, err := handler.GetServiceManager().GetService(v.UUID)
if err != nil {
httputil.ReturnError(r, w, 500, fmt.Sprintf("get services by tenantID %s error, %v",v.UUID, err))
return
}
totalResInfo, _ := handler.GetTenantManager().TotalMemCPU(services)
usedResInfo, _ := handler.GetTenantManager().StatsMemCPU(services)
var res api_model.TenantResource
res.UUID=v.UUID
res.Name=v.Name
res.EID=v.EID
res.AllocatedCPU=totalResInfo.CPU
res.AllocatedMEM=totalResInfo.MEM
res.UsedCPU=usedResInfo.CPU
res.UsedMEM=usedResInfo.MEM
result=append(result,&res)
}
pList := api_model.TenantResList(result)
sort.Sort(pList)
2017-12-07 00:35:44 +08:00
var resultList []*api_model.TenantResource
if curPage*pageLen<len(rep) {
resultList=pList[(curPage-1)*pageLen:curPage*pageLen]
}else{
resultList=pList[(curPage-1)*pageLen:len(rep)]
}
2017-12-06 23:51:34 +08:00
var ret api_model.PagedTenantResList
ret.List=resultList
2017-12-07 00:14:38 +08:00
ret.Length=len(resultList)
2017-12-06 23:51:34 +08:00
httputil.ReturnSuccess(r, w, ret)
2017-12-06 15:26:21 +08:00
return
}
2017-11-07 11:40:44 +08:00
//SumTenants 统计租户数量
func (t *TenantStruct) SumTenants(w http.ResponseWriter, r *http.Request) {
// swagger:operation GET /v2/resources/tenants/sum v2 sumTenants
//
2017-11-27 18:01:04 +08:00
// 获取租户数量
2017-11-07 11:40:44 +08:00
//
// get tenant resources
//
// ---
// produces:
// - application/json
// - application/xml
//
// responses:
// default:
// schema:
// "$ref": "#/responses/commandResponse"
// description: 统一返回格式
s, err := handler.GetTenantManager().TenantsSum()
if err != nil {
httputil.ReturnError(r, w, 500, fmt.Sprintf("sum tenants error, %v", err))
return
}
rc := make(map[string]int)
rc["num"] = s
httputil.ReturnSuccess(r, w, rc)
}
//Tenant Tenant
func (t *TenantStruct) Tenant(w http.ResponseWriter, r *http.Request) {
switch r.Method {
case "POST":
t.AddTenant(w, r)
}
}
//AddTenant AddTenant
func (t *TenantStruct) AddTenant(w http.ResponseWriter, r *http.Request) {
// swagger:operation POST /v2/tenants v2 addTenant
//
2017-11-27 18:01:04 +08:00
// 添加租户信息
2017-11-07 11:40:44 +08:00
//
// add tenant
//
// ---
// consumes:
// - application/json
// - application/x-protobuf
//
// produces:
// - application/json
// - application/xml
//
// responses:
// default:
// schema:
// "$ref": "#/responses/commandResponse"
// description: 统一返回格式
var ts api_model.AddTenantStruct
ok := httputil.ValidatorRequestStructAndErrorResponse(r, w, &ts.Body, nil)
if !ok {
return
}
var dbts dbmodel.Tenants
if ts.Body.Eid != "" {
//新接口
//TODO:生成tenant_id and tenant_name
id, name, errN := handler.GetServiceManager().CreateTenandIDAndName(ts.Body.Eid)
if errN != nil {
httputil.ReturnError(r, w, 500, fmt.Sprintf("create tenant error, %v", errN))
return
}
dbts.EID = ts.Body.Eid
dbts.Name = name
dbts.UUID = id
if err := handler.GetServiceManager().CreateTenant(&dbts); err != nil {
if strings.HasSuffix(err.Error(), "is exist") {
httputil.ReturnError(r, w, 400, err.Error())
return
}
httputil.ReturnError(r, w, 500, fmt.Sprintf("create tenant error, %v", err))
return
}
rc := make(map[string]string)
rc["tenant_id"] = id
rc["tenang_name"] = name
httputil.ReturnSuccess(r, w, rc)
return
}
if ts.Body.TenantID != "" && ts.Body.TenantName != "" {
//兼容旧接口
dbts.Name = ts.Body.TenantName
dbts.UUID = ts.Body.TenantID
if err := handler.GetServiceManager().CreateTenant(&dbts); err != nil {
if strings.HasSuffix(err.Error(), "is exist") {
httputil.ReturnError(r, w, 400, err.Error())
return
}
httputil.ReturnError(r, w, 500, fmt.Sprintf("create tenant error, %v", err))
return
}
httputil.ReturnSuccess(r, w, nil)
return
}
if ts.Body.Eid == "" && (ts.Body.TenantID == "" || ts.Body.TenantName == "") {
httputil.ReturnError(r, w, 400, "args error")
return
}
httputil.ReturnError(r, w, 400, "args error, need eid or tenatn_id / tenant_name")
return
}
//DeleteTenant DeleteTenant
func (t *TenantStruct) DeleteTenant(w http.ResponseWriter, r *http.Request) {
w.Write([]byte("delete tenant"))
}
//UpdateTenant UpdateTenant
func (t *TenantStruct) UpdateTenant(w http.ResponseWriter, r *http.Request) {
w.Write([]byte("update tenant"))
}
//ServicesInfo GetServiceInfo
func (t *TenantStruct) ServicesInfo(w http.ResponseWriter, r *http.Request) {
// swagger:operation GET /v2/tenants/{tenant_name}/services v2 getServiceInfo
//
// 获取租户所有应用信息
//
2017-11-27 18:01:04 +08:00
// get services info in tenant
2017-11-07 11:40:44 +08:00
//
// ---
// consumes:
// - application/json
// - application/x-protobuf
//
// produces:
// - application/json
// - application/xml
//
// responses:
// default:
// schema:
// "$ref": "#/responses/commandResponse"
// description: 统一返回格式
tenantID := r.Context().Value(middleware.ContextKey("tenant_id")).(string)
services, err := handler.GetServiceManager().GetService(tenantID)
if err != nil {
httputil.ReturnError(r, w, 500, "get tenant services error")
return
}
httputil.ReturnSuccess(r, w, services)
return
}
//CreateService create Service
func (t *TenantStruct) CreateService(w http.ResponseWriter, r *http.Request) {
// swagger:operation POST /v2/tenants/{tenant_name}/services v2 createService
//
// 创建应用
//
2017-11-27 18:01:04 +08:00
// create service
2017-11-07 11:40:44 +08:00
//
// ---
// consumes:
// - application/json
// - application/x-protobuf
//
// produces:
// - application/json
// - application/xml
//
// responses:
// default:
// schema:
// "$ref": "#/responses/commandResponse"
// description: 统一返回格式
logrus.Debugf("trans create service service")
var ss api_model.ServiceStruct
body, err := ioutil.ReadAll(r.Body)
if err != nil {
httputil.ReturnError(r, w, 500, err.Error())
return
}
err = ffjson.Unmarshal(body, &ss)
if err != nil {
httputil.ReturnError(r, w, 500, err.Error())
return
}
logrus.Debugf("data is %v", ss)
tenantID := r.Context().Value(middleware.ContextKey("tenant_id")).(string)
ss.TenantID = tenantID
logrus.Debugf("begin to create service")
if err := handler.GetServiceManager().ServiceCreate(&ss); err != nil {
httputil.ReturnError(r, w, 500, fmt.Sprintf("create service error, %v", err))
return
}
httputil.ReturnSuccess(r, w, nil)
}
//UpdateService create Service
func (t *TenantStruct) UpdateService(w http.ResponseWriter, r *http.Request) {
// swagger:operation PUT /v2/tenants/{tenant_name}/services/{service_alias} v2 updateService
//
// 应用更新
//
2017-11-27 18:01:04 +08:00
// update service
2017-11-07 11:40:44 +08:00
//
// ---
// consumes:
// - application/json
// - application/x-protobuf
//
// produces:
// - application/json
// - application/xml
//
// responses:
// default:
// schema:
// "$ref": "#/responses/commandResponse"
// description: 统一返回格式
logrus.Debugf("trans update service service")
//目前提供三个元素的修改
rules := validator.MapData{
"container_cmd": []string{},
"image_name": []string{},
"container_memory": []string{},
}
//对应前方三个元素
//mm := []string{"image_name", "container_cmd", "container_memory"}
data, ok := httputil.ValidatorRequestMapAndErrorResponse(r, w, rules, nil)
if !ok {
return
}
serviceID := r.Context().Value(middleware.ContextKey("service_id")).(string)
data["service_id"] = serviceID
logrus.Debugf("begin to update service")
if err := handler.GetServiceManager().ServiceUpdate(data); err != nil {
httputil.ReturnError(r, w, 500, fmt.Sprintf("update service error, %v", err))
return
}
httputil.ReturnSuccess(r, w, nil)
}
//SetLanguage SetLanguage
func (t *TenantStruct) SetLanguage(w http.ResponseWriter, r *http.Request) {
// swagger:operation POST /v2/tenants/{tenant_name}/services/{service_alias}/language v2 setLanguage
//
// 设置应用语言
//
2017-11-27 18:01:04 +08:00
// set language
2017-11-07 11:40:44 +08:00
//
// ---
// consumes:
// - application/json
// - application/x-protobuf
//
// produces:
// - application/json
// - application/xml
//
// responses:
// default:
// schema:
// "$ref": "#/responses/commandResponse"
// description: 统一返回格式
logrus.Debugf("trans language set service")
rules := validator.MapData{
"language": []string{"required"},
}
langS := &api_model.LanguageSet{}
data, ok := httputil.ValidatorRequestMapAndErrorResponse(r, w, rules, nil)
if !ok {
return
}
langS.Language = data["language"].(string)
langS.ServiceID = r.Context().Value(middleware.ContextKey("service_id")).(string)
if err := handler.GetServiceManager().LanguageSet(langS); err != nil {
httputil.ReturnError(r, w, 500, "set language error.")
return
}
httputil.ReturnSuccess(r, w, nil)
return
}
//StatusService StatusService
func (t *TenantStruct) StatusService(w http.ResponseWriter, r *http.Request) {
// swagger:operation GET /v2/tenants/{tenant_name}/services/{service_alias}/status v2 serviceStatus
//
// 获取应用状态
//
// get service status
//
// ---
// consumes:
// - application/json
// - application/x-protobuf
//
// produces:
// - application/json
// - application/xml
//
// responses:
// default:
// schema:
// "$ref": "#/responses/commandResponse"
// description: 统一返回格式
serviceID := r.Context().Value(middleware.ContextKey("service_id")).(string)
statusList, err := handler.GetServiceManager().GetStatus(serviceID)
if err != nil {
httputil.ReturnError(r, w, 500, fmt.Sprintf("get service list error,%v", err))
return
}
httputil.ReturnSuccess(r, w, statusList)
return
}
//PostStatusService PostStatusService
func (t *TenantStruct) PostStatusService(w http.ResponseWriter, r *http.Request) {
logrus.Debugf("in status service serviceID")
}
//StatusServiceList service list status
func (t *TenantStruct) StatusServiceList(w http.ResponseWriter, r *http.Request) {
// swagger:operation POST /v2/tenants/{tenant_name}/services_status v2 serviceStatuslist
//
// 获取应用状态
//
// get service statuslist
//
// ---
// consumes:
// - application/json
// - application/x-protobuf
//
// produces:
// - application/json
// - application/xml
//
// responses:
// default:
// schema:
// "$ref": "#/responses/commandResponse"
// description: 统一返回格式
var services api_model.StatusServiceListStruct
ok := httputil.ValidatorRequestStructAndErrorResponse(r, w, &services.Body, nil)
if !ok {
return
}
//logrus.Info(services.Body.ServiceIDs)
serviceList := services.Body.ServiceIDs
tenantID := r.Context().Value(middleware.ContextKey("tenant_id")).(string)
statusList, err := handler.GetServiceManager().GetServicesStatus(tenantID, serviceList)
if err != nil {
if err.Error() == gorm.ErrRecordNotFound.Error() {
httputil.ReturnError(r, w, 404, err.Error())
return
}
httputil.ReturnError(r, w, 500, err.Error())
return
}
var info = make([]map[string]string, 0)
for _, s := range statusList {
info = append(info, map[string]string{"service_id": s.ServiceID, "status": s.Status, "status_cn": TransStatus(s.Status)})
}
httputil.ReturnSuccess(r, w, info)
}
//ServiceLabel label
func (t *TenantStruct) ServiceLabel(w http.ResponseWriter, r *http.Request) {
switch r.Method {
case "PUT":
t.UpdateServiceLabel(w, r)
case "POST":
t.AddServiceLabel(w, r)
}
}
//AddServiceLabel AddServiceLabel
func (t *TenantStruct) AddServiceLabel(w http.ResponseWriter, r *http.Request) {
// swagger:operation POST /v2/tenants/{tenant_name}/services/{service_alias}/service-label v2 addServiceLabel
//
// 添加应用标签
//
2017-11-27 18:01:04 +08:00
// add service label
2017-11-07 11:40:44 +08:00
//
// ---
// consumes:
// - application/json
// - application/x-protobuf
//
// produces:
// - application/json
// - application/xml
//
// responses:
// default:
// schema:
// "$ref": "#/responses/commandResponse"
// description: 统一返回格式
rules := validator.MapData{
"label_values": []string{"required"},
}
data, ok := httputil.ValidatorRequestMapAndErrorResponse(r, w, rules, nil)
if !ok {
return
}
var valueList []string
valueList = append(valueList, data["label_values"].(string))
serviceID := r.Context().Value(middleware.ContextKey("service_id")).(string)
if err := handler.GetServiceManager().AddLabel("service", serviceID, valueList); err != nil {
httputil.ReturnError(r, w, 500, fmt.Sprintf("add service label error, %v", err))
return
}
httputil.ReturnSuccess(r, w, nil)
}
//UpdateServiceLabel UpdateServiceLabel
func (t *TenantStruct) UpdateServiceLabel(w http.ResponseWriter, r *http.Request) {
// swagger:operation PUT /v2/tenants/{tenant_name}/services/{service_alias}/service-label v2 updateServiceLabel
//
// 更新应用标签
//
2017-11-27 18:01:04 +08:00
// delete service label
2017-11-07 11:40:44 +08:00
//
// ---
// consumes:
// - application/json
// - application/x-protobuf
//
// produces:
// - application/json
// - application/xml
//
// responses:
// default:
// schema:
// "$ref": "#/responses/commandResponse"
// description: 统一返回格式
rules := validator.MapData{
"label_values": []string{"required"},
}
data, ok := httputil.ValidatorRequestMapAndErrorResponse(r, w, rules, nil)
if !ok {
return
}
value := data["label_values"].(string)
serviceID := r.Context().Value(middleware.ContextKey("service_id")).(string)
if err := handler.GetServiceManager().UpdateServiceLabel(serviceID, value); err != nil {
httputil.ReturnError(r, w, 500, fmt.Sprintf("update service label error, %v", err))
return
}
httputil.ReturnSuccess(r, w, nil)
}
//NodeLabel label
func (t *TenantStruct) NodeLabel(w http.ResponseWriter, r *http.Request) {
switch r.Method {
case "DELETE":
t.DeleteNodeLabel(w, r)
case "POST":
t.AddNodeLabel(w, r)
}
}
//AddNodeLabel AddNodeLabel
func (t *TenantStruct) AddNodeLabel(w http.ResponseWriter, r *http.Request) {
// swagger:operation POST /v2/tenants/{tenant_name}/services/{service_alias}/node-label v2 addNodeLabel
//
// 添加节点标签
//
2017-11-27 18:01:04 +08:00
// add node label
2017-11-07 11:40:44 +08:00
//
// ---
// consumes:
// - application/json
// - application/x-protobuf
//
// produces:
// - application/json
// - application/xml
//
// responses:
// default:
// schema:
// "$ref": "#/responses/commandResponse"
// description: 统一返回格式
var labels api_model.AddNodeLabelStruct
ok := httputil.ValidatorRequestStructAndErrorResponse(r, w, &labels.Body, nil)
if !ok {
return
}
//logrus.Info(labels.Body.LabelValues)
valueList := labels.Body.LabelValues
serviceID := r.Context().Value(middleware.ContextKey("service_id")).(string)
if err := handler.GetServiceManager().AddLabel("node", serviceID, valueList); err != nil {
httputil.ReturnError(r, w, 500, fmt.Sprintf("add node label failure, %v", err))
return
}
httputil.ReturnSuccess(r, w, 200)
}
//DeleteNodeLabel DeleteLabel
func (t *TenantStruct) DeleteNodeLabel(w http.ResponseWriter, r *http.Request) {
// swagger:operation DELETE /v2/tenants/{tenant_name}/services/{service_alias}/node-label v2 deleteNodeLabel
//
// 删除节点标签
//
2017-11-27 18:01:04 +08:00
// delete node label
2017-11-07 11:40:44 +08:00
//
// ---
// consumes:
// - application/json
// - application/x-protobuf
//
// produces:
// - application/json
// - application/xml
//
// responses:
// default:
// schema:
// "$ref": "#/responses/commandResponse"
// description: 统一返回格式
var labels api_model.AddNodeLabelStruct
ok := httputil.ValidatorRequestStructAndErrorResponse(r, w, &labels.Body, nil)
if !ok {
return
}
//logrus.Info(labels.Body.LabelValues)
valueList := labels.Body.LabelValues
serviceID := r.Context().Value(middleware.ContextKey("service_id")).(string)
if err := handler.GetServiceManager().DeleteLabel("node", serviceID, valueList); err != nil {
httputil.ReturnError(r, w, 500, fmt.Sprintf("delete node label failure, %v", err))
return
}
httputil.ReturnSuccess(r, w, nil)
}
//StatusContainerID StatusContainerID
func (t *TenantStruct) StatusContainerID(w http.ResponseWriter, r *http.Request) {
w.Write([]byte("status container IDs list"))
}
//SingleServiceInfo SingleServiceInfo
func (t *TenantStruct) SingleServiceInfo(w http.ResponseWriter, r *http.Request) {
switch r.Method {
case "DELETE":
t.DeleteSingleServiceInfo(w, r)
case "GET":
t.GetSingleServiceInfo(w, r)
}
}
//GetSingleServiceInfo GetSingleServiceInfo
func (t *TenantStruct) GetSingleServiceInfo(w http.ResponseWriter, r *http.Request) {
// swagger:operation GET /v2/tenants/{tenant_name}/services/{service_alias} v2 getService
//
// 获取应用信息
//
2017-11-27 18:01:04 +08:00
// get service info
2017-11-07 11:40:44 +08:00
//
// ---
// consumes:
// - application/json
// - application/x-protobuf
//
// produces:
// - application/json
// - application/xml
//
// responses:
// default:
// schema:
// "$ref": "#/responses/commandResponse"
// description: 统一返回格式
tenantID := r.Context().Value(middleware.ContextKey("tenant_id")).(string)
serviceID := r.Context().Value(middleware.ContextKey("service_id")).(string)
tenantName := r.Context().Value(middleware.ContextKey("tenant_name")).(string)
serviceName := r.Context().Value(middleware.ContextKey("service_alias")).(string)
result := make(map[string]string)
result["tenantName"] = tenantName
result["serviceAlias"] = serviceName
result["tenantId"] = tenantID
result["serviceId"] = serviceID
httputil.ReturnSuccess(r, w, result)
}
//DeleteSingleServiceInfo DeleteService
// swagger:operation DELETE /v2/tenants/{tenant_name}/services/{service_alias} v2 deleteService
//
// 删除应用
//
// delete service
//
// ---
// consumes:
// - application/json
// - application/x-protobuf
//
// produces:
// - application/json
// - application/xml
//
// responses:
// default:
// schema:
// "$ref": "#/responses/commandResponse"
// description: 统一返回格式
2017-11-07 11:40:44 +08:00
func (t *TenantStruct) DeleteSingleServiceInfo(w http.ResponseWriter, r *http.Request) {
serviceID := r.Context().Value(middleware.ContextKey("service_id")).(string)
if err := handler.GetServiceManager().TransServieToDelete(serviceID); err != nil {
if err == fmt.Errorf("unclosed") {
httputil.ReturnError(r, w, 400, fmt.Sprintf("Service must be closed"))
return
}
httputil.ReturnError(r, w, 500, fmt.Sprintf("delete service error, %v", err))
return
}
httputil.ReturnSuccess(r, w, nil)
}
//Dependency Dependency
func (t *TenantStruct) Dependency(w http.ResponseWriter, r *http.Request) {
switch r.Method {
case "DELETE":
t.DeleteDependency(w, r)
case "POST":
t.AddDependency(w, r)
}
}
//AddDependency AddDependency
// swagger:operation POST /v2/tenants/{tenant_name}/services/{service_alias}/dependency v2 addDependency
//
// 增加应用依赖关系
//
2017-11-27 18:01:04 +08:00
// add dependency
//
// ---
// consumes:
// - application/json
// - application/x-protobuf
//
// produces:
// - application/json
// - application/xml
//
// responses:
// default:
// schema:
// "$ref": "#/responses/commandResponse"
// description: 统一返回格式
2017-11-07 11:40:44 +08:00
func (t *TenantStruct) AddDependency(w http.ResponseWriter, r *http.Request) {
rules := validator.MapData{
"dep_service_id": []string{"required"},
"dep_service_type": []string{"required"},
"dep_order": []string{},
}
data, ok := httputil.ValidatorRequestMapAndErrorResponse(r, w, rules, nil)
if !ok {
return
}
ds := &api_model.DependService{
TenantID: r.Context().Value(middleware.ContextKey("tenant_id")).(string),
ServiceID: r.Context().Value(middleware.ContextKey("service_id")).(string),
DepServiceID: data["dep_service_id"].(string),
DepServiceType: data["dep_service_type"].(string),
}
if err := handler.GetServiceManager().ServiceDepend("add", ds); err != nil {
httputil.ReturnError(r, w, 500, fmt.Sprintf("add dependency error, %v", err))
return
}
httputil.ReturnSuccess(r, w, nil)
}
//DeleteDependency DeleteDependency
// swagger:operation DELETE /v2/tenants/{tenant_name}/services/{service_alias}/dependency v2 deleteDependency
//
// 删除应用依赖关系
//
2017-11-27 18:01:04 +08:00
// delete dependency
//
// ---
// consumes:
// - application/json
// - application/x-protobuf
//
// produces:
// - application/json
// - application/xml
//
// responses:
// default:
// schema:
// "$ref": "#/responses/commandResponse"
// description: 统一返回格式
2017-11-07 11:40:44 +08:00
func (t *TenantStruct) DeleteDependency(w http.ResponseWriter, r *http.Request) {
logrus.Debugf("trans delete depend service ")
rules := validator.MapData{
"dep_service_id": []string{"required"},
"dep_service_type": []string{},
"dep_order": []string{},
}
data, ok := httputil.ValidatorRequestMapAndErrorResponse(r, w, rules, nil)
if !ok {
return
}
ds := &api_model.DependService{
TenantID: r.Context().Value(middleware.ContextKey("tenant_id")).(string),
ServiceID: r.Context().Value(middleware.ContextKey("service_id")).(string),
DepServiceID: data["dep_service_id"].(string),
}
if err := handler.GetServiceManager().ServiceDepend("delete", ds); err != nil {
httputil.ReturnError(r, w, 500, fmt.Sprintf("delete dependency error, %v", err))
return
}
httputil.ReturnSuccess(r, w, nil)
}
//Env Env
func (t *TenantStruct) Env(w http.ResponseWriter, r *http.Request) {
switch r.Method {
case "DELETE":
t.DeleteEnv(w, r)
case "POST":
t.AddEnv(w, r)
}
}
//AddEnv AddEnv
// swagger:operation POST /v2/tenants/{tenant_name}/services/{service_alias}/env v2 addEnv
//
// 增加环境变量
//
2017-11-27 18:01:04 +08:00
// add env var
//
// ---
// consumes:
// - application/json
// - application/x-protobuf
//
// produces:
// - application/json
// - application/xml
//
// responses:
// default:
// schema:
// "$ref": "#/responses/commandResponse"
// description: 统一返回格式
2017-11-07 11:40:44 +08:00
func (t *TenantStruct) AddEnv(w http.ResponseWriter, r *http.Request) {
var envM api_model.AddTenantServiceEnvVar
if !httputil.ValidatorRequestStructAndErrorResponse(r, w, &envM, nil) {
return
}
tenantID := r.Context().Value(middleware.ContextKey("tenant_id")).(string)
serviceID := r.Context().Value(middleware.ContextKey("service_id")).(string)
var envD dbmodel.TenantServiceEnvVar
envD.AttrName = envM.AttrName
envD.AttrValue = envM.AttrValue
envD.TenantID = tenantID
envD.ServiceID = serviceID
envD.ContainerPort = envM.ContainerPort
envD.IsChange = envM.IsChange
envD.Name = envM.Name
envD.Scope = envM.Scope
if err := handler.GetServiceManager().EnvAttr("add", &envD); err != nil {
logrus.Errorf("Add env error, %v", err)
httputil.ReturnError(r, w, 500, fmt.Sprintf("Add env error, %v", err))
return
}
httputil.ReturnSuccess(r, w, nil)
}
//DeleteEnv DeleteEnv
// swagger:operation DELETE /v2/tenants/{tenant_name}/services/{service_alias}/env v2 deleteEnv
//
// 删除环境变量
//
2017-11-27 18:01:04 +08:00
// delete env var
//
// ---
// consumes:
// - application/json
// - application/x-protobuf
//
// produces:
// - application/json
// - application/xml
//
// responses:
// default:
// schema:
// "$ref": "#/responses/commandResponse"
// description: 统一返回格式
2017-11-07 11:40:44 +08:00
func (t *TenantStruct) DeleteEnv(w http.ResponseWriter, r *http.Request) {
var envM api_model.DelTenantServiceEnvVar
if !httputil.ValidatorRequestStructAndErrorResponse(r, w, &envM, nil) {
return
}
tenantID := r.Context().Value(middleware.ContextKey("tenant_id")).(string)
serviceID := r.Context().Value(middleware.ContextKey("service_id")).(string)
envM.TenantID = tenantID
envM.ServiceID = serviceID
var envD dbmodel.TenantServiceEnvVar
envD.AttrName = envM.AttrName
envD.AttrValue = envM.AttrValue
envD.TenantID = tenantID
envD.ServiceID = serviceID
envD.ContainerPort = envM.ContainerPort
envD.IsChange = envM.IsChange
envD.Name = envM.Name
envD.Scope = envM.Scope
if err := handler.GetServiceManager().EnvAttr("delete", &envD); err != nil {
logrus.Errorf("delete env error, %v", err)
if err.Error() == gorm.ErrRecordNotFound.Error() {
httputil.ReturnError(r, w, 404, "service port "+err.Error())
return
}
httputil.ReturnError(r, w, 500, fmt.Sprintf("Delete env error, %v", err))
return
}
httputil.ReturnSuccess(r, w, nil)
}
//Ports 应用端口控制器
func (t *TenantStruct) Ports(w http.ResponseWriter, r *http.Request) {
switch r.Method {
case "DELETE":
t.deletePortController(w, r)
case "POST":
t.addPortController(w, r)
case "PUT":
t.updatePortController(w, r)
}
}
//PutPorts PortVar
// swagger:operation PUT /v2/tenants/{tenant_name}/services/{service_alias}/ports v2 updatePort
//
// 更新应用端口信息(旧)
//
// update port
//
// ---
// consumes:
// - application/json
// - application/x-protobuf
//
// produces:
// - application/json
// - application/xml
//
// responses:
// default:
// schema:
// "$ref": "#/responses/commandResponse"
// description: 统一返回格式
func (t *TenantStruct) PutPorts(w http.ResponseWriter, r *http.Request) {
tenantID := r.Context().Value(middleware.ContextKey("tenant_id")).(string)
serviceID := r.Context().Value(middleware.ContextKey("service_id")).(string)
var ports api_model.ServicePorts
if ok := httputil.ValidatorRequestStructAndErrorResponse(r, w, &ports, nil); !ok {
return
}
if err := handler.GetServiceManager().PortVar("update", tenantID, serviceID, &ports, 0); err != nil {
logrus.Errorf("update port error. %v", err)
httputil.ReturnError(r, w, 500, err.Error())
return
}
httputil.ReturnSuccess(r, w, nil)
}
2017-11-07 11:40:44 +08:00
//AddPortVar PortVar
// swagger:operation POST /v2/tenants/{tenant_name}/services/{service_alias}/ports v2 addPort
//
// 增加应用端口,默认关闭对内和对外选项,需要开启使用相应接口
//
2017-11-27 18:01:04 +08:00
// add port
//
// ---
// consumes:
// - application/json
// - application/x-protobuf
//
// produces:
// - application/json
// - application/xml
//
// responses:
// default:
// schema:
// "$ref": "#/responses/commandResponse"
// description: 统一返回格式
2017-11-07 11:40:44 +08:00
func (t *TenantStruct) addPortController(w http.ResponseWriter, r *http.Request) {
tenantID := r.Context().Value(middleware.ContextKey("tenant_id")).(string)
serviceID := r.Context().Value(middleware.ContextKey("service_id")).(string)
var ports api_model.ServicePorts
if ok := httputil.ValidatorRequestStructAndErrorResponse(r, w, &ports, nil); !ok {
return
}
if err := handler.GetServiceManager().PortVar("add", tenantID, serviceID, &ports, 0); err != nil {
2017-11-07 11:40:44 +08:00
logrus.Errorf("add port error. %v", err)
httputil.ReturnError(r, w, 500, err.Error())
return
}
httputil.ReturnSuccess(r, w, ports.Port)
}
//UpdatePortVar PortVar
// swagger:operation PUT /v2/tenants/{tenant_name}/services/{service_alias}/ports/{port} v2 updatePort
//
// 更新应用端口信息
//
2017-11-27 18:01:04 +08:00
// update port
//
// ---
// consumes:
// - application/json
// - application/x-protobuf
//
// produces:
// - application/json
// - application/xml
//
// responses:
// default:
// schema:
// "$ref": "#/responses/commandResponse"
// description: 统一返回格式
2017-11-07 11:40:44 +08:00
func (t *TenantStruct) updatePortController(w http.ResponseWriter, r *http.Request) {
tenantID := r.Context().Value(middleware.ContextKey("tenant_id")).(string)
serviceID := r.Context().Value(middleware.ContextKey("service_id")).(string)
portStr := chi.URLParam(r, "port")
oldPort, err := strconv.Atoi(portStr)
if err != nil {
httputil.ReturnError(r, w, 400, "port must be a number")
return
}
2017-11-07 11:40:44 +08:00
var ports api_model.ServicePorts
if ok := httputil.ValidatorRequestStructAndErrorResponse(r, w, &ports, nil); !ok {
return
}
if err := handler.GetServiceManager().PortVar("update", tenantID, serviceID, &ports, oldPort); err != nil {
2017-11-07 11:40:44 +08:00
logrus.Errorf("update port error. %v", err)
httputil.ReturnError(r, w, 500, err.Error())
return
}
httputil.ReturnSuccess(r, w, nil)
}
//DeletePortVar PortVar
// swagger:operation DELETE /v2/tenants/{tenant_name}/services/{service_alias}/ports/{port} v2 deletePort
//
// 删除端口变量
//
// delete port
//
// ---
// Consumes:
// - application/json
// - application/x-protobuf
//
// produces:
// - application/json
// - application/xml
//
// responses:
// default:
// schema:
// "$ref": "#/responses/commandResponse"
// description: 统一返回格式
func (t *TenantStruct) deletePortController(w http.ResponseWriter, r *http.Request) {
tenantID := r.Context().Value(middleware.ContextKey("tenant_id")).(string)
serviceID := r.Context().Value(middleware.ContextKey("service_id")).(string)
portStr := chi.URLParam(r, "port")
oldPort, err := strconv.Atoi(portStr)
if err != nil {
httputil.ReturnError(r, w, 400, "port must be a number")
return
}
var port = &api_model.TenantServicesPort{
TenantID: tenantID,
ServiceID: serviceID,
}
var ports api_model.ServicePorts
ports.Port = append(ports.Port, port)
if err := handler.GetServiceManager().PortVar("delete", tenantID, serviceID, &ports, oldPort); err != nil {
if err.Error() == gorm.ErrRecordNotFound.Error() {
httputil.ReturnError(r, w, 404, "port can not found")
return
}
httputil.ReturnError(r, w, 500, err.Error())
return
}
httputil.ReturnSuccess(r, w, nil)
}
2017-11-07 11:40:44 +08:00
//PortOuterController 开关端口对外服务
// swagger:operation PUT /v2/tenants/{tenant_name}/services/{service_alias}/ports/{port}/outer v2 PortOuterController
//
// 开关端口对外服务,应用无需重启自动生效
//
2017-11-27 18:01:04 +08:00
// add port
//
// ---
// Consumes:
// - application/json
// - application/x-protobuf
//
// produces:
// - application/json
// - application/xml
//
// responses:
// default:
// schema:
// "$ref": "#/responses/commandResponse"
// description: 统一返回格式
2017-11-07 11:40:44 +08:00
func (t *TenantStruct) PortOuterController(w http.ResponseWriter, r *http.Request) {
var data api_model.ServicePortInnerOrOuter
if !httputil.ValidatorRequestStructAndErrorResponse(r, w, &(data.Body), nil) {
return
}
serviceID := r.Context().Value(middleware.ContextKey("service_id")).(string)
tenantName := r.Context().Value(middleware.ContextKey("tenant_name")).(string)
portStr := chi.URLParam(r, "port")
containerPort, err := strconv.Atoi(portStr)
if err != nil {
httputil.ReturnError(r, w, 400, "port must be a number")
return
}
vsPort, protocol, errV := handler.GetServiceManager().PortOuter(tenantName, serviceID, data.Body.Operation, containerPort)
if errV != nil {
if strings.HasSuffix(errV.Error(), gorm.ErrRecordNotFound.Error()) {
httputil.ReturnError(r, w, 404, errV.Error())
return
}
httputil.ReturnError(r, w, 500, errV.Error())
return
}
rc := make(map[string]string)
domain := os.Getenv("EX_DOMAIN")
if domain == "" {
httputil.ReturnError(r, w, 500, "have no EX_DOMAIN")
return
}
mm := strings.Split(domain, ":")
if protocol == "http" || protocol == "https" {
rc["domain"] = mm[0]
if len(mm) == 2 {
rc["port"] = mm[1]
} else {
rc["port"] = "10080"
}
} else if vsPort != nil && vsPort.Port != 0 {
rc["domain"] = mm[0]
rc["port"] = fmt.Sprintf("%v", vsPort.Port)
}
httputil.ReturnSuccess(r, w, rc)
}
//PortInnerController 开关端口对内服务
// swagger:operation PUT /v2/tenants/{tenant_name}/services/{service_alias}/ports/{port}/inner v2 PortInnerController
//
// 开关对内服务,应用无需重启,自动生效
//
2017-11-27 18:01:04 +08:00
// add port
//
// ---
// Consumes:
// - application/json
// - application/x-protobuf
//
// produces:
// - application/json
// - application/xml
//
// responses:
// default:
// schema:
// "$ref": "#/responses/commandResponse"
// description: 统一返回格式
2017-11-07 11:40:44 +08:00
func (t *TenantStruct) PortInnerController(w http.ResponseWriter, r *http.Request) {
var data api_model.ServicePortInnerOrOuter
if !httputil.ValidatorRequestStructAndErrorResponse(r, w, &(data.Body), nil) {
return
}
serviceID := r.Context().Value(middleware.ContextKey("service_id")).(string)
tenantName := r.Context().Value(middleware.ContextKey("tenant_name")).(string)
portStr := chi.URLParam(r, "port")
containerPort, err := strconv.Atoi(portStr)
if err != nil {
httputil.ReturnError(r, w, 400, "port must be a number")
return
}
if err := handler.GetServiceManager().PortInner(tenantName, serviceID, data.Body.Operation, containerPort); err != nil {
if err.Error() == gorm.ErrRecordNotFound.Error() {
httputil.ReturnError(r, w, 404, "service port "+err.Error())
return
} else if err.Error() == "already open" || err.Error() == "already close" {
httputil.Return(r, w, 200, httputil.ResponseBody{Msg: err.Error()})
return
} else {
httputil.ReturnError(r, w, 500, err.Error())
return
}
}
httputil.ReturnSuccess(r, w, nil)
}
//Pods pods
// swagger:operation GET /v2/tenants/{tenant_name}/services/{service_alias}/pods v2 getPodsInfo
//
// 获取pods信息
//
2017-11-27 18:01:04 +08:00
// get pods info
//
// ---
// consumes:
// - application/json
// - application/x-protobuf
//
// produces:
// - application/json
// - application/xml
//
// responses:
// default:
// schema:
// "$ref": "#/responses/commandResponse"
// description: 统一返回格式
2017-11-07 11:40:44 +08:00
func (t *TenantStruct) Pods(w http.ResponseWriter, r *http.Request) {
serviceID := r.Context().Value(middleware.ContextKey("service_id")).(string)
pods, err := handler.GetServiceManager().GetPods(serviceID)
if err != nil {
if err.Error() == gorm.ErrRecordNotFound.Error() {
httputil.ReturnError(r, w, 404, fmt.Sprintf("get pods error, %v", err))
return
}
httputil.ReturnError(r, w, 500, fmt.Sprintf("get pods error, %v", err))
return
}
httputil.ReturnSuccess(r, w, pods)
}
//Probe probe
func (t *TenantStruct) Probe(w http.ResponseWriter, r *http.Request) {
switch r.Method {
case "PUT":
t.UpdateProbe(w, r)
case "DELETE":
t.DeleteProbe(w, r)
case "POST":
t.AddProbe(w, r)
}
}
//AddProbe add probe
func (t *TenantStruct) AddProbe(w http.ResponseWriter, r *http.Request) {
// swagger:operation POST /v2/tenants/{tenant_name}/services/{service_alias}/probe v2 addProbe
//
// 增加应用探针
//
2017-11-27 18:01:04 +08:00
// add probe
2017-11-07 11:40:44 +08:00
//
// ---
// consumes:
// - application/json
// - application/x-protobuf
//
// produces:
// - application/json
// - application/xml
//
// responses:
// default:
// schema:
// "$ref": "#/responses/commandResponse"
// description: 统一返回格式
logrus.Debugf("trans add probe dependency service ")
serviceID := r.Context().Value(middleware.ContextKey("service_id")).(string)
var tsp api_model.ServiceProbe
if ok := httputil.ValidatorRequestStructAndErrorResponse(r, w, &tsp, nil); !ok {
return
}
var tspD dbmodel.ServiceProbe
tspD.ServiceID = serviceID
tspD.Cmd = tsp.Cmd
tspD.FailureThreshold = tsp.FailureThreshold
tspD.HTTPHeader = tsp.HTTPHeader
tspD.InitialDelaySecond = tsp.InitialDelaySecond
tspD.IsUsed = tsp.IsUsed
tspD.Mode = tsp.Mode
tspD.Path = tsp.Path
tspD.PeriodSecond = tsp.PeriodSecond
tspD.Port = tsp.Port
tspD.ProbeID = tsp.ProbeID
tspD.Scheme = tsp.Scheme
tspD.SuccessThreshold = tsp.SuccessThreshold
tspD.TimeoutSecond = tsp.TimeoutSecond
//注意端口问题
if err := handler.GetServiceManager().ServiceProbe(&tspD, "add"); err != nil {
httputil.ReturnError(r, w, 500, fmt.Sprintf("add service probe error, %v", err))
return
}
}
//UpdateProbe update probe
func (t *TenantStruct) UpdateProbe(w http.ResponseWriter, r *http.Request) {
// swagger:operation PUT /v2/tenants/{tenant_name}/services/{service_alias}/probe v2 updateProbe
//
// 更新应用探针信息, *注意此处为全量更新
//
2017-11-27 18:01:04 +08:00
// update probe
2017-11-07 11:40:44 +08:00
//
// ---
// consumes:
// - application/json
// - application/x-protobuf
//
// produces:
// - application/json
// - application/xml
//
// responses:
// default:
// schema:
// "$ref": "#/responses/commandResponse"
// description: 统一返回格式
logrus.Debugf("trans update probe dependency service ")
serviceID := r.Context().Value(middleware.ContextKey("service_id")).(string)
var tsp api_model.ServiceProbe
if ok := httputil.ValidatorRequestStructAndErrorResponse(r, w, &tsp, nil); !ok {
return
}
var tspD dbmodel.ServiceProbe
tspD.ServiceID = serviceID
tspD.Cmd = tsp.Cmd
tspD.FailureThreshold = tsp.FailureThreshold
tspD.HTTPHeader = tsp.HTTPHeader
tspD.InitialDelaySecond = tsp.InitialDelaySecond
tspD.IsUsed = tsp.IsUsed
tspD.Mode = tsp.Mode
tspD.Path = tsp.Path
tspD.PeriodSecond = tsp.PeriodSecond
tspD.Port = tsp.Port
tspD.ProbeID = tsp.ProbeID
tspD.Scheme = tsp.Scheme
tspD.SuccessThreshold = tsp.SuccessThreshold
tspD.TimeoutSecond = tsp.TimeoutSecond
tspD.ServiceID = serviceID
//注意端口问题
if err := handler.GetServiceManager().ServiceProbe(&tspD, "update"); err != nil {
if err.Error() == gorm.ErrRecordNotFound.Error() {
httputil.ReturnError(r, w, 404, fmt.Sprintf("update prob error, %v", err))
return
}
httputil.ReturnError(r, w, 500, fmt.Sprintf("update service probe error, %v", err))
return
}
}
//DeleteProbe delete probe
func (t *TenantStruct) DeleteProbe(w http.ResponseWriter, r *http.Request) {
// swagger:operation DELETE /v2/tenants/{tenant_name}/services/{service_alias}/probe v2 deleteProbe
//
// 删除应用探针
//
2017-11-27 18:01:04 +08:00
// delete probe
2017-11-07 11:40:44 +08:00
//
// ---
// consumes:
// - application/json
// - application/x-protobuf
//
// produces:
// - application/json
// - application/xml
//
// responses:
// default:
// schema:
// "$ref": "#/responses/commandResponse"
// description: 统一返回格式
logrus.Debugf("trans delete probe dependency service ")
serviceID := r.Context().Value(middleware.ContextKey("service_id")).(string)
var tsp api_model.ServiceProbe
if ok := httputil.ValidatorRequestStructAndErrorResponse(r, w, &tsp, nil); !ok {
return
}
var tspD dbmodel.ServiceProbe
tspD.ServiceID = serviceID
tspD.ProbeID = tsp.ProbeID
//注意端口问题
if err := handler.GetServiceManager().ServiceProbe(&tspD, "delete"); err != nil {
if err.Error() == gorm.ErrRecordNotFound.Error() {
httputil.ReturnError(r, w, 404, fmt.Sprintf("delete prob error, %v", err))
return
}
httputil.ReturnError(r, w, 500, fmt.Sprintf("delete service probe error, %v", err))
return
}
}
//Port Port
func (t *TenantStruct) Port(w http.ResponseWriter, r *http.Request) {
switch r.Method {
case "PUT":
t.UpdatePort(w, r)
case "DELETE":
t.DeletePort(w, r)
case "POST":
t.AddPort(w, r)
}
}
//AddPort add port
func (t *TenantStruct) AddPort(w http.ResponseWriter, r *http.Request) {
}
//DeletePort delete port
func (t *TenantStruct) DeletePort(w http.ResponseWriter, r *http.Request) {
}
//UpdatePort Update port
func (t *TenantStruct) UpdatePort(w http.ResponseWriter, r *http.Request) {
}
//SingleTenantResources SingleTenantResources
func (t *TenantStruct) SingleTenantResources(w http.ResponseWriter, r *http.Request) {
// swagger:operation GET /v2/tenants/{tenant_name}/resources v2 singletenantResources
//
2017-11-27 18:01:04 +08:00
// 指定租户资源使用情况
2017-11-07 11:40:44 +08:00
//
// get tenant resources
//
// ---
// produces:
// - application/json
// - application/xml
// parameters:
// - name: tenant_name
// in: path
// description: tenant name
// required: true
// type: string
//
// responses:
// default:
// schema:
// "$ref": "#/responses/commandResponse"
// description: 统一返回格式
tenantID := r.Context().Value(middleware.ContextKey("tenant_id")).(string)
//11ms
services, err := handler.GetServiceManager().GetService(tenantID)
if err != nil {
msg := httputil.ResponseBody{
Msg: fmt.Sprintf("get service error, %v", err),
}
httputil.Return(r, w, 500, msg)
}
//19ms
statsInfo, _ := handler.GetTenantManager().StatsMemCPU(services)
//900ms
statsInfo.UUID = tenantID
logrus.Debugf("stats info is %v", statsInfo)
httputil.ReturnSuccess(r, w, statsInfo)
return
}