mirror of
https://github.com/yunionio/cloudpods.git
synced 2026-06-25 07:27:18 +08:00
214 lines
8.2 KiB
Go
214 lines
8.2 KiB
Go
// Copyright 2019 Yunion
|
|
//
|
|
// Licensed under the Apache License, Version 2.0 (the "License");
|
|
// you may not use this file except in compliance with the License.
|
|
// You may obtain a copy of the License at
|
|
//
|
|
// http://www.apache.org/licenses/LICENSE-2.0
|
|
//
|
|
// Unless required by applicable law or agreed to in writing, software
|
|
// distributed under the License is distributed on an "AS IS" BASIS,
|
|
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
// See the License for the specific language governing permissions and
|
|
// limitations under the License.
|
|
|
|
package models
|
|
|
|
import (
|
|
"context"
|
|
"database/sql"
|
|
"fmt"
|
|
|
|
"yunion.io/x/jsonutils"
|
|
"yunion.io/x/log"
|
|
"yunion.io/x/pkg/util/regutils"
|
|
|
|
api "yunion.io/x/onecloud/pkg/apis/compute"
|
|
"yunion.io/x/onecloud/pkg/cloudcommon/db"
|
|
"yunion.io/x/onecloud/pkg/httperrors"
|
|
"yunion.io/x/onecloud/pkg/mcclient"
|
|
)
|
|
|
|
type SBaremetalagentManager struct {
|
|
db.SStandaloneResourceBaseManager
|
|
}
|
|
|
|
type SBaremetalagent struct {
|
|
db.SStandaloneResourceBase
|
|
|
|
Status string `width:"36" charset:"ascii" nullable:"false" default:"disable" list:"user" create:"optional"`
|
|
AccessIp string `width:"16" charset:"ascii" nullable:"false" list:"admin" update:"admin" create:"admin_required"`
|
|
ManagerUri string `width:"256" charset:"ascii" nullable:"true" list:"admin" update:"admin" create:"admin_required"`
|
|
ZoneId string `width:"128" charset:"ascii" nullable:"false" list:"admin" update:"admin" create:"admin_required"`
|
|
|
|
AgentType string `width:"32" charset:"ascii" nullable:"true" default:"baremetal" list:"admin" update:"admin" create:"admin_optional"`
|
|
|
|
Version string `width:"64" charset:"ascii" list:"admin" update:"admin" create:"admin_optional"` // Column(VARCHAR(64, charset='ascii'))
|
|
}
|
|
|
|
var BaremetalagentManager *SBaremetalagentManager
|
|
|
|
func init() {
|
|
BaremetalagentManager = &SBaremetalagentManager{
|
|
SStandaloneResourceBaseManager: db.NewStandaloneResourceBaseManager(SBaremetalagent{},
|
|
"baremetalagents_tbl",
|
|
"baremetalagent",
|
|
"baremetalagents",
|
|
)}
|
|
BaremetalagentManager.SetVirtualObject(BaremetalagentManager)
|
|
}
|
|
|
|
func (self *SBaremetalagentManager) AllowListItems(ctx context.Context, userCred mcclient.TokenCredential, query jsonutils.JSONObject) bool {
|
|
return db.IsAdminAllowList(userCred, self)
|
|
}
|
|
|
|
func (self *SBaremetalagentManager) AllowCreateItem(ctx context.Context, userCred mcclient.TokenCredential, query jsonutils.JSONObject, data jsonutils.JSONObject) bool {
|
|
return db.IsAdminAllowCreate(userCred, self)
|
|
}
|
|
|
|
func (self *SBaremetalagent) AllowGetDetails(ctx context.Context, userCred mcclient.TokenCredential, query jsonutils.JSONObject) bool {
|
|
return db.IsAdminAllowGet(userCred, self)
|
|
}
|
|
|
|
func (self *SBaremetalagent) AllowUpdateItem(ctx context.Context, userCred mcclient.TokenCredential) bool {
|
|
return db.IsAdminAllowUpdate(userCred, self)
|
|
}
|
|
|
|
func (self *SBaremetalagent) AllowDeleteItem(ctx context.Context, userCred mcclient.TokenCredential, query jsonutils.JSONObject, data jsonutils.JSONObject) bool {
|
|
return db.IsAdminAllowDelete(userCred, self)
|
|
}
|
|
|
|
func (self *SBaremetalagent) ValidateDeleteCondition(ctx context.Context) error {
|
|
if self.Status == api.BAREMETAL_AGENT_ENABLED {
|
|
return fmt.Errorf("Cannot delete in status %s", self.Status)
|
|
}
|
|
return self.SStandaloneResourceBase.ValidateDeleteCondition(ctx)
|
|
}
|
|
|
|
func (self *SBaremetalagent) ValidateUpdateData(ctx context.Context, userCred mcclient.TokenCredential, query jsonutils.JSONObject, data *jsonutils.JSONDict) (*jsonutils.JSONDict, error) {
|
|
mangerUri, err := data.GetString("manager_uri")
|
|
if err == nil {
|
|
count, err := BaremetalagentManager.Query().Equals("manager_uri", mangerUri).
|
|
NotEquals("id", self.Id).CountWithError()
|
|
if err != nil {
|
|
return nil, httperrors.NewInternalServerError("check agent uniqness fail %s", err)
|
|
}
|
|
if count > 0 {
|
|
return nil, httperrors.NewConflictError("Conflict manager_uri %s", mangerUri)
|
|
}
|
|
}
|
|
return self.SStandaloneResourceBase.ValidateUpdateData(ctx, userCred, query, data)
|
|
}
|
|
|
|
func (manager *SBaremetalagentManager) ValidateCreateData(ctx context.Context, userCred mcclient.TokenCredential, ownerId mcclient.IIdentityProvider, query jsonutils.JSONObject, data *jsonutils.JSONDict) (*jsonutils.JSONDict, error) {
|
|
mangerUri, _ := data.GetString("manager_uri")
|
|
count, err := manager.Query().Equals("manager_uri", mangerUri).CountWithError()
|
|
if err != nil {
|
|
return nil, httperrors.NewInternalServerError("check agent uniqness fail %s", err)
|
|
}
|
|
if count > 0 {
|
|
return nil, httperrors.NewDuplicateResourceError("Duplicate manager_uri %s", mangerUri)
|
|
}
|
|
return manager.SStandaloneResourceBaseManager.ValidateCreateData(ctx, userCred, ownerId, query, data)
|
|
}
|
|
|
|
func (self *SBaremetalagent) AllowPerformEnable(ctx context.Context, userCred mcclient.TokenCredential, query jsonutils.JSONObject, data jsonutils.JSONObject) bool {
|
|
return db.IsAdminAllowPerform(userCred, self, "enable")
|
|
}
|
|
|
|
func (self *SBaremetalagent) PerformEnable(ctx context.Context, userCred mcclient.TokenCredential, query jsonutils.JSONObject, data jsonutils.JSONObject) (jsonutils.JSONObject, error) {
|
|
if self.Status != api.BAREMETAL_AGENT_ENABLED {
|
|
db.Update(self, func() error {
|
|
self.Status = api.BAREMETAL_AGENT_ENABLED
|
|
return nil
|
|
})
|
|
db.OpsLog.LogEvent(self, db.ACT_ENABLE, "", userCred)
|
|
}
|
|
return nil, nil
|
|
}
|
|
|
|
func (self *SBaremetalagent) AllowPerformDisable(ctx context.Context, userCred mcclient.TokenCredential, query jsonutils.JSONObject, data jsonutils.JSONObject) bool {
|
|
return db.IsAdminAllowPerform(userCred, self, "disable")
|
|
}
|
|
|
|
func (self *SBaremetalagent) PerformDisable(ctx context.Context, userCred mcclient.TokenCredential, query jsonutils.JSONObject, data jsonutils.JSONObject) (jsonutils.JSONObject, error) {
|
|
if self.Status != api.BAREMETAL_AGENT_DISABLED {
|
|
db.Update(self, func() error {
|
|
self.Status = api.BAREMETAL_AGENT_DISABLED
|
|
return nil
|
|
})
|
|
db.OpsLog.LogEvent(self, db.ACT_DISABLE, "", userCred)
|
|
}
|
|
return nil, nil
|
|
}
|
|
|
|
func (self *SBaremetalagent) AllowPerformOnline(ctx context.Context, userCred mcclient.TokenCredential, query jsonutils.JSONObject, data jsonutils.JSONObject) bool {
|
|
return db.IsAdminAllowPerform(userCred, self, "online")
|
|
}
|
|
|
|
func (self *SBaremetalagent) PerformOnline(ctx context.Context, userCred mcclient.TokenCredential, query jsonutils.JSONObject, data jsonutils.JSONObject) (jsonutils.JSONObject, error) {
|
|
if self.Status == api.BAREMETAL_AGENT_OFFLINE {
|
|
db.Update(self, func() error {
|
|
self.Status = api.BAREMETAL_AGENT_ENABLED
|
|
return nil
|
|
})
|
|
db.OpsLog.LogEvent(self, db.ACT_ONLINE, "", userCred)
|
|
}
|
|
return nil, nil
|
|
}
|
|
|
|
func (self *SBaremetalagent) AllowPerformOffline(ctx context.Context, userCred mcclient.TokenCredential, query jsonutils.JSONObject, data jsonutils.JSONObject) bool {
|
|
return db.IsAdminAllowPerform(userCred, self, "offline")
|
|
}
|
|
|
|
func (self *SBaremetalagent) PerformOffline(ctx context.Context, userCred mcclient.TokenCredential, query jsonutils.JSONObject, data jsonutils.JSONObject) (jsonutils.JSONObject, error) {
|
|
if self.Status == api.BAREMETAL_AGENT_ENABLED {
|
|
db.Update(self, func() error {
|
|
self.Status = api.BAREMETAL_AGENT_OFFLINE
|
|
return nil
|
|
})
|
|
db.OpsLog.LogEvent(self, db.ACT_OFFLINE, "", userCred)
|
|
}
|
|
return nil, nil
|
|
}
|
|
|
|
func (self *SBaremetalagent) GetZone() *SZone {
|
|
if len(self.ZoneId) > 0 && regutils.MatchUUIDExact(self.ZoneId) {
|
|
return ZoneManager.FetchZoneById(self.ZoneId)
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (self *SBaremetalagent) GetExtraDetails(ctx context.Context, userCred mcclient.TokenCredential, query jsonutils.JSONObject) (*jsonutils.JSONDict, error) {
|
|
extra, err := self.SStandaloneResourceBase.GetExtraDetails(ctx, userCred, query)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
zone := self.GetZone()
|
|
if zone != nil {
|
|
extra.Set("zone", jsonutils.NewString(zone.GetName()))
|
|
}
|
|
return extra, nil
|
|
}
|
|
|
|
func (manager *SBaremetalagentManager) GetAgent(agentType api.TAgentType, zoneId string) *SBaremetalagent {
|
|
q := manager.Query().Equals("agent_type", agentType).Equals("zone_id", zoneId).Asc("created_at")
|
|
agents := make([]SBaremetalagent, 0)
|
|
err := db.FetchModelObjects(manager, q, &agents)
|
|
if err != nil {
|
|
if err != sql.ErrNoRows {
|
|
log.Errorf("GetAgent query fail %s", err)
|
|
}
|
|
return nil
|
|
}
|
|
if len(agents) == 0 {
|
|
return nil
|
|
}
|
|
for i := range agents {
|
|
if agents[i].Status == api.BAREMETAL_AGENT_ENABLED {
|
|
return &agents[i]
|
|
}
|
|
}
|
|
return &agents[0]
|
|
}
|