代码拉取完成,页面将自动刷新
// Code generated by ent, DO NOT EDIT.
package ent
import (
"context"
"errors"
"fmt"
"time"
"entgo.io/ent/dialect/sql"
"entgo.io/ent/dialect/sql/sqlgraph"
"entgo.io/ent/schema/field"
"gitee.com/yuLingNet/simple-iot-basic-rpc/ent/company"
"gitee.com/yuLingNet/simple-iot-basic-rpc/ent/device"
"gitee.com/yuLingNet/simple-iot-basic-rpc/ent/member"
"gitee.com/yuLingNet/simple-iot-basic-rpc/ent/predicate"
uuid "github.com/gofrs/uuid/v5"
)
// CompanyUpdate is the builder for updating Company entities.
type CompanyUpdate struct {
config
hooks []Hook
mutation *CompanyMutation
modifiers []func(*sql.UpdateBuilder)
}
// Where appends a list predicates to the CompanyUpdate builder.
func (cu *CompanyUpdate) Where(ps ...predicate.Company) *CompanyUpdate {
cu.mutation.Where(ps...)
return cu
}
// SetUpdatedAt sets the "updated_at" field.
func (cu *CompanyUpdate) SetUpdatedAt(t time.Time) *CompanyUpdate {
cu.mutation.SetUpdatedAt(t)
return cu
}
// SetStatus sets the "status" field.
func (cu *CompanyUpdate) SetStatus(u uint8) *CompanyUpdate {
cu.mutation.ResetStatus()
cu.mutation.SetStatus(u)
return cu
}
// SetNillableStatus sets the "status" field if the given value is not nil.
func (cu *CompanyUpdate) SetNillableStatus(u *uint8) *CompanyUpdate {
if u != nil {
cu.SetStatus(*u)
}
return cu
}
// AddStatus adds u to the "status" field.
func (cu *CompanyUpdate) AddStatus(u int8) *CompanyUpdate {
cu.mutation.AddStatus(u)
return cu
}
// ClearStatus clears the value of the "status" field.
func (cu *CompanyUpdate) ClearStatus() *CompanyUpdate {
cu.mutation.ClearStatus()
return cu
}
// SetSort sets the "sort" field.
func (cu *CompanyUpdate) SetSort(u uint32) *CompanyUpdate {
cu.mutation.ResetSort()
cu.mutation.SetSort(u)
return cu
}
// SetNillableSort sets the "sort" field if the given value is not nil.
func (cu *CompanyUpdate) SetNillableSort(u *uint32) *CompanyUpdate {
if u != nil {
cu.SetSort(*u)
}
return cu
}
// AddSort adds u to the "sort" field.
func (cu *CompanyUpdate) AddSort(u int32) *CompanyUpdate {
cu.mutation.AddSort(u)
return cu
}
// SetCode sets the "code" field.
func (cu *CompanyUpdate) SetCode(i int32) *CompanyUpdate {
cu.mutation.ResetCode()
cu.mutation.SetCode(i)
return cu
}
// SetNillableCode sets the "code" field if the given value is not nil.
func (cu *CompanyUpdate) SetNillableCode(i *int32) *CompanyUpdate {
if i != nil {
cu.SetCode(*i)
}
return cu
}
// AddCode adds i to the "code" field.
func (cu *CompanyUpdate) AddCode(i int32) *CompanyUpdate {
cu.mutation.AddCode(i)
return cu
}
// SetNickname sets the "nickname" field.
func (cu *CompanyUpdate) SetNickname(s string) *CompanyUpdate {
cu.mutation.SetNickname(s)
return cu
}
// SetNillableNickname sets the "nickname" field if the given value is not nil.
func (cu *CompanyUpdate) SetNillableNickname(s *string) *CompanyUpdate {
if s != nil {
cu.SetNickname(*s)
}
return cu
}
// SetAddress sets the "address" field.
func (cu *CompanyUpdate) SetAddress(s string) *CompanyUpdate {
cu.mutation.SetAddress(s)
return cu
}
// SetNillableAddress sets the "address" field if the given value is not nil.
func (cu *CompanyUpdate) SetNillableAddress(s *string) *CompanyUpdate {
if s != nil {
cu.SetAddress(*s)
}
return cu
}
// ClearAddress clears the value of the "address" field.
func (cu *CompanyUpdate) ClearAddress() *CompanyUpdate {
cu.mutation.ClearAddress()
return cu
}
// SetRemark sets the "remark" field.
func (cu *CompanyUpdate) SetRemark(s string) *CompanyUpdate {
cu.mutation.SetRemark(s)
return cu
}
// SetNillableRemark sets the "remark" field if the given value is not nil.
func (cu *CompanyUpdate) SetNillableRemark(s *string) *CompanyUpdate {
if s != nil {
cu.SetRemark(*s)
}
return cu
}
// ClearRemark clears the value of the "remark" field.
func (cu *CompanyUpdate) ClearRemark() *CompanyUpdate {
cu.mutation.ClearRemark()
return cu
}
// AddDeviceIDs adds the "devices" edge to the Device entity by IDs.
func (cu *CompanyUpdate) AddDeviceIDs(ids ...uuid.UUID) *CompanyUpdate {
cu.mutation.AddDeviceIDs(ids...)
return cu
}
// AddDevices adds the "devices" edges to the Device entity.
func (cu *CompanyUpdate) AddDevices(d ...*Device) *CompanyUpdate {
ids := make([]uuid.UUID, len(d))
for i := range d {
ids[i] = d[i].ID
}
return cu.AddDeviceIDs(ids...)
}
// AddMemberIDs adds the "members" edge to the Member entity by IDs.
func (cu *CompanyUpdate) AddMemberIDs(ids ...uuid.UUID) *CompanyUpdate {
cu.mutation.AddMemberIDs(ids...)
return cu
}
// AddMembers adds the "members" edges to the Member entity.
func (cu *CompanyUpdate) AddMembers(m ...*Member) *CompanyUpdate {
ids := make([]uuid.UUID, len(m))
for i := range m {
ids[i] = m[i].ID
}
return cu.AddMemberIDs(ids...)
}
// Mutation returns the CompanyMutation object of the builder.
func (cu *CompanyUpdate) Mutation() *CompanyMutation {
return cu.mutation
}
// ClearDevices clears all "devices" edges to the Device entity.
func (cu *CompanyUpdate) ClearDevices() *CompanyUpdate {
cu.mutation.ClearDevices()
return cu
}
// RemoveDeviceIDs removes the "devices" edge to Device entities by IDs.
func (cu *CompanyUpdate) RemoveDeviceIDs(ids ...uuid.UUID) *CompanyUpdate {
cu.mutation.RemoveDeviceIDs(ids...)
return cu
}
// RemoveDevices removes "devices" edges to Device entities.
func (cu *CompanyUpdate) RemoveDevices(d ...*Device) *CompanyUpdate {
ids := make([]uuid.UUID, len(d))
for i := range d {
ids[i] = d[i].ID
}
return cu.RemoveDeviceIDs(ids...)
}
// ClearMembers clears all "members" edges to the Member entity.
func (cu *CompanyUpdate) ClearMembers() *CompanyUpdate {
cu.mutation.ClearMembers()
return cu
}
// RemoveMemberIDs removes the "members" edge to Member entities by IDs.
func (cu *CompanyUpdate) RemoveMemberIDs(ids ...uuid.UUID) *CompanyUpdate {
cu.mutation.RemoveMemberIDs(ids...)
return cu
}
// RemoveMembers removes "members" edges to Member entities.
func (cu *CompanyUpdate) RemoveMembers(m ...*Member) *CompanyUpdate {
ids := make([]uuid.UUID, len(m))
for i := range m {
ids[i] = m[i].ID
}
return cu.RemoveMemberIDs(ids...)
}
// Save executes the query and returns the number of nodes affected by the update operation.
func (cu *CompanyUpdate) Save(ctx context.Context) (int, error) {
cu.defaults()
return withHooks(ctx, cu.sqlSave, cu.mutation, cu.hooks)
}
// SaveX is like Save, but panics if an error occurs.
func (cu *CompanyUpdate) SaveX(ctx context.Context) int {
affected, err := cu.Save(ctx)
if err != nil {
panic(err)
}
return affected
}
// Exec executes the query.
func (cu *CompanyUpdate) Exec(ctx context.Context) error {
_, err := cu.Save(ctx)
return err
}
// ExecX is like Exec, but panics if an error occurs.
func (cu *CompanyUpdate) ExecX(ctx context.Context) {
if err := cu.Exec(ctx); err != nil {
panic(err)
}
}
// defaults sets the default values of the builder before save.
func (cu *CompanyUpdate) defaults() {
if _, ok := cu.mutation.UpdatedAt(); !ok {
v := company.UpdateDefaultUpdatedAt()
cu.mutation.SetUpdatedAt(v)
}
}
// Modify adds a statement modifier for attaching custom logic to the UPDATE statement.
func (cu *CompanyUpdate) Modify(modifiers ...func(u *sql.UpdateBuilder)) *CompanyUpdate {
cu.modifiers = append(cu.modifiers, modifiers...)
return cu
}
func (cu *CompanyUpdate) sqlSave(ctx context.Context) (n int, err error) {
_spec := sqlgraph.NewUpdateSpec(company.Table, company.Columns, sqlgraph.NewFieldSpec(company.FieldID, field.TypeUint64))
if ps := cu.mutation.predicates; len(ps) > 0 {
_spec.Predicate = func(selector *sql.Selector) {
for i := range ps {
ps[i](selector)
}
}
}
if value, ok := cu.mutation.UpdatedAt(); ok {
_spec.SetField(company.FieldUpdatedAt, field.TypeTime, value)
}
if value, ok := cu.mutation.Status(); ok {
_spec.SetField(company.FieldStatus, field.TypeUint8, value)
}
if value, ok := cu.mutation.AddedStatus(); ok {
_spec.AddField(company.FieldStatus, field.TypeUint8, value)
}
if cu.mutation.StatusCleared() {
_spec.ClearField(company.FieldStatus, field.TypeUint8)
}
if value, ok := cu.mutation.Sort(); ok {
_spec.SetField(company.FieldSort, field.TypeUint32, value)
}
if value, ok := cu.mutation.AddedSort(); ok {
_spec.AddField(company.FieldSort, field.TypeUint32, value)
}
if value, ok := cu.mutation.Code(); ok {
_spec.SetField(company.FieldCode, field.TypeInt32, value)
}
if value, ok := cu.mutation.AddedCode(); ok {
_spec.AddField(company.FieldCode, field.TypeInt32, value)
}
if value, ok := cu.mutation.Nickname(); ok {
_spec.SetField(company.FieldNickname, field.TypeString, value)
}
if value, ok := cu.mutation.Address(); ok {
_spec.SetField(company.FieldAddress, field.TypeString, value)
}
if cu.mutation.AddressCleared() {
_spec.ClearField(company.FieldAddress, field.TypeString)
}
if value, ok := cu.mutation.Remark(); ok {
_spec.SetField(company.FieldRemark, field.TypeString, value)
}
if cu.mutation.RemarkCleared() {
_spec.ClearField(company.FieldRemark, field.TypeString)
}
if cu.mutation.DevicesCleared() {
edge := &sqlgraph.EdgeSpec{
Rel: sqlgraph.M2M,
Inverse: false,
Table: company.DevicesTable,
Columns: company.DevicesPrimaryKey,
Bidi: false,
Target: &sqlgraph.EdgeTarget{
IDSpec: sqlgraph.NewFieldSpec(device.FieldID, field.TypeUUID),
},
}
_spec.Edges.Clear = append(_spec.Edges.Clear, edge)
}
if nodes := cu.mutation.RemovedDevicesIDs(); len(nodes) > 0 && !cu.mutation.DevicesCleared() {
edge := &sqlgraph.EdgeSpec{
Rel: sqlgraph.M2M,
Inverse: false,
Table: company.DevicesTable,
Columns: company.DevicesPrimaryKey,
Bidi: false,
Target: &sqlgraph.EdgeTarget{
IDSpec: sqlgraph.NewFieldSpec(device.FieldID, field.TypeUUID),
},
}
for _, k := range nodes {
edge.Target.Nodes = append(edge.Target.Nodes, k)
}
_spec.Edges.Clear = append(_spec.Edges.Clear, edge)
}
if nodes := cu.mutation.DevicesIDs(); len(nodes) > 0 {
edge := &sqlgraph.EdgeSpec{
Rel: sqlgraph.M2M,
Inverse: false,
Table: company.DevicesTable,
Columns: company.DevicesPrimaryKey,
Bidi: false,
Target: &sqlgraph.EdgeTarget{
IDSpec: sqlgraph.NewFieldSpec(device.FieldID, field.TypeUUID),
},
}
for _, k := range nodes {
edge.Target.Nodes = append(edge.Target.Nodes, k)
}
_spec.Edges.Add = append(_spec.Edges.Add, edge)
}
if cu.mutation.MembersCleared() {
edge := &sqlgraph.EdgeSpec{
Rel: sqlgraph.O2M,
Inverse: false,
Table: company.MembersTable,
Columns: []string{company.MembersColumn},
Bidi: false,
Target: &sqlgraph.EdgeTarget{
IDSpec: sqlgraph.NewFieldSpec(member.FieldID, field.TypeUUID),
},
}
_spec.Edges.Clear = append(_spec.Edges.Clear, edge)
}
if nodes := cu.mutation.RemovedMembersIDs(); len(nodes) > 0 && !cu.mutation.MembersCleared() {
edge := &sqlgraph.EdgeSpec{
Rel: sqlgraph.O2M,
Inverse: false,
Table: company.MembersTable,
Columns: []string{company.MembersColumn},
Bidi: false,
Target: &sqlgraph.EdgeTarget{
IDSpec: sqlgraph.NewFieldSpec(member.FieldID, field.TypeUUID),
},
}
for _, k := range nodes {
edge.Target.Nodes = append(edge.Target.Nodes, k)
}
_spec.Edges.Clear = append(_spec.Edges.Clear, edge)
}
if nodes := cu.mutation.MembersIDs(); len(nodes) > 0 {
edge := &sqlgraph.EdgeSpec{
Rel: sqlgraph.O2M,
Inverse: false,
Table: company.MembersTable,
Columns: []string{company.MembersColumn},
Bidi: false,
Target: &sqlgraph.EdgeTarget{
IDSpec: sqlgraph.NewFieldSpec(member.FieldID, field.TypeUUID),
},
}
for _, k := range nodes {
edge.Target.Nodes = append(edge.Target.Nodes, k)
}
_spec.Edges.Add = append(_spec.Edges.Add, edge)
}
_spec.AddModifiers(cu.modifiers...)
if n, err = sqlgraph.UpdateNodes(ctx, cu.driver, _spec); err != nil {
if _, ok := err.(*sqlgraph.NotFoundError); ok {
err = &NotFoundError{company.Label}
} else if sqlgraph.IsConstraintError(err) {
err = &ConstraintError{msg: err.Error(), wrap: err}
}
return 0, err
}
cu.mutation.done = true
return n, nil
}
// CompanyUpdateOne is the builder for updating a single Company entity.
type CompanyUpdateOne struct {
config
fields []string
hooks []Hook
mutation *CompanyMutation
modifiers []func(*sql.UpdateBuilder)
}
// SetUpdatedAt sets the "updated_at" field.
func (cuo *CompanyUpdateOne) SetUpdatedAt(t time.Time) *CompanyUpdateOne {
cuo.mutation.SetUpdatedAt(t)
return cuo
}
// SetStatus sets the "status" field.
func (cuo *CompanyUpdateOne) SetStatus(u uint8) *CompanyUpdateOne {
cuo.mutation.ResetStatus()
cuo.mutation.SetStatus(u)
return cuo
}
// SetNillableStatus sets the "status" field if the given value is not nil.
func (cuo *CompanyUpdateOne) SetNillableStatus(u *uint8) *CompanyUpdateOne {
if u != nil {
cuo.SetStatus(*u)
}
return cuo
}
// AddStatus adds u to the "status" field.
func (cuo *CompanyUpdateOne) AddStatus(u int8) *CompanyUpdateOne {
cuo.mutation.AddStatus(u)
return cuo
}
// ClearStatus clears the value of the "status" field.
func (cuo *CompanyUpdateOne) ClearStatus() *CompanyUpdateOne {
cuo.mutation.ClearStatus()
return cuo
}
// SetSort sets the "sort" field.
func (cuo *CompanyUpdateOne) SetSort(u uint32) *CompanyUpdateOne {
cuo.mutation.ResetSort()
cuo.mutation.SetSort(u)
return cuo
}
// SetNillableSort sets the "sort" field if the given value is not nil.
func (cuo *CompanyUpdateOne) SetNillableSort(u *uint32) *CompanyUpdateOne {
if u != nil {
cuo.SetSort(*u)
}
return cuo
}
// AddSort adds u to the "sort" field.
func (cuo *CompanyUpdateOne) AddSort(u int32) *CompanyUpdateOne {
cuo.mutation.AddSort(u)
return cuo
}
// SetCode sets the "code" field.
func (cuo *CompanyUpdateOne) SetCode(i int32) *CompanyUpdateOne {
cuo.mutation.ResetCode()
cuo.mutation.SetCode(i)
return cuo
}
// SetNillableCode sets the "code" field if the given value is not nil.
func (cuo *CompanyUpdateOne) SetNillableCode(i *int32) *CompanyUpdateOne {
if i != nil {
cuo.SetCode(*i)
}
return cuo
}
// AddCode adds i to the "code" field.
func (cuo *CompanyUpdateOne) AddCode(i int32) *CompanyUpdateOne {
cuo.mutation.AddCode(i)
return cuo
}
// SetNickname sets the "nickname" field.
func (cuo *CompanyUpdateOne) SetNickname(s string) *CompanyUpdateOne {
cuo.mutation.SetNickname(s)
return cuo
}
// SetNillableNickname sets the "nickname" field if the given value is not nil.
func (cuo *CompanyUpdateOne) SetNillableNickname(s *string) *CompanyUpdateOne {
if s != nil {
cuo.SetNickname(*s)
}
return cuo
}
// SetAddress sets the "address" field.
func (cuo *CompanyUpdateOne) SetAddress(s string) *CompanyUpdateOne {
cuo.mutation.SetAddress(s)
return cuo
}
// SetNillableAddress sets the "address" field if the given value is not nil.
func (cuo *CompanyUpdateOne) SetNillableAddress(s *string) *CompanyUpdateOne {
if s != nil {
cuo.SetAddress(*s)
}
return cuo
}
// ClearAddress clears the value of the "address" field.
func (cuo *CompanyUpdateOne) ClearAddress() *CompanyUpdateOne {
cuo.mutation.ClearAddress()
return cuo
}
// SetRemark sets the "remark" field.
func (cuo *CompanyUpdateOne) SetRemark(s string) *CompanyUpdateOne {
cuo.mutation.SetRemark(s)
return cuo
}
// SetNillableRemark sets the "remark" field if the given value is not nil.
func (cuo *CompanyUpdateOne) SetNillableRemark(s *string) *CompanyUpdateOne {
if s != nil {
cuo.SetRemark(*s)
}
return cuo
}
// ClearRemark clears the value of the "remark" field.
func (cuo *CompanyUpdateOne) ClearRemark() *CompanyUpdateOne {
cuo.mutation.ClearRemark()
return cuo
}
// AddDeviceIDs adds the "devices" edge to the Device entity by IDs.
func (cuo *CompanyUpdateOne) AddDeviceIDs(ids ...uuid.UUID) *CompanyUpdateOne {
cuo.mutation.AddDeviceIDs(ids...)
return cuo
}
// AddDevices adds the "devices" edges to the Device entity.
func (cuo *CompanyUpdateOne) AddDevices(d ...*Device) *CompanyUpdateOne {
ids := make([]uuid.UUID, len(d))
for i := range d {
ids[i] = d[i].ID
}
return cuo.AddDeviceIDs(ids...)
}
// AddMemberIDs adds the "members" edge to the Member entity by IDs.
func (cuo *CompanyUpdateOne) AddMemberIDs(ids ...uuid.UUID) *CompanyUpdateOne {
cuo.mutation.AddMemberIDs(ids...)
return cuo
}
// AddMembers adds the "members" edges to the Member entity.
func (cuo *CompanyUpdateOne) AddMembers(m ...*Member) *CompanyUpdateOne {
ids := make([]uuid.UUID, len(m))
for i := range m {
ids[i] = m[i].ID
}
return cuo.AddMemberIDs(ids...)
}
// Mutation returns the CompanyMutation object of the builder.
func (cuo *CompanyUpdateOne) Mutation() *CompanyMutation {
return cuo.mutation
}
// ClearDevices clears all "devices" edges to the Device entity.
func (cuo *CompanyUpdateOne) ClearDevices() *CompanyUpdateOne {
cuo.mutation.ClearDevices()
return cuo
}
// RemoveDeviceIDs removes the "devices" edge to Device entities by IDs.
func (cuo *CompanyUpdateOne) RemoveDeviceIDs(ids ...uuid.UUID) *CompanyUpdateOne {
cuo.mutation.RemoveDeviceIDs(ids...)
return cuo
}
// RemoveDevices removes "devices" edges to Device entities.
func (cuo *CompanyUpdateOne) RemoveDevices(d ...*Device) *CompanyUpdateOne {
ids := make([]uuid.UUID, len(d))
for i := range d {
ids[i] = d[i].ID
}
return cuo.RemoveDeviceIDs(ids...)
}
// ClearMembers clears all "members" edges to the Member entity.
func (cuo *CompanyUpdateOne) ClearMembers() *CompanyUpdateOne {
cuo.mutation.ClearMembers()
return cuo
}
// RemoveMemberIDs removes the "members" edge to Member entities by IDs.
func (cuo *CompanyUpdateOne) RemoveMemberIDs(ids ...uuid.UUID) *CompanyUpdateOne {
cuo.mutation.RemoveMemberIDs(ids...)
return cuo
}
// RemoveMembers removes "members" edges to Member entities.
func (cuo *CompanyUpdateOne) RemoveMembers(m ...*Member) *CompanyUpdateOne {
ids := make([]uuid.UUID, len(m))
for i := range m {
ids[i] = m[i].ID
}
return cuo.RemoveMemberIDs(ids...)
}
// Where appends a list predicates to the CompanyUpdate builder.
func (cuo *CompanyUpdateOne) Where(ps ...predicate.Company) *CompanyUpdateOne {
cuo.mutation.Where(ps...)
return cuo
}
// Select allows selecting one or more fields (columns) of the returned entity.
// The default is selecting all fields defined in the entity schema.
func (cuo *CompanyUpdateOne) Select(field string, fields ...string) *CompanyUpdateOne {
cuo.fields = append([]string{field}, fields...)
return cuo
}
// Save executes the query and returns the updated Company entity.
func (cuo *CompanyUpdateOne) Save(ctx context.Context) (*Company, error) {
cuo.defaults()
return withHooks(ctx, cuo.sqlSave, cuo.mutation, cuo.hooks)
}
// SaveX is like Save, but panics if an error occurs.
func (cuo *CompanyUpdateOne) SaveX(ctx context.Context) *Company {
node, err := cuo.Save(ctx)
if err != nil {
panic(err)
}
return node
}
// Exec executes the query on the entity.
func (cuo *CompanyUpdateOne) Exec(ctx context.Context) error {
_, err := cuo.Save(ctx)
return err
}
// ExecX is like Exec, but panics if an error occurs.
func (cuo *CompanyUpdateOne) ExecX(ctx context.Context) {
if err := cuo.Exec(ctx); err != nil {
panic(err)
}
}
// defaults sets the default values of the builder before save.
func (cuo *CompanyUpdateOne) defaults() {
if _, ok := cuo.mutation.UpdatedAt(); !ok {
v := company.UpdateDefaultUpdatedAt()
cuo.mutation.SetUpdatedAt(v)
}
}
// Modify adds a statement modifier for attaching custom logic to the UPDATE statement.
func (cuo *CompanyUpdateOne) Modify(modifiers ...func(u *sql.UpdateBuilder)) *CompanyUpdateOne {
cuo.modifiers = append(cuo.modifiers, modifiers...)
return cuo
}
func (cuo *CompanyUpdateOne) sqlSave(ctx context.Context) (_node *Company, err error) {
_spec := sqlgraph.NewUpdateSpec(company.Table, company.Columns, sqlgraph.NewFieldSpec(company.FieldID, field.TypeUint64))
id, ok := cuo.mutation.ID()
if !ok {
return nil, &ValidationError{Name: "id", err: errors.New(`ent: missing "Company.id" for update`)}
}
_spec.Node.ID.Value = id
if fields := cuo.fields; len(fields) > 0 {
_spec.Node.Columns = make([]string, 0, len(fields))
_spec.Node.Columns = append(_spec.Node.Columns, company.FieldID)
for _, f := range fields {
if !company.ValidColumn(f) {
return nil, &ValidationError{Name: f, err: fmt.Errorf("ent: invalid field %q for query", f)}
}
if f != company.FieldID {
_spec.Node.Columns = append(_spec.Node.Columns, f)
}
}
}
if ps := cuo.mutation.predicates; len(ps) > 0 {
_spec.Predicate = func(selector *sql.Selector) {
for i := range ps {
ps[i](selector)
}
}
}
if value, ok := cuo.mutation.UpdatedAt(); ok {
_spec.SetField(company.FieldUpdatedAt, field.TypeTime, value)
}
if value, ok := cuo.mutation.Status(); ok {
_spec.SetField(company.FieldStatus, field.TypeUint8, value)
}
if value, ok := cuo.mutation.AddedStatus(); ok {
_spec.AddField(company.FieldStatus, field.TypeUint8, value)
}
if cuo.mutation.StatusCleared() {
_spec.ClearField(company.FieldStatus, field.TypeUint8)
}
if value, ok := cuo.mutation.Sort(); ok {
_spec.SetField(company.FieldSort, field.TypeUint32, value)
}
if value, ok := cuo.mutation.AddedSort(); ok {
_spec.AddField(company.FieldSort, field.TypeUint32, value)
}
if value, ok := cuo.mutation.Code(); ok {
_spec.SetField(company.FieldCode, field.TypeInt32, value)
}
if value, ok := cuo.mutation.AddedCode(); ok {
_spec.AddField(company.FieldCode, field.TypeInt32, value)
}
if value, ok := cuo.mutation.Nickname(); ok {
_spec.SetField(company.FieldNickname, field.TypeString, value)
}
if value, ok := cuo.mutation.Address(); ok {
_spec.SetField(company.FieldAddress, field.TypeString, value)
}
if cuo.mutation.AddressCleared() {
_spec.ClearField(company.FieldAddress, field.TypeString)
}
if value, ok := cuo.mutation.Remark(); ok {
_spec.SetField(company.FieldRemark, field.TypeString, value)
}
if cuo.mutation.RemarkCleared() {
_spec.ClearField(company.FieldRemark, field.TypeString)
}
if cuo.mutation.DevicesCleared() {
edge := &sqlgraph.EdgeSpec{
Rel: sqlgraph.M2M,
Inverse: false,
Table: company.DevicesTable,
Columns: company.DevicesPrimaryKey,
Bidi: false,
Target: &sqlgraph.EdgeTarget{
IDSpec: sqlgraph.NewFieldSpec(device.FieldID, field.TypeUUID),
},
}
_spec.Edges.Clear = append(_spec.Edges.Clear, edge)
}
if nodes := cuo.mutation.RemovedDevicesIDs(); len(nodes) > 0 && !cuo.mutation.DevicesCleared() {
edge := &sqlgraph.EdgeSpec{
Rel: sqlgraph.M2M,
Inverse: false,
Table: company.DevicesTable,
Columns: company.DevicesPrimaryKey,
Bidi: false,
Target: &sqlgraph.EdgeTarget{
IDSpec: sqlgraph.NewFieldSpec(device.FieldID, field.TypeUUID),
},
}
for _, k := range nodes {
edge.Target.Nodes = append(edge.Target.Nodes, k)
}
_spec.Edges.Clear = append(_spec.Edges.Clear, edge)
}
if nodes := cuo.mutation.DevicesIDs(); len(nodes) > 0 {
edge := &sqlgraph.EdgeSpec{
Rel: sqlgraph.M2M,
Inverse: false,
Table: company.DevicesTable,
Columns: company.DevicesPrimaryKey,
Bidi: false,
Target: &sqlgraph.EdgeTarget{
IDSpec: sqlgraph.NewFieldSpec(device.FieldID, field.TypeUUID),
},
}
for _, k := range nodes {
edge.Target.Nodes = append(edge.Target.Nodes, k)
}
_spec.Edges.Add = append(_spec.Edges.Add, edge)
}
if cuo.mutation.MembersCleared() {
edge := &sqlgraph.EdgeSpec{
Rel: sqlgraph.O2M,
Inverse: false,
Table: company.MembersTable,
Columns: []string{company.MembersColumn},
Bidi: false,
Target: &sqlgraph.EdgeTarget{
IDSpec: sqlgraph.NewFieldSpec(member.FieldID, field.TypeUUID),
},
}
_spec.Edges.Clear = append(_spec.Edges.Clear, edge)
}
if nodes := cuo.mutation.RemovedMembersIDs(); len(nodes) > 0 && !cuo.mutation.MembersCleared() {
edge := &sqlgraph.EdgeSpec{
Rel: sqlgraph.O2M,
Inverse: false,
Table: company.MembersTable,
Columns: []string{company.MembersColumn},
Bidi: false,
Target: &sqlgraph.EdgeTarget{
IDSpec: sqlgraph.NewFieldSpec(member.FieldID, field.TypeUUID),
},
}
for _, k := range nodes {
edge.Target.Nodes = append(edge.Target.Nodes, k)
}
_spec.Edges.Clear = append(_spec.Edges.Clear, edge)
}
if nodes := cuo.mutation.MembersIDs(); len(nodes) > 0 {
edge := &sqlgraph.EdgeSpec{
Rel: sqlgraph.O2M,
Inverse: false,
Table: company.MembersTable,
Columns: []string{company.MembersColumn},
Bidi: false,
Target: &sqlgraph.EdgeTarget{
IDSpec: sqlgraph.NewFieldSpec(member.FieldID, field.TypeUUID),
},
}
for _, k := range nodes {
edge.Target.Nodes = append(edge.Target.Nodes, k)
}
_spec.Edges.Add = append(_spec.Edges.Add, edge)
}
_spec.AddModifiers(cuo.modifiers...)
_node = &Company{config: cuo.config}
_spec.Assign = _node.assignValues
_spec.ScanValues = _node.scanValues
if err = sqlgraph.UpdateNode(ctx, cuo.driver, _spec); err != nil {
if _, ok := err.(*sqlgraph.NotFoundError); ok {
err = &NotFoundError{company.Label}
} else if sqlgraph.IsConstraintError(err) {
err = &ConstraintError{msg: err.Error(), wrap: err}
}
return nil, err
}
cuo.mutation.done = true
return _node, nil
}
此处可能存在不合适展示的内容,页面不予展示。您可通过相关编辑功能自查并修改。
如您确认内容无涉及 不当用语 / 纯广告导流 / 暴力 / 低俗色情 / 侵权 / 盗版 / 虚假 / 无价值内容或违法国家有关法律法规的内容,可点击提交进行申诉,我们将尽快为您处理。