This feature allows API Keys to have their own quota limits and expiration times, independent of the user's balance. Backend: - Add quota, quota_used, expires_at fields to api_key schema - Implement IsExpired() and IsQuotaExhausted() checks in middleware - Add ResetQuota and ClearExpiration API endpoints - Integrate quota billing in gateway handlers (OpenAI, Anthropic, Gemini) - Include quota/expiration fields in auth cache for performance - Expiration check returns 403, quota exhausted returns 429 Frontend: - Add quota and expiration inputs to key create/edit dialog - Add quick-select buttons for expiration (+7, +30, +90 days) - Add reset quota confirmation dialog - Add expires_at column to keys list - Add i18n translations for new features (en/zh) Migration: - Add 045_add_api_key_quota.sql for new columns
242 lines
8.7 KiB
Go
242 lines
8.7 KiB
Go
// Code generated by ent, DO NOT EDIT.
|
|
|
|
package apikey
|
|
|
|
import (
|
|
"time"
|
|
|
|
"entgo.io/ent"
|
|
"entgo.io/ent/dialect/sql"
|
|
"entgo.io/ent/dialect/sql/sqlgraph"
|
|
)
|
|
|
|
const (
|
|
// Label holds the string label denoting the apikey type in the database.
|
|
Label = "api_key"
|
|
// FieldID holds the string denoting the id field in the database.
|
|
FieldID = "id"
|
|
// FieldCreatedAt holds the string denoting the created_at field in the database.
|
|
FieldCreatedAt = "created_at"
|
|
// FieldUpdatedAt holds the string denoting the updated_at field in the database.
|
|
FieldUpdatedAt = "updated_at"
|
|
// FieldDeletedAt holds the string denoting the deleted_at field in the database.
|
|
FieldDeletedAt = "deleted_at"
|
|
// FieldUserID holds the string denoting the user_id field in the database.
|
|
FieldUserID = "user_id"
|
|
// FieldKey holds the string denoting the key field in the database.
|
|
FieldKey = "key"
|
|
// FieldName holds the string denoting the name field in the database.
|
|
FieldName = "name"
|
|
// FieldGroupID holds the string denoting the group_id field in the database.
|
|
FieldGroupID = "group_id"
|
|
// FieldStatus holds the string denoting the status field in the database.
|
|
FieldStatus = "status"
|
|
// FieldIPWhitelist holds the string denoting the ip_whitelist field in the database.
|
|
FieldIPWhitelist = "ip_whitelist"
|
|
// FieldIPBlacklist holds the string denoting the ip_blacklist field in the database.
|
|
FieldIPBlacklist = "ip_blacklist"
|
|
// FieldQuota holds the string denoting the quota field in the database.
|
|
FieldQuota = "quota"
|
|
// FieldQuotaUsed holds the string denoting the quota_used field in the database.
|
|
FieldQuotaUsed = "quota_used"
|
|
// FieldExpiresAt holds the string denoting the expires_at field in the database.
|
|
FieldExpiresAt = "expires_at"
|
|
// EdgeUser holds the string denoting the user edge name in mutations.
|
|
EdgeUser = "user"
|
|
// EdgeGroup holds the string denoting the group edge name in mutations.
|
|
EdgeGroup = "group"
|
|
// EdgeUsageLogs holds the string denoting the usage_logs edge name in mutations.
|
|
EdgeUsageLogs = "usage_logs"
|
|
// Table holds the table name of the apikey in the database.
|
|
Table = "api_keys"
|
|
// UserTable is the table that holds the user relation/edge.
|
|
UserTable = "api_keys"
|
|
// UserInverseTable is the table name for the User entity.
|
|
// It exists in this package in order to avoid circular dependency with the "user" package.
|
|
UserInverseTable = "users"
|
|
// UserColumn is the table column denoting the user relation/edge.
|
|
UserColumn = "user_id"
|
|
// GroupTable is the table that holds the group relation/edge.
|
|
GroupTable = "api_keys"
|
|
// GroupInverseTable is the table name for the Group entity.
|
|
// It exists in this package in order to avoid circular dependency with the "group" package.
|
|
GroupInverseTable = "groups"
|
|
// GroupColumn is the table column denoting the group relation/edge.
|
|
GroupColumn = "group_id"
|
|
// UsageLogsTable is the table that holds the usage_logs relation/edge.
|
|
UsageLogsTable = "usage_logs"
|
|
// UsageLogsInverseTable is the table name for the UsageLog entity.
|
|
// It exists in this package in order to avoid circular dependency with the "usagelog" package.
|
|
UsageLogsInverseTable = "usage_logs"
|
|
// UsageLogsColumn is the table column denoting the usage_logs relation/edge.
|
|
UsageLogsColumn = "api_key_id"
|
|
)
|
|
|
|
// Columns holds all SQL columns for apikey fields.
|
|
var Columns = []string{
|
|
FieldID,
|
|
FieldCreatedAt,
|
|
FieldUpdatedAt,
|
|
FieldDeletedAt,
|
|
FieldUserID,
|
|
FieldKey,
|
|
FieldName,
|
|
FieldGroupID,
|
|
FieldStatus,
|
|
FieldIPWhitelist,
|
|
FieldIPBlacklist,
|
|
FieldQuota,
|
|
FieldQuotaUsed,
|
|
FieldExpiresAt,
|
|
}
|
|
|
|
// ValidColumn reports if the column name is valid (part of the table columns).
|
|
func ValidColumn(column string) bool {
|
|
for i := range Columns {
|
|
if column == Columns[i] {
|
|
return true
|
|
}
|
|
}
|
|
return false
|
|
}
|
|
|
|
// Note that the variables below are initialized by the runtime
|
|
// package on the initialization of the application. Therefore,
|
|
// it should be imported in the main as follows:
|
|
//
|
|
// import _ "github.com/Wei-Shaw/sub2api/ent/runtime"
|
|
var (
|
|
Hooks [1]ent.Hook
|
|
Interceptors [1]ent.Interceptor
|
|
// DefaultCreatedAt holds the default value on creation for the "created_at" field.
|
|
DefaultCreatedAt func() time.Time
|
|
// DefaultUpdatedAt holds the default value on creation for the "updated_at" field.
|
|
DefaultUpdatedAt func() time.Time
|
|
// UpdateDefaultUpdatedAt holds the default value on update for the "updated_at" field.
|
|
UpdateDefaultUpdatedAt func() time.Time
|
|
// KeyValidator is a validator for the "key" field. It is called by the builders before save.
|
|
KeyValidator func(string) error
|
|
// NameValidator is a validator for the "name" field. It is called by the builders before save.
|
|
NameValidator func(string) error
|
|
// DefaultStatus holds the default value on creation for the "status" field.
|
|
DefaultStatus string
|
|
// StatusValidator is a validator for the "status" field. It is called by the builders before save.
|
|
StatusValidator func(string) error
|
|
// DefaultQuota holds the default value on creation for the "quota" field.
|
|
DefaultQuota float64
|
|
// DefaultQuotaUsed holds the default value on creation for the "quota_used" field.
|
|
DefaultQuotaUsed float64
|
|
)
|
|
|
|
// OrderOption defines the ordering options for the APIKey queries.
|
|
type OrderOption func(*sql.Selector)
|
|
|
|
// ByID orders the results by the id field.
|
|
func ByID(opts ...sql.OrderTermOption) OrderOption {
|
|
return sql.OrderByField(FieldID, opts...).ToFunc()
|
|
}
|
|
|
|
// ByCreatedAt orders the results by the created_at field.
|
|
func ByCreatedAt(opts ...sql.OrderTermOption) OrderOption {
|
|
return sql.OrderByField(FieldCreatedAt, opts...).ToFunc()
|
|
}
|
|
|
|
// ByUpdatedAt orders the results by the updated_at field.
|
|
func ByUpdatedAt(opts ...sql.OrderTermOption) OrderOption {
|
|
return sql.OrderByField(FieldUpdatedAt, opts...).ToFunc()
|
|
}
|
|
|
|
// ByDeletedAt orders the results by the deleted_at field.
|
|
func ByDeletedAt(opts ...sql.OrderTermOption) OrderOption {
|
|
return sql.OrderByField(FieldDeletedAt, opts...).ToFunc()
|
|
}
|
|
|
|
// ByUserID orders the results by the user_id field.
|
|
func ByUserID(opts ...sql.OrderTermOption) OrderOption {
|
|
return sql.OrderByField(FieldUserID, opts...).ToFunc()
|
|
}
|
|
|
|
// ByKey orders the results by the key field.
|
|
func ByKey(opts ...sql.OrderTermOption) OrderOption {
|
|
return sql.OrderByField(FieldKey, opts...).ToFunc()
|
|
}
|
|
|
|
// ByName orders the results by the name field.
|
|
func ByName(opts ...sql.OrderTermOption) OrderOption {
|
|
return sql.OrderByField(FieldName, opts...).ToFunc()
|
|
}
|
|
|
|
// ByGroupID orders the results by the group_id field.
|
|
func ByGroupID(opts ...sql.OrderTermOption) OrderOption {
|
|
return sql.OrderByField(FieldGroupID, opts...).ToFunc()
|
|
}
|
|
|
|
// ByStatus orders the results by the status field.
|
|
func ByStatus(opts ...sql.OrderTermOption) OrderOption {
|
|
return sql.OrderByField(FieldStatus, opts...).ToFunc()
|
|
}
|
|
|
|
// ByQuota orders the results by the quota field.
|
|
func ByQuota(opts ...sql.OrderTermOption) OrderOption {
|
|
return sql.OrderByField(FieldQuota, opts...).ToFunc()
|
|
}
|
|
|
|
// ByQuotaUsed orders the results by the quota_used field.
|
|
func ByQuotaUsed(opts ...sql.OrderTermOption) OrderOption {
|
|
return sql.OrderByField(FieldQuotaUsed, opts...).ToFunc()
|
|
}
|
|
|
|
// ByExpiresAt orders the results by the expires_at field.
|
|
func ByExpiresAt(opts ...sql.OrderTermOption) OrderOption {
|
|
return sql.OrderByField(FieldExpiresAt, opts...).ToFunc()
|
|
}
|
|
|
|
// ByUserField orders the results by user field.
|
|
func ByUserField(field string, opts ...sql.OrderTermOption) OrderOption {
|
|
return func(s *sql.Selector) {
|
|
sqlgraph.OrderByNeighborTerms(s, newUserStep(), sql.OrderByField(field, opts...))
|
|
}
|
|
}
|
|
|
|
// ByGroupField orders the results by group field.
|
|
func ByGroupField(field string, opts ...sql.OrderTermOption) OrderOption {
|
|
return func(s *sql.Selector) {
|
|
sqlgraph.OrderByNeighborTerms(s, newGroupStep(), sql.OrderByField(field, opts...))
|
|
}
|
|
}
|
|
|
|
// ByUsageLogsCount orders the results by usage_logs count.
|
|
func ByUsageLogsCount(opts ...sql.OrderTermOption) OrderOption {
|
|
return func(s *sql.Selector) {
|
|
sqlgraph.OrderByNeighborsCount(s, newUsageLogsStep(), opts...)
|
|
}
|
|
}
|
|
|
|
// ByUsageLogs orders the results by usage_logs terms.
|
|
func ByUsageLogs(term sql.OrderTerm, terms ...sql.OrderTerm) OrderOption {
|
|
return func(s *sql.Selector) {
|
|
sqlgraph.OrderByNeighborTerms(s, newUsageLogsStep(), append([]sql.OrderTerm{term}, terms...)...)
|
|
}
|
|
}
|
|
func newUserStep() *sqlgraph.Step {
|
|
return sqlgraph.NewStep(
|
|
sqlgraph.From(Table, FieldID),
|
|
sqlgraph.To(UserInverseTable, FieldID),
|
|
sqlgraph.Edge(sqlgraph.M2O, true, UserTable, UserColumn),
|
|
)
|
|
}
|
|
func newGroupStep() *sqlgraph.Step {
|
|
return sqlgraph.NewStep(
|
|
sqlgraph.From(Table, FieldID),
|
|
sqlgraph.To(GroupInverseTable, FieldID),
|
|
sqlgraph.Edge(sqlgraph.M2O, true, GroupTable, GroupColumn),
|
|
)
|
|
}
|
|
func newUsageLogsStep() *sqlgraph.Step {
|
|
return sqlgraph.NewStep(
|
|
sqlgraph.From(Table, FieldID),
|
|
sqlgraph.To(UsageLogsInverseTable, FieldID),
|
|
sqlgraph.Edge(sqlgraph.O2M, false, UsageLogsTable, UsageLogsColumn),
|
|
)
|
|
}
|