mirror of
https://gitee.com/rainbond/Rainbond.git
synced 2024-12-05 05:07:38 +08:00
246 lines
6.5 KiB
Go
246 lines
6.5 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 (
|
|
"io/ioutil"
|
|
"net/http"
|
|
"strconv"
|
|
"time"
|
|
|
|
"github.com/jinzhu/gorm"
|
|
|
|
"github.com/Sirupsen/logrus"
|
|
"github.com/bitly/go-simplejson"
|
|
"github.com/go-chi/chi"
|
|
"github.com/goodrain/rainbond/db"
|
|
httputil "github.com/goodrain/rainbond/util/http"
|
|
)
|
|
|
|
//Event GetLogs
|
|
func (e *TenantStruct) Event(w http.ResponseWriter, r *http.Request) {
|
|
// swagger:operation GET /v2/tenants/{tenant_name}/event v2 getevents
|
|
//
|
|
// 获取指定event_ids详细信息
|
|
//
|
|
// get events
|
|
//
|
|
// ---
|
|
// produces:
|
|
// - application/json
|
|
// - application/xml
|
|
//
|
|
// responses:
|
|
// default:
|
|
// schema:
|
|
// "$ref": "#/responses/commandResponse"
|
|
// description: 统一返回格式
|
|
b, _ := ioutil.ReadAll(r.Body)
|
|
defer r.Body.Close()
|
|
j, err := simplejson.NewJson(b)
|
|
if err != nil {
|
|
logrus.Errorf("error decode json,details %s", err.Error())
|
|
httputil.ReturnError(r, w, 400, "bad request")
|
|
return
|
|
}
|
|
eventIDS, err := j.Get("event_ids").StringArray()
|
|
if err != nil {
|
|
logrus.Errorf("error get event_id in json,details %s", err.Error())
|
|
httputil.ReturnError(r, w, 400, "bad request")
|
|
return
|
|
}
|
|
serviceEvents, err := db.GetManager().ServiceEventDao().GetEventByEventIDs(eventIDS)
|
|
if err != nil {
|
|
logrus.Warnf("can't find event by given id ,details %s", err.Error())
|
|
httputil.ReturnError(r, w, 500, err.Error())
|
|
}
|
|
httputil.ReturnSuccess(r, w, serviceEvents)
|
|
}
|
|
|
|
//GetNotificationEvents GetNotificationEvent
|
|
//support query from start and end time or all
|
|
// swagger:operation GET /v2/notificationEvent v2/notificationEvent getevents
|
|
//
|
|
// 获取数据中心通知事件
|
|
//
|
|
// get events
|
|
//
|
|
// ---
|
|
// produces:
|
|
// - application/json
|
|
// - application/xml
|
|
//
|
|
// responses:
|
|
// default:
|
|
// schema:
|
|
// "$ref": "#/responses/commandResponse"
|
|
// description: 统一返回格式
|
|
func GetNotificationEvents(w http.ResponseWriter, r *http.Request) {
|
|
var startTime, endTime time.Time
|
|
start := r.FormValue("start")
|
|
end := r.FormValue("end")
|
|
if si, err := strconv.Atoi(start); err == nil {
|
|
startTime = time.Unix(int64(si), 0)
|
|
}
|
|
if ei, err := strconv.Atoi(end); err == nil {
|
|
endTime = time.Unix(int64(ei), 0)
|
|
}
|
|
res, err := db.GetManager().NotificationEventDao().GetNotificationEventByTime(startTime, endTime)
|
|
if err != nil {
|
|
logrus.Errorf(err.Error())
|
|
httputil.ReturnError(r, w, 500, err.Error())
|
|
return
|
|
}
|
|
for _, v := range res {
|
|
service, err := db.GetManager().TenantServiceDao().GetServiceByID(v.KindID)
|
|
if err != nil {
|
|
if err == gorm.ErrRecordNotFound {
|
|
v.ServiceName = ""
|
|
v.TenantName = ""
|
|
continue
|
|
} else {
|
|
logrus.Errorf(err.Error())
|
|
httputil.ReturnError(r, w, 500, err.Error())
|
|
return
|
|
}
|
|
}
|
|
tenant, err := db.GetManager().TenantDao().GetTenantByUUID(service.TenantID)
|
|
if err != nil {
|
|
if err == gorm.ErrRecordNotFound {
|
|
v.ServiceName = ""
|
|
v.TenantName = ""
|
|
continue
|
|
} else {
|
|
logrus.Errorf(err.Error())
|
|
httputil.ReturnError(r, w, 500, err.Error())
|
|
return
|
|
}
|
|
}
|
|
v.ServiceName = service.ServiceAlias
|
|
v.TenantName = tenant.Name
|
|
}
|
|
httputil.ReturnSuccess(r, w, res)
|
|
}
|
|
|
|
|
|
//Handle Handle
|
|
// swagger:parameters handlenotify
|
|
type Handle struct {
|
|
Body struct {
|
|
//in: body
|
|
//handle message
|
|
HandleMessage string `json:"handle_message" validate:"handle_message"`
|
|
}
|
|
}
|
|
|
|
//HandleNotificationEvent HandleNotificationEvent
|
|
// swagger:operation PUT /v2/notificationEvent/{hash} v2/notificationEvent handlenotify
|
|
//
|
|
// 处理通知事件
|
|
//
|
|
// get events
|
|
//
|
|
// ---
|
|
// produces:
|
|
// - application/json
|
|
// - application/xml
|
|
//
|
|
// responses:
|
|
// default:
|
|
// schema:
|
|
// "$ref": "#/responses/commandResponse"
|
|
// description: 统一返回格式
|
|
func HandleNotificationEvent(w http.ResponseWriter, r *http.Request) {
|
|
serviceAlias := chi.URLParam(r, "serviceAlias")
|
|
if serviceAlias == "" {
|
|
httputil.ReturnError(r, w, 400, "ServiceAlias id do not empty")
|
|
return
|
|
}
|
|
var handle Handle
|
|
ok := httputil.ValidatorRequestStructAndErrorResponse(r, w, &handle.Body, nil)
|
|
if !ok {
|
|
return
|
|
}
|
|
service,err := db.GetManager().TenantServiceDao().GetServiceByServiceAlias(serviceAlias)
|
|
if err != nil {
|
|
if err == gorm.ErrRecordNotFound {
|
|
httputil.ReturnError(r, w, 404, "not found")
|
|
return
|
|
}
|
|
httputil.ReturnError(r, w, 500, err.Error())
|
|
return
|
|
}
|
|
eventList, err := db.GetManager().NotificationEventDao().GetNotificationEventByKind("service",service.ServiceID)
|
|
if err != nil {
|
|
httputil.ReturnError(r, w, 500, err.Error())
|
|
return
|
|
}
|
|
for _,event := range eventList{
|
|
event.IsHandle = true
|
|
event.HandleMessage = handle.Body.HandleMessage
|
|
err = db.GetManager().NotificationEventDao().UpdateModel(event)
|
|
if err != nil {
|
|
if err == gorm.ErrRecordNotFound {
|
|
httputil.ReturnError(r, w, 404, "not found")
|
|
return
|
|
}
|
|
httputil.ReturnError(r, w, 500, err.Error())
|
|
return
|
|
}
|
|
}
|
|
|
|
httputil.ReturnSuccess(r, w, nil)
|
|
}
|
|
|
|
//GetNotificationEvent GetNotificationEvent
|
|
// swagger:operation GET /v2/notificationEvent/{hash} v2/notificationEvent getevents
|
|
//
|
|
// 获取通知事件
|
|
//
|
|
// get events
|
|
//
|
|
// ---
|
|
// produces:
|
|
// - application/json
|
|
// - application/xml
|
|
//
|
|
// responses:
|
|
// default:
|
|
// schema:
|
|
// "$ref": "#/responses/commandResponse"
|
|
// description: 统一返回格式
|
|
func GetNotificationEvent(w http.ResponseWriter, r *http.Request) {
|
|
|
|
hash := chi.URLParam(r, "hash")
|
|
if hash == "" {
|
|
httputil.ReturnError(r, w, 400, "hash id do not empty")
|
|
return
|
|
}
|
|
event, err := db.GetManager().NotificationEventDao().GetNotificationEventByHash(hash)
|
|
if err != nil {
|
|
if err == gorm.ErrRecordNotFound {
|
|
httputil.ReturnError(r, w, 404, "not found")
|
|
return
|
|
}
|
|
httputil.ReturnError(r, w, 500, err.Error())
|
|
return
|
|
}
|
|
httputil.ReturnSuccess(r, w, event)
|
|
}
|