2018-05-09 15:51:04 +08:00
|
|
|
// RAINBOND, Application Management Platform
|
|
|
|
// Copyright (C) 2014-2017 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 (
|
2019-08-21 15:07:20 +08:00
|
|
|
"fmt"
|
2018-05-09 15:51:04 +08:00
|
|
|
"net/http"
|
2019-08-30 20:06:32 +08:00
|
|
|
"strings"
|
2018-05-09 15:51:04 +08:00
|
|
|
|
2019-08-21 15:07:20 +08:00
|
|
|
"github.com/go-chi/chi"
|
2018-11-22 14:33:29 +08:00
|
|
|
"github.com/goodrain/rainbond/api/handler"
|
2021-06-17 22:59:05 +08:00
|
|
|
ctxutil "github.com/goodrain/rainbond/api/util/ctx"
|
2019-09-03 14:19:49 +08:00
|
|
|
"github.com/goodrain/rainbond/db"
|
|
|
|
"github.com/goodrain/rainbond/db/model"
|
2018-05-09 15:51:04 +08:00
|
|
|
httputil "github.com/goodrain/rainbond/util/http"
|
2019-09-03 14:19:49 +08:00
|
|
|
"github.com/goodrain/rainbond/worker/server"
|
2020-11-25 16:39:38 +08:00
|
|
|
"github.com/sirupsen/logrus"
|
2018-05-09 15:51:04 +08:00
|
|
|
)
|
|
|
|
|
2019-08-21 15:07:20 +08:00
|
|
|
// PodController is an implementation of PodInterface
|
|
|
|
type PodController struct{}
|
|
|
|
|
2018-05-09 15:51:04 +08:00
|
|
|
//Pods get some service pods
|
|
|
|
// swagger:operation GET /v2/tenants/{tenant_name}/pods v2/tenants pods
|
|
|
|
//
|
|
|
|
// 获取一些应用的Pod信息
|
|
|
|
//
|
|
|
|
// get some service pods
|
|
|
|
//
|
|
|
|
// ---
|
|
|
|
// consumes:
|
|
|
|
// - application/json
|
|
|
|
// - application/x-protobuf
|
|
|
|
//
|
|
|
|
// produces:
|
|
|
|
// - application/json
|
|
|
|
// - application/xml
|
|
|
|
//
|
|
|
|
// responses:
|
|
|
|
// default:
|
|
|
|
// schema:
|
|
|
|
// "$ref": "#/responses/commandResponse"
|
|
|
|
// description: get some service pods
|
|
|
|
func Pods(w http.ResponseWriter, r *http.Request) {
|
2019-08-30 20:06:32 +08:00
|
|
|
serviceIDs := strings.Split(r.FormValue("service_ids"), ",")
|
|
|
|
if serviceIDs == nil || len(serviceIDs) == 0 {
|
2021-06-17 22:59:05 +08:00
|
|
|
tenant := r.Context().Value(ctxutil.ContextKey("tenant")).(*model.Tenants)
|
2019-08-30 20:06:32 +08:00
|
|
|
services, _ := db.GetManager().TenantServiceDao().GetServicesByTenantID(tenant.UUID)
|
|
|
|
for _, s := range services {
|
|
|
|
serviceIDs = append(serviceIDs, s.ServiceID)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
var allpods []*handler.K8sPodInfo
|
2020-07-29 19:08:01 +08:00
|
|
|
podinfo, err := handler.GetServiceManager().GetMultiServicePods(serviceIDs)
|
|
|
|
if err != nil {
|
|
|
|
logrus.Errorf("get service pod failure %s", err.Error())
|
|
|
|
}
|
|
|
|
if podinfo != nil {
|
2019-09-03 14:19:49 +08:00
|
|
|
var pods []*handler.K8sPodInfo
|
|
|
|
if podinfo.OldPods != nil {
|
|
|
|
pods = append(podinfo.NewPods, podinfo.OldPods...)
|
|
|
|
} else {
|
|
|
|
pods = podinfo.NewPods
|
|
|
|
}
|
|
|
|
for _, pod := range pods {
|
2019-08-30 20:06:32 +08:00
|
|
|
allpods = append(allpods, pod)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
httputil.ReturnSuccess(r, w, allpods)
|
2018-05-09 15:51:04 +08:00
|
|
|
}
|
2019-08-21 15:07:20 +08:00
|
|
|
|
2021-05-26 10:00:36 +08:00
|
|
|
// PodNums reutrns the number of pods for components.
|
|
|
|
func PodNums(w http.ResponseWriter, r *http.Request) {
|
|
|
|
componentIDs := strings.Split(r.FormValue("service_ids"), ",")
|
|
|
|
podNums, err := handler.GetServiceManager().GetComponentPodNums(r.Context(), componentIDs)
|
|
|
|
if err != nil {
|
|
|
|
httputil.ReturnBcodeError(r, w, err)
|
|
|
|
return
|
|
|
|
}
|
|
|
|
httputil.ReturnSuccess(r, w, podNums)
|
|
|
|
}
|
|
|
|
|
2019-08-21 15:07:20 +08:00
|
|
|
// PodDetail -
|
|
|
|
func (p *PodController) PodDetail(w http.ResponseWriter, r *http.Request) {
|
|
|
|
podName := chi.URLParam(r, "pod_name")
|
2021-06-23 10:07:06 +08:00
|
|
|
namespace := r.Context().Value(ctxutil.ContextKey("tenant_id")).(string)
|
2021-04-19 11:52:52 +08:00
|
|
|
pd, err := handler.GetPodHandler().PodDetail(namespace, podName)
|
2019-08-21 15:07:20 +08:00
|
|
|
if err != nil {
|
|
|
|
logrus.Errorf("error getting pod detail: %v", err)
|
2019-08-23 07:33:26 +08:00
|
|
|
if err == server.ErrPodNotFound {
|
|
|
|
httputil.ReturnError(r, w, 404, fmt.Sprintf("error getting pod detail: %v", err))
|
|
|
|
return
|
|
|
|
}
|
2019-08-21 15:07:20 +08:00
|
|
|
httputil.ReturnError(r, w, 500, fmt.Sprintf("error getting pod detail: %v", err))
|
|
|
|
return
|
|
|
|
}
|
|
|
|
httputil.ReturnSuccess(r, w, pd)
|
|
|
|
}
|