test
This commit is contained in:
@@ -1,10 +1,12 @@
|
||||
package resources
|
||||
|
||||
import (
|
||||
"encoding/json"
|
||||
"errors"
|
||||
"fmt"
|
||||
"slices"
|
||||
|
||||
"cloud.o-forge.io/core/oc-lib/config"
|
||||
"cloud.o-forge.io/core/oc-lib/dbs"
|
||||
"cloud.o-forge.io/core/oc-lib/logs"
|
||||
"cloud.o-forge.io/core/oc-lib/models/common/models"
|
||||
@@ -110,6 +112,40 @@ func (dca *ResourceMongoAccessor[T]) LoadOne(id string) (utils.DBObject, int, er
|
||||
return data, code, err
|
||||
}
|
||||
|
||||
var workspaceResourceTypes = []tools.DataType{
|
||||
tools.COMPUTE_RESOURCE,
|
||||
tools.DATA_RESOURCE,
|
||||
tools.PROCESSING_RESOURCE,
|
||||
tools.STORAGE_RESOURCE,
|
||||
tools.WORKFLOW_RESOURCE,
|
||||
tools.SERVICE_RESOURCE,
|
||||
}
|
||||
|
||||
func emitResourceNATS(method tools.NATSMethod, dt tools.DataType, payload []byte) {
|
||||
if !slices.Contains(workspaceResourceTypes, dt) {
|
||||
return
|
||||
}
|
||||
tools.NewNATSCaller().SetNATSPub(method, tools.NATSResponse{
|
||||
FromApp: config.GetAppName(),
|
||||
Datatype: dt,
|
||||
Method: int(method),
|
||||
Payload: payload,
|
||||
})
|
||||
}
|
||||
|
||||
func (dca *ResourceMongoAccessor[T]) DeleteOne(id string) (utils.DBObject, int, error) {
|
||||
data, code, err := dca.AbstractAccessor.LoadOne(id)
|
||||
if err != nil {
|
||||
return data, code, err
|
||||
}
|
||||
res, code, err := dca.AbstractAccessor.DeleteOne(id)
|
||||
if err == nil && data != nil {
|
||||
b, _ := json.Marshal(data)
|
||||
go emitResourceNATS(tools.REMOVE_RESOURCE, dca.GetType(), b)
|
||||
}
|
||||
return res, code, err
|
||||
}
|
||||
|
||||
func (dca *ResourceMongoAccessor[T]) UpdateOne(set map[string]interface{}, id string) (utils.DBObject, int, error) {
|
||||
if dca.GetType() == tools.COMPUTE_RESOURCE {
|
||||
delete(set, "architecture")
|
||||
@@ -193,9 +229,119 @@ func (dca *ResourceMongoAccessor[T]) StoreOne(data utils.DBObject) (utils.DBObje
|
||||
"resources_id": idsToUpdate,
|
||||
}, i)
|
||||
}
|
||||
if err == nil && res != nil {
|
||||
b, _ := json.Marshal(res)
|
||||
go emitResourceNATS(tools.CREATE_RESOURCE, dca.GetType(), b)
|
||||
}
|
||||
return res, code, err
|
||||
}
|
||||
|
||||
// PurgedResourcePayload holds a silently-deleted resource's type and serialized payload.
|
||||
type PurgedResourcePayload struct {
|
||||
DT tools.DataType
|
||||
Payload []byte
|
||||
}
|
||||
|
||||
// purgeByType searches and silently deletes all resources of type T created by creatorID.
|
||||
// Uses AbstractAccessor.DeleteOne directly to bypass the NATS-emitting override.
|
||||
func purgeByType[T ResourceInterface](dt tools.DataType, creatorID string) []PurgedResourcePayload {
|
||||
a := NewAccessor[T](dt, nil)
|
||||
if a == nil {
|
||||
return nil
|
||||
}
|
||||
res, _, _ := a.Search(&dbs.Filters{
|
||||
And: map[string][]dbs.Filter{
|
||||
"creator_id": {{Operator: dbs.EQUAL.String(), Value: creatorID}},
|
||||
},
|
||||
}, "", false, 0, 10000)
|
||||
var result []PurgedResourcePayload
|
||||
for _, item := range res {
|
||||
b, err := json.Marshal(item)
|
||||
if err != nil {
|
||||
continue
|
||||
}
|
||||
a.AbstractAccessor.DeleteOne(item.GetID())
|
||||
result = append(result, PurgedResourcePayload{DT: dt, Payload: b})
|
||||
}
|
||||
return result
|
||||
}
|
||||
|
||||
// PurgeCreatorResources deletes all catalog resources created by creatorPeerID from
|
||||
// the DB without emitting NATS. Used for non-blacklist peer privilege downgrades where
|
||||
// workspace state should be left untouched.
|
||||
func PurgeCreatorResources(creatorPeerID string) []PurgedResourcePayload {
|
||||
var result []PurgedResourcePayload
|
||||
result = append(result, purgeByType[*ComputeResource](tools.COMPUTE_RESOURCE, creatorPeerID)...)
|
||||
result = append(result, purgeByType[*DataResource](tools.DATA_RESOURCE, creatorPeerID)...)
|
||||
result = append(result, purgeByType[*ProcessingResource](tools.PROCESSING_RESOURCE, creatorPeerID)...)
|
||||
result = append(result, purgeByType[*StorageResource](tools.STORAGE_RESOURCE, creatorPeerID)...)
|
||||
result = append(result, purgeByType[*WorkflowResource](tools.WORKFLOW_RESOURCE, creatorPeerID)...)
|
||||
result = append(result, purgeByType[*ServiceResource](tools.SERVICE_RESOURCE, creatorPeerID)...)
|
||||
return result
|
||||
}
|
||||
|
||||
// FilterMapFromResourcePayload deserializes a resource payload by DataType, zeros out
|
||||
// the AbstractInstanciatedResource (and its AbstractResource / Instances sub-fields),
|
||||
// then marshals back to get only the concrete type's own JSON fields.
|
||||
// Returns nil for WORKFLOW_RESOURCE and unknown types.
|
||||
// JSON keys only — not BSON paths.
|
||||
func FilterMapFromResourcePayload(dt tools.DataType, payload []byte) map[string]interface{} {
|
||||
var m map[string]interface{}
|
||||
switch dt {
|
||||
case tools.COMPUTE_RESOURCE:
|
||||
var r ComputeResource
|
||||
if json.Unmarshal(payload, &r) != nil {
|
||||
return nil
|
||||
}
|
||||
r.AbstractInstanciatedResource = AbstractInstanciatedResource[*ComputeResourceInstance]{}
|
||||
b, _ := json.Marshal(r)
|
||||
json.Unmarshal(b, &m)
|
||||
case tools.DATA_RESOURCE:
|
||||
var r DataResource
|
||||
if json.Unmarshal(payload, &r) != nil {
|
||||
return nil
|
||||
}
|
||||
r.AbstractInstanciatedResource = AbstractInstanciatedResource[*DataInstance]{}
|
||||
b, _ := json.Marshal(r)
|
||||
json.Unmarshal(b, &m)
|
||||
case tools.PROCESSING_RESOURCE:
|
||||
var r ProcessingResource
|
||||
if json.Unmarshal(payload, &r) != nil {
|
||||
return nil
|
||||
}
|
||||
r.AbstractInstanciatedResource = AbstractInstanciatedResource[*ProcessingInstance]{}
|
||||
b, _ := json.Marshal(r)
|
||||
json.Unmarshal(b, &m)
|
||||
case tools.STORAGE_RESOURCE:
|
||||
var r StorageResource
|
||||
if json.Unmarshal(payload, &r) != nil {
|
||||
return nil
|
||||
}
|
||||
r.AbstractInstanciatedResource = AbstractInstanciatedResource[*StorageResourceInstance]{}
|
||||
b, _ := json.Marshal(r)
|
||||
json.Unmarshal(b, &m)
|
||||
case tools.SERVICE_RESOURCE:
|
||||
var r ServiceResource
|
||||
if json.Unmarshal(payload, &r) != nil {
|
||||
return nil
|
||||
}
|
||||
r.AbstractInstanciatedResource = AbstractInstanciatedResource[*ServiceInstance]{}
|
||||
b, _ := json.Marshal(r)
|
||||
json.Unmarshal(b, &m)
|
||||
case tools.WORKFLOW_RESOURCE:
|
||||
var r WorkflowResource
|
||||
if json.Unmarshal(payload, &r) != nil {
|
||||
return nil
|
||||
}
|
||||
r.AbstractResource = AbstractResource{}
|
||||
b, _ := json.Marshal(r)
|
||||
json.Unmarshal(b, &m)
|
||||
default:
|
||||
return nil
|
||||
}
|
||||
return m
|
||||
}
|
||||
|
||||
func (dca *ResourceMongoAccessor[T]) CopyOne(data utils.DBObject) (utils.DBObject, int, error) {
|
||||
return dca.StoreOne(data)
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user