// Code generated by gorm.io/gen. DO NOT EDIT.
// Code generated by gorm.io/gen. DO NOT EDIT.
// Code generated by gorm.io/gen. DO NOT EDIT.

package query

import (
	"context"
	"database/sql"

	"gorm.io/gorm"

	"gorm.io/gen"

	"gorm.io/plugin/dbresolver"
)

var (
	Q                     = new(Query)
	ScaAuthMenu           *scaAuthMenu
	ScaAuthPermissionRule *scaAuthPermissionRule
	ScaAuthRole           *scaAuthRole
	ScaAuthUser           *scaAuthUser
	ScaAuthUserDevice     *scaAuthUserDevice
	ScaAuthUserSocial     *scaAuthUserSocial
	ScaCommentLike        *scaCommentLike
	ScaCommentReply       *scaCommentReply
	ScaMessageReport      *scaMessageReport
	ScaStorageAlbum       *scaStorageAlbum
	ScaStorageConfig      *scaStorageConfig
	ScaStorageExtra       *scaStorageExtra
	ScaStorageImgBed      *scaStorageImgBed
	ScaStorageInfo        *scaStorageInfo
	ScaStorageLocation    *scaStorageLocation
	ScaStorageShare       *scaStorageShare
	ScaStorageShareVisit  *scaStorageShareVisit
	ScaStorageThumb       *scaStorageThumb
	ScaUserFollow         *scaUserFollow
	ScaUserLevel          *scaUserLevel
	ScaUserMessage        *scaUserMessage
)

func SetDefault(db *gorm.DB, opts ...gen.DOOption) {
	*Q = *Use(db, opts...)
	ScaAuthMenu = &Q.ScaAuthMenu
	ScaAuthPermissionRule = &Q.ScaAuthPermissionRule
	ScaAuthRole = &Q.ScaAuthRole
	ScaAuthUser = &Q.ScaAuthUser
	ScaAuthUserDevice = &Q.ScaAuthUserDevice
	ScaAuthUserSocial = &Q.ScaAuthUserSocial
	ScaCommentLike = &Q.ScaCommentLike
	ScaCommentReply = &Q.ScaCommentReply
	ScaMessageReport = &Q.ScaMessageReport
	ScaStorageAlbum = &Q.ScaStorageAlbum
	ScaStorageConfig = &Q.ScaStorageConfig
	ScaStorageExtra = &Q.ScaStorageExtra
	ScaStorageImgBed = &Q.ScaStorageImgBed
	ScaStorageInfo = &Q.ScaStorageInfo
	ScaStorageLocation = &Q.ScaStorageLocation
	ScaStorageShare = &Q.ScaStorageShare
	ScaStorageShareVisit = &Q.ScaStorageShareVisit
	ScaStorageThumb = &Q.ScaStorageThumb
	ScaUserFollow = &Q.ScaUserFollow
	ScaUserLevel = &Q.ScaUserLevel
	ScaUserMessage = &Q.ScaUserMessage
}

func Use(db *gorm.DB, opts ...gen.DOOption) *Query {
	return &Query{
		db:                    db,
		ScaAuthMenu:           newScaAuthMenu(db, opts...),
		ScaAuthPermissionRule: newScaAuthPermissionRule(db, opts...),
		ScaAuthRole:           newScaAuthRole(db, opts...),
		ScaAuthUser:           newScaAuthUser(db, opts...),
		ScaAuthUserDevice:     newScaAuthUserDevice(db, opts...),
		ScaAuthUserSocial:     newScaAuthUserSocial(db, opts...),
		ScaCommentLike:        newScaCommentLike(db, opts...),
		ScaCommentReply:       newScaCommentReply(db, opts...),
		ScaMessageReport:      newScaMessageReport(db, opts...),
		ScaStorageAlbum:       newScaStorageAlbum(db, opts...),
		ScaStorageConfig:      newScaStorageConfig(db, opts...),
		ScaStorageExtra:       newScaStorageExtra(db, opts...),
		ScaStorageImgBed:      newScaStorageImgBed(db, opts...),
		ScaStorageInfo:        newScaStorageInfo(db, opts...),
		ScaStorageLocation:    newScaStorageLocation(db, opts...),
		ScaStorageShare:       newScaStorageShare(db, opts...),
		ScaStorageShareVisit:  newScaStorageShareVisit(db, opts...),
		ScaStorageThumb:       newScaStorageThumb(db, opts...),
		ScaUserFollow:         newScaUserFollow(db, opts...),
		ScaUserLevel:          newScaUserLevel(db, opts...),
		ScaUserMessage:        newScaUserMessage(db, opts...),
	}
}

