mirror of
https://github.com/yunionio/cloudpods.git
synced 2026-07-02 23:04:22 +08:00
79 lines
3.0 KiB
Go
79 lines
3.0 KiB
Go
package tasks
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
|
|
"yunion.io/x/jsonutils"
|
|
|
|
"yunion.io/x/onecloud/pkg/cloudcommon/db"
|
|
"yunion.io/x/onecloud/pkg/cloudcommon/db/taskman"
|
|
"yunion.io/x/onecloud/pkg/compute/models"
|
|
)
|
|
|
|
type StorageCacheImageTask struct {
|
|
taskman.STask
|
|
}
|
|
|
|
func init() {
|
|
taskman.RegisterTask(StorageCacheImageTask{})
|
|
}
|
|
|
|
func (self *StorageCacheImageTask) OnInit(ctx context.Context, obj db.IStandaloneModel, data jsonutils.JSONObject) {
|
|
imageId, _ := self.Params.GetString("image_id")
|
|
// isForce := jsonutils.QueryBoolean(self.Params, "is_force", false)
|
|
|
|
storageCache := obj.(*models.SStoragecache)
|
|
scimg := models.StoragecachedimageManager.Register(ctx, self.UserCred, storageCache.Id, imageId)
|
|
if scimg.Status != models.CACHED_IMAGE_STATUS_READY {
|
|
scimg.SetStatus(self.UserCred, models.CACHED_IMAGE_STATUS_CACHING, "storage_cache_image_task")
|
|
}
|
|
|
|
db.OpsLog.LogEvent(storageCache, db.ACT_CACHING_IMAGE, imageId, self.UserCred)
|
|
|
|
self.SetStage("on_image_cache_complete", nil)
|
|
|
|
host, _ := storageCache.GetHost()
|
|
err := host.GetHostDriver().CheckAndSetCacheImage(ctx, host, storageCache, self)
|
|
if err != nil {
|
|
errData := taskman.Error2TaskData(err)
|
|
self.OnImageCacheCompleteFailed(ctx, storageCache, errData)
|
|
}
|
|
|
|
}
|
|
|
|
func (self *StorageCacheImageTask) OnImageCacheComplete(ctx context.Context, obj db.IStandaloneModel, data jsonutils.JSONObject) {
|
|
storageCache := obj.(*models.SStoragecache)
|
|
imageId, _ := self.Params.GetString("image_id")
|
|
scimg := models.StoragecachedimageManager.Register(ctx, self.UserCred, storageCache.Id, imageId)
|
|
// extImgId, _ := data.GetString("image_id")
|
|
self.OnCacheSucc(ctx, storageCache, imageId, scimg)
|
|
}
|
|
|
|
func (self *StorageCacheImageTask) OnImageCacheCompleteFailed(ctx context.Context, obj db.IStandaloneModel, data jsonutils.JSONObject) {
|
|
storageCache := obj.(*models.SStoragecache)
|
|
imageId, _ := self.Params.GetString("image_id")
|
|
scimg := models.StoragecachedimageManager.Register(ctx, self.UserCred, storageCache.Id, imageId)
|
|
err := fmt.Errorf(data.String())
|
|
self.OnCacheFailed(ctx, storageCache, imageId, scimg, err)
|
|
}
|
|
|
|
func (self *StorageCacheImageTask) OnCacheFailed(ctx context.Context, cache *models.SStoragecache, imageId string, scimg *models.SStoragecachedimage, err error) {
|
|
scimg.SetStatus(self.UserCred, models.CACHED_IMAGE_STATUS_CACHE_FAILED, err.Error())
|
|
/* if len(extImgId) > 0 && scimg.ExternalId != extImgId {
|
|
scimg.SetExternalId(extImgId)
|
|
}*/
|
|
body := jsonutils.NewDict()
|
|
body.Add(jsonutils.NewString(err.Error()), "reason")
|
|
body.Add(jsonutils.NewString(imageId), "image_id")
|
|
db.OpsLog.LogEvent(cache, db.ACT_CACHE_IMAGE_FAIL, body, self.UserCred)
|
|
self.SetStageFailed(ctx, err.Error())
|
|
}
|
|
|
|
func (self *StorageCacheImageTask) OnCacheSucc(ctx context.Context, cache *models.SStoragecache, imageId string, scimg *models.SStoragecachedimage) {
|
|
scimg.SetStatus(self.UserCred, models.CACHED_IMAGE_STATUS_READY, "cached")
|
|
models.CachedimageManager.ImageAddRefCount(imageId)
|
|
db.OpsLog.LogEvent(cache, db.ACT_CACHED_IMAGE, imageId, self.UserCred)
|
|
self.SetStageComplete(ctx, nil)
|
|
}
|