batchimageitem_update.go 38 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117111811191120112111221123112411251126112711281129113011311132
  1. // Code generated by ent, DO NOT EDIT.
  2. package ent
  3. import (
  4. "context"
  5. "errors"
  6. "fmt"
  7. "time"
  8. "entgo.io/ent/dialect/sql"
  9. "entgo.io/ent/dialect/sql/sqlgraph"
  10. "entgo.io/ent/schema/field"
  11. "github.com/Wei-Shaw/sub2api/ent/batchimageitem"
  12. "github.com/Wei-Shaw/sub2api/ent/predicate"
  13. )
  14. // BatchImageItemUpdate is the builder for updating BatchImageItem entities.
  15. type BatchImageItemUpdate struct {
  16. config
  17. hooks []Hook
  18. mutation *BatchImageItemMutation
  19. }
  20. // Where appends a list predicates to the BatchImageItemUpdate builder.
  21. func (_u *BatchImageItemUpdate) Where(ps ...predicate.BatchImageItem) *BatchImageItemUpdate {
  22. _u.mutation.Where(ps...)
  23. return _u
  24. }
  25. // SetJobID sets the "job_id" field.
  26. func (_u *BatchImageItemUpdate) SetJobID(v string) *BatchImageItemUpdate {
  27. _u.mutation.SetJobID(v)
  28. return _u
  29. }
  30. // SetNillableJobID sets the "job_id" field if the given value is not nil.
  31. func (_u *BatchImageItemUpdate) SetNillableJobID(v *string) *BatchImageItemUpdate {
  32. if v != nil {
  33. _u.SetJobID(*v)
  34. }
  35. return _u
  36. }
  37. // SetCustomID sets the "custom_id" field.
  38. func (_u *BatchImageItemUpdate) SetCustomID(v string) *BatchImageItemUpdate {
  39. _u.mutation.SetCustomID(v)
  40. return _u
  41. }
  42. // SetNillableCustomID sets the "custom_id" field if the given value is not nil.
  43. func (_u *BatchImageItemUpdate) SetNillableCustomID(v *string) *BatchImageItemUpdate {
  44. if v != nil {
  45. _u.SetCustomID(*v)
  46. }
  47. return _u
  48. }
  49. // SetStatus sets the "status" field.
  50. func (_u *BatchImageItemUpdate) SetStatus(v string) *BatchImageItemUpdate {
  51. _u.mutation.SetStatus(v)
  52. return _u
  53. }
  54. // SetNillableStatus sets the "status" field if the given value is not nil.
  55. func (_u *BatchImageItemUpdate) SetNillableStatus(v *string) *BatchImageItemUpdate {
  56. if v != nil {
  57. _u.SetStatus(*v)
  58. }
  59. return _u
  60. }
  61. // SetRequestHash sets the "request_hash" field.
  62. func (_u *BatchImageItemUpdate) SetRequestHash(v string) *BatchImageItemUpdate {
  63. _u.mutation.SetRequestHash(v)
  64. return _u
  65. }
  66. // SetNillableRequestHash sets the "request_hash" field if the given value is not nil.
  67. func (_u *BatchImageItemUpdate) SetNillableRequestHash(v *string) *BatchImageItemUpdate {
  68. if v != nil {
  69. _u.SetRequestHash(*v)
  70. }
  71. return _u
  72. }
  73. // ClearRequestHash clears the value of the "request_hash" field.
  74. func (_u *BatchImageItemUpdate) ClearRequestHash() *BatchImageItemUpdate {
  75. _u.mutation.ClearRequestHash()
  76. return _u
  77. }
  78. // SetPromptPreview sets the "prompt_preview" field.
  79. func (_u *BatchImageItemUpdate) SetPromptPreview(v string) *BatchImageItemUpdate {
  80. _u.mutation.SetPromptPreview(v)
  81. return _u
  82. }
  83. // SetNillablePromptPreview sets the "prompt_preview" field if the given value is not nil.
  84. func (_u *BatchImageItemUpdate) SetNillablePromptPreview(v *string) *BatchImageItemUpdate {
  85. if v != nil {
  86. _u.SetPromptPreview(*v)
  87. }
  88. return _u
  89. }
  90. // ClearPromptPreview clears the value of the "prompt_preview" field.
  91. func (_u *BatchImageItemUpdate) ClearPromptPreview() *BatchImageItemUpdate {
  92. _u.mutation.ClearPromptPreview()
  93. return _u
  94. }
  95. // SetProviderSourceObject sets the "provider_source_object" field.
  96. func (_u *BatchImageItemUpdate) SetProviderSourceObject(v string) *BatchImageItemUpdate {
  97. _u.mutation.SetProviderSourceObject(v)
  98. return _u
  99. }
  100. // SetNillableProviderSourceObject sets the "provider_source_object" field if the given value is not nil.
  101. func (_u *BatchImageItemUpdate) SetNillableProviderSourceObject(v *string) *BatchImageItemUpdate {
  102. if v != nil {
  103. _u.SetProviderSourceObject(*v)
  104. }
  105. return _u
  106. }
  107. // ClearProviderSourceObject clears the value of the "provider_source_object" field.
  108. func (_u *BatchImageItemUpdate) ClearProviderSourceObject() *BatchImageItemUpdate {
  109. _u.mutation.ClearProviderSourceObject()
  110. return _u
  111. }
  112. // SetSourceLineNumber sets the "source_line_number" field.
  113. func (_u *BatchImageItemUpdate) SetSourceLineNumber(v int) *BatchImageItemUpdate {
  114. _u.mutation.ResetSourceLineNumber()
  115. _u.mutation.SetSourceLineNumber(v)
  116. return _u
  117. }
  118. // SetNillableSourceLineNumber sets the "source_line_number" field if the given value is not nil.
  119. func (_u *BatchImageItemUpdate) SetNillableSourceLineNumber(v *int) *BatchImageItemUpdate {
  120. if v != nil {
  121. _u.SetSourceLineNumber(*v)
  122. }
  123. return _u
  124. }
  125. // AddSourceLineNumber adds value to the "source_line_number" field.
  126. func (_u *BatchImageItemUpdate) AddSourceLineNumber(v int) *BatchImageItemUpdate {
  127. _u.mutation.AddSourceLineNumber(v)
  128. return _u
  129. }
  130. // ClearSourceLineNumber clears the value of the "source_line_number" field.
  131. func (_u *BatchImageItemUpdate) ClearSourceLineNumber() *BatchImageItemUpdate {
  132. _u.mutation.ClearSourceLineNumber()
  133. return _u
  134. }
  135. // SetSourceByteOffset sets the "source_byte_offset" field.
  136. func (_u *BatchImageItemUpdate) SetSourceByteOffset(v int64) *BatchImageItemUpdate {
  137. _u.mutation.ResetSourceByteOffset()
  138. _u.mutation.SetSourceByteOffset(v)
  139. return _u
  140. }
  141. // SetNillableSourceByteOffset sets the "source_byte_offset" field if the given value is not nil.
  142. func (_u *BatchImageItemUpdate) SetNillableSourceByteOffset(v *int64) *BatchImageItemUpdate {
  143. if v != nil {
  144. _u.SetSourceByteOffset(*v)
  145. }
  146. return _u
  147. }
  148. // AddSourceByteOffset adds value to the "source_byte_offset" field.
  149. func (_u *BatchImageItemUpdate) AddSourceByteOffset(v int64) *BatchImageItemUpdate {
  150. _u.mutation.AddSourceByteOffset(v)
  151. return _u
  152. }
  153. // ClearSourceByteOffset clears the value of the "source_byte_offset" field.
  154. func (_u *BatchImageItemUpdate) ClearSourceByteOffset() *BatchImageItemUpdate {
  155. _u.mutation.ClearSourceByteOffset()
  156. return _u
  157. }
  158. // SetSourceByteLength sets the "source_byte_length" field.
  159. func (_u *BatchImageItemUpdate) SetSourceByteLength(v int64) *BatchImageItemUpdate {
  160. _u.mutation.ResetSourceByteLength()
  161. _u.mutation.SetSourceByteLength(v)
  162. return _u
  163. }
  164. // SetNillableSourceByteLength sets the "source_byte_length" field if the given value is not nil.
  165. func (_u *BatchImageItemUpdate) SetNillableSourceByteLength(v *int64) *BatchImageItemUpdate {
  166. if v != nil {
  167. _u.SetSourceByteLength(*v)
  168. }
  169. return _u
  170. }
  171. // AddSourceByteLength adds value to the "source_byte_length" field.
  172. func (_u *BatchImageItemUpdate) AddSourceByteLength(v int64) *BatchImageItemUpdate {
  173. _u.mutation.AddSourceByteLength(v)
  174. return _u
  175. }
  176. // ClearSourceByteLength clears the value of the "source_byte_length" field.
  177. func (_u *BatchImageItemUpdate) ClearSourceByteLength() *BatchImageItemUpdate {
  178. _u.mutation.ClearSourceByteLength()
  179. return _u
  180. }
  181. // SetMimeType sets the "mime_type" field.
  182. func (_u *BatchImageItemUpdate) SetMimeType(v string) *BatchImageItemUpdate {
  183. _u.mutation.SetMimeType(v)
  184. return _u
  185. }
  186. // SetNillableMimeType sets the "mime_type" field if the given value is not nil.
  187. func (_u *BatchImageItemUpdate) SetNillableMimeType(v *string) *BatchImageItemUpdate {
  188. if v != nil {
  189. _u.SetMimeType(*v)
  190. }
  191. return _u
  192. }
  193. // ClearMimeType clears the value of the "mime_type" field.
  194. func (_u *BatchImageItemUpdate) ClearMimeType() *BatchImageItemUpdate {
  195. _u.mutation.ClearMimeType()
  196. return _u
  197. }
  198. // SetFileExtension sets the "file_extension" field.
  199. func (_u *BatchImageItemUpdate) SetFileExtension(v string) *BatchImageItemUpdate {
  200. _u.mutation.SetFileExtension(v)
  201. return _u
  202. }
  203. // SetNillableFileExtension sets the "file_extension" field if the given value is not nil.
  204. func (_u *BatchImageItemUpdate) SetNillableFileExtension(v *string) *BatchImageItemUpdate {
  205. if v != nil {
  206. _u.SetFileExtension(*v)
  207. }
  208. return _u
  209. }
  210. // ClearFileExtension clears the value of the "file_extension" field.
  211. func (_u *BatchImageItemUpdate) ClearFileExtension() *BatchImageItemUpdate {
  212. _u.mutation.ClearFileExtension()
  213. return _u
  214. }
  215. // SetImageCount sets the "image_count" field.
  216. func (_u *BatchImageItemUpdate) SetImageCount(v int) *BatchImageItemUpdate {
  217. _u.mutation.ResetImageCount()
  218. _u.mutation.SetImageCount(v)
  219. return _u
  220. }
  221. // SetNillableImageCount sets the "image_count" field if the given value is not nil.
  222. func (_u *BatchImageItemUpdate) SetNillableImageCount(v *int) *BatchImageItemUpdate {
  223. if v != nil {
  224. _u.SetImageCount(*v)
  225. }
  226. return _u
  227. }
  228. // AddImageCount adds value to the "image_count" field.
  229. func (_u *BatchImageItemUpdate) AddImageCount(v int) *BatchImageItemUpdate {
  230. _u.mutation.AddImageCount(v)
  231. return _u
  232. }
  233. // SetErrorCode sets the "error_code" field.
  234. func (_u *BatchImageItemUpdate) SetErrorCode(v string) *BatchImageItemUpdate {
  235. _u.mutation.SetErrorCode(v)
  236. return _u
  237. }
  238. // SetNillableErrorCode sets the "error_code" field if the given value is not nil.
  239. func (_u *BatchImageItemUpdate) SetNillableErrorCode(v *string) *BatchImageItemUpdate {
  240. if v != nil {
  241. _u.SetErrorCode(*v)
  242. }
  243. return _u
  244. }
  245. // ClearErrorCode clears the value of the "error_code" field.
  246. func (_u *BatchImageItemUpdate) ClearErrorCode() *BatchImageItemUpdate {
  247. _u.mutation.ClearErrorCode()
  248. return _u
  249. }
  250. // SetErrorMessage sets the "error_message" field.
  251. func (_u *BatchImageItemUpdate) SetErrorMessage(v string) *BatchImageItemUpdate {
  252. _u.mutation.SetErrorMessage(v)
  253. return _u
  254. }
  255. // SetNillableErrorMessage sets the "error_message" field if the given value is not nil.
  256. func (_u *BatchImageItemUpdate) SetNillableErrorMessage(v *string) *BatchImageItemUpdate {
  257. if v != nil {
  258. _u.SetErrorMessage(*v)
  259. }
  260. return _u
  261. }
  262. // ClearErrorMessage clears the value of the "error_message" field.
  263. func (_u *BatchImageItemUpdate) ClearErrorMessage() *BatchImageItemUpdate {
  264. _u.mutation.ClearErrorMessage()
  265. return _u
  266. }
  267. // SetBilledAmount sets the "billed_amount" field.
  268. func (_u *BatchImageItemUpdate) SetBilledAmount(v float64) *BatchImageItemUpdate {
  269. _u.mutation.ResetBilledAmount()
  270. _u.mutation.SetBilledAmount(v)
  271. return _u
  272. }
  273. // SetNillableBilledAmount sets the "billed_amount" field if the given value is not nil.
  274. func (_u *BatchImageItemUpdate) SetNillableBilledAmount(v *float64) *BatchImageItemUpdate {
  275. if v != nil {
  276. _u.SetBilledAmount(*v)
  277. }
  278. return _u
  279. }
  280. // AddBilledAmount adds value to the "billed_amount" field.
  281. func (_u *BatchImageItemUpdate) AddBilledAmount(v float64) *BatchImageItemUpdate {
  282. _u.mutation.AddBilledAmount(v)
  283. return _u
  284. }
  285. // ClearBilledAmount clears the value of the "billed_amount" field.
  286. func (_u *BatchImageItemUpdate) ClearBilledAmount() *BatchImageItemUpdate {
  287. _u.mutation.ClearBilledAmount()
  288. return _u
  289. }
  290. // SetIndexedAt sets the "indexed_at" field.
  291. func (_u *BatchImageItemUpdate) SetIndexedAt(v time.Time) *BatchImageItemUpdate {
  292. _u.mutation.SetIndexedAt(v)
  293. return _u
  294. }
  295. // SetNillableIndexedAt sets the "indexed_at" field if the given value is not nil.
  296. func (_u *BatchImageItemUpdate) SetNillableIndexedAt(v *time.Time) *BatchImageItemUpdate {
  297. if v != nil {
  298. _u.SetIndexedAt(*v)
  299. }
  300. return _u
  301. }
  302. // ClearIndexedAt clears the value of the "indexed_at" field.
  303. func (_u *BatchImageItemUpdate) ClearIndexedAt() *BatchImageItemUpdate {
  304. _u.mutation.ClearIndexedAt()
  305. return _u
  306. }
  307. // Mutation returns the BatchImageItemMutation object of the builder.
  308. func (_u *BatchImageItemUpdate) Mutation() *BatchImageItemMutation {
  309. return _u.mutation
  310. }
  311. // Save executes the query and returns the number of nodes affected by the update operation.
  312. func (_u *BatchImageItemUpdate) Save(ctx context.Context) (int, error) {
  313. return withHooks(ctx, _u.sqlSave, _u.mutation, _u.hooks)
  314. }
  315. // SaveX is like Save, but panics if an error occurs.
  316. func (_u *BatchImageItemUpdate) SaveX(ctx context.Context) int {
  317. affected, err := _u.Save(ctx)
  318. if err != nil {
  319. panic(err)
  320. }
  321. return affected
  322. }
  323. // Exec executes the query.
  324. func (_u *BatchImageItemUpdate) Exec(ctx context.Context) error {
  325. _, err := _u.Save(ctx)
  326. return err
  327. }
  328. // ExecX is like Exec, but panics if an error occurs.
  329. func (_u *BatchImageItemUpdate) ExecX(ctx context.Context) {
  330. if err := _u.Exec(ctx); err != nil {
  331. panic(err)
  332. }
  333. }
  334. // check runs all checks and user-defined validators on the builder.
  335. func (_u *BatchImageItemUpdate) check() error {
  336. if v, ok := _u.mutation.JobID(); ok {
  337. if err := batchimageitem.JobIDValidator(v); err != nil {
  338. return &ValidationError{Name: "job_id", err: fmt.Errorf(`ent: validator failed for field "BatchImageItem.job_id": %w`, err)}
  339. }
  340. }
  341. if v, ok := _u.mutation.CustomID(); ok {
  342. if err := batchimageitem.CustomIDValidator(v); err != nil {
  343. return &ValidationError{Name: "custom_id", err: fmt.Errorf(`ent: validator failed for field "BatchImageItem.custom_id": %w`, err)}
  344. }
  345. }
  346. if v, ok := _u.mutation.Status(); ok {
  347. if err := batchimageitem.StatusValidator(v); err != nil {
  348. return &ValidationError{Name: "status", err: fmt.Errorf(`ent: validator failed for field "BatchImageItem.status": %w`, err)}
  349. }
  350. }
  351. if v, ok := _u.mutation.RequestHash(); ok {
  352. if err := batchimageitem.RequestHashValidator(v); err != nil {
  353. return &ValidationError{Name: "request_hash", err: fmt.Errorf(`ent: validator failed for field "BatchImageItem.request_hash": %w`, err)}
  354. }
  355. }
  356. if v, ok := _u.mutation.ProviderSourceObject(); ok {
  357. if err := batchimageitem.ProviderSourceObjectValidator(v); err != nil {
  358. return &ValidationError{Name: "provider_source_object", err: fmt.Errorf(`ent: validator failed for field "BatchImageItem.provider_source_object": %w`, err)}
  359. }
  360. }
  361. if v, ok := _u.mutation.MimeType(); ok {
  362. if err := batchimageitem.MimeTypeValidator(v); err != nil {
  363. return &ValidationError{Name: "mime_type", err: fmt.Errorf(`ent: validator failed for field "BatchImageItem.mime_type": %w`, err)}
  364. }
  365. }
  366. if v, ok := _u.mutation.FileExtension(); ok {
  367. if err := batchimageitem.FileExtensionValidator(v); err != nil {
  368. return &ValidationError{Name: "file_extension", err: fmt.Errorf(`ent: validator failed for field "BatchImageItem.file_extension": %w`, err)}
  369. }
  370. }
  371. if v, ok := _u.mutation.ErrorCode(); ok {
  372. if err := batchimageitem.ErrorCodeValidator(v); err != nil {
  373. return &ValidationError{Name: "error_code", err: fmt.Errorf(`ent: validator failed for field "BatchImageItem.error_code": %w`, err)}
  374. }
  375. }
  376. return nil
  377. }
  378. func (_u *BatchImageItemUpdate) sqlSave(ctx context.Context) (_node int, err error) {
  379. if err := _u.check(); err != nil {
  380. return _node, err
  381. }
  382. _spec := sqlgraph.NewUpdateSpec(batchimageitem.Table, batchimageitem.Columns, sqlgraph.NewFieldSpec(batchimageitem.FieldID, field.TypeInt64))
  383. if ps := _u.mutation.predicates; len(ps) > 0 {
  384. _spec.Predicate = func(selector *sql.Selector) {
  385. for i := range ps {
  386. ps[i](selector)
  387. }
  388. }
  389. }
  390. if value, ok := _u.mutation.JobID(); ok {
  391. _spec.SetField(batchimageitem.FieldJobID, field.TypeString, value)
  392. }
  393. if value, ok := _u.mutation.CustomID(); ok {
  394. _spec.SetField(batchimageitem.FieldCustomID, field.TypeString, value)
  395. }
  396. if value, ok := _u.mutation.Status(); ok {
  397. _spec.SetField(batchimageitem.FieldStatus, field.TypeString, value)
  398. }
  399. if value, ok := _u.mutation.RequestHash(); ok {
  400. _spec.SetField(batchimageitem.FieldRequestHash, field.TypeString, value)
  401. }
  402. if _u.mutation.RequestHashCleared() {
  403. _spec.ClearField(batchimageitem.FieldRequestHash, field.TypeString)
  404. }
  405. if value, ok := _u.mutation.PromptPreview(); ok {
  406. _spec.SetField(batchimageitem.FieldPromptPreview, field.TypeString, value)
  407. }
  408. if _u.mutation.PromptPreviewCleared() {
  409. _spec.ClearField(batchimageitem.FieldPromptPreview, field.TypeString)
  410. }
  411. if value, ok := _u.mutation.ProviderSourceObject(); ok {
  412. _spec.SetField(batchimageitem.FieldProviderSourceObject, field.TypeString, value)
  413. }
  414. if _u.mutation.ProviderSourceObjectCleared() {
  415. _spec.ClearField(batchimageitem.FieldProviderSourceObject, field.TypeString)
  416. }
  417. if value, ok := _u.mutation.SourceLineNumber(); ok {
  418. _spec.SetField(batchimageitem.FieldSourceLineNumber, field.TypeInt, value)
  419. }
  420. if value, ok := _u.mutation.AddedSourceLineNumber(); ok {
  421. _spec.AddField(batchimageitem.FieldSourceLineNumber, field.TypeInt, value)
  422. }
  423. if _u.mutation.SourceLineNumberCleared() {
  424. _spec.ClearField(batchimageitem.FieldSourceLineNumber, field.TypeInt)
  425. }
  426. if value, ok := _u.mutation.SourceByteOffset(); ok {
  427. _spec.SetField(batchimageitem.FieldSourceByteOffset, field.TypeInt64, value)
  428. }
  429. if value, ok := _u.mutation.AddedSourceByteOffset(); ok {
  430. _spec.AddField(batchimageitem.FieldSourceByteOffset, field.TypeInt64, value)
  431. }
  432. if _u.mutation.SourceByteOffsetCleared() {
  433. _spec.ClearField(batchimageitem.FieldSourceByteOffset, field.TypeInt64)
  434. }
  435. if value, ok := _u.mutation.SourceByteLength(); ok {
  436. _spec.SetField(batchimageitem.FieldSourceByteLength, field.TypeInt64, value)
  437. }
  438. if value, ok := _u.mutation.AddedSourceByteLength(); ok {
  439. _spec.AddField(batchimageitem.FieldSourceByteLength, field.TypeInt64, value)
  440. }
  441. if _u.mutation.SourceByteLengthCleared() {
  442. _spec.ClearField(batchimageitem.FieldSourceByteLength, field.TypeInt64)
  443. }
  444. if value, ok := _u.mutation.MimeType(); ok {
  445. _spec.SetField(batchimageitem.FieldMimeType, field.TypeString, value)
  446. }
  447. if _u.mutation.MimeTypeCleared() {
  448. _spec.ClearField(batchimageitem.FieldMimeType, field.TypeString)
  449. }
  450. if value, ok := _u.mutation.FileExtension(); ok {
  451. _spec.SetField(batchimageitem.FieldFileExtension, field.TypeString, value)
  452. }
  453. if _u.mutation.FileExtensionCleared() {
  454. _spec.ClearField(batchimageitem.FieldFileExtension, field.TypeString)
  455. }
  456. if value, ok := _u.mutation.ImageCount(); ok {
  457. _spec.SetField(batchimageitem.FieldImageCount, field.TypeInt, value)
  458. }
  459. if value, ok := _u.mutation.AddedImageCount(); ok {
  460. _spec.AddField(batchimageitem.FieldImageCount, field.TypeInt, value)
  461. }
  462. if value, ok := _u.mutation.ErrorCode(); ok {
  463. _spec.SetField(batchimageitem.FieldErrorCode, field.TypeString, value)
  464. }
  465. if _u.mutation.ErrorCodeCleared() {
  466. _spec.ClearField(batchimageitem.FieldErrorCode, field.TypeString)
  467. }
  468. if value, ok := _u.mutation.ErrorMessage(); ok {
  469. _spec.SetField(batchimageitem.FieldErrorMessage, field.TypeString, value)
  470. }
  471. if _u.mutation.ErrorMessageCleared() {
  472. _spec.ClearField(batchimageitem.FieldErrorMessage, field.TypeString)
  473. }
  474. if value, ok := _u.mutation.BilledAmount(); ok {
  475. _spec.SetField(batchimageitem.FieldBilledAmount, field.TypeFloat64, value)
  476. }
  477. if value, ok := _u.mutation.AddedBilledAmount(); ok {
  478. _spec.AddField(batchimageitem.FieldBilledAmount, field.TypeFloat64, value)
  479. }
  480. if _u.mutation.BilledAmountCleared() {
  481. _spec.ClearField(batchimageitem.FieldBilledAmount, field.TypeFloat64)
  482. }
  483. if value, ok := _u.mutation.IndexedAt(); ok {
  484. _spec.SetField(batchimageitem.FieldIndexedAt, field.TypeTime, value)
  485. }
  486. if _u.mutation.IndexedAtCleared() {
  487. _spec.ClearField(batchimageitem.FieldIndexedAt, field.TypeTime)
  488. }
  489. if _node, err = sqlgraph.UpdateNodes(ctx, _u.driver, _spec); err != nil {
  490. if _, ok := err.(*sqlgraph.NotFoundError); ok {
  491. err = &NotFoundError{batchimageitem.Label}
  492. } else if sqlgraph.IsConstraintError(err) {
  493. err = &ConstraintError{msg: err.Error(), wrap: err}
  494. }
  495. return 0, err
  496. }
  497. _u.mutation.done = true
  498. return _node, nil
  499. }
  500. // BatchImageItemUpdateOne is the builder for updating a single BatchImageItem entity.
  501. type BatchImageItemUpdateOne struct {
  502. config
  503. fields []string
  504. hooks []Hook
  505. mutation *BatchImageItemMutation
  506. }
  507. // SetJobID sets the "job_id" field.
  508. func (_u *BatchImageItemUpdateOne) SetJobID(v string) *BatchImageItemUpdateOne {
  509. _u.mutation.SetJobID(v)
  510. return _u
  511. }
  512. // SetNillableJobID sets the "job_id" field if the given value is not nil.
  513. func (_u *BatchImageItemUpdateOne) SetNillableJobID(v *string) *BatchImageItemUpdateOne {
  514. if v != nil {
  515. _u.SetJobID(*v)
  516. }
  517. return _u
  518. }
  519. // SetCustomID sets the "custom_id" field.
  520. func (_u *BatchImageItemUpdateOne) SetCustomID(v string) *BatchImageItemUpdateOne {
  521. _u.mutation.SetCustomID(v)
  522. return _u
  523. }
  524. // SetNillableCustomID sets the "custom_id" field if the given value is not nil.
  525. func (_u *BatchImageItemUpdateOne) SetNillableCustomID(v *string) *BatchImageItemUpdateOne {
  526. if v != nil {
  527. _u.SetCustomID(*v)
  528. }
  529. return _u
  530. }
  531. // SetStatus sets the "status" field.
  532. func (_u *BatchImageItemUpdateOne) SetStatus(v string) *BatchImageItemUpdateOne {
  533. _u.mutation.SetStatus(v)
  534. return _u
  535. }
  536. // SetNillableStatus sets the "status" field if the given value is not nil.
  537. func (_u *BatchImageItemUpdateOne) SetNillableStatus(v *string) *BatchImageItemUpdateOne {
  538. if v != nil {
  539. _u.SetStatus(*v)
  540. }
  541. return _u
  542. }
  543. // SetRequestHash sets the "request_hash" field.
  544. func (_u *BatchImageItemUpdateOne) SetRequestHash(v string) *BatchImageItemUpdateOne {
  545. _u.mutation.SetRequestHash(v)
  546. return _u
  547. }
  548. // SetNillableRequestHash sets the "request_hash" field if the given value is not nil.
  549. func (_u *BatchImageItemUpdateOne) SetNillableRequestHash(v *string) *BatchImageItemUpdateOne {
  550. if v != nil {
  551. _u.SetRequestHash(*v)
  552. }
  553. return _u
  554. }
  555. // ClearRequestHash clears the value of the "request_hash" field.
  556. func (_u *BatchImageItemUpdateOne) ClearRequestHash() *BatchImageItemUpdateOne {
  557. _u.mutation.ClearRequestHash()
  558. return _u
  559. }
  560. // SetPromptPreview sets the "prompt_preview" field.
  561. func (_u *BatchImageItemUpdateOne) SetPromptPreview(v string) *BatchImageItemUpdateOne {
  562. _u.mutation.SetPromptPreview(v)
  563. return _u
  564. }
  565. // SetNillablePromptPreview sets the "prompt_preview" field if the given value is not nil.
  566. func (_u *BatchImageItemUpdateOne) SetNillablePromptPreview(v *string) *BatchImageItemUpdateOne {
  567. if v != nil {
  568. _u.SetPromptPreview(*v)
  569. }
  570. return _u
  571. }
  572. // ClearPromptPreview clears the value of the "prompt_preview" field.
  573. func (_u *BatchImageItemUpdateOne) ClearPromptPreview() *BatchImageItemUpdateOne {
  574. _u.mutation.ClearPromptPreview()
  575. return _u
  576. }
  577. // SetProviderSourceObject sets the "provider_source_object" field.
  578. func (_u *BatchImageItemUpdateOne) SetProviderSourceObject(v string) *BatchImageItemUpdateOne {
  579. _u.mutation.SetProviderSourceObject(v)
  580. return _u
  581. }
  582. // SetNillableProviderSourceObject sets the "provider_source_object" field if the given value is not nil.
  583. func (_u *BatchImageItemUpdateOne) SetNillableProviderSourceObject(v *string) *BatchImageItemUpdateOne {
  584. if v != nil {
  585. _u.SetProviderSourceObject(*v)
  586. }
  587. return _u
  588. }
  589. // ClearProviderSourceObject clears the value of the "provider_source_object" field.
  590. func (_u *BatchImageItemUpdateOne) ClearProviderSourceObject() *BatchImageItemUpdateOne {
  591. _u.mutation.ClearProviderSourceObject()
  592. return _u
  593. }
  594. // SetSourceLineNumber sets the "source_line_number" field.
  595. func (_u *BatchImageItemUpdateOne) SetSourceLineNumber(v int) *BatchImageItemUpdateOne {
  596. _u.mutation.ResetSourceLineNumber()
  597. _u.mutation.SetSourceLineNumber(v)
  598. return _u
  599. }
  600. // SetNillableSourceLineNumber sets the "source_line_number" field if the given value is not nil.
  601. func (_u *BatchImageItemUpdateOne) SetNillableSourceLineNumber(v *int) *BatchImageItemUpdateOne {
  602. if v != nil {
  603. _u.SetSourceLineNumber(*v)
  604. }
  605. return _u
  606. }
  607. // AddSourceLineNumber adds value to the "source_line_number" field.
  608. func (_u *BatchImageItemUpdateOne) AddSourceLineNumber(v int) *BatchImageItemUpdateOne {
  609. _u.mutation.AddSourceLineNumber(v)
  610. return _u
  611. }
  612. // ClearSourceLineNumber clears the value of the "source_line_number" field.
  613. func (_u *BatchImageItemUpdateOne) ClearSourceLineNumber() *BatchImageItemUpdateOne {
  614. _u.mutation.ClearSourceLineNumber()
  615. return _u
  616. }
  617. // SetSourceByteOffset sets the "source_byte_offset" field.
  618. func (_u *BatchImageItemUpdateOne) SetSourceByteOffset(v int64) *BatchImageItemUpdateOne {
  619. _u.mutation.ResetSourceByteOffset()
  620. _u.mutation.SetSourceByteOffset(v)
  621. return _u
  622. }
  623. // SetNillableSourceByteOffset sets the "source_byte_offset" field if the given value is not nil.
  624. func (_u *BatchImageItemUpdateOne) SetNillableSourceByteOffset(v *int64) *BatchImageItemUpdateOne {
  625. if v != nil {
  626. _u.SetSourceByteOffset(*v)
  627. }
  628. return _u
  629. }
  630. // AddSourceByteOffset adds value to the "source_byte_offset" field.
  631. func (_u *BatchImageItemUpdateOne) AddSourceByteOffset(v int64) *BatchImageItemUpdateOne {
  632. _u.mutation.AddSourceByteOffset(v)
  633. return _u
  634. }
  635. // ClearSourceByteOffset clears the value of the "source_byte_offset" field.
  636. func (_u *BatchImageItemUpdateOne) ClearSourceByteOffset() *BatchImageItemUpdateOne {
  637. _u.mutation.ClearSourceByteOffset()
  638. return _u
  639. }
  640. // SetSourceByteLength sets the "source_byte_length" field.
  641. func (_u *BatchImageItemUpdateOne) SetSourceByteLength(v int64) *BatchImageItemUpdateOne {
  642. _u.mutation.ResetSourceByteLength()
  643. _u.mutation.SetSourceByteLength(v)
  644. return _u
  645. }
  646. // SetNillableSourceByteLength sets the "source_byte_length" field if the given value is not nil.
  647. func (_u *BatchImageItemUpdateOne) SetNillableSourceByteLength(v *int64) *BatchImageItemUpdateOne {
  648. if v != nil {
  649. _u.SetSourceByteLength(*v)
  650. }
  651. return _u
  652. }
  653. // AddSourceByteLength adds value to the "source_byte_length" field.
  654. func (_u *BatchImageItemUpdateOne) AddSourceByteLength(v int64) *BatchImageItemUpdateOne {
  655. _u.mutation.AddSourceByteLength(v)
  656. return _u
  657. }
  658. // ClearSourceByteLength clears the value of the "source_byte_length" field.
  659. func (_u *BatchImageItemUpdateOne) ClearSourceByteLength() *BatchImageItemUpdateOne {
  660. _u.mutation.ClearSourceByteLength()
  661. return _u
  662. }
  663. // SetMimeType sets the "mime_type" field.
  664. func (_u *BatchImageItemUpdateOne) SetMimeType(v string) *BatchImageItemUpdateOne {
  665. _u.mutation.SetMimeType(v)
  666. return _u
  667. }
  668. // SetNillableMimeType sets the "mime_type" field if the given value is not nil.
  669. func (_u *BatchImageItemUpdateOne) SetNillableMimeType(v *string) *BatchImageItemUpdateOne {
  670. if v != nil {
  671. _u.SetMimeType(*v)
  672. }
  673. return _u
  674. }
  675. // ClearMimeType clears the value of the "mime_type" field.
  676. func (_u *BatchImageItemUpdateOne) ClearMimeType() *BatchImageItemUpdateOne {
  677. _u.mutation.ClearMimeType()
  678. return _u
  679. }
  680. // SetFileExtension sets the "file_extension" field.
  681. func (_u *BatchImageItemUpdateOne) SetFileExtension(v string) *BatchImageItemUpdateOne {
  682. _u.mutation.SetFileExtension(v)
  683. return _u
  684. }
  685. // SetNillableFileExtension sets the "file_extension" field if the given value is not nil.
  686. func (_u *BatchImageItemUpdateOne) SetNillableFileExtension(v *string) *BatchImageItemUpdateOne {
  687. if v != nil {
  688. _u.SetFileExtension(*v)
  689. }
  690. return _u
  691. }
  692. // ClearFileExtension clears the value of the "file_extension" field.
  693. func (_u *BatchImageItemUpdateOne) ClearFileExtension() *BatchImageItemUpdateOne {
  694. _u.mutation.ClearFileExtension()
  695. return _u
  696. }
  697. // SetImageCount sets the "image_count" field.
  698. func (_u *BatchImageItemUpdateOne) SetImageCount(v int) *BatchImageItemUpdateOne {
  699. _u.mutation.ResetImageCount()
  700. _u.mutation.SetImageCount(v)
  701. return _u
  702. }
  703. // SetNillableImageCount sets the "image_count" field if the given value is not nil.
  704. func (_u *BatchImageItemUpdateOne) SetNillableImageCount(v *int) *BatchImageItemUpdateOne {
  705. if v != nil {
  706. _u.SetImageCount(*v)
  707. }
  708. return _u
  709. }
  710. // AddImageCount adds value to the "image_count" field.
  711. func (_u *BatchImageItemUpdateOne) AddImageCount(v int) *BatchImageItemUpdateOne {
  712. _u.mutation.AddImageCount(v)
  713. return _u
  714. }
  715. // SetErrorCode sets the "error_code" field.
  716. func (_u *BatchImageItemUpdateOne) SetErrorCode(v string) *BatchImageItemUpdateOne {
  717. _u.mutation.SetErrorCode(v)
  718. return _u
  719. }
  720. // SetNillableErrorCode sets the "error_code" field if the given value is not nil.
  721. func (_u *BatchImageItemUpdateOne) SetNillableErrorCode(v *string) *BatchImageItemUpdateOne {
  722. if v != nil {
  723. _u.SetErrorCode(*v)
  724. }
  725. return _u
  726. }
  727. // ClearErrorCode clears the value of the "error_code" field.
  728. func (_u *BatchImageItemUpdateOne) ClearErrorCode() *BatchImageItemUpdateOne {
  729. _u.mutation.ClearErrorCode()
  730. return _u
  731. }
  732. // SetErrorMessage sets the "error_message" field.
  733. func (_u *BatchImageItemUpdateOne) SetErrorMessage(v string) *BatchImageItemUpdateOne {
  734. _u.mutation.SetErrorMessage(v)
  735. return _u
  736. }
  737. // SetNillableErrorMessage sets the "error_message" field if the given value is not nil.
  738. func (_u *BatchImageItemUpdateOne) SetNillableErrorMessage(v *string) *BatchImageItemUpdateOne {
  739. if v != nil {
  740. _u.SetErrorMessage(*v)
  741. }
  742. return _u
  743. }
  744. // ClearErrorMessage clears the value of the "error_message" field.
  745. func (_u *BatchImageItemUpdateOne) ClearErrorMessage() *BatchImageItemUpdateOne {
  746. _u.mutation.ClearErrorMessage()
  747. return _u
  748. }
  749. // SetBilledAmount sets the "billed_amount" field.
  750. func (_u *BatchImageItemUpdateOne) SetBilledAmount(v float64) *BatchImageItemUpdateOne {
  751. _u.mutation.ResetBilledAmount()
  752. _u.mutation.SetBilledAmount(v)
  753. return _u
  754. }
  755. // SetNillableBilledAmount sets the "billed_amount" field if the given value is not nil.
  756. func (_u *BatchImageItemUpdateOne) SetNillableBilledAmount(v *float64) *BatchImageItemUpdateOne {
  757. if v != nil {
  758. _u.SetBilledAmount(*v)
  759. }
  760. return _u
  761. }
  762. // AddBilledAmount adds value to the "billed_amount" field.
  763. func (_u *BatchImageItemUpdateOne) AddBilledAmount(v float64) *BatchImageItemUpdateOne {
  764. _u.mutation.AddBilledAmount(v)
  765. return _u
  766. }
  767. // ClearBilledAmount clears the value of the "billed_amount" field.
  768. func (_u *BatchImageItemUpdateOne) ClearBilledAmount() *BatchImageItemUpdateOne {
  769. _u.mutation.ClearBilledAmount()
  770. return _u
  771. }
  772. // SetIndexedAt sets the "indexed_at" field.
  773. func (_u *BatchImageItemUpdateOne) SetIndexedAt(v time.Time) *BatchImageItemUpdateOne {
  774. _u.mutation.SetIndexedAt(v)
  775. return _u
  776. }
  777. // SetNillableIndexedAt sets the "indexed_at" field if the given value is not nil.
  778. func (_u *BatchImageItemUpdateOne) SetNillableIndexedAt(v *time.Time) *BatchImageItemUpdateOne {
  779. if v != nil {
  780. _u.SetIndexedAt(*v)
  781. }
  782. return _u
  783. }
  784. // ClearIndexedAt clears the value of the "indexed_at" field.
  785. func (_u *BatchImageItemUpdateOne) ClearIndexedAt() *BatchImageItemUpdateOne {
  786. _u.mutation.ClearIndexedAt()
  787. return _u
  788. }
  789. // Mutation returns the BatchImageItemMutation object of the builder.
  790. func (_u *BatchImageItemUpdateOne) Mutation() *BatchImageItemMutation {
  791. return _u.mutation
  792. }
  793. // Where appends a list predicates to the BatchImageItemUpdate builder.
  794. func (_u *BatchImageItemUpdateOne) Where(ps ...predicate.BatchImageItem) *BatchImageItemUpdateOne {
  795. _u.mutation.Where(ps...)
  796. return _u
  797. }
  798. // Select allows selecting one or more fields (columns) of the returned entity.
  799. // The default is selecting all fields defined in the entity schema.
  800. func (_u *BatchImageItemUpdateOne) Select(field string, fields ...string) *BatchImageItemUpdateOne {
  801. _u.fields = append([]string{field}, fields...)
  802. return _u
  803. }
  804. // Save executes the query and returns the updated BatchImageItem entity.
  805. func (_u *BatchImageItemUpdateOne) Save(ctx context.Context) (*BatchImageItem, error) {
  806. return withHooks(ctx, _u.sqlSave, _u.mutation, _u.hooks)
  807. }
  808. // SaveX is like Save, but panics if an error occurs.
  809. func (_u *BatchImageItemUpdateOne) SaveX(ctx context.Context) *BatchImageItem {
  810. node, err := _u.Save(ctx)
  811. if err != nil {
  812. panic(err)
  813. }
  814. return node
  815. }
  816. // Exec executes the query on the entity.
  817. func (_u *BatchImageItemUpdateOne) Exec(ctx context.Context) error {
  818. _, err := _u.Save(ctx)
  819. return err
  820. }
  821. // ExecX is like Exec, but panics if an error occurs.
  822. func (_u *BatchImageItemUpdateOne) ExecX(ctx context.Context) {
  823. if err := _u.Exec(ctx); err != nil {
  824. panic(err)
  825. }
  826. }
  827. // check runs all checks and user-defined validators on the builder.
  828. func (_u *BatchImageItemUpdateOne) check() error {
  829. if v, ok := _u.mutation.JobID(); ok {
  830. if err := batchimageitem.JobIDValidator(v); err != nil {
  831. return &ValidationError{Name: "job_id", err: fmt.Errorf(`ent: validator failed for field "BatchImageItem.job_id": %w`, err)}
  832. }
  833. }
  834. if v, ok := _u.mutation.CustomID(); ok {
  835. if err := batchimageitem.CustomIDValidator(v); err != nil {
  836. return &ValidationError{Name: "custom_id", err: fmt.Errorf(`ent: validator failed for field "BatchImageItem.custom_id": %w`, err)}
  837. }
  838. }
  839. if v, ok := _u.mutation.Status(); ok {
  840. if err := batchimageitem.StatusValidator(v); err != nil {
  841. return &ValidationError{Name: "status", err: fmt.Errorf(`ent: validator failed for field "BatchImageItem.status": %w`, err)}
  842. }
  843. }
  844. if v, ok := _u.mutation.RequestHash(); ok {
  845. if err := batchimageitem.RequestHashValidator(v); err != nil {
  846. return &ValidationError{Name: "request_hash", err: fmt.Errorf(`ent: validator failed for field "BatchImageItem.request_hash": %w`, err)}
  847. }
  848. }
  849. if v, ok := _u.mutation.ProviderSourceObject(); ok {
  850. if err := batchimageitem.ProviderSourceObjectValidator(v); err != nil {
  851. return &ValidationError{Name: "provider_source_object", err: fmt.Errorf(`ent: validator failed for field "BatchImageItem.provider_source_object": %w`, err)}
  852. }
  853. }
  854. if v, ok := _u.mutation.MimeType(); ok {
  855. if err := batchimageitem.MimeTypeValidator(v); err != nil {
  856. return &ValidationError{Name: "mime_type", err: fmt.Errorf(`ent: validator failed for field "BatchImageItem.mime_type": %w`, err)}
  857. }
  858. }
  859. if v, ok := _u.mutation.FileExtension(); ok {
  860. if err := batchimageitem.FileExtensionValidator(v); err != nil {
  861. return &ValidationError{Name: "file_extension", err: fmt.Errorf(`ent: validator failed for field "BatchImageItem.file_extension": %w`, err)}
  862. }
  863. }
  864. if v, ok := _u.mutation.ErrorCode(); ok {
  865. if err := batchimageitem.ErrorCodeValidator(v); err != nil {
  866. return &ValidationError{Name: "error_code", err: fmt.Errorf(`ent: validator failed for field "BatchImageItem.error_code": %w`, err)}
  867. }
  868. }
  869. return nil
  870. }
  871. func (_u *BatchImageItemUpdateOne) sqlSave(ctx context.Context) (_node *BatchImageItem, err error) {
  872. if err := _u.check(); err != nil {
  873. return _node, err
  874. }
  875. _spec := sqlgraph.NewUpdateSpec(batchimageitem.Table, batchimageitem.Columns, sqlgraph.NewFieldSpec(batchimageitem.FieldID, field.TypeInt64))
  876. id, ok := _u.mutation.ID()
  877. if !ok {
  878. return nil, &ValidationError{Name: "id", err: errors.New(`ent: missing "BatchImageItem.id" for update`)}
  879. }
  880. _spec.Node.ID.Value = id
  881. if fields := _u.fields; len(fields) > 0 {
  882. _spec.Node.Columns = make([]string, 0, len(fields))
  883. _spec.Node.Columns = append(_spec.Node.Columns, batchimageitem.FieldID)
  884. for _, f := range fields {
  885. if !batchimageitem.ValidColumn(f) {
  886. return nil, &ValidationError{Name: f, err: fmt.Errorf("ent: invalid field %q for query", f)}
  887. }
  888. if f != batchimageitem.FieldID {
  889. _spec.Node.Columns = append(_spec.Node.Columns, f)
  890. }
  891. }
  892. }
  893. if ps := _u.mutation.predicates; len(ps) > 0 {
  894. _spec.Predicate = func(selector *sql.Selector) {
  895. for i := range ps {
  896. ps[i](selector)
  897. }
  898. }
  899. }
  900. if value, ok := _u.mutation.JobID(); ok {
  901. _spec.SetField(batchimageitem.FieldJobID, field.TypeString, value)
  902. }
  903. if value, ok := _u.mutation.CustomID(); ok {
  904. _spec.SetField(batchimageitem.FieldCustomID, field.TypeString, value)
  905. }
  906. if value, ok := _u.mutation.Status(); ok {
  907. _spec.SetField(batchimageitem.FieldStatus, field.TypeString, value)
  908. }
  909. if value, ok := _u.mutation.RequestHash(); ok {
  910. _spec.SetField(batchimageitem.FieldRequestHash, field.TypeString, value)
  911. }
  912. if _u.mutation.RequestHashCleared() {
  913. _spec.ClearField(batchimageitem.FieldRequestHash, field.TypeString)
  914. }
  915. if value, ok := _u.mutation.PromptPreview(); ok {
  916. _spec.SetField(batchimageitem.FieldPromptPreview, field.TypeString, value)
  917. }
  918. if _u.mutation.PromptPreviewCleared() {
  919. _spec.ClearField(batchimageitem.FieldPromptPreview, field.TypeString)
  920. }
  921. if value, ok := _u.mutation.ProviderSourceObject(); ok {
  922. _spec.SetField(batchimageitem.FieldProviderSourceObject, field.TypeString, value)
  923. }
  924. if _u.mutation.ProviderSourceObjectCleared() {
  925. _spec.ClearField(batchimageitem.FieldProviderSourceObject, field.TypeString)
  926. }
  927. if value, ok := _u.mutation.SourceLineNumber(); ok {
  928. _spec.SetField(batchimageitem.FieldSourceLineNumber, field.TypeInt, value)
  929. }
  930. if value, ok := _u.mutation.AddedSourceLineNumber(); ok {
  931. _spec.AddField(batchimageitem.FieldSourceLineNumber, field.TypeInt, value)
  932. }
  933. if _u.mutation.SourceLineNumberCleared() {
  934. _spec.ClearField(batchimageitem.FieldSourceLineNumber, field.TypeInt)
  935. }
  936. if value, ok := _u.mutation.SourceByteOffset(); ok {
  937. _spec.SetField(batchimageitem.FieldSourceByteOffset, field.TypeInt64, value)
  938. }
  939. if value, ok := _u.mutation.AddedSourceByteOffset(); ok {
  940. _spec.AddField(batchimageitem.FieldSourceByteOffset, field.TypeInt64, value)
  941. }
  942. if _u.mutation.SourceByteOffsetCleared() {
  943. _spec.ClearField(batchimageitem.FieldSourceByteOffset, field.TypeInt64)
  944. }
  945. if value, ok := _u.mutation.SourceByteLength(); ok {
  946. _spec.SetField(batchimageitem.FieldSourceByteLength, field.TypeInt64, value)
  947. }
  948. if value, ok := _u.mutation.AddedSourceByteLength(); ok {
  949. _spec.AddField(batchimageitem.FieldSourceByteLength, field.TypeInt64, value)
  950. }
  951. if _u.mutation.SourceByteLengthCleared() {
  952. _spec.ClearField(batchimageitem.FieldSourceByteLength, field.TypeInt64)
  953. }
  954. if value, ok := _u.mutation.MimeType(); ok {
  955. _spec.SetField(batchimageitem.FieldMimeType, field.TypeString, value)
  956. }
  957. if _u.mutation.MimeTypeCleared() {
  958. _spec.ClearField(batchimageitem.FieldMimeType, field.TypeString)
  959. }
  960. if value, ok := _u.mutation.FileExtension(); ok {
  961. _spec.SetField(batchimageitem.FieldFileExtension, field.TypeString, value)
  962. }
  963. if _u.mutation.FileExtensionCleared() {
  964. _spec.ClearField(batchimageitem.FieldFileExtension, field.TypeString)
  965. }
  966. if value, ok := _u.mutation.ImageCount(); ok {
  967. _spec.SetField(batchimageitem.FieldImageCount, field.TypeInt, value)
  968. }
  969. if value, ok := _u.mutation.AddedImageCount(); ok {
  970. _spec.AddField(batchimageitem.FieldImageCount, field.TypeInt, value)
  971. }
  972. if value, ok := _u.mutation.ErrorCode(); ok {
  973. _spec.SetField(batchimageitem.FieldErrorCode, field.TypeString, value)
  974. }
  975. if _u.mutation.ErrorCodeCleared() {
  976. _spec.ClearField(batchimageitem.FieldErrorCode, field.TypeString)
  977. }
  978. if value, ok := _u.mutation.ErrorMessage(); ok {
  979. _spec.SetField(batchimageitem.FieldErrorMessage, field.TypeString, value)
  980. }
  981. if _u.mutation.ErrorMessageCleared() {
  982. _spec.ClearField(batchimageitem.FieldErrorMessage, field.TypeString)
  983. }
  984. if value, ok := _u.mutation.BilledAmount(); ok {
  985. _spec.SetField(batchimageitem.FieldBilledAmount, field.TypeFloat64, value)
  986. }
  987. if value, ok := _u.mutation.AddedBilledAmount(); ok {
  988. _spec.AddField(batchimageitem.FieldBilledAmount, field.TypeFloat64, value)
  989. }
  990. if _u.mutation.BilledAmountCleared() {
  991. _spec.ClearField(batchimageitem.FieldBilledAmount, field.TypeFloat64)
  992. }
  993. if value, ok := _u.mutation.IndexedAt(); ok {
  994. _spec.SetField(batchimageitem.FieldIndexedAt, field.TypeTime, value)
  995. }
  996. if _u.mutation.IndexedAtCleared() {
  997. _spec.ClearField(batchimageitem.FieldIndexedAt, field.TypeTime)
  998. }
  999. _node = &BatchImageItem{config: _u.config}
  1000. _spec.Assign = _node.assignValues
  1001. _spec.ScanValues = _node.scanValues
  1002. if err = sqlgraph.UpdateNode(ctx, _u.driver, _spec); err != nil {
  1003. if _, ok := err.(*sqlgraph.NotFoundError); ok {
  1004. err = &NotFoundError{batchimageitem.Label}
  1005. } else if sqlgraph.IsConstraintError(err) {
  1006. err = &ConstraintError{msg: err.Error(), wrap: err}
  1007. }
  1008. return nil, err
  1009. }
  1010. _u.mutation.done = true
  1011. return _node, nil
  1012. }