mirror of
https://gitee.com/rainbond/Rainbond.git
synced 2024-12-05 05:07:38 +08:00
115 lines
3.2 KiB
Go
115 lines
3.2 KiB
Go
// 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 dao
|
|
|
|
import (
|
|
"time"
|
|
|
|
"github.com/goodrain/rainbond/pkg/db/model"
|
|
"github.com/jinzhu/gorm"
|
|
|
|
"encoding/json"
|
|
|
|
"github.com/Sirupsen/logrus"
|
|
)
|
|
|
|
//AddModel AddModel
|
|
func (c *EventDaoImpl) AddModel(mo model.Interface) error {
|
|
result := mo.(*model.ServiceEvent)
|
|
var oldResult model.ServiceEvent
|
|
if ok := c.DB.Where("event_id=?", result.EventID).Find(&oldResult).RecordNotFound(); ok {
|
|
if err := c.DB.Create(result).Error; err != nil {
|
|
return err
|
|
}
|
|
} else {
|
|
logrus.Infoln("event result is exist")
|
|
return c.UpdateModel(mo)
|
|
}
|
|
return nil
|
|
}
|
|
|
|
//UpdateModel UpdateModel
|
|
func (c *EventDaoImpl) UpdateModel(mo model.Interface) error {
|
|
result := mo.(*model.ServiceEvent)
|
|
|
|
var oldResult model.ServiceEvent
|
|
if ok := c.DB.Where("event_id=?", result.EventID).Find(&oldResult).RecordNotFound(); !ok {
|
|
finalUpdateEvent(result, &oldResult)
|
|
oldB, _ := json.Marshal(oldResult)
|
|
logrus.Infof("update event to %s", string(oldB))
|
|
if err := c.DB.Save(oldResult).Error; err != nil {
|
|
return err
|
|
}
|
|
}
|
|
return nil
|
|
}
|
|
func finalUpdateEvent(target *model.ServiceEvent, old *model.ServiceEvent) {
|
|
if target.CodeVersion != "" {
|
|
old.CodeVersion = target.CodeVersion
|
|
}
|
|
if target.OptType != "" {
|
|
old.OptType = target.OptType
|
|
}
|
|
|
|
if target.Status != "" {
|
|
old.Status = target.Status
|
|
}
|
|
if target.Message != "" {
|
|
old.Message = target.Message
|
|
}
|
|
old.FinalStatus = "complete"
|
|
if target.FinalStatus != "" {
|
|
old.FinalStatus = target.FinalStatus
|
|
}
|
|
|
|
old.EndTime = time.Now().String()
|
|
if old.Status == "failure" && old.OptType == "callback" {
|
|
old.DeployVersion = old.OldDeployVersion
|
|
}
|
|
}
|
|
|
|
//EventDaoImpl EventLogMessageDaoImpl
|
|
type EventDaoImpl struct {
|
|
DB *gorm.DB
|
|
}
|
|
|
|
//GetEventByEventID get event log message
|
|
func (c *EventDaoImpl) GetEventByEventID(eventID string) (*model.ServiceEvent, error) {
|
|
var result model.ServiceEvent
|
|
if err := c.DB.Where("event_id=?", eventID).Find(&result).Error; err != nil {
|
|
if err == gorm.ErrRecordNotFound {
|
|
//return messageRaw, nil
|
|
}
|
|
return nil, err
|
|
}
|
|
return &result, nil
|
|
}
|
|
|
|
//GetEventByServiceID get event log message
|
|
func (c *EventDaoImpl) GetEventByServiceID(serviceID string) ([]*model.ServiceEvent, error) {
|
|
var result []*model.ServiceEvent
|
|
if err := c.DB.Where("service_id=?", serviceID).Find(&result).Order("start_time DESC").Error; err != nil {
|
|
if err == gorm.ErrRecordNotFound {
|
|
//return messageRaw, nil
|
|
}
|
|
return nil, err
|
|
}
|
|
return result, nil
|
|
}
|