12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121112211231124112511261127112811291130113111321133113411351136113711381139114011411142114311441145114611471148114911501151115211531154115511561157115811591160116111621163116411651166116711681169117011711172117311741175117611771178117911801181118211831184 |
- // Code generated by ent, DO NOT EDIT.
- package ent
- import (
- "context"
- "errors"
- "fmt"
- "time"
- "wechat-api/ent/batchmsg"
- "wechat-api/ent/predicate"
- "entgo.io/ent/dialect/sql"
- "entgo.io/ent/dialect/sql/sqlgraph"
- "entgo.io/ent/schema/field"
- )
- // BatchMsgUpdate is the builder for updating BatchMsg entities.
- type BatchMsgUpdate struct {
- config
- hooks []Hook
- mutation *BatchMsgMutation
- }
- // Where appends a list predicates to the BatchMsgUpdate builder.
- func (bmu *BatchMsgUpdate) Where(ps ...predicate.BatchMsg) *BatchMsgUpdate {
- bmu.mutation.Where(ps...)
- return bmu
- }
- // SetUpdatedAt sets the "updated_at" field.
- func (bmu *BatchMsgUpdate) SetUpdatedAt(t time.Time) *BatchMsgUpdate {
- bmu.mutation.SetUpdatedAt(t)
- return bmu
- }
- // SetDeletedAt sets the "deleted_at" field.
- func (bmu *BatchMsgUpdate) SetDeletedAt(t time.Time) *BatchMsgUpdate {
- bmu.mutation.SetDeletedAt(t)
- return bmu
- }
- // SetNillableDeletedAt sets the "deleted_at" field if the given value is not nil.
- func (bmu *BatchMsgUpdate) SetNillableDeletedAt(t *time.Time) *BatchMsgUpdate {
- if t != nil {
- bmu.SetDeletedAt(*t)
- }
- return bmu
- }
- // ClearDeletedAt clears the value of the "deleted_at" field.
- func (bmu *BatchMsgUpdate) ClearDeletedAt() *BatchMsgUpdate {
- bmu.mutation.ClearDeletedAt()
- return bmu
- }
- // SetStatus sets the "status" field.
- func (bmu *BatchMsgUpdate) SetStatus(u uint8) *BatchMsgUpdate {
- bmu.mutation.ResetStatus()
- bmu.mutation.SetStatus(u)
- return bmu
- }
- // SetNillableStatus sets the "status" field if the given value is not nil.
- func (bmu *BatchMsgUpdate) SetNillableStatus(u *uint8) *BatchMsgUpdate {
- if u != nil {
- bmu.SetStatus(*u)
- }
- return bmu
- }
- // AddStatus adds u to the "status" field.
- func (bmu *BatchMsgUpdate) AddStatus(u int8) *BatchMsgUpdate {
- bmu.mutation.AddStatus(u)
- return bmu
- }
- // ClearStatus clears the value of the "status" field.
- func (bmu *BatchMsgUpdate) ClearStatus() *BatchMsgUpdate {
- bmu.mutation.ClearStatus()
- return bmu
- }
- // SetBatchNo sets the "batch_no" field.
- func (bmu *BatchMsgUpdate) SetBatchNo(s string) *BatchMsgUpdate {
- bmu.mutation.SetBatchNo(s)
- return bmu
- }
- // SetNillableBatchNo sets the "batch_no" field if the given value is not nil.
- func (bmu *BatchMsgUpdate) SetNillableBatchNo(s *string) *BatchMsgUpdate {
- if s != nil {
- bmu.SetBatchNo(*s)
- }
- return bmu
- }
- // ClearBatchNo clears the value of the "batch_no" field.
- func (bmu *BatchMsgUpdate) ClearBatchNo() *BatchMsgUpdate {
- bmu.mutation.ClearBatchNo()
- return bmu
- }
- // SetTaskName sets the "task_name" field.
- func (bmu *BatchMsgUpdate) SetTaskName(s string) *BatchMsgUpdate {
- bmu.mutation.SetTaskName(s)
- return bmu
- }
- // SetNillableTaskName sets the "task_name" field if the given value is not nil.
- func (bmu *BatchMsgUpdate) SetNillableTaskName(s *string) *BatchMsgUpdate {
- if s != nil {
- bmu.SetTaskName(*s)
- }
- return bmu
- }
- // ClearTaskName clears the value of the "task_name" field.
- func (bmu *BatchMsgUpdate) ClearTaskName() *BatchMsgUpdate {
- bmu.mutation.ClearTaskName()
- return bmu
- }
- // SetFromwxid sets the "fromwxid" field.
- func (bmu *BatchMsgUpdate) SetFromwxid(s string) *BatchMsgUpdate {
- bmu.mutation.SetFromwxid(s)
- return bmu
- }
- // SetNillableFromwxid sets the "fromwxid" field if the given value is not nil.
- func (bmu *BatchMsgUpdate) SetNillableFromwxid(s *string) *BatchMsgUpdate {
- if s != nil {
- bmu.SetFromwxid(*s)
- }
- return bmu
- }
- // ClearFromwxid clears the value of the "fromwxid" field.
- func (bmu *BatchMsgUpdate) ClearFromwxid() *BatchMsgUpdate {
- bmu.mutation.ClearFromwxid()
- return bmu
- }
- // SetMsg sets the "msg" field.
- func (bmu *BatchMsgUpdate) SetMsg(s string) *BatchMsgUpdate {
- bmu.mutation.SetMsg(s)
- return bmu
- }
- // SetNillableMsg sets the "msg" field if the given value is not nil.
- func (bmu *BatchMsgUpdate) SetNillableMsg(s *string) *BatchMsgUpdate {
- if s != nil {
- bmu.SetMsg(*s)
- }
- return bmu
- }
- // ClearMsg clears the value of the "msg" field.
- func (bmu *BatchMsgUpdate) ClearMsg() *BatchMsgUpdate {
- bmu.mutation.ClearMsg()
- return bmu
- }
- // SetTag sets the "tag" field.
- func (bmu *BatchMsgUpdate) SetTag(s string) *BatchMsgUpdate {
- bmu.mutation.SetTag(s)
- return bmu
- }
- // SetNillableTag sets the "tag" field if the given value is not nil.
- func (bmu *BatchMsgUpdate) SetNillableTag(s *string) *BatchMsgUpdate {
- if s != nil {
- bmu.SetTag(*s)
- }
- return bmu
- }
- // ClearTag clears the value of the "tag" field.
- func (bmu *BatchMsgUpdate) ClearTag() *BatchMsgUpdate {
- bmu.mutation.ClearTag()
- return bmu
- }
- // SetTagids sets the "tagids" field.
- func (bmu *BatchMsgUpdate) SetTagids(s string) *BatchMsgUpdate {
- bmu.mutation.SetTagids(s)
- return bmu
- }
- // SetNillableTagids sets the "tagids" field if the given value is not nil.
- func (bmu *BatchMsgUpdate) SetNillableTagids(s *string) *BatchMsgUpdate {
- if s != nil {
- bmu.SetTagids(*s)
- }
- return bmu
- }
- // ClearTagids clears the value of the "tagids" field.
- func (bmu *BatchMsgUpdate) ClearTagids() *BatchMsgUpdate {
- bmu.mutation.ClearTagids()
- return bmu
- }
- // SetTotal sets the "total" field.
- func (bmu *BatchMsgUpdate) SetTotal(i int32) *BatchMsgUpdate {
- bmu.mutation.ResetTotal()
- bmu.mutation.SetTotal(i)
- return bmu
- }
- // SetNillableTotal sets the "total" field if the given value is not nil.
- func (bmu *BatchMsgUpdate) SetNillableTotal(i *int32) *BatchMsgUpdate {
- if i != nil {
- bmu.SetTotal(*i)
- }
- return bmu
- }
- // AddTotal adds i to the "total" field.
- func (bmu *BatchMsgUpdate) AddTotal(i int32) *BatchMsgUpdate {
- bmu.mutation.AddTotal(i)
- return bmu
- }
- // ClearTotal clears the value of the "total" field.
- func (bmu *BatchMsgUpdate) ClearTotal() *BatchMsgUpdate {
- bmu.mutation.ClearTotal()
- return bmu
- }
- // SetSuccess sets the "success" field.
- func (bmu *BatchMsgUpdate) SetSuccess(i int32) *BatchMsgUpdate {
- bmu.mutation.ResetSuccess()
- bmu.mutation.SetSuccess(i)
- return bmu
- }
- // SetNillableSuccess sets the "success" field if the given value is not nil.
- func (bmu *BatchMsgUpdate) SetNillableSuccess(i *int32) *BatchMsgUpdate {
- if i != nil {
- bmu.SetSuccess(*i)
- }
- return bmu
- }
- // AddSuccess adds i to the "success" field.
- func (bmu *BatchMsgUpdate) AddSuccess(i int32) *BatchMsgUpdate {
- bmu.mutation.AddSuccess(i)
- return bmu
- }
- // ClearSuccess clears the value of the "success" field.
- func (bmu *BatchMsgUpdate) ClearSuccess() *BatchMsgUpdate {
- bmu.mutation.ClearSuccess()
- return bmu
- }
- // SetFail sets the "fail" field.
- func (bmu *BatchMsgUpdate) SetFail(i int32) *BatchMsgUpdate {
- bmu.mutation.ResetFail()
- bmu.mutation.SetFail(i)
- return bmu
- }
- // SetNillableFail sets the "fail" field if the given value is not nil.
- func (bmu *BatchMsgUpdate) SetNillableFail(i *int32) *BatchMsgUpdate {
- if i != nil {
- bmu.SetFail(*i)
- }
- return bmu
- }
- // AddFail adds i to the "fail" field.
- func (bmu *BatchMsgUpdate) AddFail(i int32) *BatchMsgUpdate {
- bmu.mutation.AddFail(i)
- return bmu
- }
- // ClearFail clears the value of the "fail" field.
- func (bmu *BatchMsgUpdate) ClearFail() *BatchMsgUpdate {
- bmu.mutation.ClearFail()
- return bmu
- }
- // SetStartTime sets the "start_time" field.
- func (bmu *BatchMsgUpdate) SetStartTime(t time.Time) *BatchMsgUpdate {
- bmu.mutation.SetStartTime(t)
- return bmu
- }
- // SetNillableStartTime sets the "start_time" field if the given value is not nil.
- func (bmu *BatchMsgUpdate) SetNillableStartTime(t *time.Time) *BatchMsgUpdate {
- if t != nil {
- bmu.SetStartTime(*t)
- }
- return bmu
- }
- // ClearStartTime clears the value of the "start_time" field.
- func (bmu *BatchMsgUpdate) ClearStartTime() *BatchMsgUpdate {
- bmu.mutation.ClearStartTime()
- return bmu
- }
- // SetStopTime sets the "stop_time" field.
- func (bmu *BatchMsgUpdate) SetStopTime(t time.Time) *BatchMsgUpdate {
- bmu.mutation.SetStopTime(t)
- return bmu
- }
- // SetNillableStopTime sets the "stop_time" field if the given value is not nil.
- func (bmu *BatchMsgUpdate) SetNillableStopTime(t *time.Time) *BatchMsgUpdate {
- if t != nil {
- bmu.SetStopTime(*t)
- }
- return bmu
- }
- // ClearStopTime clears the value of the "stop_time" field.
- func (bmu *BatchMsgUpdate) ClearStopTime() *BatchMsgUpdate {
- bmu.mutation.ClearStopTime()
- return bmu
- }
- // SetSendTime sets the "send_time" field.
- func (bmu *BatchMsgUpdate) SetSendTime(t time.Time) *BatchMsgUpdate {
- bmu.mutation.SetSendTime(t)
- return bmu
- }
- // SetNillableSendTime sets the "send_time" field if the given value is not nil.
- func (bmu *BatchMsgUpdate) SetNillableSendTime(t *time.Time) *BatchMsgUpdate {
- if t != nil {
- bmu.SetSendTime(*t)
- }
- return bmu
- }
- // ClearSendTime clears the value of the "send_time" field.
- func (bmu *BatchMsgUpdate) ClearSendTime() *BatchMsgUpdate {
- bmu.mutation.ClearSendTime()
- return bmu
- }
- // SetType sets the "type" field.
- func (bmu *BatchMsgUpdate) SetType(i int32) *BatchMsgUpdate {
- bmu.mutation.ResetType()
- bmu.mutation.SetType(i)
- return bmu
- }
- // SetNillableType sets the "type" field if the given value is not nil.
- func (bmu *BatchMsgUpdate) SetNillableType(i *int32) *BatchMsgUpdate {
- if i != nil {
- bmu.SetType(*i)
- }
- return bmu
- }
- // AddType adds i to the "type" field.
- func (bmu *BatchMsgUpdate) AddType(i int32) *BatchMsgUpdate {
- bmu.mutation.AddType(i)
- return bmu
- }
- // ClearType clears the value of the "type" field.
- func (bmu *BatchMsgUpdate) ClearType() *BatchMsgUpdate {
- bmu.mutation.ClearType()
- return bmu
- }
- // SetOrganizationID sets the "organization_id" field.
- func (bmu *BatchMsgUpdate) SetOrganizationID(u uint64) *BatchMsgUpdate {
- bmu.mutation.ResetOrganizationID()
- bmu.mutation.SetOrganizationID(u)
- return bmu
- }
- // SetNillableOrganizationID sets the "organization_id" field if the given value is not nil.
- func (bmu *BatchMsgUpdate) SetNillableOrganizationID(u *uint64) *BatchMsgUpdate {
- if u != nil {
- bmu.SetOrganizationID(*u)
- }
- return bmu
- }
- // AddOrganizationID adds u to the "organization_id" field.
- func (bmu *BatchMsgUpdate) AddOrganizationID(u int64) *BatchMsgUpdate {
- bmu.mutation.AddOrganizationID(u)
- return bmu
- }
- // Mutation returns the BatchMsgMutation object of the builder.
- func (bmu *BatchMsgUpdate) Mutation() *BatchMsgMutation {
- return bmu.mutation
- }
- // Save executes the query and returns the number of nodes affected by the update operation.
- func (bmu *BatchMsgUpdate) Save(ctx context.Context) (int, error) {
- if err := bmu.defaults(); err != nil {
- return 0, err
- }
- return withHooks(ctx, bmu.sqlSave, bmu.mutation, bmu.hooks)
- }
- // SaveX is like Save, but panics if an error occurs.
- func (bmu *BatchMsgUpdate) SaveX(ctx context.Context) int {
- affected, err := bmu.Save(ctx)
- if err != nil {
- panic(err)
- }
- return affected
- }
- // Exec executes the query.
- func (bmu *BatchMsgUpdate) Exec(ctx context.Context) error {
- _, err := bmu.Save(ctx)
- return err
- }
- // ExecX is like Exec, but panics if an error occurs.
- func (bmu *BatchMsgUpdate) ExecX(ctx context.Context) {
- if err := bmu.Exec(ctx); err != nil {
- panic(err)
- }
- }
- // defaults sets the default values of the builder before save.
- func (bmu *BatchMsgUpdate) defaults() error {
- if _, ok := bmu.mutation.UpdatedAt(); !ok {
- if batchmsg.UpdateDefaultUpdatedAt == nil {
- return fmt.Errorf("ent: uninitialized batchmsg.UpdateDefaultUpdatedAt (forgotten import ent/runtime?)")
- }
- v := batchmsg.UpdateDefaultUpdatedAt()
- bmu.mutation.SetUpdatedAt(v)
- }
- return nil
- }
- // check runs all checks and user-defined validators on the builder.
- func (bmu *BatchMsgUpdate) check() error {
- if v, ok := bmu.mutation.OrganizationID(); ok {
- if err := batchmsg.OrganizationIDValidator(v); err != nil {
- return &ValidationError{Name: "organization_id", err: fmt.Errorf(`ent: validator failed for field "BatchMsg.organization_id": %w`, err)}
- }
- }
- return nil
- }
- func (bmu *BatchMsgUpdate) sqlSave(ctx context.Context) (n int, err error) {
- if err := bmu.check(); err != nil {
- return n, err
- }
- _spec := sqlgraph.NewUpdateSpec(batchmsg.Table, batchmsg.Columns, sqlgraph.NewFieldSpec(batchmsg.FieldID, field.TypeUint64))
- if ps := bmu.mutation.predicates; len(ps) > 0 {
- _spec.Predicate = func(selector *sql.Selector) {
- for i := range ps {
- ps[i](selector)
- }
- }
- }
- if value, ok := bmu.mutation.UpdatedAt(); ok {
- _spec.SetField(batchmsg.FieldUpdatedAt, field.TypeTime, value)
- }
- if value, ok := bmu.mutation.DeletedAt(); ok {
- _spec.SetField(batchmsg.FieldDeletedAt, field.TypeTime, value)
- }
- if bmu.mutation.DeletedAtCleared() {
- _spec.ClearField(batchmsg.FieldDeletedAt, field.TypeTime)
- }
- if value, ok := bmu.mutation.Status(); ok {
- _spec.SetField(batchmsg.FieldStatus, field.TypeUint8, value)
- }
- if value, ok := bmu.mutation.AddedStatus(); ok {
- _spec.AddField(batchmsg.FieldStatus, field.TypeUint8, value)
- }
- if bmu.mutation.StatusCleared() {
- _spec.ClearField(batchmsg.FieldStatus, field.TypeUint8)
- }
- if value, ok := bmu.mutation.BatchNo(); ok {
- _spec.SetField(batchmsg.FieldBatchNo, field.TypeString, value)
- }
- if bmu.mutation.BatchNoCleared() {
- _spec.ClearField(batchmsg.FieldBatchNo, field.TypeString)
- }
- if value, ok := bmu.mutation.TaskName(); ok {
- _spec.SetField(batchmsg.FieldTaskName, field.TypeString, value)
- }
- if bmu.mutation.TaskNameCleared() {
- _spec.ClearField(batchmsg.FieldTaskName, field.TypeString)
- }
- if value, ok := bmu.mutation.Fromwxid(); ok {
- _spec.SetField(batchmsg.FieldFromwxid, field.TypeString, value)
- }
- if bmu.mutation.FromwxidCleared() {
- _spec.ClearField(batchmsg.FieldFromwxid, field.TypeString)
- }
- if value, ok := bmu.mutation.Msg(); ok {
- _spec.SetField(batchmsg.FieldMsg, field.TypeString, value)
- }
- if bmu.mutation.MsgCleared() {
- _spec.ClearField(batchmsg.FieldMsg, field.TypeString)
- }
- if value, ok := bmu.mutation.Tag(); ok {
- _spec.SetField(batchmsg.FieldTag, field.TypeString, value)
- }
- if bmu.mutation.TagCleared() {
- _spec.ClearField(batchmsg.FieldTag, field.TypeString)
- }
- if value, ok := bmu.mutation.Tagids(); ok {
- _spec.SetField(batchmsg.FieldTagids, field.TypeString, value)
- }
- if bmu.mutation.TagidsCleared() {
- _spec.ClearField(batchmsg.FieldTagids, field.TypeString)
- }
- if value, ok := bmu.mutation.Total(); ok {
- _spec.SetField(batchmsg.FieldTotal, field.TypeInt32, value)
- }
- if value, ok := bmu.mutation.AddedTotal(); ok {
- _spec.AddField(batchmsg.FieldTotal, field.TypeInt32, value)
- }
- if bmu.mutation.TotalCleared() {
- _spec.ClearField(batchmsg.FieldTotal, field.TypeInt32)
- }
- if value, ok := bmu.mutation.Success(); ok {
- _spec.SetField(batchmsg.FieldSuccess, field.TypeInt32, value)
- }
- if value, ok := bmu.mutation.AddedSuccess(); ok {
- _spec.AddField(batchmsg.FieldSuccess, field.TypeInt32, value)
- }
- if bmu.mutation.SuccessCleared() {
- _spec.ClearField(batchmsg.FieldSuccess, field.TypeInt32)
- }
- if value, ok := bmu.mutation.Fail(); ok {
- _spec.SetField(batchmsg.FieldFail, field.TypeInt32, value)
- }
- if value, ok := bmu.mutation.AddedFail(); ok {
- _spec.AddField(batchmsg.FieldFail, field.TypeInt32, value)
- }
- if bmu.mutation.FailCleared() {
- _spec.ClearField(batchmsg.FieldFail, field.TypeInt32)
- }
- if value, ok := bmu.mutation.StartTime(); ok {
- _spec.SetField(batchmsg.FieldStartTime, field.TypeTime, value)
- }
- if bmu.mutation.StartTimeCleared() {
- _spec.ClearField(batchmsg.FieldStartTime, field.TypeTime)
- }
- if value, ok := bmu.mutation.StopTime(); ok {
- _spec.SetField(batchmsg.FieldStopTime, field.TypeTime, value)
- }
- if bmu.mutation.StopTimeCleared() {
- _spec.ClearField(batchmsg.FieldStopTime, field.TypeTime)
- }
- if value, ok := bmu.mutation.SendTime(); ok {
- _spec.SetField(batchmsg.FieldSendTime, field.TypeTime, value)
- }
- if bmu.mutation.SendTimeCleared() {
- _spec.ClearField(batchmsg.FieldSendTime, field.TypeTime)
- }
- if value, ok := bmu.mutation.GetType(); ok {
- _spec.SetField(batchmsg.FieldType, field.TypeInt32, value)
- }
- if value, ok := bmu.mutation.AddedType(); ok {
- _spec.AddField(batchmsg.FieldType, field.TypeInt32, value)
- }
- if bmu.mutation.TypeCleared() {
- _spec.ClearField(batchmsg.FieldType, field.TypeInt32)
- }
- if value, ok := bmu.mutation.OrganizationID(); ok {
- _spec.SetField(batchmsg.FieldOrganizationID, field.TypeUint64, value)
- }
- if value, ok := bmu.mutation.AddedOrganizationID(); ok {
- _spec.AddField(batchmsg.FieldOrganizationID, field.TypeUint64, value)
- }
- if n, err = sqlgraph.UpdateNodes(ctx, bmu.driver, _spec); err != nil {
- if _, ok := err.(*sqlgraph.NotFoundError); ok {
- err = &NotFoundError{batchmsg.Label}
- } else if sqlgraph.IsConstraintError(err) {
- err = &ConstraintError{msg: err.Error(), wrap: err}
- }
- return 0, err
- }
- bmu.mutation.done = true
- return n, nil
- }
- // BatchMsgUpdateOne is the builder for updating a single BatchMsg entity.
- type BatchMsgUpdateOne struct {
- config
- fields []string
- hooks []Hook
- mutation *BatchMsgMutation
- }
- // SetUpdatedAt sets the "updated_at" field.
- func (bmuo *BatchMsgUpdateOne) SetUpdatedAt(t time.Time) *BatchMsgUpdateOne {
- bmuo.mutation.SetUpdatedAt(t)
- return bmuo
- }
- // SetDeletedAt sets the "deleted_at" field.
- func (bmuo *BatchMsgUpdateOne) SetDeletedAt(t time.Time) *BatchMsgUpdateOne {
- bmuo.mutation.SetDeletedAt(t)
- return bmuo
- }
- // SetNillableDeletedAt sets the "deleted_at" field if the given value is not nil.
- func (bmuo *BatchMsgUpdateOne) SetNillableDeletedAt(t *time.Time) *BatchMsgUpdateOne {
- if t != nil {
- bmuo.SetDeletedAt(*t)
- }
- return bmuo
- }
- // ClearDeletedAt clears the value of the "deleted_at" field.
- func (bmuo *BatchMsgUpdateOne) ClearDeletedAt() *BatchMsgUpdateOne {
- bmuo.mutation.ClearDeletedAt()
- return bmuo
- }
- // SetStatus sets the "status" field.
- func (bmuo *BatchMsgUpdateOne) SetStatus(u uint8) *BatchMsgUpdateOne {
- bmuo.mutation.ResetStatus()
- bmuo.mutation.SetStatus(u)
- return bmuo
- }
- // SetNillableStatus sets the "status" field if the given value is not nil.
- func (bmuo *BatchMsgUpdateOne) SetNillableStatus(u *uint8) *BatchMsgUpdateOne {
- if u != nil {
- bmuo.SetStatus(*u)
- }
- return bmuo
- }
- // AddStatus adds u to the "status" field.
- func (bmuo *BatchMsgUpdateOne) AddStatus(u int8) *BatchMsgUpdateOne {
- bmuo.mutation.AddStatus(u)
- return bmuo
- }
- // ClearStatus clears the value of the "status" field.
- func (bmuo *BatchMsgUpdateOne) ClearStatus() *BatchMsgUpdateOne {
- bmuo.mutation.ClearStatus()
- return bmuo
- }
- // SetBatchNo sets the "batch_no" field.
- func (bmuo *BatchMsgUpdateOne) SetBatchNo(s string) *BatchMsgUpdateOne {
- bmuo.mutation.SetBatchNo(s)
- return bmuo
- }
- // SetNillableBatchNo sets the "batch_no" field if the given value is not nil.
- func (bmuo *BatchMsgUpdateOne) SetNillableBatchNo(s *string) *BatchMsgUpdateOne {
- if s != nil {
- bmuo.SetBatchNo(*s)
- }
- return bmuo
- }
- // ClearBatchNo clears the value of the "batch_no" field.
- func (bmuo *BatchMsgUpdateOne) ClearBatchNo() *BatchMsgUpdateOne {
- bmuo.mutation.ClearBatchNo()
- return bmuo
- }
- // SetTaskName sets the "task_name" field.
- func (bmuo *BatchMsgUpdateOne) SetTaskName(s string) *BatchMsgUpdateOne {
- bmuo.mutation.SetTaskName(s)
- return bmuo
- }
- // SetNillableTaskName sets the "task_name" field if the given value is not nil.
- func (bmuo *BatchMsgUpdateOne) SetNillableTaskName(s *string) *BatchMsgUpdateOne {
- if s != nil {
- bmuo.SetTaskName(*s)
- }
- return bmuo
- }
- // ClearTaskName clears the value of the "task_name" field.
- func (bmuo *BatchMsgUpdateOne) ClearTaskName() *BatchMsgUpdateOne {
- bmuo.mutation.ClearTaskName()
- return bmuo
- }
- // SetFromwxid sets the "fromwxid" field.
- func (bmuo *BatchMsgUpdateOne) SetFromwxid(s string) *BatchMsgUpdateOne {
- bmuo.mutation.SetFromwxid(s)
- return bmuo
- }
- // SetNillableFromwxid sets the "fromwxid" field if the given value is not nil.
- func (bmuo *BatchMsgUpdateOne) SetNillableFromwxid(s *string) *BatchMsgUpdateOne {
- if s != nil {
- bmuo.SetFromwxid(*s)
- }
- return bmuo
- }
- // ClearFromwxid clears the value of the "fromwxid" field.
- func (bmuo *BatchMsgUpdateOne) ClearFromwxid() *BatchMsgUpdateOne {
- bmuo.mutation.ClearFromwxid()
- return bmuo
- }
- // SetMsg sets the "msg" field.
- func (bmuo *BatchMsgUpdateOne) SetMsg(s string) *BatchMsgUpdateOne {
- bmuo.mutation.SetMsg(s)
- return bmuo
- }
- // SetNillableMsg sets the "msg" field if the given value is not nil.
- func (bmuo *BatchMsgUpdateOne) SetNillableMsg(s *string) *BatchMsgUpdateOne {
- if s != nil {
- bmuo.SetMsg(*s)
- }
- return bmuo
- }
- // ClearMsg clears the value of the "msg" field.
- func (bmuo *BatchMsgUpdateOne) ClearMsg() *BatchMsgUpdateOne {
- bmuo.mutation.ClearMsg()
- return bmuo
- }
- // SetTag sets the "tag" field.
- func (bmuo *BatchMsgUpdateOne) SetTag(s string) *BatchMsgUpdateOne {
- bmuo.mutation.SetTag(s)
- return bmuo
- }
- // SetNillableTag sets the "tag" field if the given value is not nil.
- func (bmuo *BatchMsgUpdateOne) SetNillableTag(s *string) *BatchMsgUpdateOne {
- if s != nil {
- bmuo.SetTag(*s)
- }
- return bmuo
- }
- // ClearTag clears the value of the "tag" field.
- func (bmuo *BatchMsgUpdateOne) ClearTag() *BatchMsgUpdateOne {
- bmuo.mutation.ClearTag()
- return bmuo
- }
- // SetTagids sets the "tagids" field.
- func (bmuo *BatchMsgUpdateOne) SetTagids(s string) *BatchMsgUpdateOne {
- bmuo.mutation.SetTagids(s)
- return bmuo
- }
- // SetNillableTagids sets the "tagids" field if the given value is not nil.
- func (bmuo *BatchMsgUpdateOne) SetNillableTagids(s *string) *BatchMsgUpdateOne {
- if s != nil {
- bmuo.SetTagids(*s)
- }
- return bmuo
- }
- // ClearTagids clears the value of the "tagids" field.
- func (bmuo *BatchMsgUpdateOne) ClearTagids() *BatchMsgUpdateOne {
- bmuo.mutation.ClearTagids()
- return bmuo
- }
- // SetTotal sets the "total" field.
- func (bmuo *BatchMsgUpdateOne) SetTotal(i int32) *BatchMsgUpdateOne {
- bmuo.mutation.ResetTotal()
- bmuo.mutation.SetTotal(i)
- return bmuo
- }
- // SetNillableTotal sets the "total" field if the given value is not nil.
- func (bmuo *BatchMsgUpdateOne) SetNillableTotal(i *int32) *BatchMsgUpdateOne {
- if i != nil {
- bmuo.SetTotal(*i)
- }
- return bmuo
- }
- // AddTotal adds i to the "total" field.
- func (bmuo *BatchMsgUpdateOne) AddTotal(i int32) *BatchMsgUpdateOne {
- bmuo.mutation.AddTotal(i)
- return bmuo
- }
- // ClearTotal clears the value of the "total" field.
- func (bmuo *BatchMsgUpdateOne) ClearTotal() *BatchMsgUpdateOne {
- bmuo.mutation.ClearTotal()
- return bmuo
- }
- // SetSuccess sets the "success" field.
- func (bmuo *BatchMsgUpdateOne) SetSuccess(i int32) *BatchMsgUpdateOne {
- bmuo.mutation.ResetSuccess()
- bmuo.mutation.SetSuccess(i)
- return bmuo
- }
- // SetNillableSuccess sets the "success" field if the given value is not nil.
- func (bmuo *BatchMsgUpdateOne) SetNillableSuccess(i *int32) *BatchMsgUpdateOne {
- if i != nil {
- bmuo.SetSuccess(*i)
- }
- return bmuo
- }
- // AddSuccess adds i to the "success" field.
- func (bmuo *BatchMsgUpdateOne) AddSuccess(i int32) *BatchMsgUpdateOne {
- bmuo.mutation.AddSuccess(i)
- return bmuo
- }
- // ClearSuccess clears the value of the "success" field.
- func (bmuo *BatchMsgUpdateOne) ClearSuccess() *BatchMsgUpdateOne {
- bmuo.mutation.ClearSuccess()
- return bmuo
- }
- // SetFail sets the "fail" field.
- func (bmuo *BatchMsgUpdateOne) SetFail(i int32) *BatchMsgUpdateOne {
- bmuo.mutation.ResetFail()
- bmuo.mutation.SetFail(i)
- return bmuo
- }
- // SetNillableFail sets the "fail" field if the given value is not nil.
- func (bmuo *BatchMsgUpdateOne) SetNillableFail(i *int32) *BatchMsgUpdateOne {
- if i != nil {
- bmuo.SetFail(*i)
- }
- return bmuo
- }
- // AddFail adds i to the "fail" field.
- func (bmuo *BatchMsgUpdateOne) AddFail(i int32) *BatchMsgUpdateOne {
- bmuo.mutation.AddFail(i)
- return bmuo
- }
- // ClearFail clears the value of the "fail" field.
- func (bmuo *BatchMsgUpdateOne) ClearFail() *BatchMsgUpdateOne {
- bmuo.mutation.ClearFail()
- return bmuo
- }
- // SetStartTime sets the "start_time" field.
- func (bmuo *BatchMsgUpdateOne) SetStartTime(t time.Time) *BatchMsgUpdateOne {
- bmuo.mutation.SetStartTime(t)
- return bmuo
- }
- // SetNillableStartTime sets the "start_time" field if the given value is not nil.
- func (bmuo *BatchMsgUpdateOne) SetNillableStartTime(t *time.Time) *BatchMsgUpdateOne {
- if t != nil {
- bmuo.SetStartTime(*t)
- }
- return bmuo
- }
- // ClearStartTime clears the value of the "start_time" field.
- func (bmuo *BatchMsgUpdateOne) ClearStartTime() *BatchMsgUpdateOne {
- bmuo.mutation.ClearStartTime()
- return bmuo
- }
- // SetStopTime sets the "stop_time" field.
- func (bmuo *BatchMsgUpdateOne) SetStopTime(t time.Time) *BatchMsgUpdateOne {
- bmuo.mutation.SetStopTime(t)
- return bmuo
- }
- // SetNillableStopTime sets the "stop_time" field if the given value is not nil.
- func (bmuo *BatchMsgUpdateOne) SetNillableStopTime(t *time.Time) *BatchMsgUpdateOne {
- if t != nil {
- bmuo.SetStopTime(*t)
- }
- return bmuo
- }
- // ClearStopTime clears the value of the "stop_time" field.
- func (bmuo *BatchMsgUpdateOne) ClearStopTime() *BatchMsgUpdateOne {
- bmuo.mutation.ClearStopTime()
- return bmuo
- }
- // SetSendTime sets the "send_time" field.
- func (bmuo *BatchMsgUpdateOne) SetSendTime(t time.Time) *BatchMsgUpdateOne {
- bmuo.mutation.SetSendTime(t)
- return bmuo
- }
- // SetNillableSendTime sets the "send_time" field if the given value is not nil.
- func (bmuo *BatchMsgUpdateOne) SetNillableSendTime(t *time.Time) *BatchMsgUpdateOne {
- if t != nil {
- bmuo.SetSendTime(*t)
- }
- return bmuo
- }
- // ClearSendTime clears the value of the "send_time" field.
- func (bmuo *BatchMsgUpdateOne) ClearSendTime() *BatchMsgUpdateOne {
- bmuo.mutation.ClearSendTime()
- return bmuo
- }
- // SetType sets the "type" field.
- func (bmuo *BatchMsgUpdateOne) SetType(i int32) *BatchMsgUpdateOne {
- bmuo.mutation.ResetType()
- bmuo.mutation.SetType(i)
- return bmuo
- }
- // SetNillableType sets the "type" field if the given value is not nil.
- func (bmuo *BatchMsgUpdateOne) SetNillableType(i *int32) *BatchMsgUpdateOne {
- if i != nil {
- bmuo.SetType(*i)
- }
- return bmuo
- }
- // AddType adds i to the "type" field.
- func (bmuo *BatchMsgUpdateOne) AddType(i int32) *BatchMsgUpdateOne {
- bmuo.mutation.AddType(i)
- return bmuo
- }
- // ClearType clears the value of the "type" field.
- func (bmuo *BatchMsgUpdateOne) ClearType() *BatchMsgUpdateOne {
- bmuo.mutation.ClearType()
- return bmuo
- }
- // SetOrganizationID sets the "organization_id" field.
- func (bmuo *BatchMsgUpdateOne) SetOrganizationID(u uint64) *BatchMsgUpdateOne {
- bmuo.mutation.ResetOrganizationID()
- bmuo.mutation.SetOrganizationID(u)
- return bmuo
- }
- // SetNillableOrganizationID sets the "organization_id" field if the given value is not nil.
- func (bmuo *BatchMsgUpdateOne) SetNillableOrganizationID(u *uint64) *BatchMsgUpdateOne {
- if u != nil {
- bmuo.SetOrganizationID(*u)
- }
- return bmuo
- }
- // AddOrganizationID adds u to the "organization_id" field.
- func (bmuo *BatchMsgUpdateOne) AddOrganizationID(u int64) *BatchMsgUpdateOne {
- bmuo.mutation.AddOrganizationID(u)
- return bmuo
- }
- // Mutation returns the BatchMsgMutation object of the builder.
- func (bmuo *BatchMsgUpdateOne) Mutation() *BatchMsgMutation {
- return bmuo.mutation
- }
- // Where appends a list predicates to the BatchMsgUpdate builder.
- func (bmuo *BatchMsgUpdateOne) Where(ps ...predicate.BatchMsg) *BatchMsgUpdateOne {
- bmuo.mutation.Where(ps...)
- return bmuo
- }
- // Select allows selecting one or more fields (columns) of the returned entity.
- // The default is selecting all fields defined in the entity schema.
- func (bmuo *BatchMsgUpdateOne) Select(field string, fields ...string) *BatchMsgUpdateOne {
- bmuo.fields = append([]string{field}, fields...)
- return bmuo
- }
- // Save executes the query and returns the updated BatchMsg entity.
- func (bmuo *BatchMsgUpdateOne) Save(ctx context.Context) (*BatchMsg, error) {
- if err := bmuo.defaults(); err != nil {
- return nil, err
- }
- return withHooks(ctx, bmuo.sqlSave, bmuo.mutation, bmuo.hooks)
- }
- // SaveX is like Save, but panics if an error occurs.
- func (bmuo *BatchMsgUpdateOne) SaveX(ctx context.Context) *BatchMsg {
- node, err := bmuo.Save(ctx)
- if err != nil {
- panic(err)
- }
- return node
- }
- // Exec executes the query on the entity.
- func (bmuo *BatchMsgUpdateOne) Exec(ctx context.Context) error {
- _, err := bmuo.Save(ctx)
- return err
- }
- // ExecX is like Exec, but panics if an error occurs.
- func (bmuo *BatchMsgUpdateOne) ExecX(ctx context.Context) {
- if err := bmuo.Exec(ctx); err != nil {
- panic(err)
- }
- }
- // defaults sets the default values of the builder before save.
- func (bmuo *BatchMsgUpdateOne) defaults() error {
- if _, ok := bmuo.mutation.UpdatedAt(); !ok {
- if batchmsg.UpdateDefaultUpdatedAt == nil {
- return fmt.Errorf("ent: uninitialized batchmsg.UpdateDefaultUpdatedAt (forgotten import ent/runtime?)")
- }
- v := batchmsg.UpdateDefaultUpdatedAt()
- bmuo.mutation.SetUpdatedAt(v)
- }
- return nil
- }
- // check runs all checks and user-defined validators on the builder.
- func (bmuo *BatchMsgUpdateOne) check() error {
- if v, ok := bmuo.mutation.OrganizationID(); ok {
- if err := batchmsg.OrganizationIDValidator(v); err != nil {
- return &ValidationError{Name: "organization_id", err: fmt.Errorf(`ent: validator failed for field "BatchMsg.organization_id": %w`, err)}
- }
- }
- return nil
- }
- func (bmuo *BatchMsgUpdateOne) sqlSave(ctx context.Context) (_node *BatchMsg, err error) {
- if err := bmuo.check(); err != nil {
- return _node, err
- }
- _spec := sqlgraph.NewUpdateSpec(batchmsg.Table, batchmsg.Columns, sqlgraph.NewFieldSpec(batchmsg.FieldID, field.TypeUint64))
- id, ok := bmuo.mutation.ID()
- if !ok {
- return nil, &ValidationError{Name: "id", err: errors.New(`ent: missing "BatchMsg.id" for update`)}
- }
- _spec.Node.ID.Value = id
- if fields := bmuo.fields; len(fields) > 0 {
- _spec.Node.Columns = make([]string, 0, len(fields))
- _spec.Node.Columns = append(_spec.Node.Columns, batchmsg.FieldID)
- for _, f := range fields {
- if !batchmsg.ValidColumn(f) {
- return nil, &ValidationError{Name: f, err: fmt.Errorf("ent: invalid field %q for query", f)}
- }
- if f != batchmsg.FieldID {
- _spec.Node.Columns = append(_spec.Node.Columns, f)
- }
- }
- }
- if ps := bmuo.mutation.predicates; len(ps) > 0 {
- _spec.Predicate = func(selector *sql.Selector) {
- for i := range ps {
- ps[i](selector)
- }
- }
- }
- if value, ok := bmuo.mutation.UpdatedAt(); ok {
- _spec.SetField(batchmsg.FieldUpdatedAt, field.TypeTime, value)
- }
- if value, ok := bmuo.mutation.DeletedAt(); ok {
- _spec.SetField(batchmsg.FieldDeletedAt, field.TypeTime, value)
- }
- if bmuo.mutation.DeletedAtCleared() {
- _spec.ClearField(batchmsg.FieldDeletedAt, field.TypeTime)
- }
- if value, ok := bmuo.mutation.Status(); ok {
- _spec.SetField(batchmsg.FieldStatus, field.TypeUint8, value)
- }
- if value, ok := bmuo.mutation.AddedStatus(); ok {
- _spec.AddField(batchmsg.FieldStatus, field.TypeUint8, value)
- }
- if bmuo.mutation.StatusCleared() {
- _spec.ClearField(batchmsg.FieldStatus, field.TypeUint8)
- }
- if value, ok := bmuo.mutation.BatchNo(); ok {
- _spec.SetField(batchmsg.FieldBatchNo, field.TypeString, value)
- }
- if bmuo.mutation.BatchNoCleared() {
- _spec.ClearField(batchmsg.FieldBatchNo, field.TypeString)
- }
- if value, ok := bmuo.mutation.TaskName(); ok {
- _spec.SetField(batchmsg.FieldTaskName, field.TypeString, value)
- }
- if bmuo.mutation.TaskNameCleared() {
- _spec.ClearField(batchmsg.FieldTaskName, field.TypeString)
- }
- if value, ok := bmuo.mutation.Fromwxid(); ok {
- _spec.SetField(batchmsg.FieldFromwxid, field.TypeString, value)
- }
- if bmuo.mutation.FromwxidCleared() {
- _spec.ClearField(batchmsg.FieldFromwxid, field.TypeString)
- }
- if value, ok := bmuo.mutation.Msg(); ok {
- _spec.SetField(batchmsg.FieldMsg, field.TypeString, value)
- }
- if bmuo.mutation.MsgCleared() {
- _spec.ClearField(batchmsg.FieldMsg, field.TypeString)
- }
- if value, ok := bmuo.mutation.Tag(); ok {
- _spec.SetField(batchmsg.FieldTag, field.TypeString, value)
- }
- if bmuo.mutation.TagCleared() {
- _spec.ClearField(batchmsg.FieldTag, field.TypeString)
- }
- if value, ok := bmuo.mutation.Tagids(); ok {
- _spec.SetField(batchmsg.FieldTagids, field.TypeString, value)
- }
- if bmuo.mutation.TagidsCleared() {
- _spec.ClearField(batchmsg.FieldTagids, field.TypeString)
- }
- if value, ok := bmuo.mutation.Total(); ok {
- _spec.SetField(batchmsg.FieldTotal, field.TypeInt32, value)
- }
- if value, ok := bmuo.mutation.AddedTotal(); ok {
- _spec.AddField(batchmsg.FieldTotal, field.TypeInt32, value)
- }
- if bmuo.mutation.TotalCleared() {
- _spec.ClearField(batchmsg.FieldTotal, field.TypeInt32)
- }
- if value, ok := bmuo.mutation.Success(); ok {
- _spec.SetField(batchmsg.FieldSuccess, field.TypeInt32, value)
- }
- if value, ok := bmuo.mutation.AddedSuccess(); ok {
- _spec.AddField(batchmsg.FieldSuccess, field.TypeInt32, value)
- }
- if bmuo.mutation.SuccessCleared() {
- _spec.ClearField(batchmsg.FieldSuccess, field.TypeInt32)
- }
- if value, ok := bmuo.mutation.Fail(); ok {
- _spec.SetField(batchmsg.FieldFail, field.TypeInt32, value)
- }
- if value, ok := bmuo.mutation.AddedFail(); ok {
- _spec.AddField(batchmsg.FieldFail, field.TypeInt32, value)
- }
- if bmuo.mutation.FailCleared() {
- _spec.ClearField(batchmsg.FieldFail, field.TypeInt32)
- }
- if value, ok := bmuo.mutation.StartTime(); ok {
- _spec.SetField(batchmsg.FieldStartTime, field.TypeTime, value)
- }
- if bmuo.mutation.StartTimeCleared() {
- _spec.ClearField(batchmsg.FieldStartTime, field.TypeTime)
- }
- if value, ok := bmuo.mutation.StopTime(); ok {
- _spec.SetField(batchmsg.FieldStopTime, field.TypeTime, value)
- }
- if bmuo.mutation.StopTimeCleared() {
- _spec.ClearField(batchmsg.FieldStopTime, field.TypeTime)
- }
- if value, ok := bmuo.mutation.SendTime(); ok {
- _spec.SetField(batchmsg.FieldSendTime, field.TypeTime, value)
- }
- if bmuo.mutation.SendTimeCleared() {
- _spec.ClearField(batchmsg.FieldSendTime, field.TypeTime)
- }
- if value, ok := bmuo.mutation.GetType(); ok {
- _spec.SetField(batchmsg.FieldType, field.TypeInt32, value)
- }
- if value, ok := bmuo.mutation.AddedType(); ok {
- _spec.AddField(batchmsg.FieldType, field.TypeInt32, value)
- }
- if bmuo.mutation.TypeCleared() {
- _spec.ClearField(batchmsg.FieldType, field.TypeInt32)
- }
- if value, ok := bmuo.mutation.OrganizationID(); ok {
- _spec.SetField(batchmsg.FieldOrganizationID, field.TypeUint64, value)
- }
- if value, ok := bmuo.mutation.AddedOrganizationID(); ok {
- _spec.AddField(batchmsg.FieldOrganizationID, field.TypeUint64, value)
- }
- _node = &BatchMsg{config: bmuo.config}
- _spec.Assign = _node.assignValues
- _spec.ScanValues = _node.scanValues
- if err = sqlgraph.UpdateNode(ctx, bmuo.driver, _spec); err != nil {
- if _, ok := err.(*sqlgraph.NotFoundError); ok {
- err = &NotFoundError{batchmsg.Label}
- } else if sqlgraph.IsConstraintError(err) {
- err = &ConstraintError{msg: err.Error(), wrap: err}
- }
- return nil, err
- }
- bmuo.mutation.done = true
- return _node, nil
- }
|