From 18cadd528dfc6d7807c94b480b5ed58c379bd450 Mon Sep 17 00:00:00 2001 From: tzwang Date: Tue, 30 Apr 2024 16:18:14 +0800 Subject: [PATCH] updated cluster resource db model Former-commit-id: 2904365c03e2bf8ae9475dd6eed360c8106af165 --- pkg/models/tclusterresourcemodel_gen.go | 37 +++++++++++++++---------- 1 file changed, 23 insertions(+), 14 deletions(-) diff --git a/pkg/models/tclusterresourcemodel_gen.go b/pkg/models/tclusterresourcemodel_gen.go index bf4068e6..1bb40107 100644 --- a/pkg/models/tclusterresourcemodel_gen.go +++ b/pkg/models/tclusterresourcemodel_gen.go @@ -35,17 +35,19 @@ type ( } TClusterResource struct { - ClusterId int64 `db:"cluster_id"` - ClusterName string `db:"cluster_name"` - ClusterType int64 `db:"cluster_type"` // 类型0->容器，1->智算，2->超算，3-虚拟机 - CpuAvail float64 `db:"cpu_avail"` - CpuTotal float64 `db:"cpu_total"` - MemAvail float64 `db:"mem_avail"` - MemTotal float64 `db:"mem_total"` - DiskAvail float64 `db:"disk_avail"` - DiskTotal float64 `db:"disk_total"` - GpuAvail float64 `db:"gpu_avail"` - GpuTotal float64 `db:"gpu_total"` + ClusterId int64 `db:"cluster_id"` + ClusterName string `db:"cluster_name"` + ClusterType int64 `db:"cluster_type"` // 类型0->容器，1->智算，2->超算，3-虚拟机 + CpuAvail float64 `db:"cpu_avail"` + CpuTotal float64 `db:"cpu_total"` + MemAvail float64 `db:"mem_avail"` + MemTotal float64 `db:"mem_total"` + DiskAvail float64 `db:"disk_avail"` + DiskTotal float64 `db:"disk_total"` + GpuAvail float64 `db:"gpu_avail"` + GpuTotal float64 `db:"gpu_total"` + CardTotal int64 `db:"card_total"` // 算力卡数量 + CardTopsTotal float64 `db:"card_tops_total"` // 算力总量tops } ) @@ -56,6 +58,13 @@ func newTClusterResourceModel(conn sqlx.SqlConn) *defaultTClusterResourceModel { } } +func (m *defaultTClusterResourceModel) withSession(session sqlx.Session) *defaultTClusterResourceModel { + return &defaultTClusterResourceModel{ + conn: sqlx.NewSqlConnFromSession(session), + table: "`t_cluster_resource`", + } +} + func (m *defaultTClusterResourceModel) Delete(ctx context.Context, clusterId int64) error { query := fmt.Sprintf("delete from %s where `cluster_id` = ?", m.table) _, err := m.conn.ExecCtx(ctx, query, clusterId) @@ -77,14 +86,14 @@ func (m *defaultTClusterResourceModel) FindOne(ctx context.Context, clusterId in } func (m *defaultTClusterResourceModel) Insert(ctx context.Context, data *TClusterResource) (sql.Result, error) { - query := fmt.Sprintf("insert into %s (%s) values (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)", m.table, tClusterResourceRowsExpectAutoSet) - ret, err := m.conn.ExecCtx(ctx, query, data.ClusterId, data.ClusterName, data.ClusterType, data.CpuAvail, data.CpuTotal, data.MemAvail, data.MemTotal, data.DiskAvail, data.DiskTotal, data.GpuAvail, data.GpuTotal) + query := fmt.Sprintf("insert into %s (%s) values (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)", m.table, tClusterResourceRowsExpectAutoSet) + ret, err := m.conn.ExecCtx(ctx, query, data.ClusterId, data.ClusterName, data.ClusterType, data.CpuAvail, data.CpuTotal, data.MemAvail, data.MemTotal, data.DiskAvail, data.DiskTotal, data.GpuAvail, data.GpuTotal, data.CardTotal, data.CardTopsTotal) return ret, err } func (m *defaultTClusterResourceModel) Update(ctx context.Context, data *TClusterResource) error { query := fmt.Sprintf("update %s set %s where `cluster_id` = ?", m.table, tClusterResourceRowsWithPlaceHolder) - _, err := m.conn.ExecCtx(ctx, query, data.ClusterName, data.ClusterType, data.CpuAvail, data.CpuTotal, data.MemAvail, data.MemTotal, data.DiskAvail, data.DiskTotal, data.GpuAvail, data.GpuTotal, data.ClusterId) + _, err := m.conn.ExecCtx(ctx, query, data.ClusterName, data.ClusterType, data.CpuAvail, data.CpuTotal, data.MemAvail, data.MemTotal, data.DiskAvail, data.DiskTotal, data.GpuAvail, data.GpuTotal, data.CardTotal, data.CardTopsTotal, data.ClusterId) return err }