64 lines
1.6 KiB
Go
64 lines
1.6 KiB
Go
package usecase
|
||
|
||
import (
|
||
"context"
|
||
|
||
"github.com/zeromicro/go-zero/core/logx"
|
||
)
|
||
|
||
/*
|
||
charge 是一次計點呼叫的預留與結算，沿用既有四個 meter，不新增第五個（RG-04）。
|
||
|
||
用法：`defer charge.Settle(ctx, &err)` 綁在具名 error 回傳上，這樣任何失敗路徑都退點，
|
||
不會出現「AI 失敗了但點數扣掉」。
|
||
*/
|
||
type charge struct {
|
||
svc *Service
|
||
uid int64
|
||
meter string
|
||
mode string
|
||
label string
|
||
source string
|
||
settled bool
|
||
}
|
||
|
||
func (s *Service) bill(ctx context.Context, uid int64, meter, label, source string) (*charge, error) {
|
||
if s == nil || s.Usage == nil {
|
||
return &charge{settled: true}, nil
|
||
}
|
||
mode, err := s.Usage.PrepareCall(ctx, uid, meter)
|
||
if err != nil {
|
||
return nil, err
|
||
}
|
||
return &charge{svc: s, uid: uid, meter: meter, mode: mode, label: label, source: source}, nil
|
||
}
|
||
|
||
func (c *charge) Settle(ctx context.Context, errp *error) {
|
||
if errp != nil && *errp != nil {
|
||
c.Release(ctx)
|
||
return
|
||
}
|
||
c.Commit(ctx)
|
||
}
|
||
|
||
// Commit 寫入用量事件。使用者已經拿到結果，所以這裡失敗只記錄不轉成錯誤。
|
||
func (c *charge) Commit(ctx context.Context) {
|
||
if c == nil || c.settled {
|
||
return
|
||
}
|
||
c.settled = true
|
||
if _, err := c.svc.Usage.RecordCall(ctx, c.uid, c.meter, c.mode, c.label, c.source); err != nil {
|
||
logx.Errorf("usage record uid=%d meter=%s source=%s: %v", c.uid, c.meter, c.source, err)
|
||
}
|
||
}
|
||
|
||
func (c *charge) Release(ctx context.Context) {
|
||
if c == nil || c.settled {
|
||
return
|
||
}
|
||
c.settled = true
|
||
if err := c.svc.Usage.ReleaseCall(ctx, c.uid, c.meter, c.mode); err != nil {
|
||
logx.Errorf("usage release uid=%d meter=%s source=%s: %v", c.uid, c.meter, c.source, err)
|
||
}
|
||
}
|