kanshan/megrez
1
0
Fork 0
mirror of https://github.com/XShengTech/MEGREZ.git synced 2026-01-14 00:57:17 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

[Fix] 🐛 Fix GPU Count Error after Instance Modify Fail #32

Browse Source
This commit is contained in:
Harry-zklcdc 2025-09-29 22:47:26 +08:00
parent f3543e2657
commit e54fa18283
1 changed files with 1 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
services/dispatcher/modify.go
View File

@ -10,12 +10,6 @@ import (
"strconv" "strconv"
) )
type modifyDataStruct struct {
CpuOnly bool `json:"cpu_only"`
GpuCount *int `json:"gpu_count"`
VolumeSize *int `json:"volume_size"`
}
func modify(serverID uint, data Data) (err error) { func modify(serverID uint, data Data) (err error) {
lc := l.Clone() lc := l.Clone()
lc.SetFunction("modify") lc.SetFunction("modify")
@ -71,6 +65,7 @@ func modify(serverID uint, data Data) (err error) {
err = instanceController.Patch(&instance, gpuCount, volumeSize, data.CpuOnly) err = instanceController.Patch(&instance, gpuCount, volumeSize, data.CpuOnly)
if err != nil { if err != nil {
ctx := context.Background() ctx := context.Background()
redis.RawDB.IncrBy(ctx, "remain_gpu:server:"+strconv.Itoa(int(serverID)), int64(gpuCount))
redis.RawDB.IncrBy(ctx, "remain_volume:server:"+strconv.Itoa(int(serverID)), int64(volumeSize-oldVolumeSize)) redis.RawDB.IncrBy(ctx, "remain_volume:server:"+strconv.Itoa(int(serverID)), int64(volumeSize-oldVolumeSize))
database.DB.Model(&instance).Update("status", models.InstanceStatusFail).Update("from_action", models.InstanceActionModify) database.DB.Model(&instance).Update("status", models.InstanceStatusFail).Update("from_action", models.InstanceActionModify)
lc.Error("patch instance error: %v", err) lc.Error("patch instance error: %v", err)