support user role attachment

modules/auth/dao/user_role.go

@@ -0,0 +1,51 @@
+package dao
+
+import (
+	"atom/database/models"
+	"atom/database/query"
+	"context"
+)
+
+type UserRoleDao interface {
+	Exists(context.Context, int) bool
+	Create(context.Context, int, int) error
+	Update(context.Context, int, int) error
+	Delete(context.Context, int, int) error
+}
+
+type userRoleDaoImpl struct {
+	query *query.Query
+}
+
+func NewUserRoleDao(query *query.Query) UserRoleDao {
+	return &userRoleDaoImpl{query: query}
+}
+
+func (dao *userRoleDaoImpl) Exists(ctx context.Context, userID int) bool {
+	userRole := dao.query.UserRole
+	count, _ := userRole.WithContext(ctx).Where(userRole.UserID.Eq(uint64(userID))).Count()
+	return count > 0
+}
+
+func (dao *userRoleDaoImpl) Create(ctx context.Context, userID, roleID int) error {
+	userRole := dao.query.UserRole
+	return userRole.WithContext(ctx).Create(&models.UserRole{
+		UserID: uint64(userID),
+		RoleID: uint64(roleID),
+	})
+}
+
+func (dao *userRoleDaoImpl) Update(ctx context.Context, userID, roleID int) error {
+	userRole := dao.query.UserRole
+	_, err := userRole.WithContext(ctx).Where(userRole.UserID.Eq(uint64(userID))).Update(userRole.RoleID, roleID)
+	return err
+}
+
+func (dao *userRoleDaoImpl) Delete(ctx context.Context, userID, roleID int) error {
+	userRole := dao.query.UserRole
+	_, err := userRole.WithContext(ctx).
+		Where(userRole.UserID.Eq(uint64(userID))).
+		Where(userRole.RoleID.Eq(uint64(roleID))).
+		Delete()
+	return err
+}