75 lines
1.9 KiB
Go
75 lines
1.9 KiB
Go
package service
|
|
|
|
import (
|
|
"context"
|
|
db "server/internal/db/sqlc"
|
|
"server/internal/model/common"
|
|
"server/internal/model/request"
|
|
"server/internal/pkg/dberr"
|
|
"server/internal/pkg/errs"
|
|
)
|
|
|
|
type CategoryService struct {
|
|
queries *db.Queries
|
|
}
|
|
|
|
func NewCategoryService(queries *db.Queries) *CategoryService {
|
|
return &CategoryService{
|
|
queries: queries,
|
|
}
|
|
}
|
|
|
|
func (s *CategoryService) ListPage(ctx context.Context, p *common.Pagination) ([]db.Category, int64, error) {
|
|
params := db.ListCategoriesParams{
|
|
Limit: p.PageSize,
|
|
Offset: (p.Page - 1) * p.PageSize,
|
|
}
|
|
|
|
total, err := s.queries.CountCategories(ctx)
|
|
if err != nil {
|
|
return nil, 0, err
|
|
}
|
|
|
|
list, err := s.queries.ListCategories(ctx, params)
|
|
if err != nil {
|
|
return nil, 0, err
|
|
}
|
|
|
|
return list, total, nil
|
|
}
|
|
|
|
func (s *CategoryService) ListAll(ctx context.Context) ([]db.Category, error) {
|
|
return s.queries.ListAllCategories(ctx)
|
|
}
|
|
|
|
func (s *CategoryService) Create(ctx context.Context, req request.CreateCategoryRequest) error {
|
|
params := db.CreateCategoryParams{
|
|
Name: req.Name,
|
|
Code: req.Code,
|
|
}
|
|
|
|
err := s.queries.CreateCategory(ctx, params)
|
|
return dberr.MapUniqueViolation(err, dberr.CategoryCodeKey, errs.ErrCategoryCodeAlreadyExists)
|
|
}
|
|
|
|
func (s *CategoryService) Update(ctx context.Context, id int32, req request.UpdateCategoryRequest) error {
|
|
params := db.UpdateCategoryParams{
|
|
ID: id,
|
|
Name: req.Name,
|
|
Code: req.Code,
|
|
}
|
|
rows, err := s.queries.UpdateCategory(ctx, params)
|
|
// 先判断数据条目是否存在
|
|
if err = dberr.MapRowsAffected(rows, err, errs.ErrCategoryNotFound); err != nil {
|
|
// 再判断code是否重复
|
|
return dberr.MapUniqueViolation(err, dberr.CategoryCodeKey, errs.ErrCategoryCodeAlreadyExists)
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func (s *CategoryService) Delete(ctx context.Context, id int32) error {
|
|
rows, err := s.queries.DeleteCategory(ctx, id)
|
|
return dberr.MapRowsAffected(rows, err, errs.ErrCategoryNotFound)
|
|
}
|