store

package
v0.1.1 Latest Latest
Warning

This package is not in the latest version of its module.

Go to latest
Published: Dec 17, 2024 License: Apache-2.0 Imports: 8 Imported by: 0

Documentation

Index

Constants

This section is empty.

Variables

This section is empty.

Functions

func NewDBTX

func NewDBTX(ctx context.Context, cfg *config.Config) *pgxpool.Pool

Types

type ApiKey

type ApiKey struct {
	ID        int64
	UserID    int64
	Name      string
	Key       string
	Spend     int32
	Status    int16
	CreatedAt time.Time
	UpdatedAt time.Time
	DeletedAt pgtype.Timestamptz
}

type CreateAPIKeyParams

type CreateAPIKeyParams struct {
	UserID int64
	Name   string
	Key    string
}

type CreateOutcomeParams

type CreateOutcomeParams struct {
	UserID           int64
	KeyID            int64
	ModelName        string
	PromptTokens     int32
	CompletionTokens int32
	TotalTokens      int32
	Rt               int32
	FeeRate          int32
	Cost             int32
}

type CreateUserParams

type CreateUserParams struct {
	Email            string
	VerificationCode string
}

type DBTX

type DBTX interface {
	Exec(context.Context, string, ...interface{}) (pgconn.CommandTag, error)
	Query(context.Context, string, ...interface{}) (pgx.Rows, error)
	QueryRow(context.Context, string, ...interface{}) pgx.Row
}

type Income

type Income struct {
	ID            int64
	UserID        int64
	RechargeType  int16
	RechargeValue int32
	CreatedAt     time.Time
	UpdatedAt     time.Time
	DeletedAt     pgtype.Timestamptz
}

type Outcome

type Outcome struct {
	ID               int64
	UserID           int64
	KeyID            int64
	ModelName        string
	PromptTokens     int32
	CompletionTokens int32
	TotalTokens      int32
	FeeRate          int32
	Cost             int32
	Rt               int32
	CreatedAt        time.Time
	UpdatedAt        time.Time
	DeletedAt        pgtype.Timestamptz
}

type Queries

type Queries struct {
	// contains filtered or unexported fields
}

func New

func New(db DBTX) *Queries

func NewQueries

func NewQueries(dbtx *pgxpool.Pool) *Queries

func (*Queries) CreateAPIKey

func (q *Queries) CreateAPIKey(ctx context.Context, arg CreateAPIKeyParams) (ApiKey, error)

func (*Queries) CreateOutcome

func (q *Queries) CreateOutcome(ctx context.Context, arg CreateOutcomeParams) (Outcome, error)

func (*Queries) CreateUser

func (q *Queries) CreateUser(ctx context.Context, arg CreateUserParams) (User, error)

func (*Queries) GetAPIKey

func (q *Queries) GetAPIKey(ctx context.Context, key string) (ApiKey, error)

func (*Queries) GetUserByEmail

func (q *Queries) GetUserByEmail(ctx context.Context, email string) (User, error)

func (*Queries) GetUserBySessionID

func (q *Queries) GetUserBySessionID(ctx context.Context, sessionID pgtype.Text) (User, error)

func (*Queries) ListAPIKeys

func (q *Queries) ListAPIKeys(ctx context.Context, userID int64) ([]ApiKey, error)

func (*Queries) UpdateSessionID

func (q *Queries) UpdateSessionID(ctx context.Context, arg UpdateSessionIDParams) error

func (*Queries) UpdateVerificationCode

func (q *Queries) UpdateVerificationCode(ctx context.Context, arg UpdateVerificationCodeParams) error

func (*Queries) WithTx

func (q *Queries) WithTx(tx pgx.Tx) *Queries

type UpdateSessionIDParams

type UpdateSessionIDParams struct {
	ID        int64
	SessionID pgtype.Text
}

type UpdateVerificationCodeParams

type UpdateVerificationCodeParams struct {
	ID               int64
	VerificationCode string
}

type User

type User struct {
	ID               int64
	Email            string
	Amount           int32
	VerificationCode string
	VerificationAt   time.Time
	SessionID        pgtype.Text
	CreatedAt        time.Time
	UpdatedAt        time.Time
	DeletedAt        pgtype.Timestamptz
}

Jump to

Keyboard shortcuts

? : This menu
/ : Search site
f or F : Jump to
y or Y : Canonical URL