added db name

This commit is contained in:
2026-03-12 17:41:39 +08:00
parent 52fa1f4818
commit bb6ec52732
2 changed files with 8 additions and 8 deletions
+7 -7
View File
@@ -14,8 +14,8 @@ func GetPermissionByResourceActionAndRole(resource, action string, roleID int) (
query := `
SELECT p.permissions_id, p.permission_name, p.description, p.resource, p.action
FROM permissions p
INNER JOIN role_permissions rp
FROM uess_user_management.permissions p
INNER JOIN uess_user_management.role_permissions rp
ON p.permissions_id = rp.permission_id
WHERE p.resource = ? AND p.action = ? AND rp.role_id = ? AND rp.is_deleted = 0
LIMIT 1
@@ -48,7 +48,7 @@ func GetPermissionByResourceActionAndRole(resource, action string, roleID int) (
func GetPolicyAttributesByPermission(permissionID int) ([]models.PolicyAttribute, error) {
query := `
SELECT id, attribute_name, attribute_type, comparison, attribute_value, permission_id
FROM policy_attributes
FROM uess_user_management.policy_attributes
WHERE permission_id = ?
`
@@ -84,7 +84,7 @@ func GetUserAttributes(userID string) (map[string]string, error) {
query := `
SELECT attribute_name, attribute_value
FROM user_attributes
FROM uess_user_management.user_attributes
WHERE users_id = ?
`
@@ -116,7 +116,7 @@ func GetUserByID(userID string) (*models.User, error) {
query := `
SELECT users_id, email_address
FROM users
FROM uess_user_management.users
WHERE users_id = ?
`
@@ -139,7 +139,7 @@ func GetUserByID(userID string) (*models.User, error) {
func GetAllPermissions() ([]models.Permission, error) {
query := `
SELECT id, permission_name, description, resource, action
FROM permissions
FROM uess_user_management.permissions
ORDER BY id
`
@@ -172,7 +172,7 @@ func GetAllPermissions() ([]models.Permission, error) {
func GetAllPolicyAttributes() (map[int][]models.PolicyAttribute, error) {
query := `
SELECT policy_attributes_id, attribute_name, attribute_type, comparison, attribute_value, permission_id
FROM policy_attributes
FROM uess_user_management.policy_attributes
ORDER BY permission_id, policy_attributes_id
`
+1 -1
View File
@@ -35,7 +35,7 @@ func TestGetPolicyAttributesByPermissionSuccess(t *testing.T) {
AddRow(1, "department", "user", "=", "engineering", 1).
AddRow(2, "level", "user", ">=", "5", 1)
mock.ExpectQuery("SELECT id, attribute_name, attribute_type, comparison, attribute_value, permission_id FROM policy_attributes WHERE permission_id = \\?").
mock.ExpectQuery("SELECT id, attribute_name, attribute_type, comparison, attribute_value, permission_id FROM uess_user_management.policy_attributes WHERE permission_id = \\?").
WithArgs(1).
WillReturnRows(rows)