user_roles.go 563 B

123456789101112131415161718192021222324252627
  1. package models
  2. import (
  3. "strconv"
  4. )
  5. type User_role struct {
  6. ID uint `gorm:"primary_key" json:"id"`
  7. UserId string `json:"user_id"`
  8. RoleId uint `json:"role_id"`
  9. }
  10. func FindRoleByUserId(userId interface{}) User_role {
  11. var uRole User_role
  12. DB.Where("user_id = ?", userId).First(&uRole)
  13. return uRole
  14. }
  15. func CreateUserRole(userId uint, roleId uint) {
  16. uRole := &User_role{
  17. UserId: strconv.Itoa(int(userId)),
  18. RoleId: roleId,
  19. }
  20. DB.Create(uRole)
  21. }
  22. func DeleteRoleByUserId(userId interface{}) {
  23. DB.Where("user_id = ?", userId).Delete(User_role{})
  24. }