role.go 993 B

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950
  1. package model
  2. import (
  3. "sparrow/pkg/models"
  4. "github.com/jinzhu/gorm"
  5. )
  6. // Role role
  7. type Role struct {
  8. db *gorm.DB
  9. }
  10. // Init init
  11. func (a *Role) Init(db *gorm.DB) *Role {
  12. a.db = db
  13. return a
  14. }
  15. // Create create
  16. func (a *Role) Create(role *models.Role) error {
  17. return a.db.Save(role).Error
  18. }
  19. // Update update
  20. func (a *Role) Update(role *models.Role) error {
  21. return a.db.Save(role).Error
  22. }
  23. // Delete delete
  24. func (a *Role) Delete(role *models.Role) error {
  25. return a.db.Delete(role).Error
  26. }
  27. // QueryAll query all roles
  28. func (a *Role) QueryAll(pi, ps int, name string) (datas []models.Role, total int, err error) {
  29. tx := a.db.Where("1=1")
  30. if name != "" {
  31. tx = tx.Where("role_name like ?", "%"+name+"%")
  32. }
  33. err = tx.Limit(ps).Offset((pi - 1) * ps).Find(&datas).Error
  34. err = tx.Model(&models.Role{}).Count(&total).Error
  35. return
  36. }
  37. // QueryRole query a role
  38. func (a *Role) QueryRole(roleid int) (data models.Role, err error) {
  39. err = a.db.Where(roleid).First(&data).Error
  40. return
  41. }