bill-server-go/internal/bill/service.go

56 lines
1.5 KiB
Go
Raw Normal View History

2023-05-08 15:07:20 +08:00
package bill
import (
"context"
"time"
)
type billService struct {
billRepository BillRepository
}
func NewBillService(r BillRepository) BillService {
return &billService{
billRepository: r,
}
}
2023-05-31 15:31:49 +08:00
// GetBills
2023-05-08 15:07:20 +08:00
// - date is {year, month, day} when return a day bills
// - date is {year, month} when return a month bills
// - date is {year} when return a year bills
// - date is {} when return all bills
2023-06-10 15:19:41 +08:00
func (b *billService) GetBills(ctx context.Context, date BDate) ([]Bill, error) {
2023-05-08 15:07:20 +08:00
if date.Year == 0 && (date.Month != 0 || date.Day != 0) {
date.Year = time.Now().Year()
} else if date.Month == 0 && date.Day != 0 {
date.Month = int(time.Now().Month())
}
switch {
case date.Year == 0:
return b.billRepository.GetBills(ctx)
case date.Month == 0:
return b.billRepository.GetBillByYear(ctx, date.Year)
case date.Day == 0:
return b.billRepository.GetBillByMonth(ctx, date.Year, date.Month)
default:
2023-05-08 21:48:34 +08:00
return b.billRepository.GetBillByDay(ctx, date.Year, date.Month, date.Day)
2023-05-08 15:07:20 +08:00
}
}
2023-05-11 21:00:29 +08:00
func (b *billService) GetBillByID(ctx context.Context, id int) (*Bill, error) {
2023-05-08 15:07:20 +08:00
return b.billRepository.GetBillByID(ctx, id)
}
2023-05-11 21:00:29 +08:00
func (b *billService) CreateBill(ctx context.Context, bill *Bill) error {
2023-05-08 15:07:20 +08:00
return b.billRepository.CreateBill(ctx, bill)
}
2023-05-11 21:00:29 +08:00
func (b *billService) UpdateBill(ctx context.Context, bill *Bill) error {
2023-05-08 15:07:20 +08:00
return b.billRepository.UpdateBill(ctx, bill)
}
2023-05-11 21:00:29 +08:00
func (b *billService) DeleteBill(ctx context.Context, id int) error {
2023-05-08 15:07:20 +08:00
return b.billRepository.DeleteBill(ctx, id)
}