This commit is contained in:
@@ -17,7 +17,7 @@ import (
|
||||
|
||||
var (
|
||||
Q = new(Query)
|
||||
MediaQuery *mediaQuery
|
||||
MediumQuery *mediumQuery
|
||||
OrderQuery *orderQuery
|
||||
PostQuery *postQuery
|
||||
UserQuery *userQuery
|
||||
@@ -26,7 +26,7 @@ var (
|
||||
|
||||
func SetDefault(db *gorm.DB, opts ...gen.DOOption) {
|
||||
*Q = *Use(db, opts...)
|
||||
MediaQuery = &Q.Media
|
||||
MediumQuery = &Q.Medium
|
||||
OrderQuery = &Q.Order
|
||||
PostQuery = &Q.Post
|
||||
UserQuery = &Q.User
|
||||
@@ -36,7 +36,7 @@ func SetDefault(db *gorm.DB, opts ...gen.DOOption) {
|
||||
func Use(db *gorm.DB, opts ...gen.DOOption) *Query {
|
||||
return &Query{
|
||||
db: db,
|
||||
Media: newMedia(db, opts...),
|
||||
Medium: newMedium(db, opts...),
|
||||
Order: newOrder(db, opts...),
|
||||
Post: newPost(db, opts...),
|
||||
User: newUser(db, opts...),
|
||||
@@ -47,7 +47,7 @@ func Use(db *gorm.DB, opts ...gen.DOOption) *Query {
|
||||
type Query struct {
|
||||
db *gorm.DB
|
||||
|
||||
Media mediaQuery
|
||||
Medium mediumQuery
|
||||
Order orderQuery
|
||||
Post postQuery
|
||||
User userQuery
|
||||
@@ -59,7 +59,7 @@ func (q *Query) Available() bool { return q.db != nil }
|
||||
func (q *Query) clone(db *gorm.DB) *Query {
|
||||
return &Query{
|
||||
db: db,
|
||||
Media: q.Media.clone(db),
|
||||
Medium: q.Medium.clone(db),
|
||||
Order: q.Order.clone(db),
|
||||
Post: q.Post.clone(db),
|
||||
User: q.User.clone(db),
|
||||
@@ -78,7 +78,7 @@ func (q *Query) WriteDB() *Query {
|
||||
func (q *Query) ReplaceDB(db *gorm.DB) *Query {
|
||||
return &Query{
|
||||
db: db,
|
||||
Media: q.Media.replaceDB(db),
|
||||
Medium: q.Medium.replaceDB(db),
|
||||
Order: q.Order.replaceDB(db),
|
||||
Post: q.Post.replaceDB(db),
|
||||
User: q.User.replaceDB(db),
|
||||
@@ -87,7 +87,7 @@ func (q *Query) ReplaceDB(db *gorm.DB) *Query {
|
||||
}
|
||||
|
||||
type queryCtx struct {
|
||||
Media *mediaQueryDo
|
||||
Medium *mediumQueryDo
|
||||
Order *orderQueryDo
|
||||
Post *postQueryDo
|
||||
User *userQueryDo
|
||||
@@ -96,7 +96,7 @@ type queryCtx struct {
|
||||
|
||||
func (q *Query) WithContext(ctx context.Context) *queryCtx {
|
||||
return &queryCtx{
|
||||
Media: q.Media.WithContext(ctx),
|
||||
Medium: q.Medium.WithContext(ctx),
|
||||
Order: q.Order.WithContext(ctx),
|
||||
Post: q.Post.WithContext(ctx),
|
||||
User: q.User.WithContext(ctx),
|
||||
|
||||
Reference in New Issue
Block a user