mirror of
https://github.com/hanzoai/commerce.git
synced 2026-10-30 15:08:42 +00:00
zeekay
1943f1c093
Multi-tenant commerce and billing: orders, subscriptions, invoices, plans, usage metering and a credit ledger. One Go binary. MIT OR Apache-2.0. Claude-Session: https://claude.ai/code/session_01SAtaDFm8VDiWrhYHNNtxMw
87 lines
2.0 KiB
Go
87 lines
2.0 KiB
Go
package meter
|
|
|
|
import (
|
|
"time"
|
|
|
|
"github.com/hanzoai/orm"
|
|
|
|
"github.com/hanzoai/commerce/datastore"
|
|
"github.com/hanzoai/commerce/models/mixin"
|
|
"github.com/hanzoai/commerce/util/json"
|
|
"github.com/hanzoai/commerce/util/val"
|
|
|
|
. "github.com/hanzoai/commerce/types"
|
|
)
|
|
|
|
var eventKind = "meter-event"
|
|
|
|
// MeterEvent records a single usage data point for a meter.
|
|
|
|
func init() { orm.Register[MeterEvent]("meter-event") }
|
|
|
|
type MeterEvent struct {
|
|
mixin.Model[MeterEvent]
|
|
|
|
MeterId string `json:"meterId"`
|
|
UserId string `json:"userId"`
|
|
Value int64 `json:"value"`
|
|
Timestamp time.Time `json:"timestamp"`
|
|
|
|
// Idempotency key for deduplication
|
|
Idempotency string `json:"idempotency,omitempty"`
|
|
|
|
// JSON-encoded dimension values, e.g. {"model":"gpt-4","region":"us"}
|
|
Dimensions Map `json:"dimensions,omitempty" datastore:"-"`
|
|
Dimensions_ string `json:"-" datastore:",noindex"`
|
|
|
|
Metadata Map `json:"metadata,omitempty" datastore:"-"`
|
|
Metadata_ string `json:"-" datastore:",noindex"`
|
|
}
|
|
|
|
func (e *MeterEvent) Defaults() {
|
|
e.Parent = e.Datastore().NewKey("synckey", "", 1, nil)
|
|
if e.Timestamp.IsZero() {
|
|
e.Timestamp = time.Now()
|
|
}
|
|
}
|
|
|
|
func (e *MeterEvent) Load(ps []datastore.Property) (err error) {
|
|
if err = datastore.LoadStruct(e, ps); err != nil {
|
|
return err
|
|
}
|
|
|
|
if len(e.Dimensions_) > 0 {
|
|
err = json.DecodeBytes([]byte(e.Dimensions_), &e.Dimensions)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
}
|
|
|
|
if len(e.Metadata_) > 0 {
|
|
err = json.DecodeBytes([]byte(e.Metadata_), &e.Metadata)
|
|
}
|
|
|
|
return err
|
|
}
|
|
|
|
func (e *MeterEvent) Save() (ps []datastore.Property, err error) {
|
|
e.Dimensions_ = string(json.EncodeBytes(&e.Dimensions))
|
|
e.Metadata_ = string(json.EncodeBytes(&e.Metadata))
|
|
return datastore.SaveStruct(e)
|
|
}
|
|
|
|
func (e *MeterEvent) Validator() *val.Validator {
|
|
return nil
|
|
}
|
|
|
|
func NewEvent(db *datastore.Datastore) *MeterEvent {
|
|
e := new(MeterEvent)
|
|
e.Init(db)
|
|
e.Defaults()
|
|
return e
|
|
}
|
|
|
|
func QueryEvents(db *datastore.Datastore) datastore.Query {
|
|
return db.Query(eventKind)
|
|
}
|