type Query struct {
	db *gorm.DB

	ScaAuthMenu           scaAuthMenu
	ScaAuthPermissionRule scaAuthPermissionRule
	ScaAuthRole           scaAuthRole
	ScaAuthUser           scaAuthUser
	ScaAuthUserDevice     scaAuthUserDevice
	ScaAuthUserSocial     scaAuthUserSocial
	ScaCommentLike        scaCommentLike
	ScaCommentReply       scaCommentReply
	ScaMessageReport      scaMessageReport
	ScaStorageAlbum       scaStorageAlbum
	ScaStorageConfig      scaStorageConfig
	ScaStorageExtra       scaStorageExtra
	ScaStorageImgBed      scaStorageImgBed
	ScaStorageInfo        scaStorageInfo
	ScaStorageLocation    scaStorageLocation
	ScaStorageShare       scaStorageShare
	ScaStorageShareVisit  scaStorageShareVisit
	ScaStorageThumb       scaStorageThumb
	ScaUserFollow         scaUserFollow
	ScaUserLevel          scaUserLevel
	ScaUserMessage        scaUserMessage
}

func (q *Query) Available() bool { return q.db != nil }

func (q *Query) clone(db *gorm.DB) *Query {
	return &Query{
		db:                    db,
		ScaAuthMenu:           q.ScaAuthMenu.clone(db),
		ScaAuthPermissionRule: q.ScaAuthPermissionRule.clone(db),
		ScaAuthRole:           q.ScaAuthRole.clone(db),
		ScaAuthUser:           q.ScaAuthUser.clone(db),
		ScaAuthUserDevice:     q.ScaAuthUserDevice.clone(db),
		ScaAuthUserSocial:     q.ScaAuthUserSocial.clone(db),
		ScaCommentLike:        q.ScaCommentLike.clone(db),
		ScaCommentReply:       q.ScaCommentReply.clone(db),
		ScaMessageReport:      q.ScaMessageReport.clone(db),
		ScaStorageAlbum:       q.ScaStorageAlbum.clone(db),
		ScaStorageConfig:      q.ScaStorageConfig.clone(db),
		ScaStorageExtra:       q.ScaStorageExtra.clone(db),
		ScaStorageImgBed:      q.ScaStorageImgBed.clone(db),
		ScaStorageInfo:        q.ScaStorageInfo.clone(db),
		ScaStorageLocation:    q.ScaStorageLocation.clone(db),
		ScaStorageShare:       q.ScaStorageShare.clone(db),
		ScaStorageShareVisit:  q.ScaStorageShareVisit.clone(db),
		ScaStorageThumb:       q.ScaStorageThumb.clone(db),
		ScaUserFollow:         q.ScaUserFollow.clone(db),
		ScaUserLevel:          q.ScaUserLevel.clone(db),
		ScaUserMessage:        q.ScaUserMessage.clone(db),
	}
}

func (q *Query) ReadDB() *Query {
	return q.ReplaceDB(q.db.Clauses(dbresolver.Read))
}

func (q *Query) WriteDB() *Query {
	return q.ReplaceDB(q.db.Clauses(dbresolver.Write))
}

func (q *Query) ReplaceDB(db *gorm.DB) *Query {
	return &Query{
		db:                    db,
		ScaAuthMenu:           q.ScaAuthMenu.replaceDB(db),
		ScaAuthPermissionRule: q.ScaAuthPermissionRule.replaceDB(db),
		ScaAuthRole:           q.ScaAuthRole.replaceDB(db),
		ScaAuthUser:           q.ScaAuthUser.replaceDB(db),
		ScaAuthUserDevice:     q.ScaAuthUserDevice.replaceDB(db),
		ScaAuthUserSocial:     q.ScaAuthUserSocial.replaceDB(db),
		ScaCommentLike:        q.ScaCommentLike.replaceDB(db),
		ScaCommentReply:       q.ScaCommentReply.replaceDB(db),
		ScaMessageReport:      q.ScaMessageReport.replaceDB(db),
		ScaStorageAlbum:       q.ScaStorageAlbum.replaceDB(db),
		ScaStorageConfig:      q.ScaStorageConfig.replaceDB(db),
		ScaStorageExtra:       q.ScaStorageExtra.replaceDB(db),
		ScaStorageImgBed:      q.ScaStorageImgBed.replaceDB(db),
		ScaStorageInfo:        q.ScaStorageInfo.replaceDB(db),
		ScaStorageLocation:    q.ScaStorageLocation.replaceDB(db),
		ScaStorageShare:       q.ScaStorageShare.replaceDB(db),
		ScaStorageShareVisit:  q.ScaStorageShareVisit.replaceDB(db),
		ScaStorageThumb:       q.ScaStorageThumb.replaceDB(db),
		ScaUserFollow:         q.ScaUserFollow.replaceDB(db),
		ScaUserLevel:          q.ScaUserLevel.replaceDB(db),
		ScaUserMessage:        q.ScaUserMessage.replaceDB(db),
	}
}

