| 123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683 |
- // Copyright 2019 Yunion
- //
- // Licensed under the Apache License, Version 2.0 (the "License");
- // you may not use this file except in compliance with the License.
- // You may obtain a copy of the License at
- //
- // http://www.apache.org/licenses/LICENSE-2.0
- //
- // Unless required by applicable law or agreed to in writing, software
- // distributed under the License is distributed on an "AS IS" BASIS,
- // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
- // See the License for the specific language governing permissions and
- // limitations under the License.
- package models
- import (
- "context"
- "database/sql"
- "fmt"
- "strings"
- "time"
- "github.com/serialx/hashring"
- "yunion.io/x/cloudmux/pkg/cloudprovider"
- "yunion.io/x/jsonutils"
- "yunion.io/x/log"
- "yunion.io/x/pkg/errors"
- "yunion.io/x/pkg/utils"
- "yunion.io/x/sqlchemy"
- api "yunion.io/x/onecloud/pkg/apis/compute"
- "yunion.io/x/onecloud/pkg/cloudcommon/db"
- "yunion.io/x/onecloud/pkg/cloudcommon/db/lockman"
- "yunion.io/x/onecloud/pkg/httperrors"
- "yunion.io/x/onecloud/pkg/mcclient"
- "yunion.io/x/onecloud/pkg/util/stringutils2"
- )
- type SStoragecachedimageManager struct {
- db.SJointResourceBaseManager
- db.SExternalizedResourceBaseManager
- SStoragecacheResourceBaseManager
- }
- var StoragecachedimageManager *SStoragecachedimageManager
- func init() {
- db.InitManager(func() {
- StoragecachedimageManager = &SStoragecachedimageManager{
- SJointResourceBaseManager: db.NewJointResourceBaseManager(
- SStoragecachedimage{},
- "storagecachedimages_tbl",
- "storagecachedimage",
- "storagecachedimages",
- StoragecacheManager,
- CachedimageManager,
- ),
- }
- StoragecachedimageManager.SetVirtualObject(StoragecachedimageManager)
- })
- }
- type SStoragecachedimage struct {
- db.SJointResourceBase
- db.SExternalizedResourceBase
- SStoragecacheResourceBase
- // 镜像缓存Id
- CachedimageId string `width:"36" charset:"ascii" nullable:"false" list:"admin" create:"admin_required"`
- // 外部Id
- // ExternalId string `width:"256" charset:"utf8" nullable:"false" get:"admin"`
- // 镜像状态
- Status string `width:"32" charset:"ascii" nullable:"false" default:"init" list:"admin" update:"admin" create:"admin_required"`
- Path string `width:"256" charset:"utf8" nullable:"true" list:"admin" update:"admin" create:"admin_optional"`
- // 上次下载时间
- LastDownload time.Time `get:"admin"`
- // 下载引用次数
- DownloadRefcnt int `get:"admin"`
- }
- func (manager *SStoragecachedimageManager) GetMasterFieldName() string {
- return "storagecache_id"
- }
- func (manager *SStoragecachedimageManager) GetSlaveFieldName() string {
- return "cachedimage_id"
- }
- func (sci *SStoragecachedimage) getStorageHostId() (string, error) {
- var s SStorage
- storage := StorageManager.Query()
- err := storage.Filter(sqlchemy.Equals(storage.Field("storagecache_id"), sci.StoragecacheId)).First(&s)
- if err != nil {
- return "", err
- }
- hosts := s.GetAllAttachingHosts()
- var hostIds = make([]string, 0)
- for _, host := range hosts {
- hostIds = append(hostIds, host.Id)
- }
- if len(hostIds) == 0 {
- return "", nil
- }
- ring := hashring.New(hostIds)
- ret, _ := ring.GetNode(sci.StoragecacheId)
- return ret, nil
- }
- func (sci *SStoragecachedimage) GetHost() (*SHost, error) {
- sc := sci.GetStoragecache()
- return sc.GetMasterHost()
- }
- func (manager *SStoragecachedimageManager) FetchCustomizeColumns(
- ctx context.Context,
- userCred mcclient.TokenCredential,
- query jsonutils.JSONObject,
- objs []interface{},
- fields stringutils2.SSortedStrings,
- isList bool,
- ) []api.StoragecachedimageDetails {
- rows := make([]api.StoragecachedimageDetails, len(objs))
- storagecacheIds := make([]string, 0)
- imageIds := make([]string, 0)
- jointRows := manager.SJointResourceBaseManager.FetchCustomizeColumns(ctx, userCred, query, objs, fields, isList)
- scRows := manager.SStoragecacheResourceBaseManager.FetchCustomizeColumns(ctx, userCred, query, objs, fields, isList)
- for i := range rows {
- rows[i] = api.StoragecachedimageDetails{
- JointResourceBaseDetails: jointRows[i],
- StoragecacheResourceInfo: scRows[i],
- }
- sci := objs[i].(*SStoragecachedimage)
- storagecacheIds = append(storagecacheIds, sci.StoragecacheId)
- imageIds = append(imageIds, sci.CachedimageId)
- }
- cachedImages := make(map[string]SCachedimage)
- err := db.FetchModelObjectsByIds(CachedimageManager, "id", imageIds, &cachedImages)
- if err != nil {
- log.Errorf("db.FetchModelObjectsByIds fail %s", err)
- }
- cdromRefs, err := manager.fetchCdromReferenceCounts(storagecacheIds, imageIds)
- if err != nil {
- log.Errorf("manager.fetchCdromReferenceCounts fail %s", err)
- }
- diskRefs, err := manager.fetchDiskReferenceCounts(storagecacheIds, imageIds)
- if err != nil {
- log.Errorf("manager.fetchDiskReferenceCounts fail %s", err)
- }
- for i := range rows {
- sci := objs[i].(*SStoragecachedimage)
- if cachedImages != nil {
- if cachedImage, ok := cachedImages[sci.CachedimageId]; ok {
- rows[i].Cachedimage = cachedImage.Name
- rows[i].Image = cachedImage.Name
- rows[i].Size = cachedImage.Size
- }
- }
- if cdromRefs != nil {
- if refMap, ok := cdromRefs[sci.StoragecacheId]; ok {
- rows[i].CdromReference = refMap[sci.CachedimageId]
- }
- }
- if diskRefs != nil {
- if refMap, ok := diskRefs[sci.StoragecacheId]; ok {
- rows[i].DiskReference = refMap[sci.CachedimageId]
- rows[i].Reference = rows[i].CdromReference + rows[i].DiskReference
- }
- }
- }
- return rows
- }
- func (sci *SStoragecachedimage) GetCachedimage() *SCachedimage {
- cachedImage, _ := CachedimageManager.FetchById(sci.CachedimageId)
- if cachedImage != nil {
- return cachedImage.(*SCachedimage)
- }
- return nil
- }
- /*func (sci *SStoragecachedimage) getExtraDetails(ctx context.Context, out api.StoragecachedimageDetails) api.StoragecachedimageDetails {
- storagecache := sci.GetStoragecache()
- if storagecache != nil {
- // out.Storagecache = storagecache.Name
- out.Storages = storagecache.getStorageNames()
- host, _ := storagecache.GetMasterHost()
- if host != nil {
- out.Host = host.GetShortDesc(ctx)
- } else {
- var err error
- hostDesc, err := storagecache.GetEsxiAgentHostDesc()
- if err != nil {
- log.Errorf("unable to GetEsxiAgentHostDesc of stroagecache: %s", err.Error())
- }
- if hostDesc != nil {
- out.Host = hostDesc
- }
- }
- }
- cachedImage := sci.GetCachedimage()
- if cachedImage != nil {
- out.Cachedimage = cachedImage.Name
- out.Image = cachedImage.GetName()
- out.Size = cachedImage.Size
- }
- out.Reference, _ = sci.getReferenceCount()
- return out
- }*/
- func (manager *SStoragecachedimageManager) fetchCdromReferenceCounts(storagecacheIds []string, imageIds []string) (map[string]map[string]int, error) {
- q := GuestcdromManager.Query()
- guests := GuestManager.Query().SubQuery()
- hostStorages := HoststorageManager.Query().SubQuery()
- storages := StorageManager.Query().SubQuery()
- q = q.Join(guests, sqlchemy.Equals(q.Field("id"), guests.Field("id")))
- q = q.Join(hostStorages, sqlchemy.Equals(guests.Field("host_id"), hostStorages.Field("host_id")))
- q = q.Join(storages, sqlchemy.Equals(hostStorages.Field("storage_id"), storages.Field("id")))
- q = q.GroupBy(q.Field("image_id"))
- q = q.GroupBy(storages.Field("storagecache_id"))
- q = q.Filter(sqlchemy.In(q.Field("image_id"), imageIds))
- q = q.Filter(sqlchemy.In(storages.Field("storagecache_id"), storagecacheIds))
- q = q.AppendField(sqlchemy.COUNT("ref_count"))
- q = q.AppendField(q.Field("image_id"))
- q = q.AppendField(storages.Field("storagecache_id"))
- return manager.fetchRefCount(q)
- }
- func (maanger *SStoragecachedimageManager) fetchRefCount(q *sqlchemy.SQuery) (map[string]map[string]int, error) {
- results := []struct {
- RefCount int `json:"ref_count"`
- ImageId string `json:"image_id"`
- StoragecacheId string `json:"storagecache_id"`
- }{}
- err := q.All(&results)
- if err != nil {
- return nil, errors.Wrap(err, "Query")
- }
- ret := make(map[string]map[string]int)
- for _, r := range results {
- if _, ok := ret[r.StoragecacheId]; !ok {
- ret[r.StoragecacheId] = make(map[string]int)
- }
- ret[r.StoragecacheId][r.ImageId] = r.RefCount
- }
- return ret, nil
- }
- func (sci *SStoragecachedimage) getCdromReferenceCount() (int, error) {
- cdroms := GuestcdromManager.Query().SubQuery()
- guests := GuestManager.Query().SubQuery()
- q := cdroms.Query()
- q = q.Join(guests, sqlchemy.Equals(cdroms.Field("id"), guests.Field("id")))
- q = q.Filter(sqlchemy.Equals(cdroms.Field("image_id"), sci.CachedimageId))
- return q.CountWithError()
- }
- func (manager *SStoragecachedimageManager) fetchDiskReferenceCounts(storagecacheIds, imageIds []string) (map[string]map[string]int, error) {
- disks := DiskManager.Query().SubQuery()
- storages := StorageManager.Query().SubQuery()
- q := disks.Query()
- q = q.Join(storages, sqlchemy.Equals(disks.Field("storage_id"), storages.Field("id")))
- q = q.GroupBy(disks.Field("template_id"))
- q = q.GroupBy(storages.Field("storagecache_id"))
- q = q.Filter(sqlchemy.In(disks.Field("template_id"), imageIds))
- q = q.Filter(sqlchemy.In(storages.Field("storagecache_id"), storagecacheIds))
- q = q.AppendField(sqlchemy.COUNT("ref_count"))
- q = q.AppendField(disks.Field("template_id").Label("image_id"))
- q = q.AppendField(storages.Field("storagecache_id"))
- return manager.fetchRefCount(q)
- }
- func (sci *SStoragecachedimage) getDiskReferenceCount() (int, error) {
- guestdisks := GuestdiskManager.Query().SubQuery()
- disks := DiskManager.Query().SubQuery()
- storages := StorageManager.Query().SubQuery()
- q := guestdisks.Query()
- q = q.Join(disks, sqlchemy.AND(sqlchemy.Equals(disks.Field("id"), guestdisks.Field("disk_id")),
- sqlchemy.IsFalse(disks.Field("deleted"))))
- q = q.Join(storages, sqlchemy.AND(sqlchemy.Equals(disks.Field("storage_id"), storages.Field("id")),
- sqlchemy.IsFalse(storages.Field("deleted"))))
- q = q.Filter(sqlchemy.Equals(storages.Field("storagecache_id"), sci.StoragecacheId))
- q = q.Filter(sqlchemy.Equals(disks.Field("template_id"), sci.CachedimageId))
- q = q.Filter(sqlchemy.NOT(sqlchemy.In(disks.Field("status"), []string{api.DISK_ALLOC_FAILED, api.DISK_INIT})))
- return q.CountWithError()
- }
- func (sci *SStoragecachedimage) getReferenceCount() (int, error) {
- totalCnt := 0
- cnt, err := sci.getCdromReferenceCount()
- if err != nil {
- return -1, err
- }
- totalCnt += cnt
- cnt, err = sci.getDiskReferenceCount()
- if err != nil {
- return -1, err
- }
- totalCnt += cnt
- return totalCnt, nil
- }
- func (manager *SStoragecachedimageManager) GetStoragecachedimage(cacheId string, imageId string) *SStoragecachedimage {
- obj, err := db.FetchJointByIds(manager, cacheId, imageId, nil)
- if err != nil {
- if err != sql.ErrNoRows {
- log.Errorf("manager.FetchByIds %s %s error %s", cacheId, imageId, err)
- }
- return nil
- }
- return obj.(*SStoragecachedimage)
- }
- func (manager *SStoragecachedimageManager) RecoverStoragecachedImage(
- ctx context.Context, userCred mcclient.TokenCredential, scId, imgId string,
- ) (*SStoragecachedimage, error) {
- lockman.LockRawObject(ctx, manager.Keyword(), "name")
- defer lockman.ReleaseRawObject(ctx, manager.Keyword(), "name")
- storagecachedImage := SStoragecachedimage{}
- storagecachedImage.SetModelManager(manager, &storagecachedImage)
- err := manager.RawQuery().Equals("storagecache_id", scId).Equals("cachedimage_id", imgId).First(&storagecachedImage)
- if err != nil {
- return nil, err
- }
- diff, err := db.Update(&storagecachedImage, func() error {
- storagecachedImage.Status = api.CACHED_IMAGE_STATUS_ACTIVE
- if storagecachedImage.Deleted == true {
- storagecachedImage.Deleted = false
- storagecachedImage.DeletedAt = time.Time{}
- storagecachedImage.UpdateVersion = 0
- }
- return nil
- })
- db.OpsLog.LogEvent(&storagecachedImage, db.ACT_UPDATE, diff, userCred)
- return &storagecachedImage, nil
- }
- func (sci *SStoragecachedimage) Delete(ctx context.Context, userCred mcclient.TokenCredential) error {
- _, err := sqlchemy.GetDB().Exec(
- fmt.Sprintf(
- "delete from %s where row_id = ?",
- sci.GetModelManager().TableSpec().Name(),
- ), sci.RowId,
- )
- return err
- }
- func (sci *SStoragecachedimage) Detach(ctx context.Context, userCred mcclient.TokenCredential) error {
- return db.DetachJoint(ctx, userCred, sci)
- }
- func (sci *SStoragecachedimage) ValidateDeleteCondition(ctx context.Context, info jsonutils.JSONObject) error {
- if sci.Status != api.CACHED_IMAGE_STATUS_CACHE_FAILED {
- cnt, err := sci.getReferenceCount()
- if err != nil {
- return httperrors.NewInternalServerError("getReferenceCount fail %s", err)
- }
- if cnt > 0 {
- return httperrors.NewNotEmptyError("Image is in use")
- }
- }
- return sci.SJointResourceBase.ValidateDeleteCondition(ctx, nil)
- }
- func (sci *SStoragecachedimage) isCachedImageInUse() error {
- if !sci.isDownloadSessionExpire() {
- return httperrors.NewResourceBusyError("Active download session not expired")
- }
- image := sci.GetCachedimage()
- if image != nil && !image.canDeleteLastCache() {
- return httperrors.NewResourceBusyError("Cannot delete the last cache")
- }
- return nil
- }
- func (sci *SStoragecachedimage) isDownloadSessionExpire() bool {
- if !sci.LastDownload.IsZero() && time.Now().Sub(sci.LastDownload) < api.DOWNLOAD_SESSION_LENGTH {
- return false
- } else {
- return true
- }
- }
- func (sci *SStoragecachedimage) markDeleting(ctx context.Context, userCred mcclient.TokenCredential, isForce bool) error {
- err := sci.ValidateDeleteCondition(ctx, nil)
- if err != nil {
- return err
- }
- if !isForce {
- err = sci.isCachedImageInUse()
- if err != nil {
- return err
- }
- }
- cache := sci.GetStoragecache()
- image := sci.GetCachedimage()
- if image != nil {
- lockman.LockJointObject(ctx, cache, image)
- defer lockman.ReleaseJointObject(ctx, cache, image)
- }
- if !isForce && !utils.IsInStringArray(sci.Status,
- []string{
- api.CACHED_IMAGE_STATUS_ACTIVE,
- api.CACHED_IMAGE_STATUS_DELETING,
- api.CACHED_IMAGE_STATUS_CACHE_FAILED,
- api.CACHED_IMAGE_STATUS_DELETE_FAILED,
- api.CACHED_IMAGE_STATUS_UNCACHE_IMAGE_FAILED,
- }) {
- return httperrors.NewInvalidStatusError("Cannot uncache in status %s", sci.Status)
- }
- _, err = db.Update(sci, func() error {
- sci.Status = api.CACHED_IMAGE_STATUS_DELETING
- return nil
- })
- return err
- }
- func (manager *SStoragecachedimageManager) Register(ctx context.Context, userCred mcclient.TokenCredential, cacheId, imageId string, status string) *SStoragecachedimage {
- lockman.LockClass(ctx, manager, db.GetLockClassKey(manager, userCred))
- defer lockman.ReleaseClass(ctx, manager, db.GetLockClassKey(manager, userCred))
- cachedimage := manager.GetStoragecachedimage(cacheId, imageId)
- if cachedimage != nil {
- return cachedimage
- }
- cachedimage = &SStoragecachedimage{}
- cachedimage.SetModelManager(manager, cachedimage)
- cachedimage.StoragecacheId = cacheId
- cachedimage.CachedimageId = imageId
- if len(status) == 0 {
- status = api.CACHED_IMAGE_STATUS_INIT
- }
- cachedimage.Status = status
- err := manager.TableSpec().Insert(ctx, cachedimage)
- if err != nil {
- log.Errorf("insert error %s", err)
- return nil
- }
- return cachedimage
- }
- func (sci *SStoragecachedimage) SetStatus(ctx context.Context, userCred mcclient.TokenCredential, status string, reason string) error {
- if sci.Status == status {
- return nil
- }
- oldStatus := sci.Status
- _, err := db.Update(sci, func() error {
- sci.Status = status
- return nil
- })
- if err != nil {
- return err
- }
- if userCred != nil {
- notes := fmt.Sprintf("%s=>%s", oldStatus, status)
- if len(reason) > 0 {
- notes = fmt.Sprintf("%s: %s", notes, reason)
- }
- db.OpsLog.LogEvent(sci, db.ACT_UPDATE_STATUS, notes, userCred)
- }
- return nil
- }
- func (sci *SStoragecachedimage) AddDownloadRefcount() error {
- _, err := db.Update(sci, func() error {
- sci.DownloadRefcnt += 1
- sci.LastDownload = time.Now()
- return nil
- })
- return err
- }
- func (sci *SStoragecachedimage) SetExternalId(externalId string) error {
- _, err := db.Update(sci, func() error {
- sci.ExternalId = externalId
- return nil
- })
- return err
- }
- func (sci SStoragecachedimage) GetExternalId() string {
- return sci.ExternalId
- }
- func (sci *SStoragecachedimage) syncRemoveCloudImage(ctx context.Context, userCred mcclient.TokenCredential) error {
- lockman.LockObject(ctx, sci)
- defer lockman.ReleaseObject(ctx, sci)
- image := sci.GetCachedimage()
- err := sci.Delete(ctx, userCred)
- if err != nil {
- return err
- }
- if image != nil {
- cnt, err := image.getStoragecacheCount()
- if err != nil {
- log.Errorf("getStoragecacheCount fail %s", err)
- return err
- }
- if cnt == 0 {
- err = image.Delete(ctx, userCred)
- if err != nil {
- log.Errorf("image delete error %s", err)
- return err
- }
- }
- }
- return nil
- }
- func (sci *SStoragecachedimage) syncWithCloudImage(ctx context.Context, userCred mcclient.TokenCredential, ownerId mcclient.IIdentityProvider, image cloudprovider.ICloudImage, managerId string) error {
- cachedImage := sci.GetCachedimage()
- if len(sci.ExternalId) == 0 {
- sci.SetExternalId(cachedImage.GetExternalId())
- }
- if len(cachedImage.ExternalId) > 0 {
- sci.SetStatus(ctx, userCred, image.GetStatus(), "")
- return cachedImage.syncWithCloudImage(ctx, userCred, ownerId, image, nil)
- } else {
- return nil
- }
- }
- func (manager *SStoragecachedimageManager) newFromCloudImage(ctx context.Context, userCred mcclient.TokenCredential, ownerId mcclient.IIdentityProvider, image cloudprovider.ICloudImage, cache *SStoragecache) error {
- var cachedImage *SCachedimage
- provider := cache.GetCloudprovider()
- imgObj, err := db.FetchByExternalId(CachedimageManager, image.GetGlobalId())
- if err != nil {
- if err != sql.ErrNoRows {
- // unhandled error
- log.Errorf("CachedimageManager.FetchByExternalId error %s", err)
- return err
- }
- // not found
- // first test if this image is uploaded by onecloud, if true, image name should be ID of onecloud image
- name := image.GetName()
- if utils.IsAscii(name) {
- if strings.HasPrefix(name, "img") {
- name = name[3:]
- }
- imgObj, err := CachedimageManager.FetchById(name)
- if err == nil && imgObj != nil {
- cachedImage = imgObj.(*SCachedimage)
- }
- }
- if cachedImage == nil {
- // no such image
- cachedImage, err = CachedimageManager.newFromCloudImage(ctx, userCred, ownerId, image, provider)
- if err != nil {
- log.Errorf("CachedimageManager.newFromCloudImage fail %s", err)
- return err
- }
- }
- } else {
- cachedImage = imgObj.(*SCachedimage)
- }
- if len(cachedImage.ExternalId) > 0 {
- cachedImage.syncWithCloudImage(ctx, userCred, ownerId, image, provider)
- }
- scimg := manager.Register(ctx, userCred, cache.GetId(), cachedImage.GetId(), image.GetStatus())
- if scimg == nil {
- return fmt.Errorf("register cached image fail")
- }
- return scimg.SetExternalId(image.GetGlobalId())
- }
- func (manager *SStoragecachedimageManager) ListItemFilter(
- ctx context.Context,
- q *sqlchemy.SQuery,
- userCred mcclient.TokenCredential,
- query api.StoragecachedimageListInput,
- ) (*sqlchemy.SQuery, error) {
- var err error
- q, err = manager.SJointResourceBaseManager.ListItemFilter(ctx, q, userCred, query.JointResourceBaseListInput)
- if err != nil {
- return nil, errors.Wrap(err, "SSchedtagJointsManager.ListItemFilter")
- }
- q, err = manager.SExternalizedResourceBaseManager.ListItemFilter(ctx, q, userCred, query.ExternalizedResourceBaseListInput)
- if err != nil {
- return nil, errors.Wrap(err, "SExternalizedResourceBaseManager.ListItemFilter")
- }
- q, err = manager.SStoragecacheResourceBaseManager.ListItemFilter(ctx, q, userCred, query.StoragecacheFilterListInput)
- if err != nil {
- return nil, errors.Wrap(err, "SStoragecacheResourceBaseManager.ListItemFilter")
- }
- if len(query.CachedimageId) > 0 {
- cachedImageObj, err := CachedimageManager.FetchByIdOrName(ctx, userCred, query.CachedimageId)
- if err != nil {
- if errors.Cause(err) == sql.ErrNoRows {
- return nil, httperrors.NewResourceNotFoundError2(CachedimageManager.Keyword(), query.CachedimageId)
- } else {
- return nil, errors.Wrap(err, "CachedimageManager.FetchByIdOrName")
- }
- }
- q = q.Equals("cachedimage_id", cachedImageObj.GetId())
- }
- if len(query.Status) > 0 {
- q = q.In("status", query.Status)
- }
- return q, nil
- }
- func (manager *SStoragecachedimageManager) OrderByExtraFields(
- ctx context.Context,
- q *sqlchemy.SQuery,
- userCred mcclient.TokenCredential,
- query api.StoragecachedimageListInput,
- ) (*sqlchemy.SQuery, error) {
- var err error
- q, err = manager.SJointResourceBaseManager.OrderByExtraFields(ctx, q, userCred, query.JointResourceBaseListInput)
- if err != nil {
- return nil, errors.Wrap(err, "SSchedtagJointsManager.OrderByExtraFields")
- }
- q, err = manager.SStoragecacheResourceBaseManager.OrderByExtraFields(ctx, q, userCred, query.StoragecacheFilterListInput)
- if err != nil {
- return nil, errors.Wrap(err, "SStoragecacheResourceBaseManager.OrderByExtraFields")
- }
- return q, nil
- }
- func (manager *SStoragecachedimageManager) InitializeData() error {
- images := []SStoragecachedimage{}
- q := manager.Query().Equals("status", "ready")
- err := db.FetchModelObjects(manager, q, &images)
- if err != nil {
- return errors.Wrapf(err, "db.FetchModelObjects")
- }
- for i := range images {
- _, err := db.Update(&images[i], func() error {
- images[i].Status = api.CACHED_IMAGE_STATUS_ACTIVE
- return nil
- })
- if err != nil {
- return errors.Wrapf(err, "db.Update(%d)", images[i].RowId)
- }
- }
- return nil
- }
|