models.go 1.8 KB

123456789101112131415161718192021222324252627282930313233343536373839404142
  1. package model
  2. import (
  3. "perms-system-server/internal/model/dept"
  4. "perms-system-server/internal/model/perm"
  5. "perms-system-server/internal/model/product"
  6. "perms-system-server/internal/model/productmember"
  7. "perms-system-server/internal/model/role"
  8. "perms-system-server/internal/model/roleperm"
  9. "perms-system-server/internal/model/user"
  10. "perms-system-server/internal/model/userperm"
  11. "perms-system-server/internal/model/userrole"
  12. "github.com/zeromicro/go-zero/core/stores/cache"
  13. "github.com/zeromicro/go-zero/core/stores/sqlx"
  14. )
  15. type Models struct {
  16. SysProductModel product.SysProductModel
  17. SysPermModel perm.SysPermModel
  18. SysRoleModel role.SysRoleModel
  19. SysRolePermModel roleperm.SysRolePermModel
  20. SysDeptModel dept.SysDeptModel
  21. SysUserModel user.SysUserModel
  22. SysUserRoleModel userrole.SysUserRoleModel
  23. SysUserPermModel userperm.SysUserPermModel
  24. SysProductMemberModel productmember.SysProductMemberModel
  25. }
  26. func NewModels(conn sqlx.SqlConn, cacheConf cache.CacheConf, cachePrefix string) *Models {
  27. return &Models{
  28. SysProductModel: product.NewSysProductModel(conn, cacheConf, cachePrefix),
  29. SysPermModel: perm.NewSysPermModel(conn, cacheConf, cachePrefix),
  30. SysRoleModel: role.NewSysRoleModel(conn, cacheConf, cachePrefix),
  31. SysRolePermModel: roleperm.NewSysRolePermModel(conn, cacheConf, cachePrefix),
  32. SysDeptModel: dept.NewSysDeptModel(conn, cacheConf, cachePrefix),
  33. SysUserModel: user.NewSysUserModel(conn, cacheConf, cachePrefix),
  34. SysUserRoleModel: userrole.NewSysUserRoleModel(conn, cacheConf, cachePrefix),
  35. SysUserPermModel: userperm.NewSysUserPermModel(conn, cacheConf, cachePrefix),
  36. SysProductMemberModel: productmember.NewSysProductMemberModel(conn, cacheConf, cachePrefix),
  37. }
  38. }