site_configs.gen.go 16 KB

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