diff --git a/controller/nucleus/databaseNetworkElementStore.go b/controller/nucleus/databaseNetworkElementStore.go index a5bc5cb0746246b4b53ea472f20ea635a2577110..77e7e5975dc3d7d08809aa01e5cdd09216fb6021 100644 --- a/controller/nucleus/databaseNetworkElementStore.go +++ b/controller/nucleus/databaseNetworkElementStore.go @@ -6,7 +6,6 @@ import ( "code.fbi.h-da.de/danet/gosdn/controller/customerrs" "code.fbi.h-da.de/danet/gosdn/controller/interfaces/networkelement" - "code.fbi.h-da.de/danet/gosdn/controller/nucleus/database" "code.fbi.h-da.de/danet/gosdn/controller/store" "go.mongodb.org/mongo-driver/bson" "go.mongodb.org/mongo-driver/bson/primitive" @@ -122,15 +121,10 @@ func (s *DatabaseNetworkElementStore) Add(ctx context.Context, networkElementToA func (s *DatabaseNetworkElementStore) Update(ctx context.Context, networkElementToUpdate networkelement.NetworkElement) (err error) { var updatedLoadedNetworkElement networkelement.LoadedNetworkElement - db, err := database.GetDatabaseConnection() - if err != nil { - return err - } - wc := writeconcern.Majority() txnOptions := options.Transaction().SetWriteConcern(wc) // Starts a session on the client - session, err := db.Client().StartSession() + session, err := s.collection.Database().Client().StartSession() if err != nil { return err } diff --git a/controller/rbac/databaseUserStore.go b/controller/rbac/databaseUserStore.go index cfa592de9761195c56f1c7e8185160d286213acc..c3b4400e9d576e4f6034d595b6385fc82805d2ad 100644 --- a/controller/rbac/databaseUserStore.go +++ b/controller/rbac/databaseUserStore.go @@ -6,7 +6,6 @@ import ( "code.fbi.h-da.de/danet/gosdn/controller/customerrs" "code.fbi.h-da.de/danet/gosdn/controller/interfaces/rbac" - "code.fbi.h-da.de/danet/gosdn/controller/nucleus/database" "code.fbi.h-da.de/danet/gosdn/controller/store" "github.com/google/uuid" log "github.com/sirupsen/logrus" @@ -134,15 +133,10 @@ func (s *DatabaseUserStore) GetAll(ctx context.Context) (loadedUsers []rbac.Load func (s *DatabaseUserStore) Update(ctx context.Context, userToUpdate rbac.User) (err error) { var updatedLoadedUser rbac.LoadedUser - db, err := database.GetDatabaseConnection() - if err != nil { - return err - } - wc := writeconcern.Majority() txnOptions := options.Transaction().SetWriteConcern(wc) // Starts a session on the client - session, err := db.Client().StartSession() + session, err := s.collection.Database().Client().StartSession() if err != nil { return err } diff --git a/controller/topology/nodes/databaseNodeStore.go b/controller/topology/nodes/databaseNodeStore.go index b0389082c7dcd44055f3c160c834ddea9bb49a4c..9c575cb6f74b71806ff13a907384230899f05ab3 100644 --- a/controller/topology/nodes/databaseNodeStore.go +++ b/controller/topology/nodes/databaseNodeStore.go @@ -6,7 +6,6 @@ import ( "time" "code.fbi.h-da.de/danet/gosdn/controller/customerrs" - "code.fbi.h-da.de/danet/gosdn/controller/nucleus/database" query "code.fbi.h-da.de/danet/gosdn/controller/store" "github.com/google/uuid" @@ -133,15 +132,10 @@ func (s *DatabaseNodeStore) Add(ctx context.Context, node Node) (err error) { func (s *DatabaseNodeStore) Update(ctx context.Context, node Node) (err error) { var updatedLoadedNodes Node - db, err := database.GetDatabaseConnection() - if err != nil { - return err - } - wc := writeconcern.Majority() txnOptions := options.Transaction().SetWriteConcern(wc) // Starts a session on the client - session, err := db.Client().StartSession() + session, err := s.collection.Database().Client().StartSession() if err != nil { return err }