standalone_anon.go 36 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131
  1. // Copyright 2019 Yunion
  2. //
  3. // Licensed under the Apache License, Version 2.0 (the "License");
  4. // you may not use this file except in compliance with the License.
  5. // You may obtain a copy of the License at
  6. //
  7. // http://www.apache.org/licenses/LICENSE-2.0
  8. //
  9. // Unless required by applicable law or agreed to in writing, software
  10. // distributed under the License is distributed on an "AS IS" BASIS,
  11. // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  12. // See the License for the specific language governing permissions and
  13. // limitations under the License.
  14. package db
  15. import (
  16. "context"
  17. "crypto/md5"
  18. "database/sql"
  19. "fmt"
  20. "strings"
  21. "time"
  22. "yunion.io/x/jsonutils"
  23. "yunion.io/x/log"
  24. "yunion.io/x/pkg/errors"
  25. "yunion.io/x/pkg/util/rbacscope"
  26. "yunion.io/x/pkg/util/stringutils"
  27. "yunion.io/x/sqlchemy"
  28. "yunion.io/x/onecloud/pkg/apis"
  29. "yunion.io/x/onecloud/pkg/cloudcommon/consts"
  30. "yunion.io/x/onecloud/pkg/cloudcommon/db/lockman"
  31. "yunion.io/x/onecloud/pkg/cloudcommon/policy"
  32. "yunion.io/x/onecloud/pkg/httperrors"
  33. "yunion.io/x/onecloud/pkg/mcclient"
  34. "yunion.io/x/onecloud/pkg/util/stringutils2"
  35. "yunion.io/x/onecloud/pkg/util/tagutils"
  36. )
  37. type UUIDGenerator func() string
  38. var (
  39. DefaultUUIDGenerator = stringutils.UUID4
  40. )
  41. type SStandaloneAnonResourceBase struct {
  42. SResourceBase
  43. // 资源UUID
  44. Id string `width:"128" charset:"ascii" primary:"true" list:"user" create:"optional" json:"id"`
  45. // 资源描述信息
  46. Description string `length:"0" charset:"utf8" get:"user" list:"user" update:"user" create:"optional" json:"description"`
  47. // 是否是模拟资源, 部分从公有云上同步的资源并不真实存在, 例如宿主机
  48. // list 接口默认不会返回这类资源,除非显示指定 is_emulate=true 过滤参数
  49. IsEmulated bool `nullable:"false" default:"false" list:"admin" create:"admin_optional" json:"is_emulated"`
  50. // 用以组织架构变更通知其他服务权限变更
  51. OrgNodeMd5 string `width:"32" charset:"ascii" nullable:"true"`
  52. }
  53. func (model *SStandaloneAnonResourceBase) BeforeInsert() {
  54. if len(model.Id) == 0 {
  55. model.Id = DefaultUUIDGenerator()
  56. }
  57. }
  58. type SStandaloneAnonResourceBaseManager struct {
  59. SResourceBaseManager
  60. SMetadataResourceBaseModelManager
  61. }
  62. func NewStandaloneAnonResourceBaseManager(
  63. dt interface{},
  64. tableName string,
  65. keyword string,
  66. keywordPlural string,
  67. ) SStandaloneAnonResourceBaseManager {
  68. return SStandaloneAnonResourceBaseManager{
  69. SResourceBaseManager: NewResourceBaseManager(dt, tableName, keyword, keywordPlural),
  70. }
  71. }
  72. func (manager *SStandaloneAnonResourceBaseManager) CreateByInsertOrUpdate() bool {
  73. return false
  74. }
  75. func (manager *SStandaloneAnonResourceBaseManager) IsStandaloneManager() bool {
  76. return true
  77. }
  78. func (manager *SStandaloneAnonResourceBaseManager) GetIStandaloneModelManager() IStandaloneModelManager {
  79. return manager.GetVirtualObject().(IStandaloneModelManager)
  80. }
  81. func (manager *SStandaloneAnonResourceBaseManager) FilterById(q *sqlchemy.SQuery, idStr string) *sqlchemy.SQuery {
  82. return q.Equals("id", idStr)
  83. }
  84. func (manager *SStandaloneAnonResourceBaseManager) FilterByName(q *sqlchemy.SQuery, name string) *sqlchemy.SQuery {
  85. return q.FilterByFalse()
  86. }
  87. func (manager *SStandaloneAnonResourceBaseManager) FilterByNotId(q *sqlchemy.SQuery, idStr string) *sqlchemy.SQuery {
  88. return q.NotEquals("id", idStr)
  89. }
  90. func (manager *SStandaloneAnonResourceBaseManager) FilterByOwner(ctx context.Context, q *sqlchemy.SQuery, man FilterByOwnerProvider, userCred mcclient.TokenCredential, owner mcclient.IIdentityProvider, scope rbacscope.TRbacScope) *sqlchemy.SQuery {
  91. if userCred != nil {
  92. result := policy.PolicyManager.Allow(scope, userCred, consts.GetServiceType(), man.KeywordPlural(), policy.PolicyActionList)
  93. if !result.ObjectTags.IsEmpty() {
  94. policyTagFilters := tagutils.STagFilters{}
  95. policyTagFilters.AddFilters(result.ObjectTags)
  96. q = ObjectIdQueryWithTagFilters(ctx, q, "id", man.Keyword(), policyTagFilters)
  97. }
  98. }
  99. return q
  100. }
  101. func (manager *SStandaloneAnonResourceBaseManager) FilterByHiddenSystemAttributes(q *sqlchemy.SQuery, userCred mcclient.TokenCredential, query jsonutils.JSONObject, scope rbacscope.TRbacScope) *sqlchemy.SQuery {
  102. q = manager.SResourceBaseManager.FilterByHiddenSystemAttributes(q, userCred, query, scope)
  103. showEmulated := jsonutils.QueryBoolean(query, "show_emulated", false)
  104. if showEmulated {
  105. var isAllow bool
  106. // TODO, add tagfilter
  107. allowScope, _ := policy.PolicyManager.AllowScope(userCred, consts.GetServiceType(), manager.KeywordPlural(), policy.PolicyActionList, "show_emulated")
  108. if !scope.HigherThan(allowScope) {
  109. isAllow = true
  110. }
  111. if !isAllow {
  112. showEmulated = false
  113. }
  114. }
  115. if !showEmulated {
  116. q = q.IsFalse("is_emulated")
  117. }
  118. return q
  119. }
  120. func (manager *SStandaloneAnonResourceBaseManager) FetchById(idStr string) (IModel, error) {
  121. return FetchById(manager.GetIStandaloneModelManager(), idStr)
  122. }
  123. func (manager *SStandaloneAnonResourceBaseManager) ListItemFilter(
  124. ctx context.Context,
  125. q *sqlchemy.SQuery,
  126. userCred mcclient.TokenCredential,
  127. input apis.StandaloneAnonResourceListInput,
  128. ) (*sqlchemy.SQuery, error) {
  129. q, err := manager.SResourceBaseManager.ListItemFilter(ctx, q, userCred, input.ResourceBaseListInput)
  130. if err != nil {
  131. return q, errors.Wrap(err, "SResourceBaseManager.ListItemFilte")
  132. }
  133. // show_emulated is handled by FilterByHiddenSystemAttributes
  134. if len(input.Ids) > 0 {
  135. q = q.In("id", input.Ids)
  136. }
  137. q = manager.SMetadataResourceBaseModelManager.ListItemFilter(ctx, manager.GetIModelManager(), q, input.MetadataResourceListInput)
  138. return q, nil
  139. }
  140. func (manager *SStandaloneAnonResourceBaseManager) QueryDistinctExtraField(q *sqlchemy.SQuery, field string) (*sqlchemy.SQuery, error) {
  141. q, err := manager.SMetadataResourceBaseModelManager.QueryDistinctExtraField(manager.GetIModelManager(), q, field)
  142. if err == nil {
  143. return q, nil
  144. }
  145. q, err = manager.SResourceBaseManager.QueryDistinctExtraField(q, field)
  146. if err == nil {
  147. return q, nil
  148. }
  149. return q, httperrors.ErrNotFound
  150. }
  151. func (manager *SStandaloneAnonResourceBaseManager) OrderByExtraFields(
  152. ctx context.Context,
  153. q *sqlchemy.SQuery,
  154. userCred mcclient.TokenCredential,
  155. input apis.StandaloneAnonResourceListInput,
  156. ) (*sqlchemy.SQuery, error) {
  157. q, err := manager.SResourceBaseManager.OrderByExtraFields(ctx, q, userCred, input.ResourceBaseListInput)
  158. if err != nil {
  159. return nil, errors.Wrap(err, "SResourceBaseManager.OrderByExtraFields")
  160. }
  161. q = manager.SMetadataResourceBaseModelManager.OrderByExtraFields(manager.GetIModelManager(), q, input.MetadataResourceListInput)
  162. return q, nil
  163. }
  164. func (model *SStandaloneAnonResourceBase) StandaloneModelManager() IStandaloneModelManager {
  165. return model.GetModelManager().(IStandaloneModelManager)
  166. }
  167. func (model SStandaloneAnonResourceBase) GetId() string {
  168. return model.Id
  169. }
  170. func (model *SStandaloneAnonResourceBase) GetIStandaloneModel() IStandaloneModel {
  171. return model.GetVirtualObject().(IStandaloneModel)
  172. }
  173. func (model *SStandaloneAnonResourceBase) GetShortDesc(ctx context.Context) *jsonutils.JSONDict {
  174. desc := model.SResourceBase.GetShortDesc(ctx)
  175. desc.Add(jsonutils.NewString(model.GetId()), "id")
  176. /*if len(model.ExternalId) > 0 {
  177. desc.Add(jsonutils.NewString(model.ExternalId), "external_id")
  178. }*/
  179. return desc
  180. }
  181. func (model *SStandaloneAnonResourceBase) GetShortDescV2(ctx context.Context) *apis.StandaloneAnonResourceShortDescDetail {
  182. desc := &apis.StandaloneAnonResourceShortDescDetail{}
  183. desc.ModelBaseShortDescDetail = *model.SResourceBase.GetShortDescV2(ctx)
  184. desc.Id = model.GetId()
  185. return desc
  186. }
  187. /*
  188. * userCred: optional
  189. */
  190. func (model *SStandaloneAnonResourceBase) GetMetadata(ctx context.Context, key string, userCred mcclient.TokenCredential) string {
  191. return Metadata.GetStringValue(ctx, model, key, userCred)
  192. }
  193. func (model *SStandaloneAnonResourceBase) GetMetadataJson(ctx context.Context, key string, userCred mcclient.TokenCredential) jsonutils.JSONObject {
  194. return Metadata.GetJsonValue(ctx, model, key, userCred)
  195. }
  196. func isUserMetadata(key string) bool {
  197. return strings.HasPrefix(key, USER_TAG_PREFIX)
  198. }
  199. func containsUserMetadata(dict map[string]interface{}) bool {
  200. for k := range dict {
  201. if isUserMetadata(k) {
  202. return true
  203. }
  204. }
  205. return false
  206. }
  207. func (model *SStandaloneAnonResourceBase) SetMetadata(ctx context.Context, key string, value interface{}, userCred mcclient.TokenCredential) error {
  208. err := Metadata.SetValue(ctx, model, key, value, userCred)
  209. if err != nil {
  210. return errors.Wrap(err, "SetValue")
  211. }
  212. if isUserMetadata(key) {
  213. model.GetIStandaloneModel().OnMetadataUpdated(ctx, userCred)
  214. }
  215. return nil
  216. }
  217. func (model *SStandaloneAnonResourceBase) SetAllMetadata(ctx context.Context, dictstore map[string]interface{}, userCred mcclient.TokenCredential) error {
  218. err := Metadata.SetValuesWithLog(ctx, model, dictstore, userCred)
  219. if err != nil {
  220. return errors.Wrap(err, "SetValuesWithLog")
  221. }
  222. if containsUserMetadata(dictstore) {
  223. model.GetIStandaloneModel().OnMetadataUpdated(ctx, userCred)
  224. }
  225. return nil
  226. }
  227. func ensurePrefix(input map[string]interface{}, prefix string) (map[string]interface{}, error) {
  228. dictStore := make(map[string]interface{}, len(input))
  229. for k, v := range input {
  230. if !strings.HasPrefix(k, prefix) {
  231. k = prefix + k
  232. }
  233. if len(k) > 64 {
  234. return nil, httperrors.NewInputParameterError("input key too long > %d", 64-len(prefix))
  235. }
  236. if len(v.(string)) > 65535 {
  237. return nil, httperrors.NewInputParameterError("input value too long > %d", 65535)
  238. }
  239. dictStore[k] = v
  240. }
  241. return dictStore, nil
  242. }
  243. func ensurePrefixString(input map[string]string, prefix string) (map[string]interface{}, error) {
  244. dictStore := make(map[string]interface{}, len(input))
  245. for k, v := range input {
  246. if !strings.HasPrefix(k, prefix) {
  247. k = prefix + k
  248. }
  249. if len(k) > 64 {
  250. return nil, httperrors.NewInputParameterError("input key too long > %d", 64-len(prefix))
  251. }
  252. if len(v) > 65535 {
  253. return nil, httperrors.NewInputParameterError("input value too long > %d", 65535)
  254. }
  255. dictStore[k] = v
  256. }
  257. return dictStore, nil
  258. }
  259. func (model *SStandaloneAnonResourceBase) SetUserMetadataValues(ctx context.Context, dictstore map[string]string, userCred mcclient.TokenCredential) error {
  260. dictStore, err := ensurePrefixString(dictstore, USER_TAG_PREFIX)
  261. if err != nil {
  262. return errors.Wrapf(err, "ensurePrefixString %s", USER_TAG_PREFIX)
  263. }
  264. err = Metadata.SetValuesWithLog(ctx, model, dictStore, userCred)
  265. if err != nil {
  266. return errors.Wrap(err, "SetValuesWithLog")
  267. }
  268. {
  269. model.GetModelManager().TableSpec().InformUpdate(ctx, model, jsonutils.Marshal(model).(*jsonutils.JSONDict))
  270. }
  271. model.GetIStandaloneModel().OnMetadataUpdated(ctx, userCred)
  272. return nil
  273. }
  274. func (model *SStandaloneAnonResourceBase) SetUserMetadataAll(ctx context.Context, dictstore map[string]string, userCred mcclient.TokenCredential) error {
  275. var err error
  276. dictStore, err := ensurePrefixString(dictstore, USER_TAG_PREFIX)
  277. if err != nil {
  278. return errors.Wrap(err, "ensurePrefix")
  279. }
  280. err = Metadata.SetAll(ctx, model, dictStore, userCred, USER_TAG_PREFIX)
  281. if err != nil {
  282. return errors.Wrap(err, "SetAll")
  283. }
  284. {
  285. model.GetModelManager().TableSpec().InformUpdate(ctx, model, jsonutils.Marshal(model).(*jsonutils.JSONDict))
  286. }
  287. model.GetIStandaloneModel().OnMetadataUpdated(ctx, userCred)
  288. return nil
  289. }
  290. func (model *SStandaloneAnonResourceBase) SetCloudMetadataAll(ctx context.Context, dictstore map[string]string, userCred mcclient.TokenCredential, readOnly bool) error {
  291. var err error
  292. dictStore, err := ensurePrefixString(dictstore, CLOUD_TAG_PREFIX)
  293. if err != nil {
  294. return errors.Wrap(err, "ensurePrefix")
  295. }
  296. if readOnly {
  297. err = Metadata.SetAllWithoutDelelte(ctx, model, dictStore, userCred)
  298. if err != nil {
  299. return errors.Wrap(err, "SetAll")
  300. }
  301. } else {
  302. err = Metadata.SetAll(ctx, model, dictStore, userCred, CLOUD_TAG_PREFIX)
  303. if err != nil {
  304. return errors.Wrap(err, "SetAll")
  305. }
  306. }
  307. userTags := map[string]interface{}{}
  308. for k, v := range dictstore {
  309. userTags[strings.Replace(k, CLOUD_TAG_PREFIX, USER_TAG_PREFIX, 1)] = v
  310. }
  311. if readOnly {
  312. return Metadata.SetAllWithoutDelelte(ctx, model, userTags, userCred)
  313. }
  314. return Metadata.SetAll(ctx, model, userTags, userCred, USER_TAG_PREFIX)
  315. }
  316. func (model *SStandaloneAnonResourceBase) SetOrganizationMetadataAll(ctx context.Context, meta map[string]string, userCred mcclient.TokenCredential) error {
  317. {
  318. dictStore, err := ensurePrefixString(meta, ORGANIZATION_TAG_PREFIX)
  319. if err != nil {
  320. return errors.Wrap(err, "ensurePrefix")
  321. }
  322. err = Metadata.SetAll(ctx, model, dictStore, userCred, ORGANIZATION_TAG_PREFIX)
  323. if err != nil {
  324. return errors.Wrap(err, "SetAllOrganization")
  325. }
  326. }
  327. Update(model, func() error {
  328. model.OrgNodeMd5 = fmt.Sprintf("%x", md5.Sum([]byte(jsonutils.Marshal(meta).String())))
  329. return nil
  330. })
  331. // 避免加入组织架构后,项目所在的层级会移除此项目
  332. //{
  333. // userTags := make(map[string]interface{})
  334. // for k, _ := range meta {
  335. // if strings.HasPrefix(k, ORGANIZATION_TAG_PREFIX) {
  336. // k = k[len(ORGANIZATION_TAG_PREFIX):]
  337. // }
  338. // k = USER_TAG_PREFIX + k
  339. // userTags[k] = "none"
  340. // }
  341. // err := Metadata.SetValuesWithLog(ctx, model, userTags, userCred)
  342. // if err != nil {
  343. // return errors.Wrap(err, "SetValuesWithLog userTags")
  344. // }
  345. //}
  346. return nil
  347. }
  348. func (model *SStandaloneAnonResourceBase) SetClassMetadataValues(ctx context.Context, dictstore map[string]string, userCred mcclient.TokenCredential) error {
  349. var err error
  350. dictStore, err := ensurePrefixString(dictstore, CLASS_TAG_PREFIX)
  351. if err != nil {
  352. return errors.Wrap(err, "ensurePrefixString")
  353. }
  354. err = Metadata.SetValuesWithLog(ctx, model, dictStore, userCred)
  355. if err != nil {
  356. return errors.Wrap(err, "SetValuesWithLog")
  357. }
  358. return nil
  359. }
  360. func (model *SStandaloneAnonResourceBase) SetClassMetadataAll(ctx context.Context, dictstore map[string]string, userCred mcclient.TokenCredential) error {
  361. var err error
  362. afterCheck, err := ensurePrefixString(dictstore, CLASS_TAG_PREFIX)
  363. if err != nil {
  364. return errors.Wrap(err, "ensurePrefix")
  365. }
  366. err = Metadata.SetAll(ctx, model, afterCheck, userCred, CLASS_TAG_PREFIX)
  367. if err != nil {
  368. return errors.Wrap(err, "SetAll")
  369. }
  370. return nil
  371. }
  372. func (model *SStandaloneAnonResourceBase) InheritTo(ctx context.Context, userCred mcclient.TokenCredential, dest IClassMetadataSetter) error {
  373. return InheritFromTo(ctx, userCred, model, dest)
  374. }
  375. type IClassMetadataSetter interface {
  376. // a setter should first be a owner
  377. IClassMetadataOwner
  378. SetClassMetadataAll(context.Context, map[string]string, mcclient.TokenCredential) error
  379. }
  380. func InheritFromTo(ctx context.Context, userCred mcclient.TokenCredential, src IClassMetadataOwner, dest IClassMetadataSetter) error {
  381. metadata, err := src.GetAllClassMetadata()
  382. if err != nil {
  383. return errors.Wrap(err, "GetAllClassMetadata")
  384. }
  385. if len(metadata) == 0 {
  386. return nil
  387. }
  388. curMeta, err := dest.GetAllClassMetadata()
  389. if err != nil {
  390. return errors.Wrap(err, "GetAllClassMetadata dest")
  391. }
  392. if len(curMeta) > 0 {
  393. // check conflict
  394. for k, v := range curMeta {
  395. if sv, ok := metadata[k]; ok {
  396. if sv != v {
  397. // duplicate value for identical key
  398. // return errors.Wrapf(httperrors.ErrConflict, "destination has another value for class key %s", k)
  399. log.Warningf("replace class metadata %s from %s to %s", k, v, sv)
  400. }
  401. } else {
  402. // no such class key
  403. metadata[k] = "None"
  404. // return errors.Wrapf(httperrors.ErrConflict, "destination has extra class key %s", k)
  405. }
  406. }
  407. }
  408. // userCred := auth.AdminCredential()
  409. return dest.SetClassMetadataAll(ctx, metadata, userCred)
  410. }
  411. type IClassMetadataOwner interface {
  412. GetAllClassMetadata() (map[string]string, error)
  413. }
  414. type AllMetadataOwner map[string]string
  415. func (w AllMetadataOwner) GetAllClassMetadata() (map[string]string, error) {
  416. ret := make(map[string]string)
  417. for k, v := range w {
  418. if strings.HasPrefix(k, CLASS_TAG_PREFIX) {
  419. ret[k[len(CLASS_TAG_PREFIX):]] = v
  420. }
  421. }
  422. return ret, nil
  423. }
  424. type ClassMetadataOwner map[string]string
  425. func (w ClassMetadataOwner) GetAllClassMetadata() (map[string]string, error) {
  426. return w, nil
  427. }
  428. func IsInSameClass(ctx context.Context, cmo1, cmo2 IClassMetadataOwner) (bool, error) {
  429. pureTags, err := cmo1.GetAllClassMetadata()
  430. if err != nil {
  431. return false, errors.Wrap(err, "GetAllPureMetadata")
  432. }
  433. pureTagsP, err := cmo2.GetAllClassMetadata()
  434. if err != nil {
  435. return false, errors.Wrap(err, "GetAllPureMetadata")
  436. }
  437. if len(pureTags) != len(pureTagsP) {
  438. return false, nil
  439. }
  440. for k, v := range pureTags {
  441. if vp, ok := pureTagsP[k]; !ok || vp != v {
  442. return false, nil
  443. }
  444. }
  445. return true, nil
  446. }
  447. func RequireSameClass(ctx context.Context, cmo1, cmo2 IClassMetadataOwner) error {
  448. same, err := IsInSameClass(ctx, cmo1, cmo2)
  449. if err != nil {
  450. return errors.Wrap(err, "IsInSameClass")
  451. }
  452. if !same {
  453. tag1, _ := cmo1.GetAllClassMetadata()
  454. tag2, _ := cmo2.GetAllClassMetadata()
  455. return errors.Wrapf(httperrors.ErrConflict, "inconsist class metadata: %s %s", tag1, tag2)
  456. }
  457. return nil
  458. }
  459. func (model *SStandaloneAnonResourceBase) IsInSameClass(ctx context.Context, pModel *SStandaloneAnonResourceBase) (bool, error) {
  460. return IsInSameClass(ctx, model, pModel)
  461. }
  462. func (model *SStandaloneAnonResourceBase) SetSysCloudMetadataAll(ctx context.Context, dictstore map[string]string, userCred mcclient.TokenCredential, readOnly bool) error {
  463. dictStore, err := ensurePrefixString(dictstore, SYS_CLOUD_TAG_PREFIX)
  464. if err != nil {
  465. return errors.Wrap(err, "ensurePrefixString")
  466. }
  467. if readOnly {
  468. err = Metadata.SetAllWithoutDelelte(ctx, model, dictStore, userCred)
  469. } else {
  470. err = Metadata.SetAll(ctx, model, dictStore, userCred, SYS_CLOUD_TAG_PREFIX)
  471. }
  472. if err != nil {
  473. return errors.Wrap(err, "SetAll")
  474. }
  475. return nil
  476. }
  477. func (model *SStandaloneAnonResourceBase) RemoveMetadata(ctx context.Context, key string, userCred mcclient.TokenCredential) error {
  478. err := Metadata.SetValue(ctx, model, key, "", userCred)
  479. if err != nil {
  480. return errors.Wrap(err, "SetValue")
  481. }
  482. return nil
  483. }
  484. func (model *SStandaloneAnonResourceBase) RemoveAllMetadata(ctx context.Context, userCred mcclient.TokenCredential) error {
  485. err := Metadata.RemoveAll(ctx, model, userCred)
  486. if err != nil {
  487. return errors.Wrap(err, "RemoveAll")
  488. }
  489. return nil
  490. }
  491. func (model *SStandaloneAnonResourceBase) GetAllMetadata(ctx context.Context, userCred mcclient.TokenCredential) (map[string]string, error) {
  492. return Metadata.GetAll(ctx, model, nil, "", userCred)
  493. }
  494. func (model *SStandaloneAnonResourceBase) GetAllUserMetadata() (map[string]string, error) {
  495. meta, err := Metadata.GetAll(nil, model, nil, USER_TAG_PREFIX, nil)
  496. if err != nil {
  497. return nil, errors.Wrap(err, "Metadata.GetAll")
  498. }
  499. ret := make(map[string]string)
  500. for k, v := range meta {
  501. ret[k[len(USER_TAG_PREFIX):]] = v
  502. }
  503. return ret, nil
  504. }
  505. func (model *SStandaloneAnonResourceBase) GetAllCloudMetadata() (map[string]string, error) {
  506. meta, err := Metadata.GetAll(nil, model, nil, CLOUD_TAG_PREFIX, nil)
  507. if err != nil {
  508. return nil, errors.Wrap(err, "Metadata.GetAll")
  509. }
  510. ret := make(map[string]string)
  511. for k, v := range meta {
  512. ret[k[len(CLOUD_TAG_PREFIX):]] = v
  513. }
  514. return ret, nil
  515. }
  516. func (model *SStandaloneAnonResourceBase) GetAllClassMetadata() (map[string]string, error) {
  517. meta, err := Metadata.GetAll(nil, model, nil, CLASS_TAG_PREFIX, nil)
  518. if err != nil {
  519. return nil, errors.Wrap(err, "Metadata.GetAll")
  520. }
  521. ret := make(map[string]string)
  522. for k, v := range meta {
  523. ret[k[len(CLASS_TAG_PREFIX):]] = v
  524. }
  525. return ret, nil
  526. }
  527. func (model *SStandaloneAnonResourceBase) GetAllOrganizationMetadata() (map[string]string, error) {
  528. meta, err := Metadata.GetAll(nil, model, nil, ORGANIZATION_TAG_PREFIX, nil)
  529. if err != nil {
  530. return nil, errors.Wrap(err, "Metadata.GetAll")
  531. }
  532. ret := make(map[string]string)
  533. for k, v := range meta {
  534. ret[k[len(ORGANIZATION_TAG_PREFIX):]] = v
  535. }
  536. return ret, nil
  537. }
  538. // 获取资源标签(元数据)
  539. func (model *SStandaloneAnonResourceBase) GetDetailsMetadata(ctx context.Context, userCred mcclient.TokenCredential, input apis.GetMetadataInput) (apis.GetMetadataOutput, error) {
  540. val, err := Metadata.GetAll(ctx, model, input.Field, input.Prefix, userCred)
  541. if err != nil {
  542. return nil, errors.Wrap(err, "Metadata.GetAll")
  543. }
  544. if len(input.Prefix) > 0 {
  545. // trim prefix from key
  546. ret := make(map[string]string)
  547. for k, v := range val {
  548. ret[k[len(input.Prefix):]] = v
  549. }
  550. val = ret
  551. }
  552. return val, nil
  553. }
  554. // +onecloud:swagger-gen-ignore
  555. func (model *SStandaloneAnonResourceBase) PerformMetadata(ctx context.Context, userCred mcclient.TokenCredential, query jsonutils.JSONObject, input apis.PerformMetadataInput) (jsonutils.JSONObject, error) {
  556. dictStore := make(map[string]interface{})
  557. for k, v := range input {
  558. // 已双下滑线开头的metadata是系统内置,普通用户不可添加,只能查看
  559. if strings.HasPrefix(k, SYS_TAG_PREFIX) && (userCred == nil || !IsAllowPerform(ctx, rbacscope.ScopeSystem, userCred, model, "metadata")) {
  560. return nil, httperrors.NewForbiddenError("not allow to set system key, please remove the underscore at the beginning")
  561. }
  562. dictStore[k] = v
  563. }
  564. err := model.SetAllMetadata(ctx, dictStore, userCred)
  565. if err != nil {
  566. return nil, errors.Wrap(err, "SetAllMetadata")
  567. }
  568. return nil, nil
  569. }
  570. // 更新资源的用户标签
  571. // +onecloud:swagger-gen-ignore
  572. func (model *SStandaloneAnonResourceBase) PerformUserMetadata(ctx context.Context, userCred mcclient.TokenCredential, query jsonutils.JSONObject, input apis.PerformUserMetadataInput) (jsonutils.JSONObject, error) {
  573. err := model.SetUserMetadataValues(ctx, input, userCred)
  574. if err != nil {
  575. return nil, errors.Wrap(err, "SetUserMetadataValues")
  576. }
  577. return nil, nil
  578. }
  579. // 全量替换资源的所有用户标签
  580. // +onecloud:swagger-gen-ignore
  581. func (model *SStandaloneAnonResourceBase) PerformSetUserMetadata(ctx context.Context, userCred mcclient.TokenCredential, query jsonutils.JSONObject, input apis.PerformSetUserMetadataInput) (jsonutils.JSONObject, error) {
  582. err := model.SetUserMetadataAll(ctx, input, userCred)
  583. if err != nil {
  584. return nil, errors.Wrap(err, "SetUserMetadataAll")
  585. }
  586. return nil, nil
  587. }
  588. // 更新资源的 class 标签
  589. // +onecloud:swagger-gen-ignore
  590. func (model *SStandaloneAnonResourceBase) PerformClassMetadata(ctx context.Context, userCred mcclient.TokenCredential, query jsonutils.JSONObject, input apis.PerformClassMetadataInput) (jsonutils.JSONObject, error) {
  591. err := model.SetClassMetadataValues(ctx, input, userCred)
  592. if err != nil {
  593. return nil, errors.Wrap(err, "SetUserMetadataValues")
  594. }
  595. return nil, nil
  596. }
  597. // 全量替换资源的所有 class 标签
  598. // +onecloud:swagger-gen-ignore
  599. func (model *SStandaloneAnonResourceBase) PerformSetClassMetadata(ctx context.Context, userCred mcclient.TokenCredential, query jsonutils.JSONObject, input apis.PerformSetClassMetadataInput) (jsonutils.JSONObject, error) {
  600. err := model.SetClassMetadataAll(ctx, input, userCred)
  601. if err != nil {
  602. return nil, errors.Wrap(err, "SetUserMetadataAll")
  603. }
  604. return nil, nil
  605. }
  606. // 全量替换资源的所有 class 标签
  607. // +onecloud:swagger-gen-ignore
  608. func (model *SStandaloneAnonResourceBase) PerformSetOrgMetadata(ctx context.Context, userCred mcclient.TokenCredential, query jsonutils.JSONObject, input apis.PerformSetClassMetadataInput) (jsonutils.JSONObject, error) {
  609. err := model.SetOrganizationMetadataAll(ctx, input, userCred)
  610. if err != nil {
  611. return nil, errors.Wrap(err, "SetUserMetadataAll")
  612. }
  613. return nil, nil
  614. }
  615. func validateDictStore(input map[string]string, prefix string) (map[string]string, error) {
  616. dictStore := make(map[string]string, len(input))
  617. for k, v := range input {
  618. if len(k) > 64-len(prefix) {
  619. return nil, httperrors.NewInputParameterError("input key too long > %d", 64-len(prefix))
  620. }
  621. if len(v) > 65535 {
  622. return nil, httperrors.NewInputParameterError("input value too long > %d", 65535)
  623. }
  624. dictStore[k] = v
  625. }
  626. return dictStore, nil
  627. }
  628. // +onecloud:swagger-gen-ignore
  629. func (model *SStandaloneAnonResourceBase) GetDetailsClassMetadata(ctx context.Context, userCred mcclient.TokenCredential, input apis.GetClassMetadataInput) (apis.GetClassMetadataOutput, error) {
  630. return model.GetAllClassMetadata()
  631. }
  632. // +onecloud:swagger-gen-ignore
  633. func (model *SStandaloneAnonResourceBase) GetDetailsOrgMetadata(ctx context.Context, userCred mcclient.TokenCredential, input apis.GetClassMetadataInput) (apis.GetClassMetadataOutput, error) {
  634. return model.GetAllOrganizationMetadata()
  635. }
  636. type sPolicyTags struct {
  637. PolicyObjectTags tagutils.TTagSetList `json:"policy_object_tags"`
  638. PolicyProjectTags tagutils.TTagSetList `json:"policy_project_tags"`
  639. PolicyDomainTags tagutils.TTagSetList `json:"policy_domain_tags"`
  640. }
  641. func (model *SStandaloneAnonResourceBase) PostUpdate(ctx context.Context, userCred mcclient.TokenCredential, query jsonutils.JSONObject, data jsonutils.JSONObject) {
  642. model.SResourceBase.PostUpdate(ctx, userCred, query, data)
  643. model.TrySaveMetadataInput(ctx, userCred, data)
  644. }
  645. func (model *SStandaloneAnonResourceBase) PostCreate(ctx context.Context, userCred mcclient.TokenCredential, ownerId mcclient.IIdentityProvider, query jsonutils.JSONObject, data jsonutils.JSONObject) {
  646. model.SResourceBase.PostCreate(ctx, userCred, ownerId, query, data)
  647. model.TrySaveMetadataInput(ctx, userCred, data)
  648. }
  649. func (model *SStandaloneAnonResourceBase) TrySaveMetadataInput(ctx context.Context, userCred mcclient.TokenCredential, data jsonutils.JSONObject) {
  650. meta := make(map[string]string)
  651. err := data.Unmarshal(&meta, "__meta__")
  652. if err == nil {
  653. model.PerformMetadata(ctx, userCred, nil, meta)
  654. }
  655. model.applyPolicyTags(ctx, userCred, data)
  656. }
  657. func (model *SStandaloneAnonResourceBase) applyPolicyTags(ctx context.Context, userCred mcclient.TokenCredential, data jsonutils.JSONObject) {
  658. tags := sPolicyTags{}
  659. data.Unmarshal(&tags)
  660. log.Debugf("applyPolicyTags: %s", jsonutils.Marshal(tags))
  661. if len(tags.PolicyObjectTags) > 0 {
  662. model.PerformMetadata(ctx, userCred, nil, tagutils.TagsetMap2MapString(tags.PolicyObjectTags.Flattern()))
  663. }
  664. if model.Keyword() == "project" && len(tags.PolicyProjectTags) > 0 {
  665. model.PerformMetadata(ctx, userCred, nil, tagutils.TagsetMap2MapString(tags.PolicyProjectTags.Flattern()))
  666. } else if model.Keyword() == "domain" && len(tags.PolicyDomainTags) > 0 {
  667. model.PerformMetadata(ctx, userCred, nil, tagutils.TagsetMap2MapString(tags.PolicyDomainTags.Flattern()))
  668. }
  669. }
  670. func (model *SStandaloneAnonResourceBase) ClearSchedDescCache() error {
  671. return nil
  672. }
  673. func (model *SStandaloneAnonResourceBase) AppendDescription(userCred mcclient.TokenCredential, msg string) error {
  674. _, err := Update(model.GetIStandaloneModel(), func() error {
  675. if len(model.Description) > 0 {
  676. model.Description += ";"
  677. }
  678. model.Description += msg
  679. return nil
  680. })
  681. if err != nil {
  682. return errors.Wrap(err, "db.Update")
  683. }
  684. OpsLog.LogEvent(model, "append_desc", msg, userCred)
  685. return nil
  686. }
  687. func (manager *SStandaloneAnonResourceBaseManager) ValidateCreateData(ctx context.Context, userCred mcclient.TokenCredential, ownerId mcclient.IIdentityProvider, query jsonutils.JSONObject, input apis.StandaloneAnonResourceCreateInput) (apis.StandaloneAnonResourceCreateInput, error) {
  688. var err error
  689. input.ResourceBaseCreateInput, err = manager.SResourceBaseManager.ValidateCreateData(ctx, userCred, ownerId, query, input.ResourceBaseCreateInput)
  690. if err != nil {
  691. return input, errors.Wrap(err, "SResourceBaseManager.ValidateCreateData")
  692. }
  693. return input, nil
  694. }
  695. func (manager *SStandaloneAnonResourceBaseManager) FetchCustomizeColumns(
  696. ctx context.Context,
  697. userCred mcclient.TokenCredential,
  698. query jsonutils.JSONObject,
  699. objs []interface{},
  700. fields stringutils2.SSortedStrings,
  701. isList bool,
  702. ) []apis.StandaloneAnonResourceDetails {
  703. ret := make([]apis.StandaloneAnonResourceDetails, len(objs))
  704. upperRet := manager.SResourceBaseManager.FetchCustomizeColumns(ctx, userCred, query, objs, fields, isList)
  705. metaRet := manager.SMetadataResourceBaseModelManager.FetchCustomizeColumns(manager.GetIModelManager(), userCred, objs, fields)
  706. for i := range objs {
  707. ret[i] = apis.StandaloneAnonResourceDetails{
  708. ResourceBaseDetails: upperRet[i],
  709. MetadataResourceInfo: metaRet[i],
  710. }
  711. }
  712. return ret
  713. }
  714. func (manager *SStandaloneAnonResourceBaseManager) GetMetadataHiddenKeys() []string {
  715. return nil
  716. }
  717. func (manager *SStandaloneAnonResourceBaseManager) GetExportExtraKeys(ctx context.Context, keys stringutils2.SSortedStrings, rowMap map[string]string) *jsonutils.JSONDict {
  718. res := manager.SResourceBaseManager.GetExportExtraKeys(ctx, keys, rowMap)
  719. metaRes := manager.SMetadataResourceBaseModelManager.GetExportExtraKeys(keys, rowMap)
  720. if metaRes.Length() > 0 {
  721. res.Update(metaRes)
  722. }
  723. return res
  724. }
  725. func (manager *SStandaloneAnonResourceBaseManager) ListItemExportKeys(ctx context.Context, q *sqlchemy.SQuery, userCred mcclient.TokenCredential, keys stringutils2.SSortedStrings) (*sqlchemy.SQuery, error) {
  726. var err error
  727. q, err = manager.SResourceBaseManager.ListItemExportKeys(ctx, q, userCred, keys)
  728. if err != nil {
  729. return nil, errors.Wrap(err, "SResourceBaseManager.ListItemExportKeys")
  730. }
  731. q = manager.SMetadataResourceBaseModelManager.ListItemExportKeys(manager.GetIModelManager(), q, keys)
  732. return q, nil
  733. }
  734. func (model *SStandaloneAnonResourceBase) ValidateUpdateData(ctx context.Context, userCred mcclient.TokenCredential, query jsonutils.JSONObject, input apis.StandaloneAnonResourceBaseUpdateInput) (apis.StandaloneAnonResourceBaseUpdateInput, error) {
  735. var err error
  736. input.ResourceBaseUpdateInput, err = model.SResourceBase.ValidateUpdateData(ctx, userCred, query, input.ResourceBaseUpdateInput)
  737. if err != nil {
  738. return input, errors.Wrap(err, "SModelBase.ValidateUpdateData")
  739. }
  740. return input, nil
  741. }
  742. func (model *SStandaloneAnonResourceBase) IsShared() bool {
  743. return false
  744. }
  745. func (model *SStandaloneAnonResourceBase) OnMetadataUpdated(ctx context.Context, userCred mcclient.TokenCredential) {
  746. // noop
  747. }
  748. type SGetResourceTagValuePairsInput struct {
  749. apis.MetadataBaseFilterInput
  750. // Return keys only
  751. KeyOnly *bool `json:"key_only"`
  752. // Order by key of tags
  753. OrderByTagKey string `json:"order_by_tag_key"`
  754. }
  755. // +onecloud:swagger-gen-ignore
  756. func (manager *SStandaloneAnonResourceBaseManager) GetPropertyTagValuePairs(
  757. ctx context.Context,
  758. userCred mcclient.TokenCredential,
  759. query jsonutils.JSONObject,
  760. ) (jsonutils.JSONObject, error) {
  761. return GetPropertyTagValuePairs(
  762. manager.GetIStandaloneModelManager(),
  763. manager.Keyword(),
  764. "id",
  765. ctx,
  766. userCred,
  767. query,
  768. )
  769. }
  770. func GetPropertyTagValuePairs(
  771. manager IModelManager,
  772. tagObjType string,
  773. tagIdField string,
  774. ctx context.Context,
  775. userCred mcclient.TokenCredential,
  776. query jsonutils.JSONObject,
  777. ) (jsonutils.JSONObject, error) {
  778. input := SGetResourceTagValuePairsInput{}
  779. err := query.Unmarshal(&input)
  780. if err != nil {
  781. return nil, errors.Wrap(err, "Unmarshal")
  782. }
  783. sq := Metadata.Query().SubQuery()
  784. keyOnly := (input.KeyOnly != nil && *input.KeyOnly)
  785. queryKeys := []string{tagIdField}
  786. if tagIdField != "id" {
  787. queryKeys = append(queryKeys, "id")
  788. }
  789. objQ := manager.Query(queryKeys...)
  790. objQ, err = ListItemQueryFilters(manager, ctx, objQ, userCred, query, policy.PolicyActionList)
  791. if err != nil {
  792. return nil, errors.Wrap(err, "ListItemQueryFilters")
  793. }
  794. objSQ := objQ.SubQuery()
  795. var queryFields []sqlchemy.IQueryField
  796. if keyOnly {
  797. queryFields = []sqlchemy.IQueryField{
  798. sq.Field("key"),
  799. sqlchemy.COUNT("count", objSQ.Field("id")),
  800. }
  801. } else {
  802. queryFields = []sqlchemy.IQueryField{
  803. sq.Field("key"),
  804. sq.Field("value"),
  805. sqlchemy.COUNT("count", objSQ.Field("id")),
  806. }
  807. }
  808. q := sq.Query()
  809. q = q.Join(objSQ, sqlchemy.AND(
  810. sqlchemy.Equals(q.Field("obj_type"), tagObjType),
  811. sqlchemy.Equals(q.Field("obj_id"), objSQ.Field(tagIdField)),
  812. ))
  813. q = q.AppendField(queryFields...)
  814. q = Metadata.metadataBaseFilter(q, input.MetadataBaseFilterInput)
  815. if keyOnly {
  816. q = q.GroupBy(q.Field("key"))
  817. } else {
  818. q = q.GroupBy(q.Field("key"), q.Field("value"))
  819. }
  820. if input.OrderByTagKey == string(sqlchemy.SQL_ORDER_DESC) {
  821. q = q.Desc(q.Field("key"))
  822. if !keyOnly {
  823. q = q.Desc(q.Field("value"))
  824. }
  825. } else {
  826. q = q.Asc(q.Field("key"))
  827. if !keyOnly {
  828. q = q.Asc(q.Field("value"))
  829. }
  830. }
  831. metadatas := make([]struct {
  832. Key string
  833. Value string
  834. Count int64
  835. }, 0)
  836. err = q.All(&metadatas)
  837. if err != nil {
  838. return nil, errors.Wrap(err, "Query.All")
  839. }
  840. return jsonutils.Marshal(metadatas), nil
  841. }
  842. type SGetResourceTagValueTreeInput struct {
  843. Keys []string `json:"key"`
  844. ShowMap *bool `json:"show_map"`
  845. }
  846. // +onecloud:swagger-gen-ignore
  847. func (manager *SStandaloneAnonResourceBaseManager) GetPropertyTagValueTree(
  848. ctx context.Context,
  849. userCred mcclient.TokenCredential,
  850. query jsonutils.JSONObject,
  851. ) (jsonutils.JSONObject, error) {
  852. return GetPropertyTagValueTree(
  853. manager.GetIStandaloneModelManager(),
  854. manager.Keyword(),
  855. "id",
  856. "",
  857. ctx,
  858. userCred,
  859. query,
  860. )
  861. }
  862. func GetPropertyTagValueTree(
  863. manager IModelManager,
  864. tagObjType string,
  865. tagIdField string,
  866. sumField string,
  867. ctx context.Context,
  868. userCred mcclient.TokenCredential,
  869. query jsonutils.JSONObject,
  870. ) (jsonutils.JSONObject, error) {
  871. input := SGetResourceTagValueTreeInput{}
  872. err := query.Unmarshal(&input)
  873. if err != nil {
  874. return nil, errors.Wrap(err, "Unmarshal")
  875. }
  876. valueMap, err := GetTagValueCountMap(manager, tagObjType, tagIdField, sumField, input.Keys, ctx, userCred, query)
  877. if err != nil {
  878. return nil, errors.Wrap(err, "AllStringAmp")
  879. }
  880. if input.ShowMap != nil && *input.ShowMap {
  881. return jsonutils.Marshal(valueMap), nil
  882. }
  883. tree := constructTree(valueMap, input.Keys)
  884. return jsonutils.Marshal(tree), nil
  885. }
  886. func GetTagValueCountMap(
  887. manager IModelManager,
  888. tagObjType string,
  889. tagIdField string,
  890. sumField string,
  891. keys []string,
  892. ctx context.Context,
  893. userCred mcclient.TokenCredential,
  894. query jsonutils.JSONObject,
  895. ) ([]map[string]string, error) {
  896. var err error
  897. objQ := manager.NewQuery(ctx, userCred, query, false)
  898. objQ, err = ListItemQueryFilters(manager, ctx, objQ, userCred, query, policy.PolicyActionList)
  899. if err != nil {
  900. return nil, errors.Wrap(err, "ListItemQueryFilters")
  901. }
  902. objSubQ := objQ.SubQuery().Query()
  903. objSubQ = objSubQ.AppendField(objSubQ.Field(tagIdField))
  904. objSubQ = objSubQ.GroupBy(objSubQ.Field(tagIdField))
  905. var sumFieldQ sqlchemy.IQueryField
  906. if len(sumField) > 0 {
  907. sumFieldQ = sqlchemy.SUM("_sub_count_", objSubQ.Field(sumField))
  908. } else {
  909. sumFieldQ = sqlchemy.COUNT("_sub_count_")
  910. }
  911. objSubQ = objSubQ.AppendField(sumFieldQ)
  912. // objSubQ.DebugQuery2("GetTagValueCountMap objSubQ")
  913. q := objSubQ.SubQuery().Query()
  914. q = q.AppendField(sqlchemy.SUM(tagValueCountKey, q.Field("_sub_count_")))
  915. metadataMan := GetMetadaManagerInContext(ctx)
  916. metadataSQ := metadataMan.Query().Equals("obj_type", tagObjType).In("key", keys).SubQuery()
  917. groupBy := make([]interface{}, 0)
  918. for i, key := range keys {
  919. valueFieldName := TagValueKey(i)
  920. subq := metadataSQ.Query().Equals("key", key).SubQuery()
  921. q = q.LeftJoin(subq, sqlchemy.Equals(q.Field(tagIdField), subq.Field("obj_id")))
  922. q = q.AppendField(
  923. sqlchemy.NewFunction(
  924. sqlchemy.NewCase().When(sqlchemy.IsNull(subq.Field("value")), sqlchemy.NewStringField(tagutils.NoValue)).Else(subq.Field("value")),
  925. valueFieldName,
  926. false,
  927. ),
  928. )
  929. groupBy = append(groupBy, q.Field(valueFieldName))
  930. }
  931. q = q.GroupBy(groupBy...)
  932. // q.DebugQuery2("GetTagValueCountMap")
  933. valueMap, err := q.AllStringMap()
  934. if err != nil {
  935. return nil, errors.Wrap(err, "AllStringAmp")
  936. }
  937. return valueMap, nil
  938. }
  939. func (manager *SStandaloneAnonResourceBaseManager) HistoryDataClean(ctx context.Context, timeBefor time.Time) (int, error) {
  940. q := manager.RawQuery("id").IsTrue("deleted").LE("deleted_at", timeBefor)
  941. rows, err := q.Rows()
  942. if err != nil {
  943. if errors.Cause(err) == sql.ErrNoRows {
  944. return 0, nil
  945. }
  946. return 0, errors.Wrap(err, "Query")
  947. }
  948. defer rows.Close()
  949. ids := []string{}
  950. for rows.Next() {
  951. var id string
  952. err := rows.Scan(&id)
  953. if err != nil {
  954. return 0, errors.Wrap(err, "rows.Scan")
  955. }
  956. ids = append(ids, id)
  957. }
  958. var purge = func(ids []string) error {
  959. vars := []interface{}{}
  960. placeholders := make([]string, len(ids))
  961. for i := range placeholders {
  962. placeholders[i] = "?"
  963. vars = append(vars, ids[i])
  964. }
  965. placeholder := strings.Join(placeholders, ",")
  966. sql := fmt.Sprintf(
  967. "delete from %s where id in (%s)",
  968. manager.TableSpec().Name(), placeholder,
  969. )
  970. lockman.LockRawObject(ctx, manager.Keyword(), "purge")
  971. defer lockman.ReleaseRawObject(ctx, manager.Keyword(), "purge")
  972. _, err = sqlchemy.GetDB().Exec(
  973. sql, vars...,
  974. )
  975. if err != nil {
  976. return errors.Wrapf(err, strings.ReplaceAll(sql, "?", "%s"), vars...)
  977. }
  978. return nil
  979. }
  980. var splitByLen = func(data []string, splitLen int) [][]string {
  981. var result [][]string
  982. for i := 0; i < len(data); i += splitLen {
  983. end := i + splitLen
  984. if end > len(data) {
  985. end = len(data)
  986. }
  987. result = append(result, data[i:end])
  988. }
  989. return result
  990. }
  991. idsArr := splitByLen(ids, 100)
  992. for i := range idsArr {
  993. err = purge(idsArr[i])
  994. if err != nil {
  995. return 0, err
  996. }
  997. }
  998. return len(ids), nil
  999. }