mirror of
https://github.com/hanzoai/o11y.git
synced 2026-10-15 12:58:33 +00:00
zeekay
b9f9cdf918
Re-sync the entire SigNoz-derived Go tree to a clean fork of latest SigNoz main (upstream 3e6339019), eliminating the version-skew that broke `go build ./...` (mixed-era packages, ~14 imported-but-absent packages, partial signoz->o11y / clickhouse->datastore rebrands, self-import cycle, duplicate package decls). - pkg/ + cmd/ wholesale replaced with latest SigNoz; imports rewritten github.com/SigNoz/signoz/ -> github.com/hanzoai/o11y/. - Adopt SigNoz's CURRENT structure (resourcefilter->telemetryresourcefilter, integrationtypes->cloudintegrationtypes, roletypes/grammar/times folded in upstream); old paths dropped. - Fork deps repathed to hanzoai + bumped: signoz-otel-collector v0.144.6, clickhouse-go-mock v0.14.1, govaluate v0.1.0; expr-lang/expr => hanzoai/expr v1.17.8 (replace). go.mod reconciled from latest SigNoz; go.sum re-tidied. - ee/ + cmd/enterprise stripped — o11y stays MIT-only; community edition imports no ee packages. - Preserved o11y additions: module path github.com/hanzoai/o11y, mount.go (cloud.Register/zip mount adapter, HIP-0106), NOTICE + LICENSE (SigNoz §4c attribution retained; upstream pkg files are headerless per SigNoz convention). - Reverted o11y cosmetic renames (o11y*/datastore*) and experimental originals (authz/iamauthz, billing, zapreceiver, zapmetricreceiver) to SigNoz canonical to kill the skew; snapshots retained in git history for re-layering. go build ./... = exit 0. go vet clean except benign "unreachable code" in generated ANTLR parser files (inherited from upstream).
146 lines
3.7 KiB
Go
146 lines
3.7 KiB
Go
package impllmpricingrule
|
|
|
|
import (
|
|
"context"
|
|
|
|
"github.com/hanzoai/o11y/pkg/errors"
|
|
"github.com/hanzoai/o11y/pkg/sqlstore"
|
|
"github.com/hanzoai/o11y/pkg/types/llmpricingruletypes"
|
|
"github.com/hanzoai/o11y/pkg/valuer"
|
|
)
|
|
|
|
type store struct {
|
|
sqlstore sqlstore.SQLStore
|
|
}
|
|
|
|
func NewStore(sqlstore sqlstore.SQLStore) llmpricingruletypes.Store {
|
|
return &store{sqlstore: sqlstore}
|
|
}
|
|
|
|
func (store *store) List(ctx context.Context, orgID valuer.UUID, offset, limit int, search string, isOverride *bool) ([]*llmpricingruletypes.LLMPricingRule, int, error) {
|
|
rules := make([]*llmpricingruletypes.LLMPricingRule, 0)
|
|
|
|
query := store.sqlstore.
|
|
BunDBCtx(ctx).
|
|
NewSelect().
|
|
Model(&rules).
|
|
Where("org_id = ?", orgID)
|
|
|
|
if search != "" {
|
|
like := "%" + search + "%"
|
|
query = query.Where("(LOWER(model) LIKE LOWER(?) OR LOWER(provider) LIKE LOWER(?))", like, like)
|
|
}
|
|
|
|
if isOverride != nil {
|
|
query = query.Where("is_override = ?", *isOverride)
|
|
}
|
|
|
|
count, err := query.
|
|
Order("created_at DESC").
|
|
Offset(offset).
|
|
Limit(limit).
|
|
ScanAndCount(ctx)
|
|
if err != nil {
|
|
return nil, 0, err
|
|
}
|
|
|
|
return rules, count, nil
|
|
}
|
|
|
|
func (store *store) Get(ctx context.Context, orgID, id valuer.UUID) (*llmpricingruletypes.LLMPricingRule, error) {
|
|
rule := new(llmpricingruletypes.LLMPricingRule)
|
|
|
|
err := store.sqlstore.
|
|
BunDBCtx(ctx).
|
|
NewSelect().
|
|
Model(rule).
|
|
Where("org_id = ?", orgID).
|
|
Where("id = ?", id).
|
|
Scan(ctx)
|
|
if err != nil {
|
|
return nil, store.sqlstore.WrapNotFoundErrf(err, llmpricingruletypes.ErrCodePricingRuleNotFound, "pricing rule %s not found in the org", id)
|
|
}
|
|
|
|
return rule, nil
|
|
}
|
|
|
|
func (store *store) GetBySourceID(ctx context.Context, orgID, sourceID valuer.UUID) (*llmpricingruletypes.LLMPricingRule, error) {
|
|
rule := new(llmpricingruletypes.LLMPricingRule)
|
|
|
|
err := store.sqlstore.
|
|
BunDBCtx(ctx).
|
|
NewSelect().
|
|
Model(rule).
|
|
Where("org_id = ?", orgID).
|
|
Where("source_id = ?", sourceID).
|
|
Scan(ctx)
|
|
if err != nil {
|
|
return nil, store.sqlstore.WrapNotFoundErrf(err, llmpricingruletypes.ErrCodePricingRuleNotFound, "pricing rule with source_id %s not found in the org", sourceID)
|
|
}
|
|
|
|
return rule, nil
|
|
}
|
|
|
|
func (store *store) Create(ctx context.Context, rule *llmpricingruletypes.LLMPricingRule) error {
|
|
_, err := store.sqlstore.
|
|
BunDBCtx(ctx).
|
|
NewInsert().
|
|
Model(rule).
|
|
Exec(ctx)
|
|
if err != nil {
|
|
return store.sqlstore.WrapAlreadyExistsErrf(err, llmpricingruletypes.ErrCodePricingRuleAlreadyExists, "pricing rule with model %s already exists", rule.Model)
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (store *store) Update(ctx context.Context, rule *llmpricingruletypes.LLMPricingRule) error {
|
|
res, err := store.sqlstore.
|
|
BunDBCtx(ctx).
|
|
NewUpdate().
|
|
Model(rule).
|
|
Where("org_id = ?", rule.OrgID).
|
|
Where("id = ?", rule.ID).
|
|
ExcludeColumn("id", "org_id", "source_id", "created_at", "created_by").
|
|
Exec(ctx)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
rowsAffected, err := res.RowsAffected()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
if rowsAffected == 0 {
|
|
return errors.Newf(errors.TypeNotFound, llmpricingruletypes.ErrCodePricingRuleNotFound, "pricing rule %s not found in the org", rule.ID)
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func (store *store) Delete(ctx context.Context, orgID, id valuer.UUID) error {
|
|
res, err := store.sqlstore.
|
|
BunDBCtx(ctx).
|
|
NewDelete().
|
|
Model((*llmpricingruletypes.LLMPricingRule)(nil)).
|
|
Where("org_id = ?", orgID).
|
|
Where("id = ?", id).
|
|
Exec(ctx)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
rowsAffected, err := res.RowsAffected()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
if rowsAffected == 0 {
|
|
return errors.Newf(errors.TypeNotFound, llmpricingruletypes.ErrCodePricingRuleNotFound, "pricing rule %s not found in the org", id)
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func (store *store) RunInTx(ctx context.Context, cb func(ctx context.Context) error) error {
|
|
return store.sqlstore.RunInTxCtx(ctx, nil, cb)
|
|
}
|