1
0

auth_tokens.gen.go 8.6 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350
  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/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=strftime('%Y-%m-%d %H:%M:%S','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=strftime('%Y-%m-%d %H:%M:%S','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) Order(conds ...field.Expr) *authTokenDo {
  123. return a.withDO(a.DO.Order(conds...))
  124. }
  125. func (a authTokenDo) Distinct(cols ...field.Expr) *authTokenDo {
  126. return a.withDO(a.DO.Distinct(cols...))
  127. }
  128. func (a authTokenDo) Omit(cols ...field.Expr) *authTokenDo {
  129. return a.withDO(a.DO.Omit(cols...))
  130. }
  131. func (a authTokenDo) Join(table schema.Tabler, on ...field.Expr) *authTokenDo {
  132. return a.withDO(a.DO.Join(table, on...))
  133. }
  134. func (a authTokenDo) LeftJoin(table schema.Tabler, on ...field.Expr) *authTokenDo {
  135. return a.withDO(a.DO.LeftJoin(table, on...))
  136. }
  137. func (a authTokenDo) RightJoin(table schema.Tabler, on ...field.Expr) *authTokenDo {
  138. return a.withDO(a.DO.RightJoin(table, on...))
  139. }
  140. func (a authTokenDo) Group(cols ...field.Expr) *authTokenDo {
  141. return a.withDO(a.DO.Group(cols...))
  142. }
  143. func (a authTokenDo) Having(conds ...gen.Condition) *authTokenDo {
  144. return a.withDO(a.DO.Having(conds...))
  145. }
  146. func (a authTokenDo) Limit(limit int) *authTokenDo {
  147. return a.withDO(a.DO.Limit(limit))
  148. }
  149. func (a authTokenDo) Offset(offset int) *authTokenDo {
  150. return a.withDO(a.DO.Offset(offset))
  151. }
  152. func (a authTokenDo) Scopes(funcs ...func(gen.Dao) gen.Dao) *authTokenDo {
  153. return a.withDO(a.DO.Scopes(funcs...))
  154. }
  155. func (a authTokenDo) Unscoped() *authTokenDo {
  156. return a.withDO(a.DO.Unscoped())
  157. }
  158. func (a authTokenDo) Create(values ...*model.AuthToken) error {
  159. if len(values) == 0 {
  160. return nil
  161. }
  162. return a.DO.Create(values)
  163. }
  164. func (a authTokenDo) CreateInBatches(values []*model.AuthToken, batchSize int) error {
  165. return a.DO.CreateInBatches(values, batchSize)
  166. }
  167. // Save : !!! underlying implementation is different with GORM
  168. // The method is equivalent to executing the statement: db.Clauses(clause.OnConflict{UpdateAll: true}).Create(values)
  169. func (a authTokenDo) Save(values ...*model.AuthToken) error {
  170. if len(values) == 0 {
  171. return nil
  172. }
  173. return a.DO.Save(values)
  174. }
  175. func (a authTokenDo) First() (*model.AuthToken, error) {
  176. if result, err := a.DO.First(); err != nil {
  177. return nil, err
  178. } else {
  179. return result.(*model.AuthToken), nil
  180. }
  181. }
  182. func (a authTokenDo) Take() (*model.AuthToken, error) {
  183. if result, err := a.DO.Take(); err != nil {
  184. return nil, err
  185. } else {
  186. return result.(*model.AuthToken), nil
  187. }
  188. }
  189. func (a authTokenDo) Last() (*model.AuthToken, error) {
  190. if result, err := a.DO.Last(); err != nil {
  191. return nil, err
  192. } else {
  193. return result.(*model.AuthToken), nil
  194. }
  195. }
  196. func (a authTokenDo) Find() ([]*model.AuthToken, error) {
  197. result, err := a.DO.Find()
  198. return result.([]*model.AuthToken), err
  199. }
  200. func (a authTokenDo) FindInBatch(batchSize int, fc func(tx gen.Dao, batch int) error) (results []*model.AuthToken, err error) {
  201. buf := make([]*model.AuthToken, 0, batchSize)
  202. err = a.DO.FindInBatches(&buf, batchSize, func(tx gen.Dao, batch int) error {
  203. defer func() { results = append(results, buf...) }()
  204. return fc(tx, batch)
  205. })
  206. return results, err
  207. }
  208. func (a authTokenDo) FindInBatches(result *[]*model.AuthToken, batchSize int, fc func(tx gen.Dao, batch int) error) error {
  209. return a.DO.FindInBatches(result, batchSize, fc)
  210. }
  211. func (a authTokenDo) Attrs(attrs ...field.AssignExpr) *authTokenDo {
  212. return a.withDO(a.DO.Attrs(attrs...))
  213. }
  214. func (a authTokenDo) Assign(attrs ...field.AssignExpr) *authTokenDo {
  215. return a.withDO(a.DO.Assign(attrs...))
  216. }
  217. func (a authTokenDo) Joins(fields ...field.RelationField) *authTokenDo {
  218. for _, _f := range fields {
  219. a = *a.withDO(a.DO.Joins(_f))
  220. }
  221. return &a
  222. }
  223. func (a authTokenDo) Preload(fields ...field.RelationField) *authTokenDo {
  224. for _, _f := range fields {
  225. a = *a.withDO(a.DO.Preload(_f))
  226. }
  227. return &a
  228. }
  229. func (a authTokenDo) FirstOrInit() (*model.AuthToken, error) {
  230. if result, err := a.DO.FirstOrInit(); err != nil {
  231. return nil, err
  232. } else {
  233. return result.(*model.AuthToken), nil
  234. }
  235. }
  236. func (a authTokenDo) FirstOrCreate() (*model.AuthToken, error) {
  237. if result, err := a.DO.FirstOrCreate(); err != nil {
  238. return nil, err
  239. } else {
  240. return result.(*model.AuthToken), nil
  241. }
  242. }
  243. func (a authTokenDo) FindByPage(offset int, limit int) (result []*model.AuthToken, count int64, err error) {
  244. result, err = a.Offset(offset).Limit(limit).Find()
  245. if err != nil {
  246. return
  247. }
  248. if size := len(result); 0 < limit && 0 < size && size < limit {
  249. count = int64(size + offset)
  250. return
  251. }
  252. count, err = a.Offset(-1).Limit(-1).Count()
  253. return
  254. }
  255. func (a authTokenDo) ScanByPage(result interface{}, offset int, limit int) (count int64, err error) {
  256. count, err = a.Count()
  257. if err != nil {
  258. return
  259. }
  260. err = a.Offset(offset).Limit(limit).Scan(result)
  261. return
  262. }
  263. func (a authTokenDo) Scan(result interface{}) (err error) {
  264. return a.DO.Scan(result)
  265. }
  266. func (a authTokenDo) Delete(models ...*model.AuthToken) (result gen.ResultInfo, err error) {
  267. return a.DO.Delete(models)
  268. }
  269. func (a *authTokenDo) withDO(do gen.Dao) *authTokenDo {
  270. a.DO = *do.(*gen.DO)
  271. return a
  272. }