Rainbond/db/mysql/dao/application.go

package dao

import (
	"github.com/goodrain/rainbond/api/util/bcode"
	"github.com/goodrain/rainbond/db/model"
	"github.com/jinzhu/gorm"
	"github.com/pkg/errors"
	"github.com/sirupsen/logrus"
)

// ApplicationDaoImpl -
type ApplicationDaoImpl struct {
	DB *gorm.DB
}

//AddModel -
func (a *ApplicationDaoImpl) AddModel(mo model.Interface) error {
	appReq, _ := mo.(*model.Application)
	var oldApp model.Application
	if err := a.DB.Where("tenant_id = ? AND app_id = ?", appReq.TenantID, appReq.AppID).Find(&oldApp).Error; err != nil {
		if err == gorm.ErrRecordNotFound {
			return a.DB.Create(appReq).Error
		}
		return err
	}
	return bcode.ErrApplicationExist
}

//UpdateModel -
func (a *ApplicationDaoImpl) UpdateModel(mo model.Interface) error {
	updateReq := mo.(*model.Application)
	return a.DB.Save(updateReq).Error
}

// ListApps -
func (a *ApplicationDaoImpl) ListApps(tenantID, appName string, page, pageSize int) ([]*model.Application, int64, error) {
	var datas []*model.Application
	offset := (page - 1) * pageSize

	db := a.DB.Where("tenant_id=?", tenantID).Order("create_time desc")
	if appName != "" {
		db = db.Where("app_name like ?", "%"+appName+"%")
	}
	var total int64
	if err := db.Model(&model.Application{}).Count(&total).Error; err != nil {
		return nil, 0, err
	}
	if err := db.Limit(pageSize).Offset(offset).Find(&datas).Error; err != nil {
		return nil, 0, err
	}
	return datas, total, nil
}

// GetAppByID -
func (a *ApplicationDaoImpl) GetAppByID(appID string) (*model.Application, error) {
	var app model.Application
	if err := a.DB.Where("app_id=?", appID).Find(&app).Error; err != nil {
		if err == gorm.ErrRecordNotFound {
			return nil, bcode.ErrApplicationNotFound
		}
		return nil, err
	}
	return &app, nil
}

// GetByServiceID -
func (a *ApplicationDaoImpl) GetByServiceID(sid string) (*model.Application, error) {
	var app model.Application
	if err := a.DB.Where("app_id = ?", a.DB.Table("tenant_services").Select("app_id").Where("service_id=?", sid).SubQuery()).Find(&app).Error; err != nil {
		if err == gorm.ErrRecordNotFound {
			return nil, bcode.ErrApplicationNotFound
		}
		return nil, err
	}
	return &app, nil
}

// DeleteApp Delete application By appID -
func (a *ApplicationDaoImpl) DeleteApp(appID string) error {
	var app model.Application
	if err := a.DB.Where("app_id=?", appID).Find(&app).Error; err != nil {
		if err == gorm.ErrRecordNotFound {
			return bcode.ErrApplicationNotFound
		}
		return err
	}
	return a.DB.Delete(&app).Error
}

// ListByAppIDs -
func (a *ApplicationDaoImpl) ListByAppIDs(appIDs []string) ([]*model.Application, error) {
	var datas []*model.Application
	if err := a.DB.Where("app_id in (?)", appIDs).Find(&datas).Error; err != nil {
		return nil, errors.Wrap(err, "list app by app_ids")
	}
	return datas, nil
}

// IsK8sAppDuplicate Verify whether the k8s app under the same team are duplicate
func (a *ApplicationDaoImpl) IsK8sAppDuplicate(tenantID, AppID, k8sApp string) bool {
	var count int64
	if err := a.DB.Model(&model.Application{}).Where("tenant_id=? and app_id <>? and k8s_app=?", tenantID, AppID, k8sApp).Count(&count).Error; err != nil {
		logrus.Errorf("judge K8s App Duplicate failed %v", err)
		return true
	}
	return count > 0
}

//GetAppByName -
func (a *ApplicationDaoImpl) GetAppByName(tenantID, appName, k8sAppName string) (*model.Application, error) {
	var app model.Application
	if err := a.DB.Where("tenant_id=? and app_name=? and k8s_app=?", tenantID, appName, k8sAppName).Find(&app).Error; err != nil {
		if err == gorm.ErrRecordNotFound {
			return nil, bcode.ErrApplicationNotFound
		}
		return nil, err
	}
	return &app, nil
}