type queryCtx struct {
	ScaAuthMenu           IScaAuthMenuDo
	ScaAuthPermissionRule IScaAuthPermissionRuleDo
	ScaAuthRole           IScaAuthRoleDo
	ScaAuthUser           IScaAuthUserDo
	ScaAuthUserDevice     IScaAuthUserDeviceDo
	ScaAuthUserSocial     IScaAuthUserSocialDo
	ScaCommentLike        IScaCommentLikeDo
	ScaCommentReply       IScaCommentReplyDo
	ScaMessageReport      IScaMessageReportDo
	ScaStorageAlbum       IScaStorageAlbumDo
	ScaStorageConfig      IScaStorageConfigDo
	ScaStorageExtra       IScaStorageExtraDo
	ScaStorageImgBed      IScaStorageImgBedDo
	ScaStorageInfo        IScaStorageInfoDo
	ScaStorageLocation    IScaStorageLocationDo
	ScaStorageShare       IScaStorageShareDo
	ScaStorageShareVisit  IScaStorageShareVisitDo
	ScaStorageThumb       IScaStorageThumbDo
	ScaUserFollow         IScaUserFollowDo
	ScaUserLevel          IScaUserLevelDo
	ScaUserMessage        IScaUserMessageDo
}

func (q *Query) WithContext(ctx context.Context) *queryCtx {
	return &queryCtx{
		ScaAuthMenu:           q.ScaAuthMenu.WithContext(ctx),
		ScaAuthPermissionRule: q.ScaAuthPermissionRule.WithContext(ctx),
		ScaAuthRole:           q.ScaAuthRole.WithContext(ctx),
		ScaAuthUser:           q.ScaAuthUser.WithContext(ctx),
		ScaAuthUserDevice:     q.ScaAuthUserDevice.WithContext(ctx),
		ScaAuthUserSocial:     q.ScaAuthUserSocial.WithContext(ctx),
		ScaCommentLike:        q.ScaCommentLike.WithContext(ctx),
		ScaCommentReply:       q.ScaCommentReply.WithContext(ctx),
		ScaMessageReport:      q.ScaMessageReport.WithContext(ctx),
		ScaStorageAlbum:       q.ScaStorageAlbum.WithContext(ctx),
		ScaStorageConfig:      q.ScaStorageConfig.WithContext(ctx),
		ScaStorageExtra:       q.ScaStorageExtra.WithContext(ctx),
		ScaStorageImgBed:      q.ScaStorageImgBed.WithContext(ctx),
		ScaStorageInfo:        q.ScaStorageInfo.WithContext(ctx),
		ScaStorageLocation:    q.ScaStorageLocation.WithContext(ctx),
		ScaStorageShare:       q.ScaStorageShare.WithContext(ctx),
		ScaStorageShareVisit:  q.ScaStorageShareVisit.WithContext(ctx),
		ScaStorageThumb:       q.ScaStorageThumb.WithContext(ctx),
		ScaUserFollow:         q.ScaUserFollow.WithContext(ctx),
		ScaUserLevel:          q.ScaUserLevel.WithContext(ctx),
		ScaUserMessage:        q.ScaUserMessage.WithContext(ctx),
	}
}

func (q *Query) Transaction(fc func(tx *Query) error, opts ...*sql.TxOptions) error {
	return q.db.Transaction(func(tx *gorm.DB) error { return fc(q.clone(tx)) }, opts...)
}

func (q *Query) Begin(opts ...*sql.TxOptions) *QueryTx {
	tx := q.db.Begin(opts...)
	return &QueryTx{Query: q.clone(tx), Error: tx.Error}
}

type QueryTx struct {
	*Query
	Error error
}

func (q *QueryTx) Commit() error {
	return q.db.Commit().Error
}

func (q *QueryTx) Rollback() error {
	return q.db.Rollback().Error
}

func (q *QueryTx) SavePoint(name string) error {
	return q.db.SavePoint(name).Error
}

func (q *QueryTx) RollbackTo(name string) error {
	return q.db.RollbackTo(name).Error
}