usagecleanuptask_update.go 22 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702
  1. // Code generated by ent, DO NOT EDIT.
  2. package ent
  3. import (
  4. "context"
  5. "encoding/json/jsontext"
  6. "errors"
  7. "fmt"
  8. "time"
  9. "entgo.io/ent/dialect/sql"
  10. "entgo.io/ent/dialect/sql/sqlgraph"
  11. "entgo.io/ent/dialect/sql/sqljson"
  12. "entgo.io/ent/schema/field"
  13. "github.com/Wei-Shaw/sub2api/ent/predicate"
  14. "github.com/Wei-Shaw/sub2api/ent/usagecleanuptask"
  15. )
  16. // UsageCleanupTaskUpdate is the builder for updating UsageCleanupTask entities.
  17. type UsageCleanupTaskUpdate struct {
  18. config
  19. hooks []Hook
  20. mutation *UsageCleanupTaskMutation
  21. }
  22. // Where appends a list predicates to the UsageCleanupTaskUpdate builder.
  23. func (_u *UsageCleanupTaskUpdate) Where(ps ...predicate.UsageCleanupTask) *UsageCleanupTaskUpdate {
  24. _u.mutation.Where(ps...)
  25. return _u
  26. }
  27. // SetUpdatedAt sets the "updated_at" field.
  28. func (_u *UsageCleanupTaskUpdate) SetUpdatedAt(v time.Time) *UsageCleanupTaskUpdate {
  29. _u.mutation.SetUpdatedAt(v)
  30. return _u
  31. }
  32. // SetStatus sets the "status" field.
  33. func (_u *UsageCleanupTaskUpdate) SetStatus(v string) *UsageCleanupTaskUpdate {
  34. _u.mutation.SetStatus(v)
  35. return _u
  36. }
  37. // SetNillableStatus sets the "status" field if the given value is not nil.
  38. func (_u *UsageCleanupTaskUpdate) SetNillableStatus(v *string) *UsageCleanupTaskUpdate {
  39. if v != nil {
  40. _u.SetStatus(*v)
  41. }
  42. return _u
  43. }
  44. // SetFilters sets the "filters" field.
  45. func (_u *UsageCleanupTaskUpdate) SetFilters(v jsontext.Value) *UsageCleanupTaskUpdate {
  46. _u.mutation.SetFilters(v)
  47. return _u
  48. }
  49. // AppendFilters appends value to the "filters" field.
  50. func (_u *UsageCleanupTaskUpdate) AppendFilters(v jsontext.Value) *UsageCleanupTaskUpdate {
  51. _u.mutation.AppendFilters(v)
  52. return _u
  53. }
  54. // SetCreatedBy sets the "created_by" field.
  55. func (_u *UsageCleanupTaskUpdate) SetCreatedBy(v int64) *UsageCleanupTaskUpdate {
  56. _u.mutation.ResetCreatedBy()
  57. _u.mutation.SetCreatedBy(v)
  58. return _u
  59. }
  60. // SetNillableCreatedBy sets the "created_by" field if the given value is not nil.
  61. func (_u *UsageCleanupTaskUpdate) SetNillableCreatedBy(v *int64) *UsageCleanupTaskUpdate {
  62. if v != nil {
  63. _u.SetCreatedBy(*v)
  64. }
  65. return _u
  66. }
  67. // AddCreatedBy adds value to the "created_by" field.
  68. func (_u *UsageCleanupTaskUpdate) AddCreatedBy(v int64) *UsageCleanupTaskUpdate {
  69. _u.mutation.AddCreatedBy(v)
  70. return _u
  71. }
  72. // SetDeletedRows sets the "deleted_rows" field.
  73. func (_u *UsageCleanupTaskUpdate) SetDeletedRows(v int64) *UsageCleanupTaskUpdate {
  74. _u.mutation.ResetDeletedRows()
  75. _u.mutation.SetDeletedRows(v)
  76. return _u
  77. }
  78. // SetNillableDeletedRows sets the "deleted_rows" field if the given value is not nil.
  79. func (_u *UsageCleanupTaskUpdate) SetNillableDeletedRows(v *int64) *UsageCleanupTaskUpdate {
  80. if v != nil {
  81. _u.SetDeletedRows(*v)
  82. }
  83. return _u
  84. }
  85. // AddDeletedRows adds value to the "deleted_rows" field.
  86. func (_u *UsageCleanupTaskUpdate) AddDeletedRows(v int64) *UsageCleanupTaskUpdate {
  87. _u.mutation.AddDeletedRows(v)
  88. return _u
  89. }
  90. // SetErrorMessage sets the "error_message" field.
  91. func (_u *UsageCleanupTaskUpdate) SetErrorMessage(v string) *UsageCleanupTaskUpdate {
  92. _u.mutation.SetErrorMessage(v)
  93. return _u
  94. }
  95. // SetNillableErrorMessage sets the "error_message" field if the given value is not nil.
  96. func (_u *UsageCleanupTaskUpdate) SetNillableErrorMessage(v *string) *UsageCleanupTaskUpdate {
  97. if v != nil {
  98. _u.SetErrorMessage(*v)
  99. }
  100. return _u
  101. }
  102. // ClearErrorMessage clears the value of the "error_message" field.
  103. func (_u *UsageCleanupTaskUpdate) ClearErrorMessage() *UsageCleanupTaskUpdate {
  104. _u.mutation.ClearErrorMessage()
  105. return _u
  106. }
  107. // SetCanceledBy sets the "canceled_by" field.
  108. func (_u *UsageCleanupTaskUpdate) SetCanceledBy(v int64) *UsageCleanupTaskUpdate {
  109. _u.mutation.ResetCanceledBy()
  110. _u.mutation.SetCanceledBy(v)
  111. return _u
  112. }
  113. // SetNillableCanceledBy sets the "canceled_by" field if the given value is not nil.
  114. func (_u *UsageCleanupTaskUpdate) SetNillableCanceledBy(v *int64) *UsageCleanupTaskUpdate {
  115. if v != nil {
  116. _u.SetCanceledBy(*v)
  117. }
  118. return _u
  119. }
  120. // AddCanceledBy adds value to the "canceled_by" field.
  121. func (_u *UsageCleanupTaskUpdate) AddCanceledBy(v int64) *UsageCleanupTaskUpdate {
  122. _u.mutation.AddCanceledBy(v)
  123. return _u
  124. }
  125. // ClearCanceledBy clears the value of the "canceled_by" field.
  126. func (_u *UsageCleanupTaskUpdate) ClearCanceledBy() *UsageCleanupTaskUpdate {
  127. _u.mutation.ClearCanceledBy()
  128. return _u
  129. }
  130. // SetCanceledAt sets the "canceled_at" field.
  131. func (_u *UsageCleanupTaskUpdate) SetCanceledAt(v time.Time) *UsageCleanupTaskUpdate {
  132. _u.mutation.SetCanceledAt(v)
  133. return _u
  134. }
  135. // SetNillableCanceledAt sets the "canceled_at" field if the given value is not nil.
  136. func (_u *UsageCleanupTaskUpdate) SetNillableCanceledAt(v *time.Time) *UsageCleanupTaskUpdate {
  137. if v != nil {
  138. _u.SetCanceledAt(*v)
  139. }
  140. return _u
  141. }
  142. // ClearCanceledAt clears the value of the "canceled_at" field.
  143. func (_u *UsageCleanupTaskUpdate) ClearCanceledAt() *UsageCleanupTaskUpdate {
  144. _u.mutation.ClearCanceledAt()
  145. return _u
  146. }
  147. // SetStartedAt sets the "started_at" field.
  148. func (_u *UsageCleanupTaskUpdate) SetStartedAt(v time.Time) *UsageCleanupTaskUpdate {
  149. _u.mutation.SetStartedAt(v)
  150. return _u
  151. }
  152. // SetNillableStartedAt sets the "started_at" field if the given value is not nil.
  153. func (_u *UsageCleanupTaskUpdate) SetNillableStartedAt(v *time.Time) *UsageCleanupTaskUpdate {
  154. if v != nil {
  155. _u.SetStartedAt(*v)
  156. }
  157. return _u
  158. }
  159. // ClearStartedAt clears the value of the "started_at" field.
  160. func (_u *UsageCleanupTaskUpdate) ClearStartedAt() *UsageCleanupTaskUpdate {
  161. _u.mutation.ClearStartedAt()
  162. return _u
  163. }
  164. // SetFinishedAt sets the "finished_at" field.
  165. func (_u *UsageCleanupTaskUpdate) SetFinishedAt(v time.Time) *UsageCleanupTaskUpdate {
  166. _u.mutation.SetFinishedAt(v)
  167. return _u
  168. }
  169. // SetNillableFinishedAt sets the "finished_at" field if the given value is not nil.
  170. func (_u *UsageCleanupTaskUpdate) SetNillableFinishedAt(v *time.Time) *UsageCleanupTaskUpdate {
  171. if v != nil {
  172. _u.SetFinishedAt(*v)
  173. }
  174. return _u
  175. }
  176. // ClearFinishedAt clears the value of the "finished_at" field.
  177. func (_u *UsageCleanupTaskUpdate) ClearFinishedAt() *UsageCleanupTaskUpdate {
  178. _u.mutation.ClearFinishedAt()
  179. return _u
  180. }
  181. // Mutation returns the UsageCleanupTaskMutation object of the builder.
  182. func (_u *UsageCleanupTaskUpdate) Mutation() *UsageCleanupTaskMutation {
  183. return _u.mutation
  184. }
  185. // Save executes the query and returns the number of nodes affected by the update operation.
  186. func (_u *UsageCleanupTaskUpdate) Save(ctx context.Context) (int, error) {
  187. _u.defaults()
  188. return withHooks(ctx, _u.sqlSave, _u.mutation, _u.hooks)
  189. }
  190. // SaveX is like Save, but panics if an error occurs.
  191. func (_u *UsageCleanupTaskUpdate) SaveX(ctx context.Context) int {
  192. affected, err := _u.Save(ctx)
  193. if err != nil {
  194. panic(err)
  195. }
  196. return affected
  197. }
  198. // Exec executes the query.
  199. func (_u *UsageCleanupTaskUpdate) Exec(ctx context.Context) error {
  200. _, err := _u.Save(ctx)
  201. return err
  202. }
  203. // ExecX is like Exec, but panics if an error occurs.
  204. func (_u *UsageCleanupTaskUpdate) ExecX(ctx context.Context) {
  205. if err := _u.Exec(ctx); err != nil {
  206. panic(err)
  207. }
  208. }
  209. // defaults sets the default values of the builder before save.
  210. func (_u *UsageCleanupTaskUpdate) defaults() {
  211. if _, ok := _u.mutation.UpdatedAt(); !ok {
  212. v := usagecleanuptask.UpdateDefaultUpdatedAt()
  213. _u.mutation.SetUpdatedAt(v)
  214. }
  215. }
  216. // check runs all checks and user-defined validators on the builder.
  217. func (_u *UsageCleanupTaskUpdate) check() error {
  218. if v, ok := _u.mutation.Status(); ok {
  219. if err := usagecleanuptask.StatusValidator(v); err != nil {
  220. return &ValidationError{Name: "status", err: fmt.Errorf(`ent: validator failed for field "UsageCleanupTask.status": %w`, err)}
  221. }
  222. }
  223. return nil
  224. }
  225. func (_u *UsageCleanupTaskUpdate) sqlSave(ctx context.Context) (_node int, err error) {
  226. if err := _u.check(); err != nil {
  227. return _node, err
  228. }
  229. _spec := sqlgraph.NewUpdateSpec(usagecleanuptask.Table, usagecleanuptask.Columns, sqlgraph.NewFieldSpec(usagecleanuptask.FieldID, field.TypeInt64))
  230. if ps := _u.mutation.predicates; len(ps) > 0 {
  231. _spec.Predicate = func(selector *sql.Selector) {
  232. for i := range ps {
  233. ps[i](selector)
  234. }
  235. }
  236. }
  237. if value, ok := _u.mutation.UpdatedAt(); ok {
  238. _spec.SetField(usagecleanuptask.FieldUpdatedAt, field.TypeTime, value)
  239. }
  240. if value, ok := _u.mutation.Status(); ok {
  241. _spec.SetField(usagecleanuptask.FieldStatus, field.TypeString, value)
  242. }
  243. if value, ok := _u.mutation.Filters(); ok {
  244. _spec.SetField(usagecleanuptask.FieldFilters, field.TypeJSON, value)
  245. }
  246. if value, ok := _u.mutation.AppendedFilters(); ok {
  247. _spec.AddModifier(func(u *sql.UpdateBuilder) {
  248. sqljson.Append(u, usagecleanuptask.FieldFilters, value)
  249. })
  250. }
  251. if value, ok := _u.mutation.CreatedBy(); ok {
  252. _spec.SetField(usagecleanuptask.FieldCreatedBy, field.TypeInt64, value)
  253. }
  254. if value, ok := _u.mutation.AddedCreatedBy(); ok {
  255. _spec.AddField(usagecleanuptask.FieldCreatedBy, field.TypeInt64, value)
  256. }
  257. if value, ok := _u.mutation.DeletedRows(); ok {
  258. _spec.SetField(usagecleanuptask.FieldDeletedRows, field.TypeInt64, value)
  259. }
  260. if value, ok := _u.mutation.AddedDeletedRows(); ok {
  261. _spec.AddField(usagecleanuptask.FieldDeletedRows, field.TypeInt64, value)
  262. }
  263. if value, ok := _u.mutation.ErrorMessage(); ok {
  264. _spec.SetField(usagecleanuptask.FieldErrorMessage, field.TypeString, value)
  265. }
  266. if _u.mutation.ErrorMessageCleared() {
  267. _spec.ClearField(usagecleanuptask.FieldErrorMessage, field.TypeString)
  268. }
  269. if value, ok := _u.mutation.CanceledBy(); ok {
  270. _spec.SetField(usagecleanuptask.FieldCanceledBy, field.TypeInt64, value)
  271. }
  272. if value, ok := _u.mutation.AddedCanceledBy(); ok {
  273. _spec.AddField(usagecleanuptask.FieldCanceledBy, field.TypeInt64, value)
  274. }
  275. if _u.mutation.CanceledByCleared() {
  276. _spec.ClearField(usagecleanuptask.FieldCanceledBy, field.TypeInt64)
  277. }
  278. if value, ok := _u.mutation.CanceledAt(); ok {
  279. _spec.SetField(usagecleanuptask.FieldCanceledAt, field.TypeTime, value)
  280. }
  281. if _u.mutation.CanceledAtCleared() {
  282. _spec.ClearField(usagecleanuptask.FieldCanceledAt, field.TypeTime)
  283. }
  284. if value, ok := _u.mutation.StartedAt(); ok {
  285. _spec.SetField(usagecleanuptask.FieldStartedAt, field.TypeTime, value)
  286. }
  287. if _u.mutation.StartedAtCleared() {
  288. _spec.ClearField(usagecleanuptask.FieldStartedAt, field.TypeTime)
  289. }
  290. if value, ok := _u.mutation.FinishedAt(); ok {
  291. _spec.SetField(usagecleanuptask.FieldFinishedAt, field.TypeTime, value)
  292. }
  293. if _u.mutation.FinishedAtCleared() {
  294. _spec.ClearField(usagecleanuptask.FieldFinishedAt, field.TypeTime)
  295. }
  296. if _node, err = sqlgraph.UpdateNodes(ctx, _u.driver, _spec); err != nil {
  297. if _, ok := err.(*sqlgraph.NotFoundError); ok {
  298. err = &NotFoundError{usagecleanuptask.Label}
  299. } else if sqlgraph.IsConstraintError(err) {
  300. err = &ConstraintError{msg: err.Error(), wrap: err}
  301. }
  302. return 0, err
  303. }
  304. _u.mutation.done = true
  305. return _node, nil
  306. }
  307. // UsageCleanupTaskUpdateOne is the builder for updating a single UsageCleanupTask entity.
  308. type UsageCleanupTaskUpdateOne struct {
  309. config
  310. fields []string
  311. hooks []Hook
  312. mutation *UsageCleanupTaskMutation
  313. }
  314. // SetUpdatedAt sets the "updated_at" field.
  315. func (_u *UsageCleanupTaskUpdateOne) SetUpdatedAt(v time.Time) *UsageCleanupTaskUpdateOne {
  316. _u.mutation.SetUpdatedAt(v)
  317. return _u
  318. }
  319. // SetStatus sets the "status" field.
  320. func (_u *UsageCleanupTaskUpdateOne) SetStatus(v string) *UsageCleanupTaskUpdateOne {
  321. _u.mutation.SetStatus(v)
  322. return _u
  323. }
  324. // SetNillableStatus sets the "status" field if the given value is not nil.
  325. func (_u *UsageCleanupTaskUpdateOne) SetNillableStatus(v *string) *UsageCleanupTaskUpdateOne {
  326. if v != nil {
  327. _u.SetStatus(*v)
  328. }
  329. return _u
  330. }
  331. // SetFilters sets the "filters" field.
  332. func (_u *UsageCleanupTaskUpdateOne) SetFilters(v jsontext.Value) *UsageCleanupTaskUpdateOne {
  333. _u.mutation.SetFilters(v)
  334. return _u
  335. }
  336. // AppendFilters appends value to the "filters" field.
  337. func (_u *UsageCleanupTaskUpdateOne) AppendFilters(v jsontext.Value) *UsageCleanupTaskUpdateOne {
  338. _u.mutation.AppendFilters(v)
  339. return _u
  340. }
  341. // SetCreatedBy sets the "created_by" field.
  342. func (_u *UsageCleanupTaskUpdateOne) SetCreatedBy(v int64) *UsageCleanupTaskUpdateOne {
  343. _u.mutation.ResetCreatedBy()
  344. _u.mutation.SetCreatedBy(v)
  345. return _u
  346. }
  347. // SetNillableCreatedBy sets the "created_by" field if the given value is not nil.
  348. func (_u *UsageCleanupTaskUpdateOne) SetNillableCreatedBy(v *int64) *UsageCleanupTaskUpdateOne {
  349. if v != nil {
  350. _u.SetCreatedBy(*v)
  351. }
  352. return _u
  353. }
  354. // AddCreatedBy adds value to the "created_by" field.
  355. func (_u *UsageCleanupTaskUpdateOne) AddCreatedBy(v int64) *UsageCleanupTaskUpdateOne {
  356. _u.mutation.AddCreatedBy(v)
  357. return _u
  358. }
  359. // SetDeletedRows sets the "deleted_rows" field.
  360. func (_u *UsageCleanupTaskUpdateOne) SetDeletedRows(v int64) *UsageCleanupTaskUpdateOne {
  361. _u.mutation.ResetDeletedRows()
  362. _u.mutation.SetDeletedRows(v)
  363. return _u
  364. }
  365. // SetNillableDeletedRows sets the "deleted_rows" field if the given value is not nil.
  366. func (_u *UsageCleanupTaskUpdateOne) SetNillableDeletedRows(v *int64) *UsageCleanupTaskUpdateOne {
  367. if v != nil {
  368. _u.SetDeletedRows(*v)
  369. }
  370. return _u
  371. }
  372. // AddDeletedRows adds value to the "deleted_rows" field.
  373. func (_u *UsageCleanupTaskUpdateOne) AddDeletedRows(v int64) *UsageCleanupTaskUpdateOne {
  374. _u.mutation.AddDeletedRows(v)
  375. return _u
  376. }
  377. // SetErrorMessage sets the "error_message" field.
  378. func (_u *UsageCleanupTaskUpdateOne) SetErrorMessage(v string) *UsageCleanupTaskUpdateOne {
  379. _u.mutation.SetErrorMessage(v)
  380. return _u
  381. }
  382. // SetNillableErrorMessage sets the "error_message" field if the given value is not nil.
  383. func (_u *UsageCleanupTaskUpdateOne) SetNillableErrorMessage(v *string) *UsageCleanupTaskUpdateOne {
  384. if v != nil {
  385. _u.SetErrorMessage(*v)
  386. }
  387. return _u
  388. }
  389. // ClearErrorMessage clears the value of the "error_message" field.
  390. func (_u *UsageCleanupTaskUpdateOne) ClearErrorMessage() *UsageCleanupTaskUpdateOne {
  391. _u.mutation.ClearErrorMessage()
  392. return _u
  393. }
  394. // SetCanceledBy sets the "canceled_by" field.
  395. func (_u *UsageCleanupTaskUpdateOne) SetCanceledBy(v int64) *UsageCleanupTaskUpdateOne {
  396. _u.mutation.ResetCanceledBy()
  397. _u.mutation.SetCanceledBy(v)
  398. return _u
  399. }
  400. // SetNillableCanceledBy sets the "canceled_by" field if the given value is not nil.
  401. func (_u *UsageCleanupTaskUpdateOne) SetNillableCanceledBy(v *int64) *UsageCleanupTaskUpdateOne {
  402. if v != nil {
  403. _u.SetCanceledBy(*v)
  404. }
  405. return _u
  406. }
  407. // AddCanceledBy adds value to the "canceled_by" field.
  408. func (_u *UsageCleanupTaskUpdateOne) AddCanceledBy(v int64) *UsageCleanupTaskUpdateOne {
  409. _u.mutation.AddCanceledBy(v)
  410. return _u
  411. }
  412. // ClearCanceledBy clears the value of the "canceled_by" field.
  413. func (_u *UsageCleanupTaskUpdateOne) ClearCanceledBy() *UsageCleanupTaskUpdateOne {
  414. _u.mutation.ClearCanceledBy()
  415. return _u
  416. }
  417. // SetCanceledAt sets the "canceled_at" field.
  418. func (_u *UsageCleanupTaskUpdateOne) SetCanceledAt(v time.Time) *UsageCleanupTaskUpdateOne {
  419. _u.mutation.SetCanceledAt(v)
  420. return _u
  421. }
  422. // SetNillableCanceledAt sets the "canceled_at" field if the given value is not nil.
  423. func (_u *UsageCleanupTaskUpdateOne) SetNillableCanceledAt(v *time.Time) *UsageCleanupTaskUpdateOne {
  424. if v != nil {
  425. _u.SetCanceledAt(*v)
  426. }
  427. return _u
  428. }
  429. // ClearCanceledAt clears the value of the "canceled_at" field.
  430. func (_u *UsageCleanupTaskUpdateOne) ClearCanceledAt() *UsageCleanupTaskUpdateOne {
  431. _u.mutation.ClearCanceledAt()
  432. return _u
  433. }
  434. // SetStartedAt sets the "started_at" field.
  435. func (_u *UsageCleanupTaskUpdateOne) SetStartedAt(v time.Time) *UsageCleanupTaskUpdateOne {
  436. _u.mutation.SetStartedAt(v)
  437. return _u
  438. }
  439. // SetNillableStartedAt sets the "started_at" field if the given value is not nil.
  440. func (_u *UsageCleanupTaskUpdateOne) SetNillableStartedAt(v *time.Time) *UsageCleanupTaskUpdateOne {
  441. if v != nil {
  442. _u.SetStartedAt(*v)
  443. }
  444. return _u
  445. }
  446. // ClearStartedAt clears the value of the "started_at" field.
  447. func (_u *UsageCleanupTaskUpdateOne) ClearStartedAt() *UsageCleanupTaskUpdateOne {
  448. _u.mutation.ClearStartedAt()
  449. return _u
  450. }
  451. // SetFinishedAt sets the "finished_at" field.
  452. func (_u *UsageCleanupTaskUpdateOne) SetFinishedAt(v time.Time) *UsageCleanupTaskUpdateOne {
  453. _u.mutation.SetFinishedAt(v)
  454. return _u
  455. }
  456. // SetNillableFinishedAt sets the "finished_at" field if the given value is not nil.
  457. func (_u *UsageCleanupTaskUpdateOne) SetNillableFinishedAt(v *time.Time) *UsageCleanupTaskUpdateOne {
  458. if v != nil {
  459. _u.SetFinishedAt(*v)
  460. }
  461. return _u
  462. }
  463. // ClearFinishedAt clears the value of the "finished_at" field.
  464. func (_u *UsageCleanupTaskUpdateOne) ClearFinishedAt() *UsageCleanupTaskUpdateOne {
  465. _u.mutation.ClearFinishedAt()
  466. return _u
  467. }
  468. // Mutation returns the UsageCleanupTaskMutation object of the builder.
  469. func (_u *UsageCleanupTaskUpdateOne) Mutation() *UsageCleanupTaskMutation {
  470. return _u.mutation
  471. }
  472. // Where appends a list predicates to the UsageCleanupTaskUpdate builder.
  473. func (_u *UsageCleanupTaskUpdateOne) Where(ps ...predicate.UsageCleanupTask) *UsageCleanupTaskUpdateOne {
  474. _u.mutation.Where(ps...)
  475. return _u
  476. }
  477. // Select allows selecting one or more fields (columns) of the returned entity.
  478. // The default is selecting all fields defined in the entity schema.
  479. func (_u *UsageCleanupTaskUpdateOne) Select(field string, fields ...string) *UsageCleanupTaskUpdateOne {
  480. _u.fields = append([]string{field}, fields...)
  481. return _u
  482. }
  483. // Save executes the query and returns the updated UsageCleanupTask entity.
  484. func (_u *UsageCleanupTaskUpdateOne) Save(ctx context.Context) (*UsageCleanupTask, error) {
  485. _u.defaults()
  486. return withHooks(ctx, _u.sqlSave, _u.mutation, _u.hooks)
  487. }
  488. // SaveX is like Save, but panics if an error occurs.
  489. func (_u *UsageCleanupTaskUpdateOne) SaveX(ctx context.Context) *UsageCleanupTask {
  490. node, err := _u.Save(ctx)
  491. if err != nil {
  492. panic(err)
  493. }
  494. return node
  495. }
  496. // Exec executes the query on the entity.
  497. func (_u *UsageCleanupTaskUpdateOne) Exec(ctx context.Context) error {
  498. _, err := _u.Save(ctx)
  499. return err
  500. }
  501. // ExecX is like Exec, but panics if an error occurs.
  502. func (_u *UsageCleanupTaskUpdateOne) ExecX(ctx context.Context) {
  503. if err := _u.Exec(ctx); err != nil {
  504. panic(err)
  505. }
  506. }
  507. // defaults sets the default values of the builder before save.
  508. func (_u *UsageCleanupTaskUpdateOne) defaults() {
  509. if _, ok := _u.mutation.UpdatedAt(); !ok {
  510. v := usagecleanuptask.UpdateDefaultUpdatedAt()
  511. _u.mutation.SetUpdatedAt(v)
  512. }
  513. }
  514. // check runs all checks and user-defined validators on the builder.
  515. func (_u *UsageCleanupTaskUpdateOne) check() error {
  516. if v, ok := _u.mutation.Status(); ok {
  517. if err := usagecleanuptask.StatusValidator(v); err != nil {
  518. return &ValidationError{Name: "status", err: fmt.Errorf(`ent: validator failed for field "UsageCleanupTask.status": %w`, err)}
  519. }
  520. }
  521. return nil
  522. }
  523. func (_u *UsageCleanupTaskUpdateOne) sqlSave(ctx context.Context) (_node *UsageCleanupTask, err error) {
  524. if err := _u.check(); err != nil {
  525. return _node, err
  526. }
  527. _spec := sqlgraph.NewUpdateSpec(usagecleanuptask.Table, usagecleanuptask.Columns, sqlgraph.NewFieldSpec(usagecleanuptask.FieldID, field.TypeInt64))
  528. id, ok := _u.mutation.ID()
  529. if !ok {
  530. return nil, &ValidationError{Name: "id", err: errors.New(`ent: missing "UsageCleanupTask.id" for update`)}
  531. }
  532. _spec.Node.ID.Value = id
  533. if fields := _u.fields; len(fields) > 0 {
  534. _spec.Node.Columns = make([]string, 0, len(fields))
  535. _spec.Node.Columns = append(_spec.Node.Columns, usagecleanuptask.FieldID)
  536. for _, f := range fields {
  537. if !usagecleanuptask.ValidColumn(f) {
  538. return nil, &ValidationError{Name: f, err: fmt.Errorf("ent: invalid field %q for query", f)}
  539. }
  540. if f != usagecleanuptask.FieldID {
  541. _spec.Node.Columns = append(_spec.Node.Columns, f)
  542. }
  543. }
  544. }
  545. if ps := _u.mutation.predicates; len(ps) > 0 {
  546. _spec.Predicate = func(selector *sql.Selector) {
  547. for i := range ps {
  548. ps[i](selector)
  549. }
  550. }
  551. }
  552. if value, ok := _u.mutation.UpdatedAt(); ok {
  553. _spec.SetField(usagecleanuptask.FieldUpdatedAt, field.TypeTime, value)
  554. }
  555. if value, ok := _u.mutation.Status(); ok {
  556. _spec.SetField(usagecleanuptask.FieldStatus, field.TypeString, value)
  557. }
  558. if value, ok := _u.mutation.Filters(); ok {
  559. _spec.SetField(usagecleanuptask.FieldFilters, field.TypeJSON, value)
  560. }
  561. if value, ok := _u.mutation.AppendedFilters(); ok {
  562. _spec.AddModifier(func(u *sql.UpdateBuilder) {
  563. sqljson.Append(u, usagecleanuptask.FieldFilters, value)
  564. })
  565. }
  566. if value, ok := _u.mutation.CreatedBy(); ok {
  567. _spec.SetField(usagecleanuptask.FieldCreatedBy, field.TypeInt64, value)
  568. }
  569. if value, ok := _u.mutation.AddedCreatedBy(); ok {
  570. _spec.AddField(usagecleanuptask.FieldCreatedBy, field.TypeInt64, value)
  571. }
  572. if value, ok := _u.mutation.DeletedRows(); ok {
  573. _spec.SetField(usagecleanuptask.FieldDeletedRows, field.TypeInt64, value)
  574. }
  575. if value, ok := _u.mutation.AddedDeletedRows(); ok {
  576. _spec.AddField(usagecleanuptask.FieldDeletedRows, field.TypeInt64, value)
  577. }
  578. if value, ok := _u.mutation.ErrorMessage(); ok {
  579. _spec.SetField(usagecleanuptask.FieldErrorMessage, field.TypeString, value)
  580. }
  581. if _u.mutation.ErrorMessageCleared() {
  582. _spec.ClearField(usagecleanuptask.FieldErrorMessage, field.TypeString)
  583. }
  584. if value, ok := _u.mutation.CanceledBy(); ok {
  585. _spec.SetField(usagecleanuptask.FieldCanceledBy, field.TypeInt64, value)
  586. }
  587. if value, ok := _u.mutation.AddedCanceledBy(); ok {
  588. _spec.AddField(usagecleanuptask.FieldCanceledBy, field.TypeInt64, value)
  589. }
  590. if _u.mutation.CanceledByCleared() {
  591. _spec.ClearField(usagecleanuptask.FieldCanceledBy, field.TypeInt64)
  592. }
  593. if value, ok := _u.mutation.CanceledAt(); ok {
  594. _spec.SetField(usagecleanuptask.FieldCanceledAt, field.TypeTime, value)
  595. }
  596. if _u.mutation.CanceledAtCleared() {
  597. _spec.ClearField(usagecleanuptask.FieldCanceledAt, field.TypeTime)
  598. }
  599. if value, ok := _u.mutation.StartedAt(); ok {
  600. _spec.SetField(usagecleanuptask.FieldStartedAt, field.TypeTime, value)
  601. }
  602. if _u.mutation.StartedAtCleared() {
  603. _spec.ClearField(usagecleanuptask.FieldStartedAt, field.TypeTime)
  604. }
  605. if value, ok := _u.mutation.FinishedAt(); ok {
  606. _spec.SetField(usagecleanuptask.FieldFinishedAt, field.TypeTime, value)
  607. }
  608. if _u.mutation.FinishedAtCleared() {
  609. _spec.ClearField(usagecleanuptask.FieldFinishedAt, field.TypeTime)
  610. }
  611. _node = &UsageCleanupTask{config: _u.config}
  612. _spec.Assign = _node.assignValues
  613. _spec.ScanValues = _node.scanValues
  614. if err = sqlgraph.UpdateNode(ctx, _u.driver, _spec); err != nil {
  615. if _, ok := err.(*sqlgraph.NotFoundError); ok {
  616. err = &NotFoundError{usagecleanuptask.Label}
  617. } else if sqlgraph.IsConstraintError(err) {
  618. err = &ConstraintError{msg: err.Error(), wrap: err}
  619. }
  620. return nil, err
  621. }
  622. _u.mutation.done = true
  623. return _node, nil
  624. }