diff --git a/adaptor/PCM-CORE/api/etc/pcmcore-api.yaml b/adaptor/PCM-CORE/api/etc/pcm-core-api-template.yaml similarity index 72% rename from adaptor/PCM-CORE/api/etc/pcmcore-api.yaml rename to adaptor/PCM-CORE/api/etc/pcm-core-api-template.yaml index 887105a..7bfdaa7 100644 --- a/adaptor/PCM-CORE/api/etc/pcmcore-api.yaml +++ b/adaptor/PCM-CORE/api/etc/pcm-core-api-template.yaml @@ -10,7 +10,7 @@ ScheduleHpcConf: CloudTopic: Schedule-Cloud-Topic DB: - DataSource: root:uJpLd6u-J?HC1@(106.53.150.192:3306)/pcm?parseTime=true + DataSource: Redis: Host: localhost:6379 diff --git a/adaptor/PCM-CORE/api/internal/handler/listcenterhandler.go b/adaptor/PCM-CORE/api/internal/handler/listcenterhandler.go new file mode 100644 index 0000000..0438d2c --- /dev/null +++ b/adaptor/PCM-CORE/api/internal/handler/listcenterhandler.go @@ -0,0 +1,21 @@ +package handler + +import ( + "net/http" + + "PCM/adaptor/PCM-CORE/api/internal/logic" + "PCM/adaptor/PCM-CORE/api/internal/svc" + "github.com/zeromicro/go-zero/rest/httpx" +) + +func listCenterHandler(svcCtx *svc.ServiceContext) http.HandlerFunc { + return func(w http.ResponseWriter, r *http.Request) { + l := logic.NewListCenterLogic(r.Context(), svcCtx) + resp, err := l.ListCenter() + if err != nil { + httpx.ErrorCtx(r.Context(), w, err) + } else { + httpx.OkJsonCtx(r.Context(), w, resp) + } + } +} diff --git a/adaptor/PCM-CORE/api/internal/handler/routes.go b/adaptor/PCM-CORE/api/internal/handler/routes.go index d6d222e..6ce226b 100644 --- a/adaptor/PCM-CORE/api/internal/handler/routes.go +++ b/adaptor/PCM-CORE/api/internal/handler/routes.go @@ -22,6 +22,11 @@ func RegisterHandlers(server *rest.Server, serverCtx *svc.ServiceContext) { Path: "/taskList", Handler: TaskListHandler(serverCtx), }, + { + Method: http.MethodGet, + Path: "/listCenter", + Handler: listCenterHandler(serverCtx), + }, }, ) } diff --git a/adaptor/PCM-CORE/api/internal/logic/listcenterlogic.go b/adaptor/PCM-CORE/api/internal/logic/listcenterlogic.go new file mode 100644 index 0000000..433a2ed --- /dev/null +++ b/adaptor/PCM-CORE/api/internal/logic/listcenterlogic.go @@ -0,0 +1,85 @@ +package logic + +import ( + "PCM/adaptor/PCM-CORE/api/internal/svc" + "PCM/adaptor/PCM-CORE/api/internal/types" + "PCM/adaptor/PCM-CORE/model" + "context" + "github.com/zeromicro/go-zero/core/logx" +) + +type ListCenterLogic struct { + logx.Logger + ctx context.Context + svcCtx *svc.ServiceContext +} + +func NewListCenterLogic(ctx context.Context, svcCtx *svc.ServiceContext) *ListCenterLogic { + return &ListCenterLogic{ + Logger: logx.WithContext(ctx), + ctx: ctx, + svcCtx: svcCtx, + } +} + +func (l *ListCenterLogic) ListCenter() (*types.ListCenterResp, error) { + var centers []types.Center + var centersModel *[]model.CenterOverview + + //var centersModel []model.ComputeCenter + var resp types.ListCenterResp + + centersModel, _ = l.svcCtx.CenterOverviewModel.FindAll(l.ctx) + + var centerModelV = *centersModel + + for _, overview := range centerModelV { + var center types.Center + center.Id = overview.Id.Int64 + center.CenterSource = overview.CenterSource.String + center.SourceId = overview.SourceId.String + center.Name = overview.Name.String + center.Description = overview.Description.String + center.Type = overview.Type.String + center.Area = overview.Area.String + center.City = overview.City.String + center.Longitude = overview.Longitude.Float64 + center.Latitude = overview.Latitude.Float64 + center.Status = overview.Status.String + center.UserNum = overview.UserNum.Int64 + center.CloudClusterNum = overview.CloudClusterNum.Int64 + center.AiClusterNum = overview.AiClusterNum.Int64 + center.HpcClusterNum = overview.HpcClusterNum.Int64 + center.CloudNodeNum = overview.CloudNodeNum.Int64 + center.AiNodeNum = overview.AiNodeNum.Int64 + center.HpcNodeNum = overview.HpcNodeNum.Int64 + center.CloudCpuNum = overview.CloudCpuNum.Int64 + center.AiCpuNum = overview.AiCpuNum.Int64 + center.HpcCpuNum = overview.HpcCpuNum.Int64 + center.CloudGpuNum = overview.CloudGpuNum.Int64 + center.AiGpuNum = overview.AiGpuNum.Int64 + center.HpcGpuNum = overview.HpcGpuNum.Int64 + center.CloudMngFlops = overview.CloudMngFlops.Int64 + center.AiMngFlops = overview.AiMngFlops.Int64 + center.HpcMngFlops = overview.HpcMngFlops.Int64 + center.CloudUmngFlops = overview.CloudUmngFlops.Int64 + center.AiUmngFlops = overview.AiUmngFlops.Int64 + center.HpcUmngFlops = overview.HpcUmngFlops.Int64 + center.CloudMngStorage = overview.CloudMngStorage.Int64 + center.AiMngStorage = overview.AiMngStorage.Int64 + center.HpcMngStorage = overview.HpcMngStorage.Int64 + center.CloudUmngStorage = overview.CloudUmngStorage.Int64 + center.AiUmngStorage = overview.AiUmngStorage.Int64 + center.HpcUmngStorage = overview.HpcUmngStorage.Int64 + center.Edwc = overview.Edwc.Bool + center.Ydyl = overview.Ydyl.Bool + center.DeletedFlag = overview.DeletedFlag.Int64 + centers = append(centers, center) + } + + resp.Code = 200 + resp.Msg = "success" + resp.Data.TotalCount = len(centers) + resp.Data.Centers = centers + return &resp, nil +} diff --git a/adaptor/PCM-CORE/api/internal/svc/servicecontext.go b/adaptor/PCM-CORE/api/internal/svc/servicecontext.go index 4cfe42b..1f2375a 100644 --- a/adaptor/PCM-CORE/api/internal/svc/servicecontext.go +++ b/adaptor/PCM-CORE/api/internal/svc/servicecontext.go @@ -15,6 +15,7 @@ type ServiceContext struct { RedisClient *redis.Client ScheduleCloudClient *kq.Pusher TaskModel model.TaskModel + CenterOverviewModel model.CenterOverviewModel SqlConn sqlx.SqlConn Db *sql.DB } @@ -30,6 +31,7 @@ func NewServiceContext(c config.Config) *ServiceContext { }), SqlConn: sqlx.NewMysql(c.DB.DataSource), TaskModel: model.NewTaskModel(sqlConn, c.Cache), + CenterOverviewModel: model.NewCenterOverviewModel(sqlConn, c.Cache), ScheduleHpcClient: kq.NewPusher(c.ScheduleHpcConf.Brokers, c.ScheduleHpcConf.HpcTopic), ScheduleCloudClient: kq.NewPusher(c.ScheduleHpcConf.Brokers, c.ScheduleHpcConf.CloudTopic), } diff --git a/adaptor/PCM-CORE/api/internal/types/types.go b/adaptor/PCM-CORE/api/internal/types/types.go index 863baa6..0747852 100644 --- a/adaptor/PCM-CORE/api/internal/types/types.go +++ b/adaptor/PCM-CORE/api/internal/types/types.go @@ -34,3 +34,56 @@ type Task struct { ServiceName string `json:"serviceName"` SynergyStatus string `json:"synergyStatus"` } + +type ListCenterResp struct { + Code int32 `json:"code"` + Msg string `json:"msg"` + Data CenterData `json:"data"` +} + +type CenterData struct { + TotalCount int `json:"totalCount"` + Centers []Center `json:"centers"` +} + +type Center struct { + Id int64 `json:"id"` + CenterSource string `json:"centerSource"` + SourceId string `json:"sourceId"` + Name string `json:"name"` + Description string `json:"description"` + Type string `json:"type"` + Area string `json:"area"` + City string `json:"city"` + Longitude float64 `json:"longitude"` + Latitude float64 `json:"latitude"` + Status string `json:"status"` + UserNum int64 `json:"userNum"` + DeletedFlag int64 `json:"deletedFlag"` + CloudClusterNum int64 `json:"cloudClusterNum"` + CloudNodeNum int64 `json:"cloudNodeNum"` + CloudCpuNum int64 `json:"cloudCpuNum"` + CloudGpuNum int64 `json:"cloudGpuNum"` + CloudMngFlops int64 `json:"cloudMngFlops"` + CloudUmngFlops int64 `json:"cloudUmngFlops"` + CloudMngStorage int64 `json:"cloudMngStorage"` + CloudUmngStorage int64 `json:"cloudUmngStorage"` + AiClusterNum int64 `json:"aiClusterNum"` + AiNodeNum int64 `json:"aiNodeNum"` + AiCpuNum int64 `json:"aiCpuNum"` + AiGpuNum int64 `json:"aiGpuNum"` + AiMngFlops int64 `json:"aiMngFlops"` + AiUmngFlops int64 `json:"aiUmngFlops"` + AiMngStorage int64 `json:"aiMngStorage"` + AiUmngStorage int64 `json:"aiUmngStorage"` + HpcClusterNum int64 `json:"hpcClusterNum"` + HpcNodeNum int64 `json:"hpcNodeNum"` + HpcCpuNum int64 `json:"hpcCpuNum"` + HpcGpuNum int64 `json:"hpcGpuNum"` + HpcMngFlops int64 `json:"hpcMngFlops"` + HpcUmngFlops int64 `json:"hpcUmngFlops"` + HpcMngStorage int64 `json:"hpcMngStorage"` + HpcUmngStorage int64 `json:"hpcUmngStorage"` + Edwc bool `json:"edwc"` + Ydyl bool `json:"ydyl"` +} diff --git a/adaptor/PCM-CORE/api/pcm-core.api b/adaptor/PCM-CORE/api/pcm-core.api index 3139cd4..75926f6 100644 --- a/adaptor/PCM-CORE/api/pcm-core.api +++ b/adaptor/PCM-CORE/api/pcm-core.api @@ -42,10 +42,66 @@ type ( } ) +type ( + listCenterResp { + Code int32 `json:"code"` + Msg string `json:"msg"` + Data CenterData `json:"data"` + } + CenterData { + TotalCount int `json:"totalCount"` + Centers []Center `json:"centers"` + } + Center { + Id int64 `json:"id"` + CenterSource string `json:"centerSource"` + SourceId string `json:"sourceId"` + Name string `json:"name"` + Description string `json:"description"` + Type string `json:"type"` + Area string `json:"area"` + City string `json:"city"` + Longitude float64 `json:"longitude"` + Latitude float64 `json:"latitude"` + Status string `json:"status"` + UserNum int64 `json:"userNum"` + DeletedFlag int64 `json:"deletedFlag"` + CloudClusterNum int64 `json:"cloudClusterNum"` + CloudNodeNum int64 `json:"cloudNodeNum"` + CloudCpuNum int64 `json:"cloudCpuNum"` + CloudGpuNum int64 `json:"cloudGpuNum"` + CloudMngFlops int64 `json:"cloudMngFlops"` + CloudUmngFlops int64 `json:"cloudUmngFlops"` + CloudMngStorage int64 `json:"cloudMngStorage"` + CloudUmngStorage int64 `json:"cloudUmngStorage"` + AiClusterNum int64 `json:"aiClusterNum"` + AiNodeNum int64 `json:"aiNodeNum"` + AiCpuNum int64 `json:"aiCpuNum"` + AiGpuNum int64 `json:"aiGpuNum"` + AiMngFlops int64 `json:"aiMngFlops"` + AiUmngFlops int64 `json:"aiUmngFlops"` + AiMngStorage int64 `json:"aiMngStorage"` + AiUmngStorage int64 `json:"aiUmngStorage"` + HpcClusterNum int64 `json:"hpcClusterNum"` + HpcNodeNum int64 `json:"hpcNodeNum"` + HpcCpuNum int64 `json:"hpcCpuNum"` + HpcGpuNum int64 `json:"hpcGpuNum"` + HpcMngFlops int64 `json:"hpcMngFlops"` + HpcUmngFlops int64 `json:"hpcUmngFlops"` + HpcMngStorage int64 `json:"hpcMngStorage"` + HpcUmngStorage int64 `json:"hpcUmngStorage"` + Edwc bool `json:"edwc"` + Ydyl bool `json:"ydyl"` + } +) + service pcmcore-api { @handler scheduleTaskHandler post /scheduleTask (scheduleTaskReq) returns (scheduleTaskResp) @handler TaskListHandler get /taskList () returns (taskListResp) + + @handler listCenterHandler + get /listCenter () returns (listCenterResp) } \ No newline at end of file diff --git a/adaptor/PCM-CORE/model/aiCenterModel.go b/adaptor/PCM-CORE/model/aiCenterModel.go new file mode 100644 index 0000000..c70e3c3 --- /dev/null +++ b/adaptor/PCM-CORE/model/aiCenterModel.go @@ -0,0 +1,27 @@ +package model + +import ( + "github.com/zeromicro/go-zero/core/stores/cache" + "github.com/zeromicro/go-zero/core/stores/sqlx" +) + +var _ AiCenterModel = (*customAiCenterModel)(nil) + +type ( + // AiCenterModel is an interface to be customized, add more methods here, + // and implement the added methods in customAiCenterModel. + AiCenterModel interface { + aiCenterModel + } + + customAiCenterModel struct { + *defaultAiCenterModel + } +) + +// NewAiCenterModel returns a model for the database table. +func NewAiCenterModel(conn sqlx.SqlConn, c cache.CacheConf) AiCenterModel { + return &customAiCenterModel{ + defaultAiCenterModel: newAiCenterModel(conn, c), + } +} diff --git a/adaptor/PCM-CORE/model/aiCenterModel_gen.go b/adaptor/PCM-CORE/model/aiCenterModel_gen.go new file mode 100644 index 0000000..362f931 --- /dev/null +++ b/adaptor/PCM-CORE/model/aiCenterModel_gen.go @@ -0,0 +1,115 @@ +// Code generated by goctl. DO NOT EDIT. + +package model + +import ( + "context" + "database/sql" + "fmt" + "strings" + + "github.com/zeromicro/go-zero/core/stores/builder" + "github.com/zeromicro/go-zero/core/stores/cache" + "github.com/zeromicro/go-zero/core/stores/sqlc" + "github.com/zeromicro/go-zero/core/stores/sqlx" + "github.com/zeromicro/go-zero/core/stringx" +) + +var ( + aiCenterFieldNames = builder.RawFieldNames(&AiCenter{}) + aiCenterRows = strings.Join(aiCenterFieldNames, ",") + aiCenterRowsExpectAutoSet = strings.Join(stringx.Remove(aiCenterFieldNames, "`create_at`", "`create_time`", "`created_at`", "`update_at`", "`update_time`", "`updated_at`"), ",") + aiCenterRowsWithPlaceHolder = strings.Join(stringx.Remove(aiCenterFieldNames, "`id`", "`create_at`", "`create_time`", "`created_at`", "`update_at`", "`update_time`", "`updated_at`"), "=?,") + "=?" + + cachePcmAiCenterIdPrefix = "cache:pcm:aiCenter:id:" +) + +type ( + aiCenterModel interface { + Insert(ctx context.Context, data *AiCenter) (sql.Result, error) + FindOne(ctx context.Context, id int64) (*AiCenter, error) + Update(ctx context.Context, data *AiCenter) error + Delete(ctx context.Context, id int64) error + } + + defaultAiCenterModel struct { + sqlc.CachedConn + table string + } + + AiCenter struct { + Id int64 `db:"id"` // 平台唯一id + ClusterNum sql.NullInt64 `db:"cluster_num"` // 集群数量 + NodeNum sql.NullInt64 `db:"node_num"` // 节点数量 + CpuNum sql.NullInt64 `db:"cpu_num"` // CPU核数 + GpuNum sql.NullInt64 `db:"gpu_num"` // GPU卡数 + ManagedFlops sql.NullString `db:"managed_flops"` // 已接入算力 + UnmanagedFlops sql.NullString `db:"unmanaged_flops"` // 未接入算力 + ManagedStorage sql.NullString `db:"managed_storage"` // 已接入存储 + UnmanagedStorage sql.NullString `db:"unmanaged_storage"` // 未接入存储 + } +) + +func newAiCenterModel(conn sqlx.SqlConn, c cache.CacheConf) *defaultAiCenterModel { + return &defaultAiCenterModel{ + CachedConn: sqlc.NewConn(conn, c), + table: "`ai_center`", + } +} + +func (m *defaultAiCenterModel) Delete(ctx context.Context, id int64) error { + pcmAiCenterIdKey := fmt.Sprintf("%s%v", cachePcmAiCenterIdPrefix, id) + _, err := m.ExecCtx(ctx, func(ctx context.Context, conn sqlx.SqlConn) (result sql.Result, err error) { + query := fmt.Sprintf("delete from %s where `id` = ?", m.table) + return conn.ExecCtx(ctx, query, id) + }, pcmAiCenterIdKey) + return err +} + +func (m *defaultAiCenterModel) FindOne(ctx context.Context, id int64) (*AiCenter, error) { + pcmAiCenterIdKey := fmt.Sprintf("%s%v", cachePcmAiCenterIdPrefix, id) + var resp AiCenter + err := m.QueryRowCtx(ctx, &resp, pcmAiCenterIdKey, func(ctx context.Context, conn sqlx.SqlConn, v interface{}) error { + query := fmt.Sprintf("select %s from %s where `id` = ? limit 1", aiCenterRows, m.table) + return conn.QueryRowCtx(ctx, v, query, id) + }) + switch err { + case nil: + return &resp, nil + case sqlc.ErrNotFound: + return nil, ErrNotFound + default: + return nil, err + } +} + +func (m *defaultAiCenterModel) Insert(ctx context.Context, data *AiCenter) (sql.Result, error) { + pcmAiCenterIdKey := fmt.Sprintf("%s%v", cachePcmAiCenterIdPrefix, data.Id) + ret, err := m.ExecCtx(ctx, func(ctx context.Context, conn sqlx.SqlConn) (result sql.Result, err error) { + query := fmt.Sprintf("insert into %s (%s) values (?, ?, ?, ?, ?, ?, ?, ?, ?)", m.table, aiCenterRowsExpectAutoSet) + return conn.ExecCtx(ctx, query, data.Id, data.ClusterNum, data.NodeNum, data.CpuNum, data.GpuNum, data.ManagedFlops, data.UnmanagedFlops, data.ManagedStorage, data.UnmanagedStorage) + }, pcmAiCenterIdKey) + return ret, err +} + +func (m *defaultAiCenterModel) Update(ctx context.Context, data *AiCenter) error { + pcmAiCenterIdKey := fmt.Sprintf("%s%v", cachePcmAiCenterIdPrefix, data.Id) + _, err := m.ExecCtx(ctx, func(ctx context.Context, conn sqlx.SqlConn) (result sql.Result, err error) { + query := fmt.Sprintf("update %s set %s where `id` = ?", m.table, aiCenterRowsWithPlaceHolder) + return conn.ExecCtx(ctx, query, data.ClusterNum, data.NodeNum, data.CpuNum, data.GpuNum, data.ManagedFlops, data.UnmanagedFlops, data.ManagedStorage, data.UnmanagedStorage, data.Id) + }, pcmAiCenterIdKey) + return err +} + +func (m *defaultAiCenterModel) formatPrimary(primary interface{}) string { + return fmt.Sprintf("%s%v", cachePcmAiCenterIdPrefix, primary) +} + +func (m *defaultAiCenterModel) queryPrimary(ctx context.Context, conn sqlx.SqlConn, v, primary interface{}) error { + query := fmt.Sprintf("select %s from %s where `id` = ? limit 1", aiCenterRows, m.table) + return conn.QueryRowCtx(ctx, v, query, primary) +} + +func (m *defaultAiCenterModel) tableName() string { + return m.table +} diff --git a/adaptor/PCM-CORE/model/centerOverViewModel.go b/adaptor/PCM-CORE/model/centerOverViewModel.go new file mode 100644 index 0000000..7c10bea --- /dev/null +++ b/adaptor/PCM-CORE/model/centerOverViewModel.go @@ -0,0 +1,27 @@ +package model + +import ( + "github.com/zeromicro/go-zero/core/stores/cache" + "github.com/zeromicro/go-zero/core/stores/sqlx" +) + +var _ CenterOverviewModel = (*customCenterOverviewModel)(nil) + +type ( + // CenterOverviewModel is an interface to be customized, add more methods here, + // and implement the added methods in customCenterOverviewModel. + CenterOverviewModel interface { + centerOverviewModel + } + + customCenterOverviewModel struct { + *defaultCenterOverviewModel + } +) + +// NewCenterOverviewModel returns a model for the database table. +func NewCenterOverviewModel(conn sqlx.SqlConn, c cache.CacheConf) CenterOverviewModel { + return &customCenterOverviewModel{ + defaultCenterOverviewModel: newCenterOverviewModel(conn, c), + } +} diff --git a/adaptor/PCM-CORE/model/centerOverviewModel_gen.go b/adaptor/PCM-CORE/model/centerOverviewModel_gen.go new file mode 100644 index 0000000..fdcc0b7 --- /dev/null +++ b/adaptor/PCM-CORE/model/centerOverviewModel_gen.go @@ -0,0 +1,91 @@ +package model + +import ( + "context" + "database/sql" + "github.com/Masterminds/squirrel" + "github.com/zeromicro/go-zero/core/stores/cache" + "github.com/zeromicro/go-zero/core/stores/sqlc" + "github.com/zeromicro/go-zero/core/stores/sqlx" +) + +type ( + centerOverviewModel interface { + FindAll(ctx context.Context) (*[]CenterOverview, error) + } + + defaultCenterOverviewModel struct { + sqlc.CachedConn + table string + } + + CenterOverview struct { + Id sql.NullInt64 `json:"id"` + CenterSource sql.NullString `json:"centerSource"` + SourceId sql.NullString `json:"sourceId"` + Name sql.NullString `json:"name"` + Description sql.NullString `json:"description"` + Type sql.NullString `json:"type"` + Area sql.NullString `json:"area"` + City sql.NullString `json:"city"` + Longitude sql.NullFloat64 `json:"longitude"` + Latitude sql.NullFloat64 `json:"latitude"` + Status sql.NullString `json:"status"` + UserNum sql.NullInt64 `json:"userNum"` + DeletedFlag sql.NullInt64 `json:"deletedFlag"` + CloudClusterNum sql.NullInt64 `json:"cloudClusterNum"` + CloudNodeNum sql.NullInt64 `json:"cloudNodeNum"` + CloudCpuNum sql.NullInt64 `json:"cloudCpuNum"` + CloudGpuNum sql.NullInt64 `json:"cloudGpuNum"` + CloudMngFlops sql.NullInt64 `json:"cloudMngFlops"` + CloudUmngFlops sql.NullInt64 `json:"cloudUmngFlops"` + CloudMngStorage sql.NullInt64 `json:"CloudMngStorage"` + CloudUmngStorage sql.NullInt64 `json:"CloudUmngStorage"` + AiClusterNum sql.NullInt64 `json:"aiClusterNum"` + AiNodeNum sql.NullInt64 `json:"aiNodeNum"` + AiCpuNum sql.NullInt64 `json:"aiCpuNum"` + AiGpuNum sql.NullInt64 `json:"aiGpuNum"` + AiMngFlops sql.NullInt64 `json:"aiMngFlops"` + AiUmngFlops sql.NullInt64 `json:"aiUmngFlops"` + AiMngStorage sql.NullInt64 `json:"aiMngStorage"` + AiUmngStorage sql.NullInt64 `json:"aiUmngStorage"` + HpcClusterNum sql.NullInt64 `json:"hpcClusterNum"` + HpcNodeNum sql.NullInt64 `json:"hpcNodeNum"` + HpcCpuNum sql.NullInt64 `json:"hpcCpuNum"` + HpcGpuNum sql.NullInt64 `json:"hpcGpuNum"` + HpcMngFlops sql.NullInt64 `json:"hpcMngFlops"` + HpcUmngFlops sql.NullInt64 `json:"hpcUmngFlops"` + HpcMngStorage sql.NullInt64 `json:"hpcMngStorage"` + HpcUmngStorage sql.NullInt64 `json:"hpcUmngStorage"` + Edwc sql.NullBool `json:"edwc"` + Ydyl sql.NullBool `json:"ydyl"` + } +) + +func newCenterOverviewModel(conn sqlx.SqlConn, c cache.CacheConf) *defaultCenterOverviewModel { + return &defaultCenterOverviewModel{ + CachedConn: sqlc.NewConn(conn, c), + table: "`compute_center`", + } +} + +func (m *defaultCenterOverviewModel) FindAll(ctx context.Context) (*[]CenterOverview, error) { + var centers []CenterOverview + + centerSelect := squirrel.Select(`cc.*`, `ac.cluster_num`, `ac.node_num`, `ac.cpu_num`, `ac.gpu_num`, `ac.managed_flops`, `ac.unmanaged_flops`, `ac.managed_storage`, `ac.unmanaged_storage`, `hc.cluster_num`, `c.node_num`, `hc.cpu_num`, `hc.gpu_num`, `hc.managed_flops`, `hc.unmanaged_flops`, `hc.managed_storage`, `hc.unmanaged_storage`, `c.cluster_num`, `c.node_num`, `c.cpu_num`, `c.gpu_num`, `c.managed_flops`, `c.unmanaged_flops`, `c.managed_storage`, `c.unmanaged_storage`, `ct.edwc`, `ct.ydyl`). + From("compute_center cc"). + LeftJoin(`ai_center ac on cc.id = ac.id`). + LeftJoin(`hpc_center hc on cc.id = hc.id`). + LeftJoin(`cloud_center c on cc.id = c.id`). + LeftJoin(`center_tag ct on cc.id = ct.id`). + Where(`cc.deleted_flag = 0`) + query, values, _ := centerSelect.ToSql() + + err := m.QueryRowsNoCacheCtx(ctx, ¢ers, query, values...) + if err != nil { + return nil, err + } + + return ¢ers, nil + +} diff --git a/adaptor/PCM-CORE/model/centerTagModel.go b/adaptor/PCM-CORE/model/centerTagModel.go new file mode 100644 index 0000000..a7a0cd6 --- /dev/null +++ b/adaptor/PCM-CORE/model/centerTagModel.go @@ -0,0 +1,27 @@ +package model + +import ( + "github.com/zeromicro/go-zero/core/stores/cache" + "github.com/zeromicro/go-zero/core/stores/sqlx" +) + +var _ CenterTagModel = (*customCenterTagModel)(nil) + +type ( + // CenterTagModel is an interface to be customized, add more methods here, + // and implement the added methods in customCenterTagModel. + CenterTagModel interface { + centerTagModel + } + + customCenterTagModel struct { + *defaultCenterTagModel + } +) + +// NewCenterTagModel returns a model for the database table. +func NewCenterTagModel(conn sqlx.SqlConn, c cache.CacheConf) CenterTagModel { + return &customCenterTagModel{ + defaultCenterTagModel: newCenterTagModel(conn, c), + } +} diff --git a/adaptor/PCM-CORE/model/centerTagModel_gen.go b/adaptor/PCM-CORE/model/centerTagModel_gen.go new file mode 100644 index 0000000..f7f4b4d --- /dev/null +++ b/adaptor/PCM-CORE/model/centerTagModel_gen.go @@ -0,0 +1,109 @@ +// Code generated by goctl. DO NOT EDIT. + +package model + +import ( + "context" + "database/sql" + "fmt" + "strings" + + "github.com/zeromicro/go-zero/core/stores/builder" + "github.com/zeromicro/go-zero/core/stores/cache" + "github.com/zeromicro/go-zero/core/stores/sqlc" + "github.com/zeromicro/go-zero/core/stores/sqlx" + "github.com/zeromicro/go-zero/core/stringx" +) + +var ( + centerTagFieldNames = builder.RawFieldNames(&CenterTag{}) + centerTagRows = strings.Join(centerTagFieldNames, ",") + centerTagRowsExpectAutoSet = strings.Join(stringx.Remove(centerTagFieldNames, "`create_at`", "`create_time`", "`created_at`", "`update_at`", "`update_time`", "`updated_at`"), ",") + centerTagRowsWithPlaceHolder = strings.Join(stringx.Remove(centerTagFieldNames, "`id`", "`create_at`", "`create_time`", "`created_at`", "`update_at`", "`update_time`", "`updated_at`"), "=?,") + "=?" + + cachePcmCenterTagIdPrefix = "cache:pcm:centerTag:id:" +) + +type ( + centerTagModel interface { + Insert(ctx context.Context, data *CenterTag) (sql.Result, error) + FindOne(ctx context.Context, id int64) (*CenterTag, error) + Update(ctx context.Context, data *CenterTag) error + Delete(ctx context.Context, id int64) error + } + + defaultCenterTagModel struct { + sqlc.CachedConn + table string + } + + CenterTag struct { + Id int64 `db:"id"` // 平台唯一id + Ydyl sql.NullInt64 `db:"ydyl"` // 一带一路标识 + Edwc sql.NullInt64 `db:"edwc"` // 东数西算标识 + } +) + +func newCenterTagModel(conn sqlx.SqlConn, c cache.CacheConf) *defaultCenterTagModel { + return &defaultCenterTagModel{ + CachedConn: sqlc.NewConn(conn, c), + table: "`center_tag`", + } +} + +func (m *defaultCenterTagModel) Delete(ctx context.Context, id int64) error { + pcmCenterTagIdKey := fmt.Sprintf("%s%v", cachePcmCenterTagIdPrefix, id) + _, err := m.ExecCtx(ctx, func(ctx context.Context, conn sqlx.SqlConn) (result sql.Result, err error) { + query := fmt.Sprintf("delete from %s where `id` = ?", m.table) + return conn.ExecCtx(ctx, query, id) + }, pcmCenterTagIdKey) + return err +} + +func (m *defaultCenterTagModel) FindOne(ctx context.Context, id int64) (*CenterTag, error) { + pcmCenterTagIdKey := fmt.Sprintf("%s%v", cachePcmCenterTagIdPrefix, id) + var resp CenterTag + err := m.QueryRowCtx(ctx, &resp, pcmCenterTagIdKey, func(ctx context.Context, conn sqlx.SqlConn, v interface{}) error { + query := fmt.Sprintf("select %s from %s where `id` = ? limit 1", centerTagRows, m.table) + return conn.QueryRowCtx(ctx, v, query, id) + }) + switch err { + case nil: + return &resp, nil + case sqlc.ErrNotFound: + return nil, ErrNotFound + default: + return nil, err + } +} + +func (m *defaultCenterTagModel) Insert(ctx context.Context, data *CenterTag) (sql.Result, error) { + pcmCenterTagIdKey := fmt.Sprintf("%s%v", cachePcmCenterTagIdPrefix, data.Id) + ret, err := m.ExecCtx(ctx, func(ctx context.Context, conn sqlx.SqlConn) (result sql.Result, err error) { + query := fmt.Sprintf("insert into %s (%s) values (?, ?, ?)", m.table, centerTagRowsExpectAutoSet) + return conn.ExecCtx(ctx, query, data.Id, data.Ydyl, data.Edwc) + }, pcmCenterTagIdKey) + return ret, err +} + +func (m *defaultCenterTagModel) Update(ctx context.Context, data *CenterTag) error { + pcmCenterTagIdKey := fmt.Sprintf("%s%v", cachePcmCenterTagIdPrefix, data.Id) + _, err := m.ExecCtx(ctx, func(ctx context.Context, conn sqlx.SqlConn) (result sql.Result, err error) { + query := fmt.Sprintf("update %s set %s where `id` = ?", m.table, centerTagRowsWithPlaceHolder) + return conn.ExecCtx(ctx, query, data.Ydyl, data.Edwc, data.Id) + }, pcmCenterTagIdKey) + return err +} + +func (m *defaultCenterTagModel) formatPrimary(primary interface{}) string { + return fmt.Sprintf("%s%v", cachePcmCenterTagIdPrefix, primary) +} + +func (m *defaultCenterTagModel) queryPrimary(ctx context.Context, conn sqlx.SqlConn, v, primary interface{}) error { + query := fmt.Sprintf("select %s from %s where `id` = ? limit 1", centerTagRows, m.table) + return conn.QueryRowCtx(ctx, v, query, primary) +} + +func (m *defaultCenterTagModel) tableName() string { + return m.table +} diff --git a/adaptor/PCM-CORE/model/cloudCenterModel.go b/adaptor/PCM-CORE/model/cloudCenterModel.go new file mode 100644 index 0000000..6e9e3ce --- /dev/null +++ b/adaptor/PCM-CORE/model/cloudCenterModel.go @@ -0,0 +1,27 @@ +package model + +import ( + "github.com/zeromicro/go-zero/core/stores/cache" + "github.com/zeromicro/go-zero/core/stores/sqlx" +) + +var _ CloudCenterModel = (*customCloudCenterModel)(nil) + +type ( + // CloudCenterModel is an interface to be customized, add more methods here, + // and implement the added methods in customCloudCenterModel. + CloudCenterModel interface { + cloudCenterModel + } + + customCloudCenterModel struct { + *defaultCloudCenterModel + } +) + +// NewCloudCenterModel returns a model for the database table. +func NewCloudCenterModel(conn sqlx.SqlConn, c cache.CacheConf) CloudCenterModel { + return &customCloudCenterModel{ + defaultCloudCenterModel: newCloudCenterModel(conn, c), + } +} diff --git a/adaptor/PCM-CORE/model/cloudCenterModel_gen.go b/adaptor/PCM-CORE/model/cloudCenterModel_gen.go new file mode 100644 index 0000000..8f1af86 --- /dev/null +++ b/adaptor/PCM-CORE/model/cloudCenterModel_gen.go @@ -0,0 +1,115 @@ +// Code generated by goctl. DO NOT EDIT. + +package model + +import ( + "context" + "database/sql" + "fmt" + "strings" + + "github.com/zeromicro/go-zero/core/stores/builder" + "github.com/zeromicro/go-zero/core/stores/cache" + "github.com/zeromicro/go-zero/core/stores/sqlc" + "github.com/zeromicro/go-zero/core/stores/sqlx" + "github.com/zeromicro/go-zero/core/stringx" +) + +var ( + cloudCenterFieldNames = builder.RawFieldNames(&CloudCenter{}) + cloudCenterRows = strings.Join(cloudCenterFieldNames, ",") + cloudCenterRowsExpectAutoSet = strings.Join(stringx.Remove(cloudCenterFieldNames, "`create_at`", "`create_time`", "`created_at`", "`update_at`", "`update_time`", "`updated_at`"), ",") + cloudCenterRowsWithPlaceHolder = strings.Join(stringx.Remove(cloudCenterFieldNames, "`id`", "`create_at`", "`create_time`", "`created_at`", "`update_at`", "`update_time`", "`updated_at`"), "=?,") + "=?" + + cachePcmCloudCenterIdPrefix = "cache:pcm:cloudCenter:id:" +) + +type ( + cloudCenterModel interface { + Insert(ctx context.Context, data *CloudCenter) (sql.Result, error) + FindOne(ctx context.Context, id int64) (*CloudCenter, error) + Update(ctx context.Context, data *CloudCenter) error + Delete(ctx context.Context, id int64) error + } + + defaultCloudCenterModel struct { + sqlc.CachedConn + table string + } + + CloudCenter struct { + Id int64 `db:"id"` // 平台唯一id + ClusterNum sql.NullInt64 `db:"cluster_num"` // 集群数量 + NodeNum sql.NullInt64 `db:"node_num"` // 节点数量 + CpuNum sql.NullInt64 `db:"cpu_num"` // CPU核数 + GpuNum sql.NullInt64 `db:"gpu_num"` // GPU卡数 + ManagedFlops sql.NullString `db:"managed_flops"` // 已接入算力 + UnmanagedFlops sql.NullString `db:"unmanaged_flops"` // 未接入算力 + ManagedStorage sql.NullString `db:"managed_storage"` // 已接入存储 + UnmanagedStorage sql.NullString `db:"unmanaged_storage"` // 未接入存储 + } +) + +func newCloudCenterModel(conn sqlx.SqlConn, c cache.CacheConf) *defaultCloudCenterModel { + return &defaultCloudCenterModel{ + CachedConn: sqlc.NewConn(conn, c), + table: "`cloud_center`", + } +} + +func (m *defaultCloudCenterModel) Delete(ctx context.Context, id int64) error { + pcmCloudCenterIdKey := fmt.Sprintf("%s%v", cachePcmCloudCenterIdPrefix, id) + _, err := m.ExecCtx(ctx, func(ctx context.Context, conn sqlx.SqlConn) (result sql.Result, err error) { + query := fmt.Sprintf("delete from %s where `id` = ?", m.table) + return conn.ExecCtx(ctx, query, id) + }, pcmCloudCenterIdKey) + return err +} + +func (m *defaultCloudCenterModel) FindOne(ctx context.Context, id int64) (*CloudCenter, error) { + pcmCloudCenterIdKey := fmt.Sprintf("%s%v", cachePcmCloudCenterIdPrefix, id) + var resp CloudCenter + err := m.QueryRowCtx(ctx, &resp, pcmCloudCenterIdKey, func(ctx context.Context, conn sqlx.SqlConn, v interface{}) error { + query := fmt.Sprintf("select %s from %s where `id` = ? limit 1", cloudCenterRows, m.table) + return conn.QueryRowCtx(ctx, v, query, id) + }) + switch err { + case nil: + return &resp, nil + case sqlc.ErrNotFound: + return nil, ErrNotFound + default: + return nil, err + } +} + +func (m *defaultCloudCenterModel) Insert(ctx context.Context, data *CloudCenter) (sql.Result, error) { + pcmCloudCenterIdKey := fmt.Sprintf("%s%v", cachePcmCloudCenterIdPrefix, data.Id) + ret, err := m.ExecCtx(ctx, func(ctx context.Context, conn sqlx.SqlConn) (result sql.Result, err error) { + query := fmt.Sprintf("insert into %s (%s) values (?, ?, ?, ?, ?, ?, ?, ?, ?)", m.table, cloudCenterRowsExpectAutoSet) + return conn.ExecCtx(ctx, query, data.Id, data.ClusterNum, data.NodeNum, data.CpuNum, data.GpuNum, data.ManagedFlops, data.UnmanagedFlops, data.ManagedStorage, data.UnmanagedStorage) + }, pcmCloudCenterIdKey) + return ret, err +} + +func (m *defaultCloudCenterModel) Update(ctx context.Context, data *CloudCenter) error { + pcmCloudCenterIdKey := fmt.Sprintf("%s%v", cachePcmCloudCenterIdPrefix, data.Id) + _, err := m.ExecCtx(ctx, func(ctx context.Context, conn sqlx.SqlConn) (result sql.Result, err error) { + query := fmt.Sprintf("update %s set %s where `id` = ?", m.table, cloudCenterRowsWithPlaceHolder) + return conn.ExecCtx(ctx, query, data.ClusterNum, data.NodeNum, data.CpuNum, data.GpuNum, data.ManagedFlops, data.UnmanagedFlops, data.ManagedStorage, data.UnmanagedStorage, data.Id) + }, pcmCloudCenterIdKey) + return err +} + +func (m *defaultCloudCenterModel) formatPrimary(primary interface{}) string { + return fmt.Sprintf("%s%v", cachePcmCloudCenterIdPrefix, primary) +} + +func (m *defaultCloudCenterModel) queryPrimary(ctx context.Context, conn sqlx.SqlConn, v, primary interface{}) error { + query := fmt.Sprintf("select %s from %s where `id` = ? limit 1", cloudCenterRows, m.table) + return conn.QueryRowCtx(ctx, v, query, primary) +} + +func (m *defaultCloudCenterModel) tableName() string { + return m.table +} diff --git a/adaptor/PCM-CORE/model/computeCenterModel.go b/adaptor/PCM-CORE/model/computeCenterModel.go new file mode 100644 index 0000000..8125c96 --- /dev/null +++ b/adaptor/PCM-CORE/model/computeCenterModel.go @@ -0,0 +1,27 @@ +package model + +import ( + "github.com/zeromicro/go-zero/core/stores/cache" + "github.com/zeromicro/go-zero/core/stores/sqlx" +) + +var _ ComputeCenterModel = (*customComputeCenterModel)(nil) + +type ( + // ComputeCenterModel is an interface to be customized, add more methods here, + // and implement the added methods in customComputeCenterModel. + ComputeCenterModel interface { + computeCenterModel + } + + customComputeCenterModel struct { + *defaultComputeCenterModel + } +) + +// NewComputeCenterModel returns a model for the database table. +func NewComputeCenterModel(conn sqlx.SqlConn, c cache.CacheConf) ComputeCenterModel { + return &customComputeCenterModel{ + defaultComputeCenterModel: newComputeCenterModel(conn, c), + } +} diff --git a/adaptor/PCM-CORE/model/computeCenterModel_gen.go b/adaptor/PCM-CORE/model/computeCenterModel_gen.go new file mode 100644 index 0000000..69b0fdb --- /dev/null +++ b/adaptor/PCM-CORE/model/computeCenterModel_gen.go @@ -0,0 +1,119 @@ +// Code generated by goctl. DO NOT EDIT. + +package model + +import ( + "context" + "database/sql" + "fmt" + "strings" + + "github.com/zeromicro/go-zero/core/stores/builder" + "github.com/zeromicro/go-zero/core/stores/cache" + "github.com/zeromicro/go-zero/core/stores/sqlc" + "github.com/zeromicro/go-zero/core/stores/sqlx" + "github.com/zeromicro/go-zero/core/stringx" +) + +var ( + computeCenterFieldNames = builder.RawFieldNames(&ComputeCenter{}) + computeCenterRows = strings.Join(computeCenterFieldNames, ",") + computeCenterRowsExpectAutoSet = strings.Join(stringx.Remove(computeCenterFieldNames, "`create_at`", "`create_time`", "`created_at`", "`update_at`", "`update_time`", "`updated_at`"), ",") + computeCenterRowsWithPlaceHolder = strings.Join(stringx.Remove(computeCenterFieldNames, "`id`", "`create_at`", "`create_time`", "`created_at`", "`update_at`", "`update_time`", "`updated_at`"), "=?,") + "=?" + + cachePcmComputeCenterIdPrefix = "cache:pcm:computeCenter:id:" +) + +type ( + computeCenterModel interface { + Insert(ctx context.Context, data *ComputeCenter) (sql.Result, error) + FindOne(ctx context.Context, id int64) (*ComputeCenter, error) + Update(ctx context.Context, data *ComputeCenter) error + Delete(ctx context.Context, id int64) error + } + + defaultComputeCenterModel struct { + sqlc.CachedConn + table string + } + + ComputeCenter struct { + Id int64 `db:"id"` // 平台唯一id + CenterSource sql.NullString `db:"center_source"` // 中心来源(鹏城,nudt,公有云等) + SourceId sql.NullString `db:"source_id"` // 数据来源原id + Name sql.NullString `db:"name"` // 中心名称 + Description sql.NullString `db:"description"` // 详细描述 + Type sql.NullString `db:"type"` // 中心类型(云算、智算 or 超算) + Area sql.NullString `db:"area"` // 资源区域 + City sql.NullString `db:"city"` // 所在城市 + Longitude sql.NullFloat64 `db:"longitude"` // 经度 + Latitude sql.NullFloat64 `db:"latitude"` // 纬度 + Status sql.NullString `db:"status"` // 接入状态 + UserNum sql.NullInt64 `db:"user_num"` // 用户数量 + DeletedFlag sql.NullInt64 `db:"deleted_flag"` // 是否删除（0-否，1-是） + } +) + +func newComputeCenterModel(conn sqlx.SqlConn, c cache.CacheConf) *defaultComputeCenterModel { + return &defaultComputeCenterModel{ + CachedConn: sqlc.NewConn(conn, c), + table: "`compute_center`", + } +} + +func (m *defaultComputeCenterModel) Delete(ctx context.Context, id int64) error { + pcmComputeCenterIdKey := fmt.Sprintf("%s%v", cachePcmComputeCenterIdPrefix, id) + _, err := m.ExecCtx(ctx, func(ctx context.Context, conn sqlx.SqlConn) (result sql.Result, err error) { + query := fmt.Sprintf("delete from %s where `id` = ?", m.table) + return conn.ExecCtx(ctx, query, id) + }, pcmComputeCenterIdKey) + return err +} + +func (m *defaultComputeCenterModel) FindOne(ctx context.Context, id int64) (*ComputeCenter, error) { + pcmComputeCenterIdKey := fmt.Sprintf("%s%v", cachePcmComputeCenterIdPrefix, id) + var resp ComputeCenter + err := m.QueryRowCtx(ctx, &resp, pcmComputeCenterIdKey, func(ctx context.Context, conn sqlx.SqlConn, v interface{}) error { + query := fmt.Sprintf("select %s from %s where `id` = ? limit 1", computeCenterRows, m.table) + return conn.QueryRowCtx(ctx, v, query, id) + }) + switch err { + case nil: + return &resp, nil + case sqlc.ErrNotFound: + return nil, ErrNotFound + default: + return nil, err + } +} + +func (m *defaultComputeCenterModel) Insert(ctx context.Context, data *ComputeCenter) (sql.Result, error) { + pcmComputeCenterIdKey := fmt.Sprintf("%s%v", cachePcmComputeCenterIdPrefix, data.Id) + ret, err := m.ExecCtx(ctx, func(ctx context.Context, conn sqlx.SqlConn) (result sql.Result, err error) { + query := fmt.Sprintf("insert into %s (%s) values (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)", m.table, computeCenterRowsExpectAutoSet) + return conn.ExecCtx(ctx, query, data.Id, data.CenterSource, data.SourceId, data.Name, data.Description, data.Type, data.Area, data.City, data.Longitude, data.Latitude, data.Status, data.UserNum, data.DeletedFlag) + }, pcmComputeCenterIdKey) + return ret, err +} + +func (m *defaultComputeCenterModel) Update(ctx context.Context, data *ComputeCenter) error { + pcmComputeCenterIdKey := fmt.Sprintf("%s%v", cachePcmComputeCenterIdPrefix, data.Id) + _, err := m.ExecCtx(ctx, func(ctx context.Context, conn sqlx.SqlConn) (result sql.Result, err error) { + query := fmt.Sprintf("update %s set %s where `id` = ?", m.table, computeCenterRowsWithPlaceHolder) + return conn.ExecCtx(ctx, query, data.CenterSource, data.SourceId, data.Name, data.Description, data.Type, data.Area, data.City, data.Longitude, data.Latitude, data.Status, data.UserNum, data.DeletedFlag, data.Id) + }, pcmComputeCenterIdKey) + return err +} + +func (m *defaultComputeCenterModel) formatPrimary(primary interface{}) string { + return fmt.Sprintf("%s%v", cachePcmComputeCenterIdPrefix, primary) +} + +func (m *defaultComputeCenterModel) queryPrimary(ctx context.Context, conn sqlx.SqlConn, v, primary interface{}) error { + query := fmt.Sprintf("select %s from %s where `id` = ? limit 1", computeCenterRows, m.table) + return conn.QueryRowCtx(ctx, v, query, primary) +} + +func (m *defaultComputeCenterModel) tableName() string { + return m.table +} diff --git a/adaptor/PCM-CORE/model/hpcCenterModel.go b/adaptor/PCM-CORE/model/hpcCenterModel.go new file mode 100644 index 0000000..7ee5ee3 --- /dev/null +++ b/adaptor/PCM-CORE/model/hpcCenterModel.go @@ -0,0 +1,27 @@ +package model + +import ( + "github.com/zeromicro/go-zero/core/stores/cache" + "github.com/zeromicro/go-zero/core/stores/sqlx" +) + +var _ HpcCenterModel = (*customHpcCenterModel)(nil) + +type ( + // HpcCenterModel is an interface to be customized, add more methods here, + // and implement the added methods in customHpcCenterModel. + HpcCenterModel interface { + hpcCenterModel + } + + customHpcCenterModel struct { + *defaultHpcCenterModel + } +) + +// NewHpcCenterModel returns a model for the database table. +func NewHpcCenterModel(conn sqlx.SqlConn, c cache.CacheConf) HpcCenterModel { + return &customHpcCenterModel{ + defaultHpcCenterModel: newHpcCenterModel(conn, c), + } +} diff --git a/adaptor/PCM-CORE/model/hpcCenterModel_gen.go b/adaptor/PCM-CORE/model/hpcCenterModel_gen.go new file mode 100644 index 0000000..4d2faa1 --- /dev/null +++ b/adaptor/PCM-CORE/model/hpcCenterModel_gen.go @@ -0,0 +1,115 @@ +// Code generated by goctl. DO NOT EDIT. + +package model + +import ( + "context" + "database/sql" + "fmt" + "strings" + + "github.com/zeromicro/go-zero/core/stores/builder" + "github.com/zeromicro/go-zero/core/stores/cache" + "github.com/zeromicro/go-zero/core/stores/sqlc" + "github.com/zeromicro/go-zero/core/stores/sqlx" + "github.com/zeromicro/go-zero/core/stringx" +) + +var ( + hpcCenterFieldNames = builder.RawFieldNames(&HpcCenter{}) + hpcCenterRows = strings.Join(hpcCenterFieldNames, ",") + hpcCenterRowsExpectAutoSet = strings.Join(stringx.Remove(hpcCenterFieldNames, "`create_at`", "`create_time`", "`created_at`", "`update_at`", "`update_time`", "`updated_at`"), ",") + hpcCenterRowsWithPlaceHolder = strings.Join(stringx.Remove(hpcCenterFieldNames, "`id`", "`create_at`", "`create_time`", "`created_at`", "`update_at`", "`update_time`", "`updated_at`"), "=?,") + "=?" + + cachePcmHpcCenterIdPrefix = "cache:pcm:hpcCenter:id:" +) + +type ( + hpcCenterModel interface { + Insert(ctx context.Context, data *HpcCenter) (sql.Result, error) + FindOne(ctx context.Context, id int64) (*HpcCenter, error) + Update(ctx context.Context, data *HpcCenter) error + Delete(ctx context.Context, id int64) error + } + + defaultHpcCenterModel struct { + sqlc.CachedConn + table string + } + + HpcCenter struct { + Id int64 `db:"id"` // 平台唯一id + ClusterNum sql.NullInt64 `db:"cluster_num"` // 集群数量 + NodeNum sql.NullInt64 `db:"node_num"` // 节点数量 + CpuNum sql.NullInt64 `db:"cpu_num"` // CPU核数 + GpuNum sql.NullInt64 `db:"gpu_num"` // GPU卡数 + ManagedFlops sql.NullString `db:"managed_flops"` // 已接入算力 + UnmanagedFlops sql.NullString `db:"unmanaged_flops"` // 未接入算力 + ManagedStorage sql.NullString `db:"managed_storage"` // 已接入存储 + UnmanagedStorage sql.NullString `db:"unmanaged_storage"` // 未接入存储 + } +) + +func newHpcCenterModel(conn sqlx.SqlConn, c cache.CacheConf) *defaultHpcCenterModel { + return &defaultHpcCenterModel{ + CachedConn: sqlc.NewConn(conn, c), + table: "`hpc_center`", + } +} + +func (m *defaultHpcCenterModel) Delete(ctx context.Context, id int64) error { + pcmHpcCenterIdKey := fmt.Sprintf("%s%v", cachePcmHpcCenterIdPrefix, id) + _, err := m.ExecCtx(ctx, func(ctx context.Context, conn sqlx.SqlConn) (result sql.Result, err error) { + query := fmt.Sprintf("delete from %s where `id` = ?", m.table) + return conn.ExecCtx(ctx, query, id) + }, pcmHpcCenterIdKey) + return err +} + +func (m *defaultHpcCenterModel) FindOne(ctx context.Context, id int64) (*HpcCenter, error) { + pcmHpcCenterIdKey := fmt.Sprintf("%s%v", cachePcmHpcCenterIdPrefix, id) + var resp HpcCenter + err := m.QueryRowCtx(ctx, &resp, pcmHpcCenterIdKey, func(ctx context.Context, conn sqlx.SqlConn, v interface{}) error { + query := fmt.Sprintf("select %s from %s where `id` = ? limit 1", hpcCenterRows, m.table) + return conn.QueryRowCtx(ctx, v, query, id) + }) + switch err { + case nil: + return &resp, nil + case sqlc.ErrNotFound: + return nil, ErrNotFound + default: + return nil, err + } +} + +func (m *defaultHpcCenterModel) Insert(ctx context.Context, data *HpcCenter) (sql.Result, error) { + pcmHpcCenterIdKey := fmt.Sprintf("%s%v", cachePcmHpcCenterIdPrefix, data.Id) + ret, err := m.ExecCtx(ctx, func(ctx context.Context, conn sqlx.SqlConn) (result sql.Result, err error) { + query := fmt.Sprintf("insert into %s (%s) values (?, ?, ?, ?, ?, ?, ?, ?, ?)", m.table, hpcCenterRowsExpectAutoSet) + return conn.ExecCtx(ctx, query, data.Id, data.ClusterNum, data.NodeNum, data.CpuNum, data.GpuNum, data.ManagedFlops, data.UnmanagedFlops, data.ManagedStorage, data.UnmanagedStorage) + }, pcmHpcCenterIdKey) + return ret, err +} + +func (m *defaultHpcCenterModel) Update(ctx context.Context, data *HpcCenter) error { + pcmHpcCenterIdKey := fmt.Sprintf("%s%v", cachePcmHpcCenterIdPrefix, data.Id) + _, err := m.ExecCtx(ctx, func(ctx context.Context, conn sqlx.SqlConn) (result sql.Result, err error) { + query := fmt.Sprintf("update %s set %s where `id` = ?", m.table, hpcCenterRowsWithPlaceHolder) + return conn.ExecCtx(ctx, query, data.ClusterNum, data.NodeNum, data.CpuNum, data.GpuNum, data.ManagedFlops, data.UnmanagedFlops, data.ManagedStorage, data.UnmanagedStorage, data.Id) + }, pcmHpcCenterIdKey) + return err +} + +func (m *defaultHpcCenterModel) formatPrimary(primary interface{}) string { + return fmt.Sprintf("%s%v", cachePcmHpcCenterIdPrefix, primary) +} + +func (m *defaultHpcCenterModel) queryPrimary(ctx context.Context, conn sqlx.SqlConn, v, primary interface{}) error { + query := fmt.Sprintf("select %s from %s where `id` = ? limit 1", hpcCenterRows, m.table) + return conn.QueryRowCtx(ctx, v, query, primary) +} + +func (m *defaultHpcCenterModel) tableName() string { + return m.table +} diff --git a/adaptor/PCM-HPC/PCM-AC/rpc/etc/hpcac.yaml b/adaptor/PCM-HPC/PCM-AC/rpc/etc/hpc-ac-template.yaml similarity index 73% rename from adaptor/PCM-HPC/PCM-AC/rpc/etc/hpcac.yaml rename to adaptor/PCM-HPC/PCM-AC/rpc/etc/hpc-ac-template.yaml index ea6535e..2030b08 100644 --- a/adaptor/PCM-HPC/PCM-AC/rpc/etc/hpcac.yaml +++ b/adaptor/PCM-HPC/PCM-AC/rpc/etc/hpc-ac-template.yaml @@ -8,8 +8,8 @@ ListenOn: 127.0.0.1:2001 ClusterUrl: "https://api01.hpccube.com:65106/hpc/openapi/v2/cluster" TokenUrl: "https://api01.hpccube.com:65102/ac/openapi/v2/tokens" StateUrl: "https://api01.hpccube.com:65102/ac/openapi/v2/tokens/state" -User: "zhijiang" -Password: "111111a" -OrgId: "313ae32df03bc116255e6808949fcf57" +User: "" +Password: "" +OrgId: "" Layout: "2006-01-02 15:04:05" EndPoint: https://api01.hpccube.com:65106 \ No newline at end of file diff --git a/adaptor/PCM-HPC/PCM-HPC-CORE/api/etc/hpccore.yaml b/adaptor/PCM-HPC/PCM-HPC-CORE/api/etc/hpc-core-api-template.yaml similarity index 100% rename from adaptor/PCM-HPC/PCM-HPC-CORE/api/etc/hpccore.yaml rename to adaptor/PCM-HPC/PCM-HPC-CORE/api/etc/hpc-core-api-template.yaml diff --git a/adaptor/PCM-HPC/PCM-HPC-CORE/api/hpccoreapi.go b/adaptor/PCM-HPC/PCM-HPC-CORE/api/hpccoreapi.go index dadadd2..ba1cfe1 100644 --- a/adaptor/PCM-HPC/PCM-HPC-CORE/api/hpccoreapi.go +++ b/adaptor/PCM-HPC/PCM-HPC-CORE/api/hpccoreapi.go @@ -10,7 +10,7 @@ import ( "github.com/zeromicro/go-zero/rest" ) -var configFile = flag.String("f", "adaptor/PCM-HPC/PCM-HPC-CORE/api/etc/hpccore.yaml", "the config file") +var configFile = flag.String("f", "adaptor/PCM-HPC/PCM-HPC-CORE/api/etc/hpc-core-api-template.yaml", "the config file") func main() { flag.Parse() diff --git a/adaptor/PCM-HPC/PCM-HPC-CORE/rpc/etc/pcmhpccore.yaml b/adaptor/PCM-HPC/PCM-HPC-CORE/rpc/etc/hpc-core-rpc-template.yaml similarity index 86% rename from adaptor/PCM-HPC/PCM-HPC-CORE/rpc/etc/pcmhpccore.yaml rename to adaptor/PCM-HPC/PCM-HPC-CORE/rpc/etc/hpc-core-rpc-template.yaml index c388b29..2960da4 100644 --- a/adaptor/PCM-HPC/PCM-HPC-CORE/rpc/etc/pcmhpccore.yaml +++ b/adaptor/PCM-HPC/PCM-HPC-CORE/rpc/etc/hpc-core-rpc-template.yaml @@ -28,4 +28,4 @@ THRpcConf: Key: hpcth.rpc DB: - DataSource: root:uJpLd6u-J?HC1@(106.53.150.192:3306)/pcm + DataSource: diff --git a/adaptor/PCM-HPC/PCM-HPC-CORE/rpc/pcmhpccore.go b/adaptor/PCM-HPC/PCM-HPC-CORE/rpc/pcmhpccore.go index 45467af..849adbe 100644 --- a/adaptor/PCM-HPC/PCM-HPC-CORE/rpc/pcmhpccore.go +++ b/adaptor/PCM-HPC/PCM-HPC-CORE/rpc/pcmhpccore.go @@ -19,7 +19,7 @@ import ( "google.golang.org/grpc/reflection" ) -var configFile = flag.String("f", "adaptor/PCM-HPC/PCM-HPC-CORE/rpc/etc/pcmhpccore.yaml", "the config file") +var configFile = flag.String("f", "adaptor/PCM-HPC/PCM-HPC-CORE/rpc/etc/hpc-core-rpc-template.yaml", "the config file") func main() { flag.Parse() diff --git a/adaptor/PCM-HPC/PCM-TH/rpc/etc/hpc-th-template.yaml b/adaptor/PCM-HPC/PCM-TH/rpc/etc/hpc-th-template.yaml new file mode 100644 index 0000000..e5017d2 --- /dev/null +++ b/adaptor/PCM-HPC/PCM-TH/rpc/etc/hpc-th-template.yaml @@ -0,0 +1,6 @@ +Name: hpcth.rpc +ListenOn: 0.0.0.0:8881 +Etcd: + Hosts: + - localhost:2379 + Key: hpcth.rpc \ No newline at end of file diff --git a/adaptor/PCM-HPC/PCM-TH/rpc/etc/hpcth.yaml b/adaptor/PCM-HPC/PCM-TH/rpc/etc/hpcth.yaml deleted file mode 100644 index 1e2b14c..0000000 --- a/adaptor/PCM-HPC/PCM-TH/rpc/etc/hpcth.yaml +++ /dev/null @@ -1,2 +0,0 @@ -Name: hpcth.rpc -ListenOn: 0.0.0.0:2002 diff --git a/adaptor/PCM-HPC/PCM-TH/rpc/hpcth.go b/adaptor/PCM-HPC/PCM-TH/rpc/hpcth.go index 60e9a63..68528c2 100644 --- a/adaptor/PCM-HPC/PCM-TH/rpc/hpcth.go +++ b/adaptor/PCM-HPC/PCM-TH/rpc/hpcth.go @@ -16,7 +16,7 @@ import ( "google.golang.org/grpc/reflection" ) -var configFile = flag.String("f", "adaptor/PCM-HPC/PCM-TH/rpc/etc/hpcth.yaml", "the config file") +var configFile = flag.String("f", "adaptor/PCM-HPC/PCM-TH/rpc/etc/hpc-th-template.yaml", "the config file") func main() { flag.Parse() diff --git a/go.mod b/go.mod index 1abc680..4f2fd57 100644 --- a/go.mod +++ b/go.mod @@ -3,6 +3,7 @@ module PCM go 1.19 require ( + github.com/Masterminds/squirrel v1.5.4 github.com/bitly/go-simplejson v0.5.0 github.com/go-redis/redis/v8 v8.11.5 github.com/go-resty/resty/v2 v2.7.0 @@ -57,6 +58,8 @@ require ( github.com/josharian/intern v1.0.0 // indirect github.com/json-iterator/go v1.1.12 // indirect github.com/klauspost/compress v1.15.15 // indirect + github.com/lann/builder v0.0.0-20180802200727-47ae307949d0 // indirect + github.com/lann/ps v0.0.0-20150810152359-62de8c46ede0 // indirect github.com/mailru/easyjson v0.7.6 // indirect github.com/mattn/go-colorable v0.1.13 // indirect github.com/mattn/go-isatty v0.0.17 // indirect diff --git a/go.sum b/go.sum index 440f2b0..f13ae0b 100644 --- a/go.sum +++ b/go.sum @@ -373,6 +373,8 @@ github.com/ClickHouse/clickhouse-go v1.5.4/go.mod h1:EaI/sW7Azgz9UATzd5ZdZHRUhHg github.com/ClickHouse/clickhouse-go/v2 v2.0.14/go.mod h1:iq2DUGgpA4BBki2CVwrF8x43zqBjdgHtbexkFkh5a6M= github.com/DATA-DOG/go-sqlmock v1.5.0 h1:Shsta01QNfFxHCfpW6YH2STWB0MudeXXEWMr20OEh60= github.com/DATA-DOG/go-sqlmock v1.5.0/go.mod h1:f/Ixk793poVmq4qj/V1dPUg2JEAKC73Q5eFN3EC/SaM= +github.com/Masterminds/squirrel v1.5.4 h1:uUcX/aBc8O7Fg9kaISIUsHXdKuqehiXAMQTYX8afzqM= +github.com/Masterminds/squirrel v1.5.4/go.mod h1:NNaOrjSoIDfDA40n7sr2tPNZRfjzjA400rg+riTZj10= github.com/NYTimes/gziphandler v0.0.0-20170623195520-56545f4a5d46/go.mod h1:3wb06e3pkSAbeQ52E9H9iFoQsEEwGN64994WTCIhntQ= github.com/OneOfOne/xxhash v1.2.2/go.mod h1:HSdplMjZKSmBqAxg5vPj2TmRDmfkzw+cTzAElWljhcU= github.com/PuerkitoBio/purell v1.1.1/go.mod h1:c11w/QuzBsJSee3cPx9rAFu61PvFxuPbtSwDGJws/X0= @@ -697,6 +699,10 @@ github.com/kr/pty v1.1.1/go.mod h1:pFQYn66WHrOpPYNljwOMqo10TkYh1fy3cYio2l3bCsQ= github.com/kr/text v0.1.0/go.mod h1:4Jbv+DJW3UT/LiOwJeYQe1efqtUx/iVham/4vfdArNI= github.com/kr/text v0.2.0 h1:5Nx0Ya0ZqY2ygV366QzturHI13Jq95ApcVaJBhpS+AY= github.com/kr/text v0.2.0/go.mod h1:eLer722TekiGuMkidMxC/pM04lWEeraHUUmBw8l2grE= +github.com/lann/builder v0.0.0-20180802200727-47ae307949d0 h1:SOEGU9fKiNWd/HOJuq6+3iTQz8KNCLtVX6idSoTLdUw= +github.com/lann/builder v0.0.0-20180802200727-47ae307949d0/go.mod h1:dXGbAdH5GtBTC4WfIxhKZfyBF/HBFgRZSWwZ9g/He9o= +github.com/lann/ps v0.0.0-20150810152359-62de8c46ede0 h1:P6pPBnrTSX3DEVR4fDembhRWSsG5rVo6hYhAB/ADZrk= +github.com/lann/ps v0.0.0-20150810152359-62de8c46ede0/go.mod h1:vmVJ0l/dxyfGW6FmdpVm2joNMFikkuWg0EoCKLGUMNw= github.com/lib/pq v1.0.0/go.mod h1:5WUZQaWbwv1U+lTReE5YruASi9Al49XbQIvNi/34Woo= github.com/lib/pq v1.2.0/go.mod h1:5WUZQaWbwv1U+lTReE5YruASi9Al49XbQIvNi/34Woo= github.com/lib/pq v1.10.7 h1:p7ZhMD+KsSRozJr34udlUrhboJwWAgCg34+/ZZNvZZw=