From e6c37f497291bd9e754c5ff11118643c327b8f71 Mon Sep 17 00:00:00 2001
From: Abin Simon <abin.simon@rafay.co>
Date: Thu, 17 Mar 2022 15:10:27 +0530
Subject: [PATCH] Fix authz issues from implementing transactions

---
 pkg/service/authz.go | 8 ++++----
 1 file changed, 4 insertions(+), 4 deletions(-)

diff --git a/pkg/service/authz.go b/pkg/service/authz.go
index 97af183..4ea6f28 100644
--- a/pkg/service/authz.go
+++ b/pkg/service/authz.go
@@ -27,15 +27,15 @@ type AuthzService interface {
 }
 
 type authzService struct {
-	dao          pg.EntityDAO
+	db           *bun.DB
 	enforcer     *casbin.CachedEnforcer
 	mappingCache map[string][]rpmUrlAction
 }
 
 func NewAuthzService(db *bun.DB, en *casbin.CachedEnforcer) AuthzService {
 	return &authzService{
-		dao:      pg.NewEntityDAO(db),
-		enforcer: en,
+		db:           db,
+		enforcer:     en,
 		mappingCache: make(map[string][]rpmUrlAction),
 	}
 }
@@ -91,7 +91,7 @@ func processRpms(rpm models.ResourcePermission) []rpmUrlAction {
 
 func (s *authzService) cacheResourceRolePermissions(ctx context.Context) error {
 	var items []models.ResourcePermission
-	entities, err := s.dao.ListAll(ctx, &items)
+	entities, err := pg.ListAll(ctx, s.db, &items)
 	if err != nil {
 		return err
 	}