2017-11-07 11:40:44 +08:00
// RAINBOND, Application Management Platform
// Copyright (C) 2014-2017 Goodrain Co., Ltd.
2017-11-07 19:11:49 +08:00
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 19:11:49 +08:00
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 19:11:49 +08:00
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 dao
import (
"fmt"
"os"
"strconv"
"time"
2017-11-07 19:11:49 +08:00
"github.com/goodrain/rainbond/pkg/db/model"
2017-11-07 11:40:44 +08:00
"github.com/Sirupsen/logrus"
"github.com/jinzhu/gorm"
)
//TenantDaoImpl 租户信息管理
type TenantDaoImpl struct {
DB * gorm . DB
}
//AddModel 添加租户
func ( t * TenantDaoImpl ) AddModel ( mo model . Interface ) error {
tenant := mo . ( * model . Tenants )
var oldTenant model . Tenants
if ok := t . DB . Where ( "uuid = ? or name=?" , tenant . UUID , tenant . Name ) . Find ( & oldTenant ) . RecordNotFound ( ) ; ok {
if err := t . DB . Create ( tenant ) . Error ; err != nil {
return err
}
} else {
return fmt . Errorf ( "tenant uuid %s is exist" , tenant . UUID )
}
return nil
}
//UpdateModel 更新租户
func ( t * TenantDaoImpl ) UpdateModel ( mo model . Interface ) error {
tenant := mo . ( * model . Tenants )
if err := t . DB . Save ( tenant ) . Error ; err != nil {
return err
}
return nil
}
//GetTenantByUUID 获取租户
func ( t * TenantDaoImpl ) GetTenantByUUID ( uuid string ) ( * model . Tenants , error ) {
var tenant model . Tenants
if err := t . DB . Where ( "uuid = ?" , uuid ) . Find ( & tenant ) . Error ; err != nil {
return nil , err
}
return & tenant , nil
}
//GetTenantIDByName 获取租户
func ( t * TenantDaoImpl ) GetTenantIDByName ( name string ) ( * model . Tenants , error ) {
var tenant model . Tenants
if err := t . DB . Where ( "name = ?" , name ) . Find ( & tenant ) . Error ; err != nil {
return nil , err
}
return & tenant , nil
}
//GetALLTenants GetALLTenants
func ( t * TenantDaoImpl ) GetALLTenants ( ) ( [ ] * model . Tenants , error ) {
var tenants [ ] * model . Tenants
if err := t . DB . Find ( & tenants ) . Error ; err != nil {
return nil , err
}
return tenants , nil
}
//TenantServicesDaoImpl 租户应用dao
type TenantServicesDaoImpl struct {
DB * gorm . DB
}
//AddModel 添加租户应用
func ( t * TenantServicesDaoImpl ) AddModel ( mo model . Interface ) error {
service := mo . ( * model . TenantServices )
var oldService model . TenantServices
if ok := t . DB . Where ( "service_alias = ? and tenant_id=?" , service . ServiceAlias , service . TenantID ) . Find ( & oldService ) . RecordNotFound ( ) ; ok {
if err := t . DB . Create ( service ) . Error ; err != nil {
return err
}
} else {
return fmt . Errorf ( "service name %s and is exist in tenant %s" , service . ServiceAlias , service . TenantID )
}
return nil
}
//UpdateModel 更新租户应用
func ( t * TenantServicesDaoImpl ) UpdateModel ( mo model . Interface ) error {
service := mo . ( * model . TenantServices )
if err := t . DB . Save ( service ) . Error ; err != nil {
return err
}
return nil
}
//GetServiceByID 获取服务通过服务id
func ( t * TenantServicesDaoImpl ) GetServiceByID ( serviceID string ) ( * model . TenantServices , error ) {
var service model . TenantServices
if err := t . DB . Where ( "service_id=?" , serviceID ) . Find ( & service ) . Error ; err != nil {
return nil , err
}
return & service , nil
}
//GetCPUAndMEM GetCPUAndMEM
func ( t * TenantServicesDaoImpl ) GetCPUAndMEM ( tenantName [ ] string ) ( [ ] * map [ string ] interface { } , error ) {
if len ( tenantName ) == 0 {
rows , err := t . DB . Raw ( "select sum(container_cpu) as cpu,sum(container_memory * replicas) as memory from tenant_services where service_id in (select service_id from tenant_service_status where status != 'closed' && status != 'undeploy')" ) . Rows ( )
if err != nil {
return nil , err
}
defer rows . Close ( )
var cpu int
var mem int
for rows . Next ( ) {
rows . Scan ( & cpu , & mem )
}
var rc [ ] * map [ string ] interface { }
res := make ( map [ string ] interface { } )
res [ "cpu" ] = cpu
res [ "memory" ] = mem
rc = append ( rc , & res )
return rc , nil
}
var rc [ ] * map [ string ] interface { }
for _ , tenant := range tenantName {
rows , err := t . DB . Raw ( "select tenant_id, sum(container_cpu) as cpu, sum(container_memory * replicas) as memory from tenant_services where service_id in (select service_id from tenant_service_status where (status != 'closed' && status != 'undeploy') && service_id in (select service_id from tenant_services where domain = (?))) group by tenant_id" , tenant ) . Rows ( )
if err != nil {
return nil , err
}
defer rows . Close ( )
for rows . Next ( ) {
var cpu int
var mem int
var id string
rows . Scan ( & id , & cpu , & mem )
res := make ( map [ string ] interface { } )
res [ "cpu" ] = cpu
res [ "memory" ] = mem
res [ "tenant_id" ] = id
logrus . Infof ( "res is $v" , res )
rc = append ( rc , & res )
}
}
return rc , nil
}
//GetServiceAliasByIDs 获取应用别名
func ( t * TenantServicesDaoImpl ) GetServiceAliasByIDs ( uids [ ] string ) ( [ ] * model . TenantServices , error ) {
var services [ ] * model . TenantServices
if err := t . DB . Where ( "service_id in (?)" , uids ) . Select ( "service_alias,service_id" ) . Find ( & services ) . Error ; err != nil {
if err == gorm . ErrRecordNotFound {
return services , nil
}
return nil , err
}
return services , nil
}
//GetServiceByTenantIDAndServiceAlias 根据租户名和服务名
func ( t * TenantServicesDaoImpl ) GetServiceByTenantIDAndServiceAlias ( tenantID , serviceName string ) ( * model . TenantServices , error ) {
var service model . TenantServices
if err := t . DB . Where ( "service_alias = ? and tenant_id=?" , serviceName , tenantID ) . Find ( & service ) . Error ; err != nil {
return nil , err
}
return & service , nil
}
//GetServicesByTenantID GetServicesByTenantID
func ( t * TenantServicesDaoImpl ) GetServicesByTenantID ( tenantID string ) ( [ ] * model . TenantServices , error ) {
var services [ ] * model . TenantServices
if err := t . DB . Where ( "tenant_id= ?" , tenantID ) . Select ( "tenant_id,service_alias,service_id,replica_id" ) . Find ( & services ) . Error ; err != nil {
if err == gorm . ErrRecordNotFound {
return services , nil
}
return nil , err
}
return services , nil
}
//GetServicesAllInfoByTenantID GetServicesAllInfoByTenantID
func ( t * TenantServicesDaoImpl ) GetServicesAllInfoByTenantID ( tenantID string ) ( [ ] * model . TenantServices , error ) {
var services [ ] * model . TenantServices
if err := t . DB . Where ( "tenant_id= ?" , tenantID ) . Find ( & services ) . Error ; err != nil {
if err == gorm . ErrRecordNotFound {
return services , nil
}
return nil , err
}
return services , nil
}
//SetTenantServiceStatus SetTenantServiceStatus
func ( t * TenantServicesDaoImpl ) SetTenantServiceStatus ( serviceID , status string ) error {
var service model . TenantServices
if status == "closed" || status == "undeploy" {
if err := t . DB . Model ( & service ) . Where ( "service_id = ?" , serviceID ) . Update ( map [ string ] interface { } { "cur_status" : status , "status" : 0 } ) . Error ; err != nil {
return err
}
} else {
if err := t . DB . Model ( & service ) . Where ( "service_id = ?" , serviceID ) . Update ( map [ string ] interface { } { "cur_status" : status , "status" : 1 } ) . Error ; err != nil {
return err
}
}
return nil
}
//DeleteServiceByServiceID DeleteServiceByServiceID
func ( t * TenantServicesDaoImpl ) DeleteServiceByServiceID ( serviceID string ) error {
ts := & model . TenantServices {
ServiceID : serviceID ,
}
if err := t . DB . Where ( "service_id = ?" , serviceID ) . Delete ( ts ) . Error ; err != nil {
return err
}
return nil
}
//TenantServicesDeleteImpl TenantServiceDeleteImpl
type TenantServicesDeleteImpl struct {
DB * gorm . DB
}
//AddModel 添加已删除的应用
func ( t * TenantServicesDeleteImpl ) AddModel ( mo model . Interface ) error {
service := mo . ( * model . TenantServicesDelete )
var oldService model . TenantServicesDelete
if ok := t . DB . Where ( "service_alias = ? and tenant_id=?" , service . ServiceAlias , service . TenantID ) . Find ( & oldService ) . RecordNotFound ( ) ; ok {
if err := t . DB . Create ( service ) . Error ; err != nil {
return err
}
} else {
return fmt . Errorf ( "service name %s and is exist in tenant %s" , service . ServiceAlias , service . TenantID )
}
return nil
}
//UpdateModel 更新租户应用
func ( t * TenantServicesDeleteImpl ) UpdateModel ( mo model . Interface ) error {
service := mo . ( * model . TenantServicesDelete )
if err := t . DB . Save ( service ) . Error ; err != nil {
return err
}
return nil
}
//TenantServicesPortDaoImpl 租户应用端口操作
type TenantServicesPortDaoImpl struct {
DB * gorm . DB
}
//AddModel 添加应用端口
func ( t * TenantServicesPortDaoImpl ) AddModel ( mo model . Interface ) error {
port := mo . ( * model . TenantServicesPort )
var oldPort model . TenantServicesPort
if ok := t . DB . Where ( "service_id = ? and container_port = ?" , port . ServiceID , port . ContainerPort ) . Find ( & oldPort ) . RecordNotFound ( ) ; ok {
if err := t . DB . Create ( port ) . Error ; err != nil {
return err
}
} else {
return fmt . Errorf ( "service port %d in service %s is exist" , port . ContainerPort , port . ServiceID )
}
return nil
}
//UpdateModel 更新租户
func ( t * TenantServicesPortDaoImpl ) UpdateModel ( mo model . Interface ) error {
port := mo . ( * model . TenantServicesPort )
if port . ID == 0 {
return fmt . Errorf ( "port id can not be empty when update " )
}
if err := t . DB . Save ( port ) . Error ; err != nil {
return err
}
return nil
}
//DeleteModel 删除端口
func ( t * TenantServicesPortDaoImpl ) DeleteModel ( serviceID string , args ... interface { } ) error {
if len ( args ) < 1 {
return fmt . Errorf ( "can not provide containerPort" )
}
containerPort := args [ 0 ] . ( int )
tsp := & model . TenantServicesPort {
ServiceID : serviceID ,
ContainerPort : containerPort ,
//Protocol: protocol,
}
if err := t . DB . Where ( "service_id=? and container_port=?" , serviceID , containerPort ) . Delete ( tsp ) . Error ; err != nil {
return err
}
return nil
}
//GetPortsByServiceID 通过服务获取port
func ( t * TenantServicesPortDaoImpl ) GetPortsByServiceID ( serviceID string ) ( [ ] * model . TenantServicesPort , error ) {
var oldPort [ ] * model . TenantServicesPort
if err := t . DB . Where ( "service_id = ?" , serviceID ) . Find ( & oldPort ) . Error ; err != nil {
if err == gorm . ErrRecordNotFound {
return oldPort , nil
}
return nil , err
}
return oldPort , nil
}
//GetOuterPorts 获取对外端口
func ( t * TenantServicesPortDaoImpl ) GetOuterPorts ( serviceID string ) ( [ ] * model . TenantServicesPort , error ) {
var oldPort [ ] * model . TenantServicesPort
if err := t . DB . Where ( "service_id = ? and is_outer_service=?" , serviceID , true ) . Find ( & oldPort ) . Error ; err != nil {
if err == gorm . ErrRecordNotFound {
return oldPort , nil
}
return nil , err
}
return oldPort , nil
}
//GetInnerPorts 获取对内端口
func ( t * T enantServicesPortDaoImpl ) GetInnerPorts ( serviceID string ) ( [ ] * model . TenantServicesPort , error ) {
var oldPort [ ] * model . TenantServicesPort
if err := t . DB . Where ( "service_id = ? and is_inner_service=?" , serviceID , true ) . Find ( & oldPort ) . Error ; err != nil {
return nil , err
}
return oldPort , nil
}
//GetPort get port
func ( t * TenantServicesPortDaoImpl ) GetPort ( serviceID string , port int ) ( * model . TenantServicesPort , error ) {
var oldPort model . TenantServicesPort
if err := t . DB . Where ( "service_id = ? and container_port=?" , serviceID , port ) . Find ( & oldPort ) . Error ; err != nil {
return nil , err
}
return & oldPort , nil
}
//DELPortsByServiceID DELPortsByServiceID
func ( t * TenantServicesPortDaoImpl ) DELPortsByServiceID ( serviceID string ) error {
ports := & model . TenantServicesPort {
ServiceID : serviceID ,
}
if err := t . DB . Where ( "service_id=?" , serviceID ) . Where ( ports ) . Error ; err != nil {
return err
}
return nil
}
//TenantServiceRelationDaoImpl TenantServiceRelationDaoImpl
type TenantServiceRelationDaoImpl struct {
DB * gorm . DB
}
//AddModel 添加应用依赖关系
func ( t * TenantServiceRelationDaoImpl ) AddModel ( mo model . Interface ) error {
relation := mo . ( * model . TenantServiceRelation )
var oldRelation model . TenantServiceRelation
if ok := t . DB . Where ( "service_id = ? and dep_service_id = ?" , relation . ServiceID , relation . DependServiceID ) . Find ( & oldRelation ) . RecordNotFound ( ) ; ok {
if err := t . DB . Create ( relation ) . Error ; err != nil {
return err
}
} else {
return fmt . Errorf ( "service %s depend service %s relation is exist" , relation . ServiceID , relation . DependServiceID )
}
return nil
}
//UpdateModel 更新应用依赖关系
func ( t * TenantServiceRelationDaoImpl ) UpdateModel ( mo model . Interface ) error {
relation := mo . ( * model . TenantServiceRelation )
if relation . ID == 0 {
return fmt . Errorf ( "relation id can not be empty when update " )
}
if err := t . DB . Save ( relation ) . Error ; err != nil {
return err
}
return nil
}
//DeleteModel 删除依赖
func ( t * TenantServiceRelationDaoImpl ) DeleteModel ( serviceID string , args ... interface { } ) error {
depServiceID := args [ 0 ] . ( string )
relation := & model . TenantServiceRelation {
ServiceID : serviceID ,
DependServiceID : depServiceID ,
}
logrus . Infof ( "service: %v, depend: %v" , serviceID , depServiceID )
if err := t . DB . Where ( "service_id=? and dep_service_id=?" , serviceID , depServiceID ) . Delete ( relation ) . Error ; err != nil {
return err
}
return nil
}
//DeleteRelationByDepID DeleteRelationByDepID
func ( t * TenantServiceRelationDaoImpl ) DeleteRelationByDepID ( serviceID , depID string ) error {
relation := & model . TenantServiceRelation {
ServiceID : serviceID ,
DependServiceID : depID ,
}
if err := t . DB . Where ( "service_id=? and dep_service_id=?" , serviceID , depID ) . Delete ( relation ) . Error ; err != nil {
return err
}
return nil
}
//GetTenantServiceRelations 获取应用依赖关系
func ( t * TenantServiceRelationDaoImpl ) GetTenantServiceRelations ( serviceID string ) ( [ ] * model . TenantServiceRelation , error ) {
var oldRelation [ ] * model . TenantServiceRelation
if err := t . DB . Where ( "service_id = ?" , serviceID ) . Find ( & oldRelation ) . Error ; err != nil {
if err == gorm . ErrRecordNotFound {
return oldRelation , nil
}
return nil , err
}
return oldRelation , nil
}
//HaveRelations 是否有依赖
func ( t * TenantServiceRelationDaoImpl ) HaveRelations ( serviceID string ) bool {
var oldRelation [ ] * model . TenantServiceRelation
if err := t . DB . Where ( "service_id = ?" , serviceID ) . Find ( & oldRelation ) . Error ; err != nil {
return false
}
if len ( oldRelation ) > 0 {
return true
}
return false
}
//DELRelationsByServiceID DELRelationsByServiceID
func ( t * TenantServiceRelationDaoImpl ) DELRelationsByServiceID ( serviceID string ) error {
relation := & model . TenantServiceRelation {
ServiceID : serviceID ,
}
if err := t . DB . Where ( "service_id=?" , serviceID ) . Delete ( relation ) . Error ; err != nil {
return err
}
return nil
}
//GetTenantServiceRelationsByDependServiceID 获取全部依赖当前服务的应用
func ( t * TenantServiceRelationDaoImpl ) GetTenantServiceRelationsByDependServiceID ( dependServiceID string ) ( [ ] * model . TenantServiceRelation , error ) {
var oldRelation [ ] * model . TenantServiceRelation
if err := t . DB . Where ( "dep_service_id = ?" , dependServiceID ) . Find ( & oldRelation ) . Error ; err != nil {
return nil , err
}
return oldRelation , nil
}
//TenantServiceEnvVarDaoImpl TenantServiceEnvVarDaoImpl
type TenantServiceEnvVarDaoImpl struct {
DB * gorm . DB
}
//AddModel 添加应用环境变量
func ( t * TenantServiceEnvVarDaoImpl ) AddModel ( mo model . Interface ) error {
relation := mo . ( * model . TenantServiceEnvVar )
var oldRelation model . TenantServiceEnvVar
if ok := t . DB . Where ( "service_id = ? and attr_name = ?" , relation . ServiceID , relation . AttrName ) . Find ( & oldRelation ) . RecordNotFound ( ) ; ok {
if err := t . DB . Create ( relation ) . Error ; err != nil {
return err
}
} else {
return fmt . Errorf ( "already exist" )
}
return nil
}
//UpdateModel 更新应用环境变量
func ( t * TenantServiceEnvVarDaoImpl ) UpdateModel ( mo model . Interface ) error {
relation := mo . ( * model . TenantServiceEnvVar )
if relation . ID == 0 {
return fmt . Errorf ( "relation id can not be empty when update" )
}
if err := t . DB . Save ( relation ) . Error ; err != nil {
return err
}
return nil
}
//DeleteModel 删除env
func ( t * TenantServiceEnvVarDaoImpl ) DeleteModel ( serviceID string , args ... interface { } ) error {
envName := args [ 0 ] . ( string )
relation := & model . TenantServiceEnvVar {
ServiceID : serviceID ,
AttrName : envName ,
}
if err := t . DB . Where ( "service_id=? and attr_name=?" , serviceID , envName ) . Delete ( relation ) . Error ; err != nil {
return err
}
return nil
}
//GetDependServiceEnvs 获取依赖服务的环境变量
func ( t * TenantServiceEnvVarDaoImpl ) GetDependServiceEnvs ( serviceIDs [ ] string , scopes [ ] string ) ( [ ] * model . TenantServiceEnvVar , error ) {
var envs [ ] * model . TenantServiceEnvVar
if err := t . DB . Where ( "service_id in (?) and scope in (?)" , serviceIDs , scopes ) . Find ( & envs ) . Error ; err != nil {
if err == gorm . ErrRecordNotFound {
return envs , nil
}
return nil , err
}
return envs , nil
}
//GetServiceEnvs 获取服务环境变量
func ( t * TenantServiceEnvVarDaoImpl ) GetServiceEnvs ( serviceID string , scopes [ ] string ) ( [ ] * model . TenantServiceEnvVar , error ) {
var envs [ ] * model . TenantServiceEnvVar
if scopes == nil {
if err := t . DB . Where ( "service_id=?" , serviceID ) . Find ( & envs ) . Error ; err != nil {
if err == gorm . ErrRecordNotFound {
return envs , nil
}
return nil , err
}
} else {
if err := t . DB . Where ( "service_id=? and scope in (?)" , serviceID , scopes ) . Find ( & envs ) . Error ; err != nil {
if err == gorm . ErrRecordNotFound {
return envs , nil
}
return nil , err
}
}
return envs , nil
}
//GetEnv 获取某个环境变量
func ( t * TenantServiceEnvVarDaoImpl ) GetEnv ( serviceID , envName string ) ( * model . TenantServiceEnvVar , error ) {
var env model . TenantServiceEnvVar
if err := t . DB . Where ( "service_id=? and attr_name=? " , serviceID , envName ) . Find ( & env ) . Error ; err != nil {
return nil , err
}
return & env , nil
}
//DELServiceEnvsByServiceID 通过serviceID 删除envs
func ( t * TenantServiceEnvVarDaoImpl ) DELServiceEnvsByServiceID ( serviceID string ) error {
env := & model . TenantServiceEnvVar {
ServiceID : serviceID ,
}
if err := t . DB . Where ( "service_id=?" , serviceID ) . Delete ( env ) . Error ; err != nil {
return err
}
return nil
}
//TenantServiceMountRelationDaoImpl 依赖存储
type TenantServiceMountRelationDaoImpl struct {
DB * gorm . DB
}
//AddModel 添加应用依赖挂载
func ( t * TenantServiceMountRelationDaoImpl ) AddModel ( mo model . Interface ) error {
relation := mo . ( * model . TenantServiceMountRelation )
var oldRelation model . TenantServiceMountRelation
if ok := t . DB . Where ( "service_id = ? and dep_service_id = ?" , relation . ServiceID , relation . DependServiceID ) . Find ( & oldRelation ) . RecordNotFound ( ) ; ok {
if err := t . DB . Create ( relation ) . Error ; err != nil {
return err
}
} else {
return fmt . Errorf ( "service %s depend service %s mount relation is exist" , relation . ServiceID , relation . DependServiceID )
}
return nil
}
//UpdateModel 更新应用依赖挂载
func ( t * TenantServiceMountRelationDaoImpl ) UpdateModel ( mo model . Interface ) error {
relation := mo . ( * model . TenantServiceMountRelation )
if relation . ID == 0 {
return fmt . Errorf ( "mount relation id can not be empty when update " )
}
if err := t . DB . Save ( relation ) . Error ; err != nil {
return err
}
return nil
}
//DElTenantServiceMountRelationByServiceAndName DElTenantServiceMountRelationByServiceAndName
func ( t * TenantServiceMountRelationDaoImpl ) DElTenantServiceMountRelationByServiceAndName ( serviceID , name string ) error {
var relation model . TenantServiceMountRelation
if err := t . DB . Where ( "service_id=? and volume_name=? " , serviceID , name ) . Find ( & relation ) . Error ; err != nil {
return err
}
if err := t . DB . Where ( "service_id=? and volume_name=? " , serviceID , name ) . Delete ( & relation ) . Error ; err != nil {
return err
}
return nil
}
//DElTenantServiceMountRelationByDepService del mount relation
func ( t * TenantServiceMountRelationDaoImpl ) DElTenantServiceMountRelationByDepService ( serviceID , depServiceID string ) error {
var relation model . TenantServiceMountRelation
if err := t . DB . Where ( "service_id=? and dep_service_id=?" , serviceID , depServiceID ) . Find ( & relation ) . Error ; err != nil {
return err
}
if err := t . DB . Where ( "service_id=? and dep_service_id=?" , serviceID , depServiceID ) . Delete ( & relation ) . Error ; err != nil {
return err
}
return nil
}
//DELTenantServiceMountRelationByServiceID DELTenantServiceMountRelationByServiceID
func ( t * TenantServiceMountRelationDaoImpl ) DELTenantServiceMountRelationByServiceID ( serviceID string ) error {
var relation model . TenantServiceMountRelation
if err := t . DB . Where ( "service_id=?" , serviceID ) . Delete ( & relation ) . Error ; err != nil {
return err
}
return nil
}
//GetTenantServiceMountRelationsByService 获取应用的所有挂载依赖
func ( t * TenantServiceMountRelationDaoImpl ) GetTenantServiceMountRelationsByService ( serviceID string ) ( [ ] * model . TenantServiceMountRelation , error ) {
var relations [ ] * model . TenantServiceMountRelation
if err := t . DB . Where ( "service_id=? " , serviceID ) . Find ( & relations ) . Error ; err != nil {
if err == gorm . ErrRecordNotFound {
return relations , nil
}
return nil , err
}
return relations , nil
}
//TenantServiceVolumeDaoImpl 应用存储
type TenantServiceVolumeDaoImpl struct {
DB * gorm . DB
}
//AddModel 添加应用挂载
func ( t * TenantServiceVolumeDaoImpl ) AddModel ( mo model . Interface ) error {
volume := mo . ( * model . TenantServiceVolume )
var oldvolume model . TenantServiceVolume
if ok := t . DB . Where ( "(volume_name=? or volume_path = ?) and service_id=?" , volume . VolumeName , volume . VolumePath , volume . ServiceID ) . Find ( & oldvolume ) . RecordNotFound ( ) ; ok {
if err := t . DB . Create ( volume ) . Error ; err != nil {
return err
}
} else {
return fmt . Errorf ( "service %s volume name %s path %s is exist " , volume . ServiceID , volume . VolumeName , volume . VolumePath )
}
return nil
}
//UpdateModel 更新应用挂载
func ( t * TenantServiceVolumeDaoImpl ) UpdateModel ( mo model . Interface ) error {
volume := mo . ( * model . TenantServiceVolume )
if volume . ID == 0 {
return fmt . Errorf ( "volume id can not be empty when update " )
}
if err := t . DB . Save ( volume ) . Error ; err != nil {
return err
}
return nil
}
//GetTenantServiceVolumesByServiceID 获取应用挂载
func ( t * TenantServiceVolumeDaoImpl ) GetTenantServiceVolumesByServiceID ( serviceID string ) ( [ ] * model . TenantServiceVolume , error ) {
var volumes [ ] * model . TenantServiceVolume
if err := t . DB . Where ( "service_id=? " , serviceID ) . Find ( & volumes ) . Error ; err != nil {
if err == gorm . ErrRecordNotFound {
return volumes , nil
}
return nil , err
}
return volumes , nil
}
//DeleteModel 删除挂载
func ( t * TenantServiceVolumeDaoImpl ) DeleteModel ( serviceID string , args ... interface { } ) error {
var volume model . TenantServiceVolume
volumeName := args [ 0 ] . ( string )
if err := t . DB . Where ( "volume_name = ? and service_id=?" , volumeName , serviceID ) . Find ( & volume ) . Error ; err != nil {
return err
}
if err := t . DB . Where ( "volume_name = ? and service_id=?" , volumeName , serviceID ) . Delete ( & volume ) . Error ; err != nil {
return err
}
return nil
}
//DeleteByServiceIDAndVolumePath 删除挂载通过挂载的目录
func ( t * TenantServiceVolumeDaoImpl ) DeleteByServiceIDAndVolumePath ( serviceID string , volumePath string ) error {
var volume model . TenantServiceVolume
if err := t . DB . Where ( "volume_path = ? and service_id=?" , volumePath , serviceID ) . Find ( & volume ) . Error ; err != nil {
return err
}
if err := t . DB . Where ( "volume_path = ? and service_id=?" , volumePath , serviceID ) . Delete ( & volume ) . Error ; err != nil {
return err
}
return nil
}
//GetVolumeByServiceIDAndName 获取存储信息
func ( t * TenantServiceVolumeDaoImpl ) GetVolumeByServiceIDAndName ( serviceID , name string ) ( * model . TenantServiceVolume , error ) {
var volume model . TenantServiceVolume
if err := t . DB . Where ( "service_id=? and volume_name=? " , serviceID , name ) . Find ( & volume ) . Error ; err != nil {
return nil , err
}
return & volume , nil
}
//DeleteTenantServiceVolumesByServiceID 删除挂载
func ( t * TenantServiceVolumeDaoImpl ) DeleteTenantServiceVolumesByServiceID ( serviceID string ) error {
var volume model . TenantServiceVolume
if err := t . DB . Where ( "service_id=? " , serviceID ) . Delete ( & volume ) . Error ; err != nil {
return err
}
return nil
}
//TenantServiceLBMappingPortDaoImpl stream服务映射
type TenantServiceLBMappingPortDaoImpl struct {
DB * gorm . DB
}
//AddModel 添加应用端口映射
func ( t * TenantServiceLBMappingPortDaoImpl ) AddModel ( mo model . Interface ) error {
mapPort := mo . ( * model . TenantServiceLBMappingPort )
var oldMapPort model . TenantServiceLBMappingPort
if ok := t . DB . Where ( "service_id=? and port=?" , mapPort . ServiceID , mapPort . ContainerPort ) . Find ( & oldMapPort ) . RecordNotFound ( ) ; ok {
if err := t . DB . Create ( mapPort ) . Error ; err != nil {
return err
}
} else {
return fmt . Errorf ( "service %s mapport %d is exist " , mapPort . ServiceID , mapPort . ContainerPort )
}
return nil
}
//UpdateModel 更新应用端口映射
func ( t * TenantServiceLBMappingPortDaoImpl ) UpdateModel ( mo model . Interface ) error {
mapPort := mo . ( * model . TenantServiceLBMappingPort )
if mapPort . ID == 0 {
return fmt . Errorf ( "mapport id can not be empty when update " )
}
if err := t . DB . Save ( mapPort ) . Error ; err != nil {
return err
}
return nil
}
//GetTenantServiceLBMappingPort 获取端口映射
func ( t * TenantServiceLBMappingPortDaoImpl ) GetTenantServiceLBMappingPort ( serviceID string , containerPort int ) ( * model . TenantServiceLBMappingPort , error ) {
var mapPort model . TenantServiceLBMappingPort
if err := t . DB . Where ( "service_id=? and container_port=?" , serviceID , containerPort ) . Find ( & mapPort ) . Error ; err != nil {
return nil , err
}
return & mapPort , nil
}
//CreateTenantServiceLBMappingPort 创建负载均衡VS端口,如果端口分配已存在,直接返回
func ( t * TenantServiceLBMappingPortDaoImpl ) CreateTenantServiceLBMappingPort ( serviceID string , containerPort int ) ( * model . TenantServiceLBMappingPort , error ) {
var mapPorts [ ] * model . TenantServiceLBMappingPort
var mapPort model . TenantServiceLBMappingPort
err := t . DB . Where ( "service_id=? and container_port=?" , serviceID , containerPort ) . Find ( & mapPort ) . Error
if err == nil {
return & mapPort , nil
}
//分配端口
var ports [ ] int
err = t . DB . Order ( "port asc" ) . Find ( & mapPorts ) . Error
if err != nil {
return nil , fmt . Errorf ( "select all exist port error,%s" , err . Error ( ) )
}
for _ , p := range mapPorts {
ports = append ( ports , p . Port )
}
maxPort , _ := strconv . Atoi ( os . Getenv ( "MIN_LB_PORT" ) )
minPort , _ := strconv . Atoi ( os . Getenv ( "MAX_LB_PORT" ) )
if minPort == 0 {
minPort = 20001
}
if maxPort == 0 {
maxPort = 35000
}
var maxUsePort int
if len ( ports ) > 0 {
maxUsePort = ports [ len ( ports ) - 1 ]
} else {
maxUsePort = 20001
}
//顺序分配端口
selectPort := maxUsePort + 1
if selectPort <= maxPort {
mp := & model . TenantServiceLBMappingPort {
ServiceID : serviceID ,
Port : selectPort ,
ContainerPort : containerPort ,
}
if err := t . DB . Save ( mp ) . Error ; err == nil {
return mp , nil
}
}
//捡漏以前端口
selectPort = minPort
errCount := 0
for _ , p := range ports {
if p == selectPort {
selectPort = selectPort + 1
continue
}
if p > selectPort {
mp := & model . TenantServiceLBMappingPort {
ServiceID : serviceID ,
Port : selectPort ,
ContainerPort : containerPort ,
}
if err := t . DB . Save ( mp ) . Error ; err != nil {
logrus . Errorf ( "save select map vs port %d error %s" , selectPort , err . Error ( ) )
errCount ++
if errCount > 2 { //尝试3次
break
}
} else {
return mp , nil
}
}
selectPort = selectPort + 1
}
if selectPort <= maxPort {
mp := & model . TenantServiceLBMappingPort {
ServiceID : serviceID ,
Port : selectPort ,
ContainerPort : containerPort ,
}
if err := t . DB . Save ( mp ) . Error ; err != nil {
logrus . Errorf ( "save select map vs port %d error %s" , selectPort , err . Error ( ) )
return nil , fmt . Errorf ( "can not select a good port for service stream port" )
}
return mp , nil
}
logrus . Errorf ( "no more lb port can be use,max port is %d" , maxPort )
return nil , fmt . Errorf ( "no more lb port can be use,max port is %d" , maxPort )
}
//GetTenantServiceLBMappingPortByService 获取端口映射
func ( t * TenantServiceLBMappingPortDaoImpl ) GetTenantServiceLBMappingPortByService ( serviceID string ) ( * model . TenantServiceLBMappingPort , error ) {
var mapPort model . TenantServiceLBMappingPort
if err := t . DB . Where ( "service_id=?" , serviceID ) . Find ( & mapPort ) . Error ; err != nil {
return nil , err
}
return & mapPort , nil
}
//DELServiceLBMappingPortByServiceID DELServiceLBMappingPortByServiceID
func ( t * TenantServiceLBMappingPortDaoImpl ) DELServiceLBMappingPortByServiceID ( serviceID string ) error {
mapPorts := & model . TenantServiceLBMappingPort {
ServiceID : serviceID ,
}
if err := t . DB . Where ( "service_id=?" , serviceID ) . Delete ( mapPorts ) . Error ; err != nil {
return err
}
return nil
}
//ServiceLabelDaoImpl ServiceLabelDaoImpl
type ServiceLabelDaoImpl struct {
DB * gorm . DB
}
//AddModel 添加应用Label
func ( t * ServiceLabelDaoImpl ) AddModel ( mo model . Interface ) error {
label := mo . ( * model . TenantServiceLable )
var oldLabel model . TenantServiceLable
if label . LabelKey == model . LabelKeyServiceType { //LabelKeyServiceType 只能有一条
if ok := t . DB . Where ( "service_id = ? and label_key=?" , label . ServiceID , label . LabelKey ) . Find ( & oldLabel ) . RecordNotFound ( ) ; ok {
if err := t . DB . Create ( label ) . Error ; err != nil {
return err
}
} else {
return fmt . Errorf ( "label key %s of service %s is exist" , label . LabelKey , label . ServiceID )
}
} else {
if ok := t . DB . Where ( "service_id = ? and label_key=? and label_value=?" , label . ServiceID , label . LabelKey , label . LabelValue ) . Find ( & oldLabel ) . RecordNotFound ( ) ; ok {
if err := t . DB . Create ( label ) . Error ; err != nil {
return err
}
} else {
return fmt . Errorf ( "label key %s value %s of service %s is exist" , label . LabelKey , label . LabelValue , label . ServiceID )
}
}
return nil
}
//UpdateModel 更新应用Label
func ( t * ServiceLabelDaoImpl ) UpdateModel ( mo model . Interface ) error {
label := mo . ( * model . TenantServiceLable )
if label . ID == 0 {
return fmt . Errorf ( "label id can not be empty when update " )
}
if err := t . DB . Save ( label ) . Error ; err != nil {
return err
}
return nil
}
//DeleteModel 删除应用label
func ( t * ServiceLabelDaoImpl ) DeleteModel ( serviceID string , args ... interface { } ) error {
label := & model . TenantServiceLable {
ServiceID : serviceID ,
LabelKey : args [ 0 ] . ( string ) ,
LabelValue : args [ 1 ] . ( string ) ,
}
if err := t . DB . Where ( "service_id=? and label_key=? and label_value=?" ,
serviceID , label . LabelKey , label . LabelValue ) . Delete ( label ) . Error ; err != nil {
return err
}
return nil
}
//GetTenantServiceLabel GetTenantServiceLabel
func ( t * ServiceLabelDaoImpl ) GetTenantServiceLabel ( serviceID string ) ( [ ] * model . TenantServiceLable , error ) {
var labels [ ] * model . TenantServiceLable
if err := t . DB . Where ( "service_id=?" , serviceID ) . Find ( & labels ) . Error ; err != nil {
if err == gorm . ErrRecordNotFound {
return labels , nil
}
return nil , err
}
return labels , nil
}
//GetTenantServiceNodeSelectorLabel GetTenantServiceNodeSelectorLabel
func ( t * ServiceLabelDaoImpl ) GetTenantServiceNodeSelectorLabel ( serviceID string ) ( [ ] * model . TenantServiceLable , error ) {
var labels [ ] * model . TenantServiceLable
if err := t . DB . Where ( "service_id=? and label_value=?" , serviceID , model . LabelKeyNodeSelector ) . Find ( & labels ) . Error ; err != nil {
if err == gorm . ErrRecordNotFound {
return labels , nil
}
return nil , err
}
return labels , nil
}
//GetTenantServiceAffinityLabel GetTenantServiceAffinityLabel
func ( t * ServiceLabelDaoImpl ) GetTenantServiceAffinityLabel ( serviceID string ) ( [ ] * model . TenantServiceLable , error ) {
var labels [ ] * model . TenantServiceLable
if err := t . DB . Where ( "service_id=? and label_key in (?)" , serviceID , [ ] string { model . LabelKeyNodeAffinity , model . LabelKeyNodeAntyAffinity ,
model . LabelKeyServiceAffinity , model . LabelKeyServiceAntyAffinity } ) . Find ( & labels ) . Error ; err != nil {
if err == gorm . ErrRecordNotFound {
return labels , nil
}
return nil , err
}
return labels , nil
}
//GetTenantServiceTypeLabel GetTenantServiceTypeLabel
func ( t * ServiceLabelDaoImpl ) GetTenantServiceTypeLabel ( serviceID string ) ( * model . TenantServiceLable , error ) {
var label model . TenantServiceLable
if err := t . DB . Where ( "service_id=? and label_key=?" , serviceID , model . LabelKeyServiceType ) . Find ( & label ) . Error ; err != nil {
if err == gorm . ErrRecordNotFound {
return nil , nil
}
return nil , err
}
return & label , nil
}
//DELTenantServiceLabelsByLabelvaluesAndServiceID DELTenantServiceLabelsByLabelvaluesAndServiceID
func ( t * ServiceLabelDaoImpl ) DELTenantServiceLabelsByLabelvaluesAndServiceID ( serviceID string , labelValues [ ] string ) error {
label := & model . TenantServiceLable {
ServiceID : serviceID ,
}
if err := t . DB . Where ( "service_id=? and label_value=? and label_key in (?)" , serviceID , "node_select" , labelValues ) . Delete ( label ) . Error ; err != nil {
return err
}
return nil
}
//ServiceStatusDaoImpl 更新应用状态
type ServiceStatusDaoImpl struct {
DB * gorm . DB
}
//AddModel 添加应用状态
func ( t * ServiceStatusDaoImpl ) AddModel ( mo model . Interface ) error {
status := mo . ( * model . TenantServiceStatus )
var oldStatus model . TenantServiceStatus
if ok := t . DB . Where ( "service_id=?" , oldStatus . ServiceID ) . Find ( & oldStatus ) . RecordNotFound ( ) ; ok {
if err := t . DB . Create ( status ) . Error ; err != nil {
return err
}
} else {
return t . UpdateModel ( mo )
}
return nil
}
//UpdateModel 更新应用状态
func ( t * ServiceStatusDaoImpl ) UpdateModel ( mo model . Interface ) error {
status := mo . ( * model . TenantServiceStatus )
if status . Status == "" {
return fmt . Errorf ( "service status is undefined when update service status" )
}
if err := t . DB . Save ( status ) . Error ; err != nil {
return err
}
return nil
}
//GetTenantServiceStatus 获取应用状态
func ( t * ServiceStatusDaoImpl ) GetTenantServiceStatus ( serviceID string ) ( * model . TenantServiceStatus , error ) {
var status model . TenantServiceStatus
if err := t . DB . Where ( "service_id=?" , serviceID ) . Find ( & status ) . Error ; err != nil {
if err == gorm . ErrRecordNotFound {
var tenantService model . TenantServices
if err := t . DB . Where ( "service_id=?" , serviceID ) . Find ( & tenantService ) . Error ; err != nil {
return nil , err
}
status . Status = tenantService . CurStatus
status . ServiceID = tenantService . ServiceID
if err := t . DB . Create ( & status ) . Error ; err != nil {
return nil , err
}
}
return nil , err
}
return & status , nil
}
//SetTenantServiceStatus 设置应用状态
func ( t * ServiceStatusDaoImpl ) SetTenantServiceStatus ( serviceID , status string ) error {
var oldStatus model . TenantServiceStatus
if ok := t . DB . Where ( "service_id=?" , serviceID ) . Find ( & oldStatus ) . RecordNotFound ( ) ; ! ok {
oldStatus . Status = status
if err := t . DB . Save ( & oldStatus ) . Error ; err != nil {
return fmt . Errorf ( "set service status failed, %v" , err )
}
} else {
oldStatus . ServiceID = serviceID
oldStatus . Status = status
oldStatus . CreatedAt = time . Now ( )
if err := t . DB . Create ( & oldStatus ) . Error ; err != nil {
return err
}
}
return nil
}
//GetRunningService GetRunningService
func ( t * ServiceStatusDaoImpl ) GetRunningService ( ) ( [ ] * model . TenantServiceStatus , error ) {
var statuss [ ] * model . TenantServiceStatus
if err := t . DB . Where ( "status in (?)" , [ ] string { "running" , "starting" } ) . Find ( & statuss ) . Error ; err != nil {
if err == gorm . ErrRecordNotFound {
return statuss , nil
}
return nil , err
}
return statuss , nil
}
//GetTenantStatus GetTenantStatus
func ( t * ServiceStatusDaoImpl ) GetTenantStatus ( tenantID string ) ( [ ] * model . TenantServiceStatus , error ) {
var statuss [ ] * model . TenantServiceStatus
if err := t . DB . Table ( "tenant_service_status" ) . Raw ( "select * from tenant_service_status where service_id in (select service_id from tenant_services where tenant_id=?)" , tenantID ) . Find ( & statuss ) . Error ; err != nil {
return nil , err
}
return statuss , nil
}
//GetTenantServicesStatus GetTenantServicesStatus
func ( t * ServiceStatusDaoImpl ) GetTenantServicesStatus ( serviceIDs [ ] string ) ( [ ] * model . TenantServiceStatus , error ) {
var statuss [ ] * model . TenantServiceStatus
if err := t . DB . Where ( "service_id in (?)" , serviceIDs ) . Find ( & statuss ) . Error ; err != nil {
return nil , err
}
return statuss , nil
}
//PluginDaoImpl PluginDaoImpl
type PluginDaoImpl struct {
DB * gorm . DB
}
//AddModel 创建插件
func ( t * PluginDaoImpl ) AddModel ( mo model . Interface ) error {
plugin := mo . ( * model . TenantPlugin )
var oldPlugin model . TenantPlugin
if ok := t . DB . Where ( "plugin_name = ?" , plugin . PluginName ) . Find ( & oldPlugin ) . RecordNotFound ( ) ; ok {
if err := t . DB . Create ( plugin ) . Error ; err != nil {
return err
}
} else {
return fmt . Errorf ( "plugin %s is exist" , plugin . PluginName )
}
return nil
}
//UpdateModel 更新插件
func ( t * PluginDaoImpl ) UpdateModel ( mo model . Interface ) error {
plugin := mo . ( * model . TenantPlugin )
if err := t . DB . Save ( plugin ) . Error ; err != nil {
return err
}
return nil
}
//GetPluginByID GetPluginByID
func ( t * PluginDaoImpl ) GetPluginByID ( id string ) ( * model . TenantPlugin , error ) {
var plugin model . TenantPlugin
if err := t . DB . Where ( "plugin_id = ? " , id ) . Find ( & plugin ) . Error ; err != nil {
return nil , err
}
return & plugin , nil
}
//DeletePluginByID DeletePluginByID
func ( t * PluginDaoImpl ) DeletePluginByID ( id string ) error {
relation := & model . TenantPlugin {
PluginID : id ,
}
if err := t . DB . Where ( "plugin_id=?" , id ) . Delete ( relation ) . Error ; err != nil {
return err
}
return nil
}
2017-11-07 19:11:49 +08:00
//GetPluginsByTenantID GetPluginsByTenantID
func ( t * PluginDaoImpl ) GetPluginsByTenantID ( tenantID string ) ( [ ] * model . TenantPlugin , error ) {
var plugins [ ] * model . TenantPlugin
if err := t . DB . Where ( "tenant_id=?" , tenantID ) . Find ( & plugins ) . Error ; err != nil {
return nil , err
}
return plugins , nil
}
2017-11-07 11:40:44 +08:00
//PluginDefaultENVDaoImpl PluginDefaultENVDaoImpl
type PluginDefaultENVDaoImpl struct {
DB * gorm . DB
}
//AddModel 添加插件默认变量
func ( t * PluginDefaultENVDaoImpl ) AddModel ( mo model . Interface ) error {
env := mo . ( * model . TenantPluginDefaultENV )
var oldENV model . TenantPluginDefaultENV
2017-11-09 18:31:31 +08:00
if ok := t . DB . Where ( "plugin_id=? and env_name = ?" , env . PluginID , env . ENVName ) . Find ( & oldENV ) . RecordNotFound ( ) ; ok {
2017-11-07 11:40:44 +08:00
if err := t . DB . Create ( env ) . Error ; err != nil {
return err
}
} else {
return fmt . Errorf ( "env %s is exist" , env . ENVName )
}
return nil
}
//UpdateModel 更新插件默认变量
func ( t * PluginDefaultENVDaoImpl ) UpdateModel ( mo model . Interface ) error {
env := mo . ( * model . TenantPluginDefaultENV )
if err := t . DB . Save ( env ) . Error ; err != nil {
return err
}
return nil
}
//GetDefaultENVByName GetDefaultENVByName
2017-11-09 18:31:31 +08:00
func ( t * PluginDefaultENVDaoImpl ) GetDefaultENVByName ( pluginID string , name string ) ( * model . TenantPluginDefaultENV , error ) {
2017-11-07 11:40:44 +08:00
var env model . TenantPluginDefaultENV
2017-11-09 18:31:31 +08:00
if err := t . DB . Where ( "plugin_id=? and env_name=?" , name ) . Find ( & env ) . Error ; err != nil {
2017-11-07 11:40:44 +08:00
return nil , err
}
return & env , nil
}
2017-11-08 19:17:21 +08:00
//GetDefaultENVSByPluginID GetDefaultENVSByPluginID
func ( t * PluginDefaultENVDaoImpl ) GetDefaultENVSByPluginID ( pluginID string ) ( [ ] * model . TenantPluginDefaultENV , error ) {
var envs [ ] * model . TenantPluginDefaultENV
if err := t . DB . Where ( "plugin_id=?" , pluginID ) . Find ( & envs ) . Error ; err != nil {
return nil , err
}
return envs , nil
}
//GetDefaultENVSByPluginIDCantBeSet GetDefaultENVSByPluginIDCantBeSet
func ( t * PluginDefaultENVDaoImpl ) GetDefaultENVSByPluginIDCantBeSet ( pluginID string ) ( [ ] * model . TenantPluginDefaultENV , error ) {
var envs [ ] * model . TenantPluginDefaultENV
2017-11-09 15:47:32 +08:00
if err := t . DB . Where ( "plugin_id=? and is_change=0" , pluginID ) . Find ( & envs ) . Error ; err != nil {
2017-11-08 19:17:21 +08:00
return nil , err
}
return envs , nil
}
2017-11-07 11:40:44 +08:00
//DeleteDefaultENVByName DeleteDefaultENVByName
2017-11-09 18:31:31 +08:00
func ( t * PluginDefaultENVDaoImpl ) DeleteDefaultENVByName ( pluginID , name string ) error {
2017-11-07 11:40:44 +08:00
relation := & model . TenantPluginDefaultENV {
ENVName : name ,
}
2017-11-09 18:31:31 +08:00
if err := t . DB . Where ( "plugin_id=? and env_name=?" , pluginID , name ) . Delete ( relation ) . Error ; err != nil {
2017-11-07 11:40:44 +08:00
return err
}
return nil
}
//DeleteAllDefaultENVByPluginID DeleteAllDefaultENVByPluginID
func ( t * PluginDefaultENVDaoImpl ) DeleteAllDefaultENVByPluginID ( id string ) error {
relation := & model . TenantPluginDefaultENV {
PluginID : id ,
}
if err := t . DB . Where ( "plugin_id=?" , id ) . Delete ( relation ) . Error ; err != nil {
return err
}
return nil
}
2017-11-07 19:11:49 +08:00
//GetDefaultEnvWhichCanBeSetByPluginID GetDefaultEnvWhichCanBeSetByPluginID
func ( t * PluginDefaultENVDaoImpl ) GetDefaultEnvWhichCanBeSetByPluginID ( pluginID string ) ( [ ] * model . TenantPluginDefaultENV , error ) {
var envs [ ] * model . TenantPluginDefaultENV
2017-11-09 15:47:32 +08:00
if err := t . DB . Where ( "plugin_id=? and is_change=1" , pluginID ) . Find ( & envs ) . Error ; err != nil {
2017-11-07 19:11:49 +08:00
return nil , err
}
return envs , nil
}
2017-11-07 11:40:44 +08:00
//PluginBuildVersionDaoImpl PluginBuildVersionDaoImpl
type PluginBuildVersionDaoImpl struct {
DB * gorm . DB
}
//AddModel 添加插件构建版本信息
func ( t * PluginBuildVersionDaoImpl ) AddModel ( mo model . Interface ) error {
version := mo . ( * model . TenantPluginBuildVersion )
var oldVersion model . TenantPluginBuildVersion
2017-11-09 18:31:31 +08:00
if ok := t . DB . Where ( "plugin_id =? and version_id = ?" , version . PluginID , version . VersionID ) . Find ( & oldVersion ) . RecordNotFound ( ) ; ok {
2017-11-07 11:40:44 +08:00
if err := t . DB . Create ( version ) . Error ; err != nil {
return err
}
} else {
return fmt . Errorf ( "plugin build version %s is exist" , version . VersionID )
}
return nil
}
//UpdateModel 更新插件默认变量
//主体信息一般不变更, 仅构建的本地镜像名与status需要变更
func ( t * PluginBuildVersionDaoImpl ) UpdateModel ( mo model . Interface ) error {
version := mo . ( * model . TenantPluginBuildVersion )
if version . ID == 0 {
return fmt . Errorf ( "id can not be empty when update build verion" )
}
if err := t . DB . Save ( version ) . Error ; err != nil {
return err
}
return nil
}
//DeleteBuildVersionByVersionID DeleteBuildVersionByVersionID
func ( t * PluginBuildVersionDaoImpl ) DeleteBuildVersionByVersionID ( versionID string ) error {
relation := & model . TenantPluginBuildVersion {
VersionID : versionID ,
}
if err := t . DB . Where ( "version_id=?" , versionID ) . Delete ( relation ) . Error ; err != nil {
return err
}
return nil
}
//DeleteBuildVersionByPluginID DeleteBuildVersionByPluginID
func ( t * PluginBuildVersionDaoImpl ) DeleteBuildVersionByPluginID ( pluginID string ) error {
relation := & model . TenantPluginBuildVersion {
PluginID : pluginID ,
}
if err := t . DB . Where ( "plugin_id=?" , pluginID ) . Delete ( relation ) . Error ; err != nil {
return err
}
return nil
}
//GetBuildVersionByPluginID GetBuildVersionByPluginID
func ( t * PluginBuildVersionDaoImpl ) GetBuildVersionByPluginID ( pluginID string ) ( [ ] * model . TenantPluginBuildVersion , error ) {
var versions [ ] * model . TenantPluginBuildVersion
if err := t . DB . Where ( "plugin_id = ? and status= ?" , pluginID , "complete" ) . Find ( & versions ) . Error ; err != nil {
return nil , err
}
return versions , nil
}
//GetBuildVersionByVersionID GetBuildVersionByVersionID
func ( t * PluginBuildVersionDaoImpl ) GetBuildVersionByVersionID ( pluginID , versionID string ) ( * model . TenantPluginBuildVersion , error ) {
var version model . TenantPluginBuildVersion
if err := t . DB . Where ( "plugin_id=? and version_id = ? " , pluginID , versionID ) . Find ( & version ) . Error ; err != nil {
return nil , err
}
return & version , nil
}
//PluginVersionEnvDaoImpl PluginVersionEnvDaoImpl
type PluginVersionEnvDaoImpl struct {
DB * gorm . DB
}
//AddModel 添加插件默认变量
func ( t * PluginVersionEnvDaoImpl ) AddModel ( mo model . Interface ) error {
env := mo . ( * model . TenantPluginVersionEnv )
var oldENV model . TenantPluginVersionEnv
2017-11-09 18:31:31 +08:00
if ok := t . DB . Where ( "plugin_id=? and env_name = ?" , env . PluginID , env . EnvName ) . Find ( & oldENV ) . RecordNotFound ( ) ; ok {
2017-11-07 11:40:44 +08:00
if err := t . DB . Create ( env ) . Error ; err != nil {
return err
}
} else {
return fmt . Errorf ( "env %s is exist" , env . EnvName )
}
return nil
}
//UpdateModel 更新插件默认变量
func ( t * PluginVersionEnvDaoImpl ) UpdateModel ( mo model . Interface ) error {
env := mo . ( * model . TenantPluginVersionEnv )
if env . ID == 0 || env . ServiceID == "" || env . PluginID == "" {
return fmt . Errorf ( "id can not be empty when update plugin version env" )
}
if err := t . DB . Save ( env ) . Error ; err != nil {
return err
}
return nil
}
//DeleteEnvByEnvName 删除单个env
func ( t * PluginVersionEnvDaoImpl ) DeleteEnvByEnvName ( envName , pluginID , serviceID string ) error {
env := & model . TenantPluginVersionEnv {
PluginID : pluginID ,
EnvName : envName ,
ServiceID : serviceID ,
}
return t . DB . Where ( "env_name=? and plugin_id=? and service_id=?" , envName , pluginID , serviceID ) . Delete ( env ) . Error
}
//DeleteEnvByPluginID 删除插件依赖关系时, 需要操作删除对应env
func ( t * PluginVersionEnvDaoImpl ) DeleteEnvByPluginID ( serviceID , pluginID string ) error {
env := & model . TenantPluginVersionEnv {
PluginID : pluginID ,
ServiceID : serviceID ,
}
return t . DB . Where ( "plugin_id=? and service_id= ?" , pluginID , serviceID ) . Delete ( env ) . Error
}
//DeleteEnvByServiceID 删除应用时,需要进行此操作
func ( t * PluginVersionEnvDaoImpl ) DeleteEnvByServiceID ( serviceID string ) error {
env := & model . TenantPluginVersionEnv {
ServiceID : serviceID ,
}
return t . DB . Where ( "service_id=?" , serviceID ) . Delete ( env ) . Error
}
//GetVersionEnvByServiceID 获取该应用下使用的某个插件依赖的插件变量
func ( t * PluginVersionEnvDaoImpl ) GetVersionEnvByServiceID ( serviceID string , pluginID string ) ( [ ] * model . TenantPluginVersionEnv , error ) {
var envs [ ] * model . TenantPluginVersionEnv
if err := t . DB . Where ( "service_id=? and plugin_id=?" , serviceID , pluginID ) . Find ( & envs ) . Error ; err != nil {
return nil , err
}
return envs , nil
}
2017-11-09 18:31:31 +08:00
//GetVersionEnvByEnvName GetVersionEnvByEnvName
func ( t * PluginVersionEnvDaoImpl ) GetVersionEnvByEnvName ( serviceID , pluginID , envName string ) ( * model . TenantPluginVersionEnv , error ) {
var env * model . TenantPluginVersionEnv
if err := t . DB . Where ( "service_id=? and plugin_id=? and env_name=?" , serviceID , pluginID , envName ) . Find ( & env ) . Error ; err != nil {
return nil , err
}
return env , nil
}
2017-11-07 11:40:44 +08:00
//TenantServicePluginRelationDaoImpl TenantServicePluginRelationDaoImpl
type TenantServicePluginRelationDaoImpl struct {
DB * gorm . DB
}
//AddModel 添加插件默认变量
func ( t * TenantServicePluginRelationDaoImpl ) AddModel ( mo model . Interface ) error {
relation := mo . ( * model . TenantServicePluginRelation )
var oldRelation model . TenantServicePluginRelation
if ok := t . DB . Where ( "service_id= ? and plugin_id=?" , relation . ServiceID , relation . PluginID ) . Find ( & oldRelation ) . RecordNotFound ( ) ; ok {
if err := t . DB . Create ( relation ) . Error ; err != nil {
return err
}
} else {
return fmt . Errorf ( "relation between %s and %s is exist" , relation . ServiceID , relation . PluginID )
}
return nil
}
//UpdateModel 更新插件默认变量 更新依赖的version id
func ( t * TenantServicePluginRelationDaoImpl ) UpdateModel ( mo model . Interface ) error {
relation := mo . ( * model . TenantServicePluginRelation )
if relation . ID == 0 {
return fmt . Errorf ( "id can not be empty when update service plugin relation" )
}
if err := t . DB . Save ( relation ) . Error ; err != nil {
return err
}
return nil
}
//DeleteRelationByServiceIDAndPluginID 删除service plugin 对应关系
func ( t * TenantServicePluginRelationDaoImpl ) DeleteRelationByServiceIDAndPluginID ( serviceID , pluginID string ) error {
relation := & model . TenantServicePluginRelation {
ServiceID : serviceID ,
PluginID : pluginID ,
}
return t . DB . Where ( "plugin_id=? and service_id=?" ,
pluginID ,
serviceID ) . Delete ( relation ) . Error
}
//DeleteALLRelationByServiceID 删除serviceID所有插件依赖 一般用于删除应用时使用
func ( t * TenantServicePluginRelationDaoImpl ) DeleteALLRelationByServiceID ( serviceID string ) error {
relation := & model . TenantServicePluginRelation {
ServiceID : serviceID ,
}
return t . DB . Where ( "service_id=?" , serviceID ) . Delete ( relation ) . Error
}
//DeleteALLRelationByPluginID 删除pluginID所有依赖 一般不要使用 会影响关联过的应用启动
func ( t * TenantServicePluginRelationDaoImpl ) DeleteALLRelationByPluginID ( pluginID string ) error {
relation := & model . TenantServicePluginRelation {
PluginID : pluginID ,
}
return t . DB . Where ( "plugin_id=?" , pluginID ) . Delete ( relation ) . Error
}
//GetALLRelationByServiceID 获取当前应用所有的插件依赖关系
func ( t * TenantServicePluginRelationDaoImpl ) GetALLRelationByServiceID ( serviceID string ) ( [ ] * model . TenantServicePluginRelation , error ) {
var relations [ ] * model . TenantServicePluginRelation
if err := t . DB . Where ( "service_id=?" , serviceID ) . Find ( & relations ) . Error ; err != nil {
return nil , err
}
return relations , nil
}
2017-11-07 19:11:49 +08:00
//GetRelateionByServiceIDAndPluginID GetRelateionByServiceIDAndPluginID
func ( t * TenantServicePluginRelationDaoImpl ) GetRelateionByServiceIDAndPluginID ( serviceID , pluginID string ) ( * model . TenantServicePluginRelation , error ) {
relation := & model . TenantServicePluginRelation {
PluginID : pluginID ,
ServiceID : serviceID ,
}
if err := t . DB . Where ( "plugin_id=? and service_id=?" , pluginID , serviceID ) . Find ( relation ) . Error ; err != nil {
return nil , err
}
return relation , nil
}