kenneth/projectx
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

gorm wire

Browse Source
This commit is contained in:
kenneth
2025-05-07 14:12:53 +08:00
parent 461531c308
commit 68606c76f9
111 changed files with 1726 additions and 5809 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
internal/erpserver/repository/system/audit_log.go
View File

@@ -9,23 +9,21 @@ import (
)
type auditLogRepository struct {
store repository.Store
repo *repository.Repository
}
var _ system.AuditLogRepository = (*auditLogRepository)(nil)
func NewAuditLogRepository(store repository.Store) *auditLogRepository {
func NewAuditLogRepository(repo *repository.Repository) system.AuditLogRepository {
return &auditLogRepository{
store: store,
repo: repo,
}
}
func (s *auditLogRepository) Create(ctx context.Context, obj *system.AuditLog) error {
return s.store.DB(ctx).Create(obj).Error
return s.repo.DB(ctx).Create(obj).Error
}
func (s *auditLogRepository) List(ctx context.Context, q dto.SearchDto) ([]*system.AuditLog, int64, error) {
query := s.store.DB(ctx).
query := s.repo.DB(ctx).
Model(&system.AuditLog{}).
Where("created_at BETWEEN ? AND ?", q.SearchTimeBegin, q.SearchTimeEnd)
if q.SearchEmail != "" {

38
internal/erpserver/repository/system/config.go
View File

@@ -9,28 +9,26 @@ import (
)
type configRepository struct {
store repository.Store
repo *repository.Repository
}
var _ system.ConfigRepository = (*configRepository)(nil)
func NewConfigRepository(store repository.Store) *configRepository {
func NewConfigRepository(repo *repository.Repository) system.ConfigRepository {
return &configRepository{
store: store,
repo: repo,
}
}
func (r *configRepository) Create(ctx context.Context, obj *system.Config) error {
return r.store.DB(ctx).Create(obj).Error
return r.repo.DB(ctx).Create(obj).Error
}
func (r *configRepository) Update(ctx context.Context, obj *system.Config) error {
return r.store.DB(ctx).Save(obj).Error
return r.repo.DB(ctx).Save(obj).Error
}
func (r *configRepository) Get(ctx context.Context, id int32) (*system.Config, error) {
var obj system.Config
err := r.store.DB(ctx).First(&obj, id).Error
err := r.repo.DB(ctx).First(&obj, id).Error
if err != nil {
return nil, err
}
@@ -39,7 +37,7 @@ func (r *configRepository) Get(ctx context.Context, id int32) (*system.Config, e
func (r *configRepository) GetByKey(ctx context.Context, key string) (*system.Config, error) {
var obj system.Config
err := r.store.DB(ctx).Where("key = ?", key).First(&obj).Error
err := r.repo.DB(ctx).Where("key = ?", key).First(&obj).Error
if err != nil {
return nil, err
}
@@ -47,5 +45,25 @@ func (r *configRepository) GetByKey(ctx context.Context, key string) (*system.Co
}
func (r *configRepository) List(ctx context.Context, q dto.SearchDto) ([]*system.Config, int64, error) {
return nil, 0, nil
query := r.repo.DB(ctx).
Model(&system.Config{}).
Where("created_at BETWEEN ? AND ?", q.SearchTimeBegin, q.SearchTimeEnd)
var count int64
err := query.Count(&count).Error
if err != nil {
return nil, 0, err
}
var configs []*system.Config
err = query.
Order("id DESC").
Offset((q.Page - 1) * q.Rows).
Limit(q.Rows).
Find(&configs).
Error
if err != nil {
return nil, 0, err
}
return configs, count, nil
}

20
internal/erpserver/repository/system/department.go
View File

@@ -9,28 +9,26 @@ import (
)
type departmentRepository struct {
store repository.Store
repo *repository.Repository
}
var _ system.DepartmentRepository = (*departmentRepository)(nil)
func NewDepartmentRepository(store repository.Store) *departmentRepository {
func NewDepartmentRepository(repo *repository.Repository) system.DepartmentRepository {
return &departmentRepository{
store: store,
repo: repo,
}
}
func (r *departmentRepository) Create(ctx context.Context, obj *system.Department) error {
return r.store.DB(ctx).Create(obj).Error
return r.repo.DB(ctx).Create(obj).Error
}
func (r *departmentRepository) Update(ctx context.Context, obj *system.Department) error {
return r.store.DB(ctx).Save(obj).Error
return r.repo.DB(ctx).Save(obj).Error
}
func (r *departmentRepository) Get(ctx context.Context, id int32) (*system.Department, error) {
var obj system.Department
err := r.store.DB(ctx).First(&obj, id).Error
err := r.repo.DB(ctx).First(&obj, id).Error
if err != nil {
return nil, err
}
@@ -39,7 +37,7 @@ func (r *departmentRepository) Get(ctx context.Context, id int32) (*system.Depar
func (r *departmentRepository) All(ctx context.Context) ([]*system.Department, error) {
var departs []*system.Department
err := r.store.DB(ctx).Find(&departs).Error
err := r.repo.DB(ctx).Find(&departs).Error
if err != nil {
return nil, err
}
@@ -47,7 +45,7 @@ func (r *departmentRepository) All(ctx context.Context) ([]*system.Department, e
}
func (r *departmentRepository) List(ctx context.Context, q dto.SearchDto) ([]*system.Department, int64, error) {
query := r.store.DB(ctx).
query := r.repo.DB(ctx).
Model(&system.Department{}).
Where("created_at BETWEEN ? AND ?", q.SearchTimeBegin, q.SearchTimeEnd)
if q.SearchID != 0 {
@@ -96,5 +94,5 @@ SET parent_path = (SELECT ',' || string_agg(cast(t.parent_id AS VARCHAR), ',') |
FROM temp
ORDER BY id) AS t)
WHERE tm.status = 0;`
return r.store.DB(ctx).Exec(query).Error
return r.repo.DB(ctx).Exec(query).Error
}

16
internal/erpserver/repository/system/login_log.go
View File

@@ -9,24 +9,22 @@ import (
)
type loginLogRepository struct {
store repository.Store
repo *repository.Repository
}
var _ system.LoginLogRepository = (*loginLogRepository)(nil)
func NewLoginLogRepository(store repository.Store) *loginLogRepository {
func NewLoginLogRepository(repo *repository.Repository) system.LoginLogRepository {
return &loginLogRepository{
store: store,
repo: repo,
}
}
func (s *loginLogRepository) Create(ctx context.Context, obj *system.LoginLog) error {
return s.store.DB(ctx).Create(obj).Error
return s.repo.DB(ctx).Create(obj).Error
}
func (s *loginLogRepository) GetLatest(ctx context.Context, email string) (*system.LoginLog, error) {
var log system.LoginLog
err := s.store.DB(ctx).
err := s.repo.DB(ctx).
Where("email = ?", email).
Order("id DESC").
First(&log).
@@ -38,7 +36,7 @@ func (s *loginLogRepository) GetLatest(ctx context.Context, email string) (*syst
}
func (s *loginLogRepository) List(ctx context.Context, q dto.SearchDto) ([]*system.LoginLog, int64, error) {
query := s.store.DB(ctx).
query := s.repo.DB(ctx).
Model(&system.LoginLog{}).
Where("created_at BETWEEN ? AND ?", q.SearchTimeBegin, q.SearchTimeEnd)
if q.SearchEmail != "" {
@@ -66,7 +64,7 @@ func (s *loginLogRepository) List(ctx context.Context, q dto.SearchDto) ([]*syst
func (s *loginLogRepository) Count(ctx context.Context, email string) (int64, error) {
var count int64
err := s.store.DB(ctx).
err := s.repo.DB(ctx).
Model(&system.LoginLog{}).
Where("email = ?", email).
Count(&count).

20
internal/erpserver/repository/system/menu.go
View File

@@ -8,28 +8,26 @@ import (
)
type menuRepository struct {
store repository.Store
repo *repository.Repository
}
var _ system.MenuRepository = (*menuRepository)(nil)
func NewMenuRepository(store repository.Store) *menuRepository {
func NewMenuRepository(repo *repository.Repository) system.MenuRepository {
return &menuRepository{
store: store,
repo: repo,
}
}
func (r *menuRepository) Create(ctx context.Context, obj *system.Menu) error {
return r.store.DB(ctx).Create(obj).Error
return r.repo.DB(ctx).Create(obj).Error
}
func (r *menuRepository) Update(ctx context.Context, obj *system.Menu) error {
return r.store.DB(ctx).Save(obj).Error
return r.repo.DB(ctx).Save(obj).Error
}
func (r *menuRepository) Get(ctx context.Context, id int32) (*system.Menu, error) {
var menu system.Menu
err := r.store.DB(ctx).Where("id = ?", id).First(&menu).Error
err := r.repo.DB(ctx).Where("id = ?", id).First(&menu).Error
if err != nil {
return nil, err
}
@@ -38,7 +36,7 @@ func (r *menuRepository) Get(ctx context.Context, id int32) (*system.Menu, error
func (r *menuRepository) GetByUrl(ctx context.Context, url string) (*system.Menu, error) {
var menu system.Menu
err := r.store.DB(ctx).Where("url = ?", url).First(&menu).Error
err := r.repo.DB(ctx).Where("url = ?", url).First(&menu).Error
if err != nil {
return nil, err
}
@@ -47,7 +45,7 @@ func (r *menuRepository) GetByUrl(ctx context.Context, url string) (*system.Menu
func (r *menuRepository) All(ctx context.Context) ([]*system.Menu, error) {
var menus []*system.Menu
err := r.store.DB(ctx).Find(&menus).Error
err := r.repo.DB(ctx).Find(&menus).Error
if err != nil {
return nil, err
}
@@ -69,5 +67,5 @@ func (r *menuRepository) RebuildParentPath(ctx context.Context) error {
FROM temp
ORDER BY id) AS t)
WHERE tm.status = 0;`
return r.store.DB(ctx).Exec(query).Error
return r.repo.DB(ctx).Exec(query).Error
}

20
internal/erpserver/repository/system/role.go
View File

@@ -9,28 +9,26 @@ import (
)
type roleRepository struct {
store repository.Store
repo *repository.Repository
}
var _ system.RoleRepository = (*roleRepository)(nil)
func NewRoleRepository(store repository.Store) *roleRepository {
func NewRoleRepository(repo *repository.Repository) system.RoleRepository {
return &roleRepository{
store: store,
repo: repo,
}
}
func (r *roleRepository) Create(ctx context.Context, obj *system.Role) error {
return r.store.DB(ctx).Create(obj).Error
return r.repo.DB(ctx).Create(obj).Error
}
func (r *roleRepository) Update(ctx context.Context, obj *system.Role) error {
return r.store.DB(ctx).Save(obj).Error
return r.repo.DB(ctx).Save(obj).Error
}
func (r *roleRepository) Get(ctx context.Context, id int32) (*system.Role, error) {
var role system.Role
err := r.store.DB(ctx).Where("id = ?", id).First(&role).Error
err := r.repo.DB(ctx).Where("id = ?", id).First(&role).Error
if err != nil {
return nil, err
}
@@ -39,7 +37,7 @@ func (r *roleRepository) Get(ctx context.Context, id int32) (*system.Role, error
func (r *roleRepository) All(ctx context.Context) ([]*system.Role, error) {
var roles []*system.Role
err := r.store.DB(ctx).Find(&roles).Error
err := r.repo.DB(ctx).Find(&roles).Error
if err != nil {
return nil, err
}
@@ -47,7 +45,7 @@ func (r *roleRepository) All(ctx context.Context) ([]*system.Role, error) {
}
func (r *roleRepository) List(ctx context.Context, q dto.SearchDto) ([]*system.Role, int64, error) {
query := r.store.DB(ctx).
query := r.repo.DB(ctx).
Model(&system.Role{}).
Where("created_at BETWEEN ? AND ?", q.SearchTimeBegin, q.SearchTimeEnd)
if q.SearchID != 0 {
@@ -96,5 +94,5 @@ func (r *roleRepository) RebuildParentPath(ctx context.Context) error {
FROM temp
ORDER BY id) AS t)
WHERE tm.status = 0;`
return r.store.DB(ctx).Exec(query).Error
return r.repo.DB(ctx).Exec(query).Error
}

14
internal/erpserver/repository/system/role_menu.go
View File

@@ -8,28 +8,26 @@ import (
)
type roleMenuRepository struct {
store repository.Store
repo *repository.Repository
}
var _ system.RoleMenuRepository = (*roleMenuRepository)(nil)
func NewRoleMenuRepository(store repository.Store) *roleMenuRepository {
func NewRoleMenuRepository(repo *repository.Repository) system.RoleMenuRepository {
return &roleMenuRepository{
store: store,
repo: repo,
}
}
func (r *roleMenuRepository) Create(ctx context.Context, obj []*system.RoleMenu) error {
return r.store.DB(ctx).Create(obj).Error
return r.repo.DB(ctx).Create(obj).Error
}
func (r *roleMenuRepository) DeleteByRoleID(ctx context.Context, roleID int32) error {
return r.store.DB(ctx).Where("role_id = ?", roleID).Delete(&system.RoleMenu{}).Error
return r.repo.DB(ctx).Where("role_id = ?", roleID).Delete(&system.RoleMenu{}).Error
}
func (r *roleMenuRepository) ListByRoleID(ctx context.Context, roleID int32) ([]*system.RoleMenu, error) {
var roleMenus []*system.RoleMenu
err := r.store.DB(ctx).Where("role_id = ?", roleID).Find(&roleMenus).Error
err := r.repo.DB(ctx).Where("role_id = ?", roleID).Find(&roleMenus).Error
if err != nil {
return nil, err
}

20
internal/erpserver/repository/system/user.go
View File

@@ -9,28 +9,26 @@ import (
)
type userRepository struct {
store repository.Store
repo *repository.Repository
}
var _ system.UserRepository = (*userRepository)(nil)
func NewUserRepository(store repository.Store) *userRepository {
func NewUserRepository(repo *repository.Repository) system.UserRepository {
return &userRepository{
store: store,
repo: repo,
}
}
func (s *userRepository) Create(ctx context.Context, obj *system.User) error {
return s.store.DB(ctx).Create(obj).Error
return s.repo.DB(ctx).Create(obj).Error
}
func (s *userRepository) Update(ctx context.Context, obj *system.User) error {
return s.store.DB(ctx).Save(obj).Error
return s.repo.DB(ctx).Save(obj).Error
}
func (s *userRepository) Get(ctx context.Context, id int32) (*system.User, error) {
var user system.User
err := s.store.DB(ctx).Where("id = ?", id).First(&user).Error
err := s.repo.DB(ctx).Where("id = ?", id).First(&user).Error
if err != nil {
return nil, err
}
@@ -39,7 +37,7 @@ func (s *userRepository) Get(ctx context.Context, id int32) (*system.User, error
func (s *userRepository) GetByEmail(ctx context.Context, email string) (*system.User, error) {
var user system.User
err := s.store.DB(ctx).Where("email = ?", email).First(&user).Error
err := s.repo.DB(ctx).Where("email = ?", email).First(&user).Error
if err != nil {
return nil, err
}
@@ -48,7 +46,7 @@ func (s *userRepository) GetByEmail(ctx context.Context, email string) (*system.
func (s *userRepository) All(ctx context.Context) ([]*system.User, error) {
var users []*system.User
err := s.store.DB(ctx).Find(&users).Error
err := s.repo.DB(ctx).Find(&users).Error
if err != nil {
return nil, err
}
@@ -56,7 +54,7 @@ func (s *userRepository) All(ctx context.Context) ([]*system.User, error) {
}
func (s *userRepository) List(ctx context.Context, q dto.SearchDto) ([]*system.User, int64, error) {
query := s.store.DB(ctx).
query := s.repo.DB(ctx).
Model(&system.User{}).
Preload("Role").
Preload("Department").