45 lines
1.1 KiB
Go
Executable File
45 lines
1.1 KiB
Go
Executable File
package dao
|
|
|
|
import (
|
|
"atom/database/models"
|
|
"atom/database/query"
|
|
"context"
|
|
)
|
|
|
|
type UserRoleDao struct {
|
|
query *query.Query
|
|
}
|
|
|
|
func NewUserRoleDao(query *query.Query) *UserRoleDao {
|
|
return &UserRoleDao{query: query}
|
|
}
|
|
|
|
func (dao *UserRoleDao) 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 *UserRoleDao) 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 *UserRoleDao) 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 *UserRoleDao) 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
|
|
}
|