site_configs.gen.go 12 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414
  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 newSiteConfig(db *gorm.DB, opts ...gen.DOOption) siteConfig {
  17. _siteConfig := siteConfig{}
  18. _siteConfig.siteConfigDo.UseDB(db, opts...)
  19. _siteConfig.siteConfigDo.UseModel(&model.SiteConfig{})
  20. tableName := _siteConfig.siteConfigDo.TableName()
  21. _siteConfig.ALL = field.NewAsterisk(tableName)
  22. _siteConfig.ID = field.NewUint64(tableName, "id")
  23. _siteConfig.CreatedAt = field.NewTime(tableName, "created_at")
  24. _siteConfig.UpdatedAt = field.NewTime(tableName, "updated_at")
  25. _siteConfig.DeletedAt = field.NewField(tableName, "deleted_at")
  26. _siteConfig.Host = field.NewString(tableName, "host")
  27. _siteConfig.Port = field.NewInt(tableName, "port")
  28. _siteConfig.Scheme = field.NewString(tableName, "scheme")
  29. _siteConfig.DisplayURL = field.NewString(tableName, "display_url")
  30. _siteConfig.CustomOrder = field.NewInt(tableName, "custom_order")
  31. _siteConfig.HealthCheckEnabled = field.NewBool(tableName, "health_check_enabled")
  32. _siteConfig.CheckInterval = field.NewInt(tableName, "check_interval")
  33. _siteConfig.Timeout = field.NewInt(tableName, "timeout")
  34. _siteConfig.UserAgent = field.NewString(tableName, "user_agent")
  35. _siteConfig.MaxRedirects = field.NewInt(tableName, "max_redirects")
  36. _siteConfig.FollowRedirects = field.NewBool(tableName, "follow_redirects")
  37. _siteConfig.CheckFavicon = field.NewBool(tableName, "check_favicon")
  38. _siteConfig.HealthCheckConfig = field.NewField(tableName, "health_check_config")
  39. _siteConfig.fillFieldMap()
  40. return _siteConfig
  41. }
  42. type siteConfig struct {
  43. siteConfigDo
  44. ALL field.Asterisk
  45. ID field.Uint64
  46. CreatedAt field.Time
  47. UpdatedAt field.Time
  48. DeletedAt field.Field
  49. Host field.String
  50. Port field.Int
  51. Scheme field.String
  52. DisplayURL field.String
  53. CustomOrder field.Int
  54. HealthCheckEnabled field.Bool
  55. CheckInterval field.Int
  56. Timeout field.Int
  57. UserAgent field.String
  58. MaxRedirects field.Int
  59. FollowRedirects field.Bool
  60. CheckFavicon field.Bool
  61. HealthCheckConfig field.Field
  62. fieldMap map[string]field.Expr
  63. }
  64. func (s siteConfig) Table(newTableName string) *siteConfig {
  65. s.siteConfigDo.UseTable(newTableName)
  66. return s.updateTableName(newTableName)
  67. }
  68. func (s siteConfig) As(alias string) *siteConfig {
  69. s.siteConfigDo.DO = *(s.siteConfigDo.As(alias).(*gen.DO))
  70. return s.updateTableName(alias)
  71. }
  72. func (s *siteConfig) updateTableName(table string) *siteConfig {
  73. s.ALL = field.NewAsterisk(table)
  74. s.ID = field.NewUint64(table, "id")
  75. s.CreatedAt = field.NewTime(table, "created_at")
  76. s.UpdatedAt = field.NewTime(table, "updated_at")
  77. s.DeletedAt = field.NewField(table, "deleted_at")
  78. s.Host = field.NewString(table, "host")
  79. s.Port = field.NewInt(table, "port")
  80. s.Scheme = field.NewString(table, "scheme")
  81. s.DisplayURL = field.NewString(table, "display_url")
  82. s.CustomOrder = field.NewInt(table, "custom_order")
  83. s.HealthCheckEnabled = field.NewBool(table, "health_check_enabled")
  84. s.CheckInterval = field.NewInt(table, "check_interval")
  85. s.Timeout = field.NewInt(table, "timeout")
  86. s.UserAgent = field.NewString(table, "user_agent")
  87. s.MaxRedirects = field.NewInt(table, "max_redirects")
  88. s.FollowRedirects = field.NewBool(table, "follow_redirects")
  89. s.CheckFavicon = field.NewBool(table, "check_favicon")
  90. s.HealthCheckConfig = field.NewField(table, "health_check_config")
  91. s.fillFieldMap()
  92. return s
  93. }
  94. func (s *siteConfig) GetFieldByName(fieldName string) (field.OrderExpr, bool) {
  95. _f, ok := s.fieldMap[fieldName]
  96. if !ok || _f == nil {
  97. return nil, false
  98. }
  99. _oe, ok := _f.(field.OrderExpr)
  100. return _oe, ok
  101. }
  102. func (s *siteConfig) fillFieldMap() {
  103. s.fieldMap = make(map[string]field.Expr, 17)
  104. s.fieldMap["id"] = s.ID
  105. s.fieldMap["created_at"] = s.CreatedAt
  106. s.fieldMap["updated_at"] = s.UpdatedAt
  107. s.fieldMap["deleted_at"] = s.DeletedAt
  108. s.fieldMap["host"] = s.Host
  109. s.fieldMap["port"] = s.Port
  110. s.fieldMap["scheme"] = s.Scheme
  111. s.fieldMap["display_url"] = s.DisplayURL
  112. s.fieldMap["custom_order"] = s.CustomOrder
  113. s.fieldMap["health_check_enabled"] = s.HealthCheckEnabled
  114. s.fieldMap["check_interval"] = s.CheckInterval
  115. s.fieldMap["timeout"] = s.Timeout
  116. s.fieldMap["user_agent"] = s.UserAgent
  117. s.fieldMap["max_redirects"] = s.MaxRedirects
  118. s.fieldMap["follow_redirects"] = s.FollowRedirects
  119. s.fieldMap["check_favicon"] = s.CheckFavicon
  120. s.fieldMap["health_check_config"] = s.HealthCheckConfig
  121. }
  122. func (s siteConfig) clone(db *gorm.DB) siteConfig {
  123. s.siteConfigDo.ReplaceConnPool(db.Statement.ConnPool)
  124. return s
  125. }
  126. func (s siteConfig) replaceDB(db *gorm.DB) siteConfig {
  127. s.siteConfigDo.ReplaceDB(db)
  128. return s
  129. }
  130. type siteConfigDo struct{ gen.DO }
  131. // FirstByID Where("id=@id")
  132. func (s siteConfigDo) FirstByID(id uint64) (result *model.SiteConfig, err error) {
  133. var params []interface{}
  134. var generateSQL strings.Builder
  135. params = append(params, id)
  136. generateSQL.WriteString("id=? ")
  137. var executeSQL *gorm.DB
  138. executeSQL = s.UnderlyingDB().Where(generateSQL.String(), params...).Take(&result) // ignore_security_alert
  139. err = executeSQL.Error
  140. return
  141. }
  142. // DeleteByID update @@table set deleted_at=strftime('%Y-%m-%d %H:%M:%S','now') where id=@id
  143. func (s siteConfigDo) DeleteByID(id uint64) (err error) {
  144. var params []interface{}
  145. var generateSQL strings.Builder
  146. params = append(params, id)
  147. generateSQL.WriteString("update site_configs set deleted_at=strftime('%Y-%m-%d %H:%M:%S','now') where id=? ")
  148. var executeSQL *gorm.DB
  149. executeSQL = s.UnderlyingDB().Exec(generateSQL.String(), params...) // ignore_security_alert
  150. err = executeSQL.Error
  151. return
  152. }
  153. func (s siteConfigDo) Debug() *siteConfigDo {
  154. return s.withDO(s.DO.Debug())
  155. }
  156. func (s siteConfigDo) WithContext(ctx context.Context) *siteConfigDo {
  157. return s.withDO(s.DO.WithContext(ctx))
  158. }
  159. func (s siteConfigDo) ReadDB() *siteConfigDo {
  160. return s.Clauses(dbresolver.Read)
  161. }
  162. func (s siteConfigDo) WriteDB() *siteConfigDo {
  163. return s.Clauses(dbresolver.Write)
  164. }
  165. func (s siteConfigDo) Session(config *gorm.Session) *siteConfigDo {
  166. return s.withDO(s.DO.Session(config))
  167. }
  168. func (s siteConfigDo) Clauses(conds ...clause.Expression) *siteConfigDo {
  169. return s.withDO(s.DO.Clauses(conds...))
  170. }
  171. func (s siteConfigDo) Returning(value interface{}, columns ...string) *siteConfigDo {
  172. return s.withDO(s.DO.Returning(value, columns...))
  173. }
  174. func (s siteConfigDo) Not(conds ...gen.Condition) *siteConfigDo {
  175. return s.withDO(s.DO.Not(conds...))
  176. }
  177. func (s siteConfigDo) Or(conds ...gen.Condition) *siteConfigDo {
  178. return s.withDO(s.DO.Or(conds...))
  179. }
  180. func (s siteConfigDo) Select(conds ...field.Expr) *siteConfigDo {
  181. return s.withDO(s.DO.Select(conds...))
  182. }
  183. func (s siteConfigDo) Where(conds ...gen.Condition) *siteConfigDo {
  184. return s.withDO(s.DO.Where(conds...))
  185. }
  186. func (s siteConfigDo) Order(conds ...field.Expr) *siteConfigDo {
  187. return s.withDO(s.DO.Order(conds...))
  188. }
  189. func (s siteConfigDo) Distinct(cols ...field.Expr) *siteConfigDo {
  190. return s.withDO(s.DO.Distinct(cols...))
  191. }
  192. func (s siteConfigDo) Omit(cols ...field.Expr) *siteConfigDo {
  193. return s.withDO(s.DO.Omit(cols...))
  194. }
  195. func (s siteConfigDo) Join(table schema.Tabler, on ...field.Expr) *siteConfigDo {
  196. return s.withDO(s.DO.Join(table, on...))
  197. }
  198. func (s siteConfigDo) LeftJoin(table schema.Tabler, on ...field.Expr) *siteConfigDo {
  199. return s.withDO(s.DO.LeftJoin(table, on...))
  200. }
  201. func (s siteConfigDo) RightJoin(table schema.Tabler, on ...field.Expr) *siteConfigDo {
  202. return s.withDO(s.DO.RightJoin(table, on...))
  203. }
  204. func (s siteConfigDo) Group(cols ...field.Expr) *siteConfigDo {
  205. return s.withDO(s.DO.Group(cols...))
  206. }
  207. func (s siteConfigDo) Having(conds ...gen.Condition) *siteConfigDo {
  208. return s.withDO(s.DO.Having(conds...))
  209. }
  210. func (s siteConfigDo) Limit(limit int) *siteConfigDo {
  211. return s.withDO(s.DO.Limit(limit))
  212. }
  213. func (s siteConfigDo) Offset(offset int) *siteConfigDo {
  214. return s.withDO(s.DO.Offset(offset))
  215. }
  216. func (s siteConfigDo) Scopes(funcs ...func(gen.Dao) gen.Dao) *siteConfigDo {
  217. return s.withDO(s.DO.Scopes(funcs...))
  218. }
  219. func (s siteConfigDo) Unscoped() *siteConfigDo {
  220. return s.withDO(s.DO.Unscoped())
  221. }
  222. func (s siteConfigDo) Create(values ...*model.SiteConfig) error {
  223. if len(values) == 0 {
  224. return nil
  225. }
  226. return s.DO.Create(values)
  227. }
  228. func (s siteConfigDo) CreateInBatches(values []*model.SiteConfig, batchSize int) error {
  229. return s.DO.CreateInBatches(values, batchSize)
  230. }
  231. // Save : !!! underlying implementation is different with GORM
  232. // The method is equivalent to executing the statement: db.Clauses(clause.OnConflict{UpdateAll: true}).Create(values)
  233. func (s siteConfigDo) Save(values ...*model.SiteConfig) error {
  234. if len(values) == 0 {
  235. return nil
  236. }
  237. return s.DO.Save(values)
  238. }
  239. func (s siteConfigDo) First() (*model.SiteConfig, error) {
  240. if result, err := s.DO.First(); err != nil {
  241. return nil, err
  242. } else {
  243. return result.(*model.SiteConfig), nil
  244. }
  245. }
  246. func (s siteConfigDo) Take() (*model.SiteConfig, error) {
  247. if result, err := s.DO.Take(); err != nil {
  248. return nil, err
  249. } else {
  250. return result.(*model.SiteConfig), nil
  251. }
  252. }
  253. func (s siteConfigDo) Last() (*model.SiteConfig, error) {
  254. if result, err := s.DO.Last(); err != nil {
  255. return nil, err
  256. } else {
  257. return result.(*model.SiteConfig), nil
  258. }
  259. }
  260. func (s siteConfigDo) Find() ([]*model.SiteConfig, error) {
  261. result, err := s.DO.Find()
  262. return result.([]*model.SiteConfig), err
  263. }
  264. func (s siteConfigDo) FindInBatch(batchSize int, fc func(tx gen.Dao, batch int) error) (results []*model.SiteConfig, err error) {
  265. buf := make([]*model.SiteConfig, 0, batchSize)
  266. err = s.DO.FindInBatches(&buf, batchSize, func(tx gen.Dao, batch int) error {
  267. defer func() { results = append(results, buf...) }()
  268. return fc(tx, batch)
  269. })
  270. return results, err
  271. }
  272. func (s siteConfigDo) FindInBatches(result *[]*model.SiteConfig, batchSize int, fc func(tx gen.Dao, batch int) error) error {
  273. return s.DO.FindInBatches(result, batchSize, fc)
  274. }
  275. func (s siteConfigDo) Attrs(attrs ...field.AssignExpr) *siteConfigDo {
  276. return s.withDO(s.DO.Attrs(attrs...))
  277. }
  278. func (s siteConfigDo) Assign(attrs ...field.AssignExpr) *siteConfigDo {
  279. return s.withDO(s.DO.Assign(attrs...))
  280. }
  281. func (s siteConfigDo) Joins(fields ...field.RelationField) *siteConfigDo {
  282. for _, _f := range fields {
  283. s = *s.withDO(s.DO.Joins(_f))
  284. }
  285. return &s
  286. }
  287. func (s siteConfigDo) Preload(fields ...field.RelationField) *siteConfigDo {
  288. for _, _f := range fields {
  289. s = *s.withDO(s.DO.Preload(_f))
  290. }
  291. return &s
  292. }
  293. func (s siteConfigDo) FirstOrInit() (*model.SiteConfig, error) {
  294. if result, err := s.DO.FirstOrInit(); err != nil {
  295. return nil, err
  296. } else {
  297. return result.(*model.SiteConfig), nil
  298. }
  299. }
  300. func (s siteConfigDo) FirstOrCreate() (*model.SiteConfig, error) {
  301. if result, err := s.DO.FirstOrCreate(); err != nil {
  302. return nil, err
  303. } else {
  304. return result.(*model.SiteConfig), nil
  305. }
  306. }
  307. func (s siteConfigDo) FindByPage(offset int, limit int) (result []*model.SiteConfig, count int64, err error) {
  308. result, err = s.Offset(offset).Limit(limit).Find()
  309. if err != nil {
  310. return
  311. }
  312. if size := len(result); 0 < limit && 0 < size && size < limit {
  313. count = int64(size + offset)
  314. return
  315. }
  316. count, err = s.Offset(-1).Limit(-1).Count()
  317. return
  318. }
  319. func (s siteConfigDo) ScanByPage(result interface{}, offset int, limit int) (count int64, err error) {
  320. count, err = s.Count()
  321. if err != nil {
  322. return
  323. }
  324. err = s.Offset(offset).Limit(limit).Scan(result)
  325. return
  326. }
  327. func (s siteConfigDo) Scan(result interface{}) (err error) {
  328. return s.DO.Scan(result)
  329. }
  330. func (s siteConfigDo) Delete(models ...*model.SiteConfig) (result gen.ResultInfo, err error) {
  331. return s.DO.Delete(models)
  332. }
  333. func (s *siteConfigDo) withDO(do gen.Dao) *siteConfigDo {
  334. s.DO = *do.(*gen.DO)
  335. return s
  336. }