支持管理员配置上游错误如何返回给客户端: - 新增 ErrorPassthroughRule 数据模型和 Ent Schema - 实现规则的 CRUD API(/admin/error-passthrough-rules) - 支持按错误码、关键词匹配,支持 any/all 匹配模式 - 支持按平台过滤(anthropic/openai/gemini/antigravity) - 支持透传或自定义响应状态码和错误消息 - 实现两级缓存(Redis + 本地内存)和多实例同步 - 集成到 gateway_handler 的错误处理流程 - 新增前端管理界面组件 - 新增单元测试覆盖核心匹配逻辑 优化: - 移除 refreshLocalCache 中的冗余排序(数据库已排序) - 后端 Validate() 增加匹配条件非空校验
690 lines
27 KiB
Go
690 lines
27 KiB
Go
// Code generated by ent, DO NOT EDIT.
|
|
|
|
package intercept
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
|
|
"entgo.io/ent/dialect/sql"
|
|
"github.com/Wei-Shaw/sub2api/ent"
|
|
"github.com/Wei-Shaw/sub2api/ent/account"
|
|
"github.com/Wei-Shaw/sub2api/ent/accountgroup"
|
|
"github.com/Wei-Shaw/sub2api/ent/announcement"
|
|
"github.com/Wei-Shaw/sub2api/ent/announcementread"
|
|
"github.com/Wei-Shaw/sub2api/ent/apikey"
|
|
"github.com/Wei-Shaw/sub2api/ent/errorpassthroughrule"
|
|
"github.com/Wei-Shaw/sub2api/ent/group"
|
|
"github.com/Wei-Shaw/sub2api/ent/predicate"
|
|
"github.com/Wei-Shaw/sub2api/ent/promocode"
|
|
"github.com/Wei-Shaw/sub2api/ent/promocodeusage"
|
|
"github.com/Wei-Shaw/sub2api/ent/proxy"
|
|
"github.com/Wei-Shaw/sub2api/ent/redeemcode"
|
|
"github.com/Wei-Shaw/sub2api/ent/setting"
|
|
"github.com/Wei-Shaw/sub2api/ent/usagecleanuptask"
|
|
"github.com/Wei-Shaw/sub2api/ent/usagelog"
|
|
"github.com/Wei-Shaw/sub2api/ent/user"
|
|
"github.com/Wei-Shaw/sub2api/ent/userallowedgroup"
|
|
"github.com/Wei-Shaw/sub2api/ent/userattributedefinition"
|
|
"github.com/Wei-Shaw/sub2api/ent/userattributevalue"
|
|
"github.com/Wei-Shaw/sub2api/ent/usersubscription"
|
|
)
|
|
|
|
// The Query interface represents an operation that queries a graph.
|
|
// By using this interface, users can write generic code that manipulates
|
|
// query builders of different types.
|
|
type Query interface {
|
|
// Type returns the string representation of the query type.
|
|
Type() string
|
|
// Limit the number of records to be returned by this query.
|
|
Limit(int)
|
|
// Offset to start from.
|
|
Offset(int)
|
|
// Unique configures the query builder to filter duplicate records.
|
|
Unique(bool)
|
|
// Order specifies how the records should be ordered.
|
|
Order(...func(*sql.Selector))
|
|
// WhereP appends storage-level predicates to the query builder. Using this method, users
|
|
// can use type-assertion to append predicates that do not depend on any generated package.
|
|
WhereP(...func(*sql.Selector))
|
|
}
|
|
|
|
// The Func type is an adapter that allows ordinary functions to be used as interceptors.
|
|
// Unlike traversal functions, interceptors are skipped during graph traversals. Note that the
|
|
// implementation of Func is different from the one defined in entgo.io/ent.InterceptFunc.
|
|
type Func func(context.Context, Query) error
|
|
|
|
// Intercept calls f(ctx, q) and then applied the next Querier.
|
|
func (f Func) Intercept(next ent.Querier) ent.Querier {
|
|
return ent.QuerierFunc(func(ctx context.Context, q ent.Query) (ent.Value, error) {
|
|
query, err := NewQuery(q)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
if err := f(ctx, query); err != nil {
|
|
return nil, err
|
|
}
|
|
return next.Query(ctx, q)
|
|
})
|
|
}
|
|
|
|
// The TraverseFunc type is an adapter to allow the use of ordinary function as Traverser.
|
|
// If f is a function with the appropriate signature, TraverseFunc(f) is a Traverser that calls f.
|
|
type TraverseFunc func(context.Context, Query) error
|
|
|
|
// Intercept is a dummy implementation of Intercept that returns the next Querier in the pipeline.
|
|
func (f TraverseFunc) Intercept(next ent.Querier) ent.Querier {
|
|
return next
|
|
}
|
|
|
|
// Traverse calls f(ctx, q).
|
|
func (f TraverseFunc) Traverse(ctx context.Context, q ent.Query) error {
|
|
query, err := NewQuery(q)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
return f(ctx, query)
|
|
}
|
|
|
|
// The APIKeyFunc type is an adapter to allow the use of ordinary function as a Querier.
|
|
type APIKeyFunc func(context.Context, *ent.APIKeyQuery) (ent.Value, error)
|
|
|
|
// Query calls f(ctx, q).
|
|
func (f APIKeyFunc) Query(ctx context.Context, q ent.Query) (ent.Value, error) {
|
|
if q, ok := q.(*ent.APIKeyQuery); ok {
|
|
return f(ctx, q)
|
|
}
|
|
return nil, fmt.Errorf("unexpected query type %T. expect *ent.APIKeyQuery", q)
|
|
}
|
|
|
|
// The TraverseAPIKey type is an adapter to allow the use of ordinary function as Traverser.
|
|
type TraverseAPIKey func(context.Context, *ent.APIKeyQuery) error
|
|
|
|
// Intercept is a dummy implementation of Intercept that returns the next Querier in the pipeline.
|
|
func (f TraverseAPIKey) Intercept(next ent.Querier) ent.Querier {
|
|
return next
|
|
}
|
|
|
|
// Traverse calls f(ctx, q).
|
|
func (f TraverseAPIKey) Traverse(ctx context.Context, q ent.Query) error {
|
|
if q, ok := q.(*ent.APIKeyQuery); ok {
|
|
return f(ctx, q)
|
|
}
|
|
return fmt.Errorf("unexpected query type %T. expect *ent.APIKeyQuery", q)
|
|
}
|
|
|
|
// The AccountFunc type is an adapter to allow the use of ordinary function as a Querier.
|
|
type AccountFunc func(context.Context, *ent.AccountQuery) (ent.Value, error)
|
|
|
|
// Query calls f(ctx, q).
|
|
func (f AccountFunc) Query(ctx context.Context, q ent.Query) (ent.Value, error) {
|
|
if q, ok := q.(*ent.AccountQuery); ok {
|
|
return f(ctx, q)
|
|
}
|
|
return nil, fmt.Errorf("unexpected query type %T. expect *ent.AccountQuery", q)
|
|
}
|
|
|
|
// The TraverseAccount type is an adapter to allow the use of ordinary function as Traverser.
|
|
type TraverseAccount func(context.Context, *ent.AccountQuery) error
|
|
|
|
// Intercept is a dummy implementation of Intercept that returns the next Querier in the pipeline.
|
|
func (f TraverseAccount) Intercept(next ent.Querier) ent.Querier {
|
|
return next
|
|
}
|
|
|
|
// Traverse calls f(ctx, q).
|
|
func (f TraverseAccount) Traverse(ctx context.Context, q ent.Query) error {
|
|
if q, ok := q.(*ent.AccountQuery); ok {
|
|
return f(ctx, q)
|
|
}
|
|
return fmt.Errorf("unexpected query type %T. expect *ent.AccountQuery", q)
|
|
}
|
|
|
|
// The AccountGroupFunc type is an adapter to allow the use of ordinary function as a Querier.
|
|
type AccountGroupFunc func(context.Context, *ent.AccountGroupQuery) (ent.Value, error)
|
|
|
|
// Query calls f(ctx, q).
|
|
func (f AccountGroupFunc) Query(ctx context.Context, q ent.Query) (ent.Value, error) {
|
|
if q, ok := q.(*ent.AccountGroupQuery); ok {
|
|
return f(ctx, q)
|
|
}
|
|
return nil, fmt.Errorf("unexpected query type %T. expect *ent.AccountGroupQuery", q)
|
|
}
|
|
|
|
// The TraverseAccountGroup type is an adapter to allow the use of ordinary function as Traverser.
|
|
type TraverseAccountGroup func(context.Context, *ent.AccountGroupQuery) error
|
|
|
|
// Intercept is a dummy implementation of Intercept that returns the next Querier in the pipeline.
|
|
func (f TraverseAccountGroup) Intercept(next ent.Querier) ent.Querier {
|
|
return next
|
|
}
|
|
|
|
// Traverse calls f(ctx, q).
|
|
func (f TraverseAccountGroup) Traverse(ctx context.Context, q ent.Query) error {
|
|
if q, ok := q.(*ent.AccountGroupQuery); ok {
|
|
return f(ctx, q)
|
|
}
|
|
return fmt.Errorf("unexpected query type %T. expect *ent.AccountGroupQuery", q)
|
|
}
|
|
|
|
// The AnnouncementFunc type is an adapter to allow the use of ordinary function as a Querier.
|
|
type AnnouncementFunc func(context.Context, *ent.AnnouncementQuery) (ent.Value, error)
|
|
|
|
// Query calls f(ctx, q).
|
|
func (f AnnouncementFunc) Query(ctx context.Context, q ent.Query) (ent.Value, error) {
|
|
if q, ok := q.(*ent.AnnouncementQuery); ok {
|
|
return f(ctx, q)
|
|
}
|
|
return nil, fmt.Errorf("unexpected query type %T. expect *ent.AnnouncementQuery", q)
|
|
}
|
|
|
|
// The TraverseAnnouncement type is an adapter to allow the use of ordinary function as Traverser.
|
|
type TraverseAnnouncement func(context.Context, *ent.AnnouncementQuery) error
|
|
|
|
// Intercept is a dummy implementation of Intercept that returns the next Querier in the pipeline.
|
|
func (f TraverseAnnouncement) Intercept(next ent.Querier) ent.Querier {
|
|
return next
|
|
}
|
|
|
|
// Traverse calls f(ctx, q).
|
|
func (f TraverseAnnouncement) Traverse(ctx context.Context, q ent.Query) error {
|
|
if q, ok := q.(*ent.AnnouncementQuery); ok {
|
|
return f(ctx, q)
|
|
}
|
|
return fmt.Errorf("unexpected query type %T. expect *ent.AnnouncementQuery", q)
|
|
}
|
|
|
|
// The AnnouncementReadFunc type is an adapter to allow the use of ordinary function as a Querier.
|
|
type AnnouncementReadFunc func(context.Context, *ent.AnnouncementReadQuery) (ent.Value, error)
|
|
|
|
// Query calls f(ctx, q).
|
|
func (f AnnouncementReadFunc) Query(ctx context.Context, q ent.Query) (ent.Value, error) {
|
|
if q, ok := q.(*ent.AnnouncementReadQuery); ok {
|
|
return f(ctx, q)
|
|
}
|
|
return nil, fmt.Errorf("unexpected query type %T. expect *ent.AnnouncementReadQuery", q)
|
|
}
|
|
|
|
// The TraverseAnnouncementRead type is an adapter to allow the use of ordinary function as Traverser.
|
|
type TraverseAnnouncementRead func(context.Context, *ent.AnnouncementReadQuery) error
|
|
|
|
// Intercept is a dummy implementation of Intercept that returns the next Querier in the pipeline.
|
|
func (f TraverseAnnouncementRead) Intercept(next ent.Querier) ent.Querier {
|
|
return next
|
|
}
|
|
|
|
// Traverse calls f(ctx, q).
|
|
func (f TraverseAnnouncementRead) Traverse(ctx context.Context, q ent.Query) error {
|
|
if q, ok := q.(*ent.AnnouncementReadQuery); ok {
|
|
return f(ctx, q)
|
|
}
|
|
return fmt.Errorf("unexpected query type %T. expect *ent.AnnouncementReadQuery", q)
|
|
}
|
|
|
|
// The ErrorPassthroughRuleFunc type is an adapter to allow the use of ordinary function as a Querier.
|
|
type ErrorPassthroughRuleFunc func(context.Context, *ent.ErrorPassthroughRuleQuery) (ent.Value, error)
|
|
|
|
// Query calls f(ctx, q).
|
|
func (f ErrorPassthroughRuleFunc) Query(ctx context.Context, q ent.Query) (ent.Value, error) {
|
|
if q, ok := q.(*ent.ErrorPassthroughRuleQuery); ok {
|
|
return f(ctx, q)
|
|
}
|
|
return nil, fmt.Errorf("unexpected query type %T. expect *ent.ErrorPassthroughRuleQuery", q)
|
|
}
|
|
|
|
// The TraverseErrorPassthroughRule type is an adapter to allow the use of ordinary function as Traverser.
|
|
type TraverseErrorPassthroughRule func(context.Context, *ent.ErrorPassthroughRuleQuery) error
|
|
|
|
// Intercept is a dummy implementation of Intercept that returns the next Querier in the pipeline.
|
|
func (f TraverseErrorPassthroughRule) Intercept(next ent.Querier) ent.Querier {
|
|
return next
|
|
}
|
|
|
|
// Traverse calls f(ctx, q).
|
|
func (f TraverseErrorPassthroughRule) Traverse(ctx context.Context, q ent.Query) error {
|
|
if q, ok := q.(*ent.ErrorPassthroughRuleQuery); ok {
|
|
return f(ctx, q)
|
|
}
|
|
return fmt.Errorf("unexpected query type %T. expect *ent.ErrorPassthroughRuleQuery", q)
|
|
}
|
|
|
|
// The GroupFunc type is an adapter to allow the use of ordinary function as a Querier.
|
|
type GroupFunc func(context.Context, *ent.GroupQuery) (ent.Value, error)
|
|
|
|
// Query calls f(ctx, q).
|
|
func (f GroupFunc) Query(ctx context.Context, q ent.Query) (ent.Value, error) {
|
|
if q, ok := q.(*ent.GroupQuery); ok {
|
|
return f(ctx, q)
|
|
}
|
|
return nil, fmt.Errorf("unexpected query type %T. expect *ent.GroupQuery", q)
|
|
}
|
|
|
|
// The TraverseGroup type is an adapter to allow the use of ordinary function as Traverser.
|
|
type TraverseGroup func(context.Context, *ent.GroupQuery) error
|
|
|
|
// Intercept is a dummy implementation of Intercept that returns the next Querier in the pipeline.
|
|
func (f TraverseGroup) Intercept(next ent.Querier) ent.Querier {
|
|
return next
|
|
}
|
|
|
|
// Traverse calls f(ctx, q).
|
|
func (f TraverseGroup) Traverse(ctx context.Context, q ent.Query) error {
|
|
if q, ok := q.(*ent.GroupQuery); ok {
|
|
return f(ctx, q)
|
|
}
|
|
return fmt.Errorf("unexpected query type %T. expect *ent.GroupQuery", q)
|
|
}
|
|
|
|
// The PromoCodeFunc type is an adapter to allow the use of ordinary function as a Querier.
|
|
type PromoCodeFunc func(context.Context, *ent.PromoCodeQuery) (ent.Value, error)
|
|
|
|
// Query calls f(ctx, q).
|
|
func (f PromoCodeFunc) Query(ctx context.Context, q ent.Query) (ent.Value, error) {
|
|
if q, ok := q.(*ent.PromoCodeQuery); ok {
|
|
return f(ctx, q)
|
|
}
|
|
return nil, fmt.Errorf("unexpected query type %T. expect *ent.PromoCodeQuery", q)
|
|
}
|
|
|
|
// The TraversePromoCode type is an adapter to allow the use of ordinary function as Traverser.
|
|
type TraversePromoCode func(context.Context, *ent.PromoCodeQuery) error
|
|
|
|
// Intercept is a dummy implementation of Intercept that returns the next Querier in the pipeline.
|
|
func (f TraversePromoCode) Intercept(next ent.Querier) ent.Querier {
|
|
return next
|
|
}
|
|
|
|
// Traverse calls f(ctx, q).
|
|
func (f TraversePromoCode) Traverse(ctx context.Context, q ent.Query) error {
|
|
if q, ok := q.(*ent.PromoCodeQuery); ok {
|
|
return f(ctx, q)
|
|
}
|
|
return fmt.Errorf("unexpected query type %T. expect *ent.PromoCodeQuery", q)
|
|
}
|
|
|
|
// The PromoCodeUsageFunc type is an adapter to allow the use of ordinary function as a Querier.
|
|
type PromoCodeUsageFunc func(context.Context, *ent.PromoCodeUsageQuery) (ent.Value, error)
|
|
|
|
// Query calls f(ctx, q).
|
|
func (f PromoCodeUsageFunc) Query(ctx context.Context, q ent.Query) (ent.Value, error) {
|
|
if q, ok := q.(*ent.PromoCodeUsageQuery); ok {
|
|
return f(ctx, q)
|
|
}
|
|
return nil, fmt.Errorf("unexpected query type %T. expect *ent.PromoCodeUsageQuery", q)
|
|
}
|
|
|
|
// The TraversePromoCodeUsage type is an adapter to allow the use of ordinary function as Traverser.
|
|
type TraversePromoCodeUsage func(context.Context, *ent.PromoCodeUsageQuery) error
|
|
|
|
// Intercept is a dummy implementation of Intercept that returns the next Querier in the pipeline.
|
|
func (f TraversePromoCodeUsage) Intercept(next ent.Querier) ent.Querier {
|
|
return next
|
|
}
|
|
|
|
// Traverse calls f(ctx, q).
|
|
func (f TraversePromoCodeUsage) Traverse(ctx context.Context, q ent.Query) error {
|
|
if q, ok := q.(*ent.PromoCodeUsageQuery); ok {
|
|
return f(ctx, q)
|
|
}
|
|
return fmt.Errorf("unexpected query type %T. expect *ent.PromoCodeUsageQuery", q)
|
|
}
|
|
|
|
// The ProxyFunc type is an adapter to allow the use of ordinary function as a Querier.
|
|
type ProxyFunc func(context.Context, *ent.ProxyQuery) (ent.Value, error)
|
|
|
|
// Query calls f(ctx, q).
|
|
func (f ProxyFunc) Query(ctx context.Context, q ent.Query) (ent.Value, error) {
|
|
if q, ok := q.(*ent.ProxyQuery); ok {
|
|
return f(ctx, q)
|
|
}
|
|
return nil, fmt.Errorf("unexpected query type %T. expect *ent.ProxyQuery", q)
|
|
}
|
|
|
|
// The TraverseProxy type is an adapter to allow the use of ordinary function as Traverser.
|
|
type TraverseProxy func(context.Context, *ent.ProxyQuery) error
|
|
|
|
// Intercept is a dummy implementation of Intercept that returns the next Querier in the pipeline.
|
|
func (f TraverseProxy) Intercept(next ent.Querier) ent.Querier {
|
|
return next
|
|
}
|
|
|
|
// Traverse calls f(ctx, q).
|
|
func (f TraverseProxy) Traverse(ctx context.Context, q ent.Query) error {
|
|
if q, ok := q.(*ent.ProxyQuery); ok {
|
|
return f(ctx, q)
|
|
}
|
|
return fmt.Errorf("unexpected query type %T. expect *ent.ProxyQuery", q)
|
|
}
|
|
|
|
// The RedeemCodeFunc type is an adapter to allow the use of ordinary function as a Querier.
|
|
type RedeemCodeFunc func(context.Context, *ent.RedeemCodeQuery) (ent.Value, error)
|
|
|
|
// Query calls f(ctx, q).
|
|
func (f RedeemCodeFunc) Query(ctx context.Context, q ent.Query) (ent.Value, error) {
|
|
if q, ok := q.(*ent.RedeemCodeQuery); ok {
|
|
return f(ctx, q)
|
|
}
|
|
return nil, fmt.Errorf("unexpected query type %T. expect *ent.RedeemCodeQuery", q)
|
|
}
|
|
|
|
// The TraverseRedeemCode type is an adapter to allow the use of ordinary function as Traverser.
|
|
type TraverseRedeemCode func(context.Context, *ent.RedeemCodeQuery) error
|
|
|
|
// Intercept is a dummy implementation of Intercept that returns the next Querier in the pipeline.
|
|
func (f TraverseRedeemCode) Intercept(next ent.Querier) ent.Querier {
|
|
return next
|
|
}
|
|
|
|
// Traverse calls f(ctx, q).
|
|
func (f TraverseRedeemCode) Traverse(ctx context.Context, q ent.Query) error {
|
|
if q, ok := q.(*ent.RedeemCodeQuery); ok {
|
|
return f(ctx, q)
|
|
}
|
|
return fmt.Errorf("unexpected query type %T. expect *ent.RedeemCodeQuery", q)
|
|
}
|
|
|
|
// The SettingFunc type is an adapter to allow the use of ordinary function as a Querier.
|
|
type SettingFunc func(context.Context, *ent.SettingQuery) (ent.Value, error)
|
|
|
|
// Query calls f(ctx, q).
|
|
func (f SettingFunc) Query(ctx context.Context, q ent.Query) (ent.Value, error) {
|
|
if q, ok := q.(*ent.SettingQuery); ok {
|
|
return f(ctx, q)
|
|
}
|
|
return nil, fmt.Errorf("unexpected query type %T. expect *ent.SettingQuery", q)
|
|
}
|
|
|
|
// The TraverseSetting type is an adapter to allow the use of ordinary function as Traverser.
|
|
type TraverseSetting func(context.Context, *ent.SettingQuery) error
|
|
|
|
// Intercept is a dummy implementation of Intercept that returns the next Querier in the pipeline.
|
|
func (f TraverseSetting) Intercept(next ent.Querier) ent.Querier {
|
|
return next
|
|
}
|
|
|
|
// Traverse calls f(ctx, q).
|
|
func (f TraverseSetting) Traverse(ctx context.Context, q ent.Query) error {
|
|
if q, ok := q.(*ent.SettingQuery); ok {
|
|
return f(ctx, q)
|
|
}
|
|
return fmt.Errorf("unexpected query type %T. expect *ent.SettingQuery", q)
|
|
}
|
|
|
|
// The UsageCleanupTaskFunc type is an adapter to allow the use of ordinary function as a Querier.
|
|
type UsageCleanupTaskFunc func(context.Context, *ent.UsageCleanupTaskQuery) (ent.Value, error)
|
|
|
|
// Query calls f(ctx, q).
|
|
func (f UsageCleanupTaskFunc) Query(ctx context.Context, q ent.Query) (ent.Value, error) {
|
|
if q, ok := q.(*ent.UsageCleanupTaskQuery); ok {
|
|
return f(ctx, q)
|
|
}
|
|
return nil, fmt.Errorf("unexpected query type %T. expect *ent.UsageCleanupTaskQuery", q)
|
|
}
|
|
|
|
// The TraverseUsageCleanupTask type is an adapter to allow the use of ordinary function as Traverser.
|
|
type TraverseUsageCleanupTask func(context.Context, *ent.UsageCleanupTaskQuery) error
|
|
|
|
// Intercept is a dummy implementation of Intercept that returns the next Querier in the pipeline.
|
|
func (f TraverseUsageCleanupTask) Intercept(next ent.Querier) ent.Querier {
|
|
return next
|
|
}
|
|
|
|
// Traverse calls f(ctx, q).
|
|
func (f TraverseUsageCleanupTask) Traverse(ctx context.Context, q ent.Query) error {
|
|
if q, ok := q.(*ent.UsageCleanupTaskQuery); ok {
|
|
return f(ctx, q)
|
|
}
|
|
return fmt.Errorf("unexpected query type %T. expect *ent.UsageCleanupTaskQuery", q)
|
|
}
|
|
|
|
// The UsageLogFunc type is an adapter to allow the use of ordinary function as a Querier.
|
|
type UsageLogFunc func(context.Context, *ent.UsageLogQuery) (ent.Value, error)
|
|
|
|
// Query calls f(ctx, q).
|
|
func (f UsageLogFunc) Query(ctx context.Context, q ent.Query) (ent.Value, error) {
|
|
if q, ok := q.(*ent.UsageLogQuery); ok {
|
|
return f(ctx, q)
|
|
}
|
|
return nil, fmt.Errorf("unexpected query type %T. expect *ent.UsageLogQuery", q)
|
|
}
|
|
|
|
// The TraverseUsageLog type is an adapter to allow the use of ordinary function as Traverser.
|
|
type TraverseUsageLog func(context.Context, *ent.UsageLogQuery) error
|
|
|
|
// Intercept is a dummy implementation of Intercept that returns the next Querier in the pipeline.
|
|
func (f TraverseUsageLog) Intercept(next ent.Querier) ent.Querier {
|
|
return next
|
|
}
|
|
|
|
// Traverse calls f(ctx, q).
|
|
func (f TraverseUsageLog) Traverse(ctx context.Context, q ent.Query) error {
|
|
if q, ok := q.(*ent.UsageLogQuery); ok {
|
|
return f(ctx, q)
|
|
}
|
|
return fmt.Errorf("unexpected query type %T. expect *ent.UsageLogQuery", q)
|
|
}
|
|
|
|
// The UserFunc type is an adapter to allow the use of ordinary function as a Querier.
|
|
type UserFunc func(context.Context, *ent.UserQuery) (ent.Value, error)
|
|
|
|
// Query calls f(ctx, q).
|
|
func (f UserFunc) Query(ctx context.Context, q ent.Query) (ent.Value, error) {
|
|
if q, ok := q.(*ent.UserQuery); ok {
|
|
return f(ctx, q)
|
|
}
|
|
return nil, fmt.Errorf("unexpected query type %T. expect *ent.UserQuery", q)
|
|
}
|
|
|
|
// The TraverseUser type is an adapter to allow the use of ordinary function as Traverser.
|
|
type TraverseUser func(context.Context, *ent.UserQuery) error
|
|
|
|
// Intercept is a dummy implementation of Intercept that returns the next Querier in the pipeline.
|
|
func (f TraverseUser) Intercept(next ent.Querier) ent.Querier {
|
|
return next
|
|
}
|
|
|
|
// Traverse calls f(ctx, q).
|
|
func (f TraverseUser) Traverse(ctx context.Context, q ent.Query) error {
|
|
if q, ok := q.(*ent.UserQuery); ok {
|
|
return f(ctx, q)
|
|
}
|
|
return fmt.Errorf("unexpected query type %T. expect *ent.UserQuery", q)
|
|
}
|
|
|
|
// The UserAllowedGroupFunc type is an adapter to allow the use of ordinary function as a Querier.
|
|
type UserAllowedGroupFunc func(context.Context, *ent.UserAllowedGroupQuery) (ent.Value, error)
|
|
|
|
// Query calls f(ctx, q).
|
|
func (f UserAllowedGroupFunc) Query(ctx context.Context, q ent.Query) (ent.Value, error) {
|
|
if q, ok := q.(*ent.UserAllowedGroupQuery); ok {
|
|
return f(ctx, q)
|
|
}
|
|
return nil, fmt.Errorf("unexpected query type %T. expect *ent.UserAllowedGroupQuery", q)
|
|
}
|
|
|
|
// The TraverseUserAllowedGroup type is an adapter to allow the use of ordinary function as Traverser.
|
|
type TraverseUserAllowedGroup func(context.Context, *ent.UserAllowedGroupQuery) error
|
|
|
|
// Intercept is a dummy implementation of Intercept that returns the next Querier in the pipeline.
|
|
func (f TraverseUserAllowedGroup) Intercept(next ent.Querier) ent.Querier {
|
|
return next
|
|
}
|
|
|
|
// Traverse calls f(ctx, q).
|
|
func (f TraverseUserAllowedGroup) Traverse(ctx context.Context, q ent.Query) error {
|
|
if q, ok := q.(*ent.UserAllowedGroupQuery); ok {
|
|
return f(ctx, q)
|
|
}
|
|
return fmt.Errorf("unexpected query type %T. expect *ent.UserAllowedGroupQuery", q)
|
|
}
|
|
|
|
// The UserAttributeDefinitionFunc type is an adapter to allow the use of ordinary function as a Querier.
|
|
type UserAttributeDefinitionFunc func(context.Context, *ent.UserAttributeDefinitionQuery) (ent.Value, error)
|
|
|
|
// Query calls f(ctx, q).
|
|
func (f UserAttributeDefinitionFunc) Query(ctx context.Context, q ent.Query) (ent.Value, error) {
|
|
if q, ok := q.(*ent.UserAttributeDefinitionQuery); ok {
|
|
return f(ctx, q)
|
|
}
|
|
return nil, fmt.Errorf("unexpected query type %T. expect *ent.UserAttributeDefinitionQuery", q)
|
|
}
|
|
|
|
// The TraverseUserAttributeDefinition type is an adapter to allow the use of ordinary function as Traverser.
|
|
type TraverseUserAttributeDefinition func(context.Context, *ent.UserAttributeDefinitionQuery) error
|
|
|
|
// Intercept is a dummy implementation of Intercept that returns the next Querier in the pipeline.
|
|
func (f TraverseUserAttributeDefinition) Intercept(next ent.Querier) ent.Querier {
|
|
return next
|
|
}
|
|
|
|
// Traverse calls f(ctx, q).
|
|
func (f TraverseUserAttributeDefinition) Traverse(ctx context.Context, q ent.Query) error {
|
|
if q, ok := q.(*ent.UserAttributeDefinitionQuery); ok {
|
|
return f(ctx, q)
|
|
}
|
|
return fmt.Errorf("unexpected query type %T. expect *ent.UserAttributeDefinitionQuery", q)
|
|
}
|
|
|
|
// The UserAttributeValueFunc type is an adapter to allow the use of ordinary function as a Querier.
|
|
type UserAttributeValueFunc func(context.Context, *ent.UserAttributeValueQuery) (ent.Value, error)
|
|
|
|
// Query calls f(ctx, q).
|
|
func (f UserAttributeValueFunc) Query(ctx context.Context, q ent.Query) (ent.Value, error) {
|
|
if q, ok := q.(*ent.UserAttributeValueQuery); ok {
|
|
return f(ctx, q)
|
|
}
|
|
return nil, fmt.Errorf("unexpected query type %T. expect *ent.UserAttributeValueQuery", q)
|
|
}
|
|
|
|
// The TraverseUserAttributeValue type is an adapter to allow the use of ordinary function as Traverser.
|
|
type TraverseUserAttributeValue func(context.Context, *ent.UserAttributeValueQuery) error
|
|
|
|
// Intercept is a dummy implementation of Intercept that returns the next Querier in the pipeline.
|
|
func (f TraverseUserAttributeValue) Intercept(next ent.Querier) ent.Querier {
|
|
return next
|
|
}
|
|
|
|
// Traverse calls f(ctx, q).
|
|
func (f TraverseUserAttributeValue) Traverse(ctx context.Context, q ent.Query) error {
|
|
if q, ok := q.(*ent.UserAttributeValueQuery); ok {
|
|
return f(ctx, q)
|
|
}
|
|
return fmt.Errorf("unexpected query type %T. expect *ent.UserAttributeValueQuery", q)
|
|
}
|
|
|
|
// The UserSubscriptionFunc type is an adapter to allow the use of ordinary function as a Querier.
|
|
type UserSubscriptionFunc func(context.Context, *ent.UserSubscriptionQuery) (ent.Value, error)
|
|
|
|
// Query calls f(ctx, q).
|
|
func (f UserSubscriptionFunc) Query(ctx context.Context, q ent.Query) (ent.Value, error) {
|
|
if q, ok := q.(*ent.UserSubscriptionQuery); ok {
|
|
return f(ctx, q)
|
|
}
|
|
return nil, fmt.Errorf("unexpected query type %T. expect *ent.UserSubscriptionQuery", q)
|
|
}
|
|
|
|
// The TraverseUserSubscription type is an adapter to allow the use of ordinary function as Traverser.
|
|
type TraverseUserSubscription func(context.Context, *ent.UserSubscriptionQuery) error
|
|
|
|
// Intercept is a dummy implementation of Intercept that returns the next Querier in the pipeline.
|
|
func (f TraverseUserSubscription) Intercept(next ent.Querier) ent.Querier {
|
|
return next
|
|
}
|
|
|
|
// Traverse calls f(ctx, q).
|
|
func (f TraverseUserSubscription) Traverse(ctx context.Context, q ent.Query) error {
|
|
if q, ok := q.(*ent.UserSubscriptionQuery); ok {
|
|
return f(ctx, q)
|
|
}
|
|
return fmt.Errorf("unexpected query type %T. expect *ent.UserSubscriptionQuery", q)
|
|
}
|
|
|
|
// NewQuery returns the generic Query interface for the given typed query.
|
|
func NewQuery(q ent.Query) (Query, error) {
|
|
switch q := q.(type) {
|
|
case *ent.APIKeyQuery:
|
|
return &query[*ent.APIKeyQuery, predicate.APIKey, apikey.OrderOption]{typ: ent.TypeAPIKey, tq: q}, nil
|
|
case *ent.AccountQuery:
|
|
return &query[*ent.AccountQuery, predicate.Account, account.OrderOption]{typ: ent.TypeAccount, tq: q}, nil
|
|
case *ent.AccountGroupQuery:
|
|
return &query[*ent.AccountGroupQuery, predicate.AccountGroup, accountgroup.OrderOption]{typ: ent.TypeAccountGroup, tq: q}, nil
|
|
case *ent.AnnouncementQuery:
|
|
return &query[*ent.AnnouncementQuery, predicate.Announcement, announcement.OrderOption]{typ: ent.TypeAnnouncement, tq: q}, nil
|
|
case *ent.AnnouncementReadQuery:
|
|
return &query[*ent.AnnouncementReadQuery, predicate.AnnouncementRead, announcementread.OrderOption]{typ: ent.TypeAnnouncementRead, tq: q}, nil
|
|
case *ent.ErrorPassthroughRuleQuery:
|
|
return &query[*ent.ErrorPassthroughRuleQuery, predicate.ErrorPassthroughRule, errorpassthroughrule.OrderOption]{typ: ent.TypeErrorPassthroughRule, tq: q}, nil
|
|
case *ent.GroupQuery:
|
|
return &query[*ent.GroupQuery, predicate.Group, group.OrderOption]{typ: ent.TypeGroup, tq: q}, nil
|
|
case *ent.PromoCodeQuery:
|
|
return &query[*ent.PromoCodeQuery, predicate.PromoCode, promocode.OrderOption]{typ: ent.TypePromoCode, tq: q}, nil
|
|
case *ent.PromoCodeUsageQuery:
|
|
return &query[*ent.PromoCodeUsageQuery, predicate.PromoCodeUsage, promocodeusage.OrderOption]{typ: ent.TypePromoCodeUsage, tq: q}, nil
|
|
case *ent.ProxyQuery:
|
|
return &query[*ent.ProxyQuery, predicate.Proxy, proxy.OrderOption]{typ: ent.TypeProxy, tq: q}, nil
|
|
case *ent.RedeemCodeQuery:
|
|
return &query[*ent.RedeemCodeQuery, predicate.RedeemCode, redeemcode.OrderOption]{typ: ent.TypeRedeemCode, tq: q}, nil
|
|
case *ent.SettingQuery:
|
|
return &query[*ent.SettingQuery, predicate.Setting, setting.OrderOption]{typ: ent.TypeSetting, tq: q}, nil
|
|
case *ent.UsageCleanupTaskQuery:
|
|
return &query[*ent.UsageCleanupTaskQuery, predicate.UsageCleanupTask, usagecleanuptask.OrderOption]{typ: ent.TypeUsageCleanupTask, tq: q}, nil
|
|
case *ent.UsageLogQuery:
|
|
return &query[*ent.UsageLogQuery, predicate.UsageLog, usagelog.OrderOption]{typ: ent.TypeUsageLog, tq: q}, nil
|
|
case *ent.UserQuery:
|
|
return &query[*ent.UserQuery, predicate.User, user.OrderOption]{typ: ent.TypeUser, tq: q}, nil
|
|
case *ent.UserAllowedGroupQuery:
|
|
return &query[*ent.UserAllowedGroupQuery, predicate.UserAllowedGroup, userallowedgroup.OrderOption]{typ: ent.TypeUserAllowedGroup, tq: q}, nil
|
|
case *ent.UserAttributeDefinitionQuery:
|
|
return &query[*ent.UserAttributeDefinitionQuery, predicate.UserAttributeDefinition, userattributedefinition.OrderOption]{typ: ent.TypeUserAttributeDefinition, tq: q}, nil
|
|
case *ent.UserAttributeValueQuery:
|
|
return &query[*ent.UserAttributeValueQuery, predicate.UserAttributeValue, userattributevalue.OrderOption]{typ: ent.TypeUserAttributeValue, tq: q}, nil
|
|
case *ent.UserSubscriptionQuery:
|
|
return &query[*ent.UserSubscriptionQuery, predicate.UserSubscription, usersubscription.OrderOption]{typ: ent.TypeUserSubscription, tq: q}, nil
|
|
default:
|
|
return nil, fmt.Errorf("unknown query type %T", q)
|
|
}
|
|
}
|
|
|
|
type query[T any, P ~func(*sql.Selector), R ~func(*sql.Selector)] struct {
|
|
typ string
|
|
tq interface {
|
|
Limit(int) T
|
|
Offset(int) T
|
|
Unique(bool) T
|
|
Order(...R) T
|
|
Where(...P) T
|
|
}
|
|
}
|
|
|
|
func (q query[T, P, R]) Type() string {
|
|
return q.typ
|
|
}
|
|
|
|
func (q query[T, P, R]) Limit(limit int) {
|
|
q.tq.Limit(limit)
|
|
}
|
|
|
|
func (q query[T, P, R]) Offset(offset int) {
|
|
q.tq.Offset(offset)
|
|
}
|
|
|
|
func (q query[T, P, R]) Unique(unique bool) {
|
|
q.tq.Unique(unique)
|
|
}
|
|
|
|
func (q query[T, P, R]) Order(orders ...func(*sql.Selector)) {
|
|
rs := make([]R, len(orders))
|
|
for i := range orders {
|
|
rs[i] = orders[i]
|
|
}
|
|
q.tq.Order(rs...)
|
|
}
|
|
|
|
func (q query[T, P, R]) WhereP(ps ...func(*sql.Selector)) {
|
|
p := make([]P, len(ps))
|
|
for i := range ps {
|
|
p[i] = ps[i]
|
|
}
|
|
q.tq.Where(p...)
|
|
}
|