JCC-CSScheduler/manager/internal/mq/image.go

25 lines
927 B
Go

package mq
import (
"gitlink.org.cn/cloudream/common/consts/errorcode"
"gitlink.org.cn/cloudream/common/pkgs/logger"
"gitlink.org.cn/cloudream/common/pkgs/mq"
mgrmq "gitlink.org.cn/cloudream/scheduler/common/pkgs/mq/manager"
)
func (svc *Service) GetImageInfo(msg *mgrmq.GetImageInfo) (*mgrmq.GetImageInfoResp, *mq.CodeMessage) {
image, err := svc.db.Image().GetByID(svc.db.SQLCtx(), msg.ImageID)
if err != nil {
logger.WithField("ImageID", msg.ImageID).Warnf("getting image by id: %s", err.Error())
return nil, mq.Failed(errorcode.OperationFailed, "get image failed")
}
pcmImages, err := svc.db.PCMImage().GetByImageID(svc.db.SQLCtx(), msg.ImageID)
if err != nil {
logger.WithField("ImageID", msg.ImageID).Warnf("getting pcm image by image id: %s", err.Error())
return nil, mq.Failed(errorcode.OperationFailed, "get pcm images failed")
}
return mq.ReplyOK(mgrmq.NewGetImageInfoResp(image, pcmImages))
}