auth_tokens.gen.go 8.7 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354
  1. // Code generated by gorm.io/gen. DO NOT EDIT.
  2. // Code generated by gorm.io/gen. DO NOT EDIT.
  3. // Code generated by gorm.io/gen. DO NOT EDIT.
  4. package query
  5. import (
  6. "context"
  7. "strings"
  8. "gorm.io/gorm"
  9. "gorm.io/gorm/clause"
  10. "gorm.io/gorm/schema"
  11. "gorm.io/gen"
  12. "gorm.io/gen/field"
  13. "gorm.io/plugin/dbresolver"
  14. "github.com/0xJacky/Nginx-UI/server/model"
  15. )
  16. func newAuthToken(db *gorm.DB, opts ...gen.DOOption) authToken {
  17. _authToken := authToken{}
  18. _authToken.authTokenDo.UseDB(db, opts...)
  19. _authToken.authTokenDo.UseModel(&model.AuthToken{})
  20. tableName := _authToken.authTokenDo.TableName()
  21. _authToken.ALL = field.NewAsterisk(tableName)
  22. _authToken.Token = field.NewString(tableName, "token")
  23. _authToken.fillFieldMap()
  24. return _authToken
  25. }
  26. type authToken struct {
  27. authTokenDo
  28. ALL field.Asterisk
  29. Token field.String
  30. fieldMap map[string]field.Expr
  31. }
  32. func (a authToken) Table(newTableName string) *authToken {
  33. a.authTokenDo.UseTable(newTableName)
  34. return a.updateTableName(newTableName)
  35. }
  36. func (a authToken) As(alias string) *authToken {
  37. a.authTokenDo.DO = *(a.authTokenDo.As(alias).(*gen.DO))
  38. return a.updateTableName(alias)
  39. }
  40. func (a *authToken) updateTableName(table string) *authToken {
  41. a.ALL = field.NewAsterisk(table)
  42. a.Token = field.NewString(table, "token")
  43. a.fillFieldMap()
  44. return a
  45. }
  46. func (a *authToken) GetFieldByName(fieldName string) (field.OrderExpr, bool) {
  47. _f, ok := a.fieldMap[fieldName]
  48. if !ok || _f == nil {
  49. return nil, false
  50. }
  51. _oe, ok := _f.(field.OrderExpr)
  52. return _oe, ok
  53. }
  54. func (a *authToken) fillFieldMap() {
  55. a.fieldMap = make(map[string]field.Expr, 1)
  56. a.fieldMap["token"] = a.Token
  57. }
  58. func (a authToken) clone(db *gorm.DB) authToken {
  59. a.authTokenDo.ReplaceConnPool(db.Statement.ConnPool)
  60. return a
  61. }
  62. func (a authToken) replaceDB(db *gorm.DB) authToken {
  63. a.authTokenDo.ReplaceDB(db)
  64. return a
  65. }
  66. type authTokenDo struct{ gen.DO }
  67. // FirstByID Where("id=@id")
  68. func (a authTokenDo) FirstByID(id int) (result *model.AuthToken, err error) {
  69. var params []interface{}
  70. var generateSQL strings.Builder
  71. params = append(params, id)
  72. generateSQL.WriteString("id=? ")
  73. var executeSQL *gorm.DB
  74. executeSQL = a.UnderlyingDB().Where(generateSQL.String(), params...).Take(&result) // ignore_security_alert
  75. err = executeSQL.Error
  76. return
  77. }
  78. // DeleteByID update @@table set deleted_at=NOW() where id=@id
  79. func (a authTokenDo) DeleteByID(id int) (err error) {
  80. var params []interface{}
  81. var generateSQL strings.Builder
  82. params = append(params, id)
  83. generateSQL.WriteString("update auth_tokens set deleted_at=NOW() where id=? ")
  84. var executeSQL *gorm.DB
  85. executeSQL = a.UnderlyingDB().Exec(generateSQL.String(), params...) // ignore_security_alert
  86. err = executeSQL.Error
  87. return
  88. }
  89. func (a authTokenDo) Debug() *authTokenDo {
  90. return a.withDO(a.DO.Debug())
  91. }
  92. func (a authTokenDo) WithContext(ctx context.Context) *authTokenDo {
  93. return a.withDO(a.DO.WithContext(ctx))
  94. }
  95. func (a authTokenDo) ReadDB() *authTokenDo {
  96. return a.Clauses(dbresolver.Read)
  97. }
  98. func (a authTokenDo) WriteDB() *authTokenDo {
  99. return a.Clauses(dbresolver.Write)
  100. }
  101. func (a authTokenDo) Session(config *gorm.Session) *authTokenDo {
  102. return a.withDO(a.DO.Session(config))
  103. }
  104. func (a authTokenDo) Clauses(conds ...clause.Expression) *authTokenDo {
  105. return a.withDO(a.DO.Clauses(conds...))
  106. }
  107. func (a authTokenDo) Returning(value interface{}, columns ...string) *authTokenDo {
  108. return a.withDO(a.DO.Returning(value, columns...))
  109. }
  110. func (a authTokenDo) Not(conds ...gen.Condition) *authTokenDo {
  111. return a.withDO(a.DO.Not(conds...))
  112. }
  113. func (a authTokenDo) Or(conds ...gen.Condition) *authTokenDo {
  114. return a.withDO(a.DO.Or(conds...))
  115. }
  116. func (a authTokenDo) Select(conds ...field.Expr) *authTokenDo {
  117. return a.withDO(a.DO.Select(conds...))
  118. }
  119. func (a authTokenDo) Where(conds ...gen.Condition) *authTokenDo {
  120. return a.withDO(a.DO.Where(conds...))
  121. }
  122. func (a authTokenDo) Exists(subquery interface{ UnderlyingDB() *gorm.DB }) *authTokenDo {
  123. return a.Where(field.CompareSubQuery(field.ExistsOp, nil, subquery.UnderlyingDB()))
  124. }
  125. func (a authTokenDo) Order(conds ...field.Expr) *authTokenDo {
  126. return a.withDO(a.DO.Order(conds...))
  127. }
  128. func (a authTokenDo) Distinct(cols ...field.Expr) *authTokenDo {
  129. return a.withDO(a.DO.Distinct(cols...))
  130. }
  131. func (a authTokenDo) Omit(cols ...field.Expr) *authTokenDo {
  132. return a.withDO(a.DO.Omit(cols...))
  133. }
  134. func (a authTokenDo) Join(table schema.Tabler, on ...field.Expr) *authTokenDo {
  135. return a.withDO(a.DO.Join(table, on...))
  136. }
  137. func (a authTokenDo) LeftJoin(table schema.Tabler, on ...field.Expr) *authTokenDo {
  138. return a.withDO(a.DO.LeftJoin(table, on...))
  139. }
  140. func (a authTokenDo) RightJoin(table schema.Tabler, on ...field.Expr) *authTokenDo {
  141. return a.withDO(a.DO.RightJoin(table, on...))
  142. }
  143. func (a authTokenDo) Group(cols ...field.Expr) *authTokenDo {
  144. return a.withDO(a.DO.Group(cols...))
  145. }
  146. func (a authTokenDo) Having(conds ...gen.Condition) *authTokenDo {
  147. return a.withDO(a.DO.Having(conds...))
  148. }
  149. func (a authTokenDo) Limit(limit int) *authTokenDo {
  150. return a.withDO(a.DO.Limit(limit))
  151. }
  152. func (a authTokenDo) Offset(offset int) *authTokenDo {
  153. return a.withDO(a.DO.Offset(offset))
  154. }
  155. func (a authTokenDo) Scopes(funcs ...func(gen.Dao) gen.Dao) *authTokenDo {
  156. return a.withDO(a.DO.Scopes(funcs...))
  157. }
  158. func (a authTokenDo) Unscoped() *authTokenDo {
  159. return a.withDO(a.DO.Unscoped())
  160. }
  161. func (a authTokenDo) Create(values ...*model.AuthToken) error {
  162. if len(values) == 0 {
  163. return nil
  164. }
  165. return a.DO.Create(values)
  166. }
  167. func (a authTokenDo) CreateInBatches(values []*model.AuthToken, batchSize int) error {
  168. return a.DO.CreateInBatches(values, batchSize)
  169. }
  170. // Save : !!! underlying implementation is different with GORM
  171. // The method is equivalent to executing the statement: db.Clauses(clause.OnConflict{UpdateAll: true}).Create(values)
  172. func (a authTokenDo) Save(values ...*model.AuthToken) error {
  173. if len(values) == 0 {
  174. return nil
  175. }
  176. return a.DO.Save(values)
  177. }
  178. func (a authTokenDo) First() (*model.AuthToken, error) {
  179. if result, err := a.DO.First(); err != nil {
  180. return nil, err
  181. } else {
  182. return result.(*model.AuthToken), nil
  183. }
  184. }
  185. func (a authTokenDo) Take() (*model.AuthToken, error) {
  186. if result, err := a.DO.Take(); err != nil {
  187. return nil, err
  188. } else {
  189. return result.(*model.AuthToken), nil
  190. }
  191. }
  192. func (a authTokenDo) Last() (*model.AuthToken, error) {
  193. if result, err := a.DO.Last(); err != nil {
  194. return nil, err
  195. } else {
  196. return result.(*model.AuthToken), nil
  197. }
  198. }
  199. func (a authTokenDo) Find() ([]*model.AuthToken, error) {
  200. result, err := a.DO.Find()
  201. return result.([]*model.AuthToken), err
  202. }
  203. func (a authTokenDo) FindInBatch(batchSize int, fc func(tx gen.Dao, batch int) error) (results []*model.AuthToken, err error) {
  204. buf := make([]*model.AuthToken, 0, batchSize)
  205. err = a.DO.FindInBatches(&buf, batchSize, func(tx gen.Dao, batch int) error {
  206. defer func() { results = append(results, buf...) }()
  207. return fc(tx, batch)
  208. })
  209. return results, err
  210. }
  211. func (a authTokenDo) FindInBatches(result *[]*model.AuthToken, batchSize int, fc func(tx gen.Dao, batch int) error) error {
  212. return a.DO.FindInBatches(result, batchSize, fc)
  213. }
  214. func (a authTokenDo) Attrs(attrs ...field.AssignExpr) *authTokenDo {
  215. return a.withDO(a.DO.Attrs(attrs...))
  216. }
  217. func (a authTokenDo) Assign(attrs ...field.AssignExpr) *authTokenDo {
  218. return a.withDO(a.DO.Assign(attrs...))
  219. }
  220. func (a authTokenDo) Joins(fields ...field.RelationField) *authTokenDo {
  221. for _, _f := range fields {
  222. a = *a.withDO(a.DO.Joins(_f))
  223. }
  224. return &a
  225. }
  226. func (a authTokenDo) Preload(fields ...field.RelationField) *authTokenDo {
  227. for _, _f := range fields {
  228. a = *a.withDO(a.DO.Preload(_f))
  229. }
  230. return &a
  231. }
  232. func (a authTokenDo) FirstOrInit() (*model.AuthToken, error) {
  233. if result, err := a.DO.FirstOrInit(); err != nil {
  234. return nil, err
  235. } else {
  236. return result.(*model.AuthToken), nil
  237. }
  238. }
  239. func (a authTokenDo) FirstOrCreate() (*model.AuthToken, error) {
  240. if result, err := a.DO.FirstOrCreate(); err != nil {
  241. return nil, err
  242. } else {
  243. return result.(*model.AuthToken), nil
  244. }
  245. }
  246. func (a authTokenDo) FindByPage(offset int, limit int) (result []*model.AuthToken, count int64, err error) {
  247. result, err = a.Offset(offset).Limit(limit).Find()
  248. if err != nil {
  249. return
  250. }
  251. if size := len(result); 0 < limit && 0 < size && size < limit {
  252. count = int64(size + offset)
  253. return
  254. }
  255. count, err = a.Offset(-1).Limit(-1).Count()
  256. return
  257. }
  258. func (a authTokenDo) ScanByPage(result interface{}, offset int, limit int) (count int64, err error) {
  259. count, err = a.Count()
  260. if err != nil {
  261. return
  262. }
  263. err = a.Offset(offset).Limit(limit).Scan(result)
  264. return
  265. }
  266. func (a authTokenDo) Scan(result interface{}) (err error) {
  267. return a.DO.Scan(result)
  268. }
  269. func (a authTokenDo) Delete(models ...*model.AuthToken) (result gen.ResultInfo, err error) {
  270. return a.DO.Delete(models)
  271. }
  272. func (a *authTokenDo) withDO(do gen.Dao) *authTokenDo {
  273. a.DO = *do.(*gen.DO)
  274. return a
  275. }