package models import ( "errors" "cloud.o-forge.io/core/deprecated-oc-catalog/models/rtype" "cloud.o-forge.io/core/deprecated-oc-catalog/services" "github.com/beego/beego/v2/core/logs" "go.mongodb.org/mongo-driver/bson" "go.mongodb.org/mongo-driver/bson/primitive" "go.mongodb.org/mongo-driver/mongo/options" ) func getObjIDFromString(id string) interface{} { objectID, err := primitive.ObjectIDFromHex(id) if err == nil { return objectID } return id } func getMultipleObjIDFromArray(ids []string) []interface{} { var ret []interface{} for _, val := range ids { ret = append(ret, getObjIDFromString(val)) } return ret } func getOneResourceByID(ID string, rType rtype.Rtype) (obj interface{}, err error) { targetDBCollection := rType.MongoCollection() var retObj interface{} // asd := rType. switch rType { case rtype.DATA: retObj = &DataModel{} case rtype.COMPUTING: retObj = &ComputingModel{} case rtype.STORAGE: retObj = &StorageModel{} case rtype.DATACENTER: retObj = &DatacenterModel{} default: message := "Rtype " + rType.String() + " is not implemented" logs.Error(message) return nil, errors.New(message) } filter := bson.M{"_id": getObjIDFromString(ID)} res := targetDBCollection.FindOne(services.MngoCtx, filter) res.Decode(retObj) if res.Err() != nil { logs.Warn("Couldn't find resource: " + res.Err().Error()) } return retObj, res.Err() } func getMultipleResourceByIDs(IDs []string, rType rtype.Rtype) (interface{}, error) { targetDBCollection := rType.MongoCollection() var retObj interface{} // asd := rType. switch rType { case rtype.DATA: retObj = &[]DataModel{} case rtype.COMPUTING: retObj = &[]ComputingModel{} case rtype.STORAGE: retObj = &[]StorageModel{} case rtype.DATACENTER: retObj = &[]DatacenterModel{} default: message := "Rtype " + rType.String() + " is not implemented" logs.Error(message) return nil, errors.New(message) } filter := bson.M{"_id": bson.M{"$in": getMultipleObjIDFromArray(IDs)}} //FIXME: Limit of find res, err := targetDBCollection.Find(services.MngoCtx, filter, options.Find().SetLimit(100), ) if err != nil { logs.Warn("Couldn't find multiple data: " + err.Error()) return nil, err } res.All(services.MngoCtx, retObj) return retObj, res.Err() } func postOneResource(retObj interface{}, rType rtype.Rtype) (ID string, err error) { targetDBCollection := rType.MongoCollection() result, err := targetDBCollection.InsertOne(services.MngoCtx, retObj) if err != nil { logs.Warn("Couldn't insert resource: " + err.Error()) return "", err } return result.InsertedID.(primitive.ObjectID).Hex(), nil }