objecttype_update.go

// Code generated by entc, DO NOT EDIT.

package ent

import (
	"context"

	"github.com/facebookincubator/ent/dialect/sql"
	"github.com/facebookincubator/ent/dialect/sql/sqlgraph"
	"github.com/facebookincubator/ent/schema/field"
	"github.com/knabben/ggql/ent/fieldtype"
	"github.com/knabben/ggql/ent/objecttype"
	"github.com/knabben/ggql/ent/predicate"
)

// ObjectTypeUpdate is the builder for updating ObjectType entities.
type ObjectTypeUpdate struct {
	config
	name          *string
	kind          *string
	description   *string
	fields        map[int]struct{}
	removedFields map[int]struct{}
	predicates    []predicate.ObjectType
}

// Where adds a new predicate for the builder.
func (otu *ObjectTypeUpdate) Where(ps ...predicate.ObjectType) *ObjectTypeUpdate {
	otu.predicates = append(otu.predicates, ps...)
	return otu
}

// SetName sets the name field.
func (otu *ObjectTypeUpdate) SetName(s string) *ObjectTypeUpdate {
	otu.name = &s
	return otu
}

// SetKind sets the kind field.
func (otu *ObjectTypeUpdate) SetKind(s string) *ObjectTypeUpdate {
	otu.kind = &s
	return otu
}

// SetDescription sets the description field.
func (otu *ObjectTypeUpdate) SetDescription(s string) *ObjectTypeUpdate {
	otu.description = &s
	return otu
}

// AddFieldIDs adds the fields edge to FieldType by ids.
func (otu *ObjectTypeUpdate) AddFieldIDs(ids ...int) *ObjectTypeUpdate {
	if otu.fields == nil {
		otu.fields = make(map[int]struct{})
	}
	for i := range ids {
		otu.fields[ids[i]] = struct{}{}
	}
	return otu
}

// AddFields adds the fields edges to FieldType.
func (otu *ObjectTypeUpdate) AddFields(f ...*FieldType) *ObjectTypeUpdate {
	ids := make([]int, len(f))
	for i := range f {
		ids[i] = f[i].ID
	}
	return otu.AddFieldIDs(ids...)
}

// RemoveFieldIDs removes the fields edge to FieldType by ids.
func (otu *ObjectTypeUpdate) RemoveFieldIDs(ids ...int) *ObjectTypeUpdate {
	if otu.removedFields == nil {
		otu.removedFields = make(map[int]struct{})
	}
	for i := range ids {
		otu.removedFields[ids[i]] = struct{}{}
	}
	return otu
}

// RemoveFields removes fields edges to FieldType.
func (otu *ObjectTypeUpdate) RemoveFields(f ...*FieldType) *ObjectTypeUpdate {
	ids := make([]int, len(f))
	for i := range f {
		ids[i] = f[i].ID
	}
	return otu.RemoveFieldIDs(ids...)
}

// Save executes the query and returns the number of rows/vertices matched by this operation.
func (otu *ObjectTypeUpdate) Save(ctx context.Context) (int, error) {
	return otu.sqlSave(ctx)
}

// SaveX is like Save, but panics if an error occurs.
func (otu *ObjectTypeUpdate) SaveX(ctx context.Context) int {
	affected, err := otu.Save(ctx)
	if err != nil {
		panic(err)
	}
	return affected
}

// Exec executes the query.
func (otu *ObjectTypeUpdate) Exec(ctx context.Context) error {
	_, err := otu.Save(ctx)
	return err
}

// ExecX is like Exec, but panics if an error occurs.
func (otu *ObjectTypeUpdate) ExecX(ctx context.Context) {
	if err := otu.Exec(ctx); err != nil {
		panic(err)
	}
}

func (otu *ObjectTypeUpdate) sqlSave(ctx context.Context) (n int, err error) {
	_spec := &sqlgraph.UpdateSpec{
		Node: &sqlgraph.NodeSpec{
			Table:   objecttype.Table,
			Columns: objecttype.Columns,
			ID: &sqlgraph.FieldSpec{
				Type:   field.TypeInt,
				Column: objecttype.FieldID,
			},
		},
	}
	if ps := otu.predicates; len(ps) > 0 {
		_spec.Predicate = func(selector *sql.Selector) {
			for i := range ps {
				ps[i](selector)
			}
		}
	}
	if value := otu.name; value != nil {
		_spec.Fields.Set = append(_spec.Fields.Set, &sqlgraph.FieldSpec{
			Type:   field.TypeString,
			Value:  *value,
			Column: objecttype.FieldName,
		})
	}
	if value := otu.kind; value != nil {
		_spec.Fields.Set = append(_spec.Fields.Set, &sqlgraph.FieldSpec{
			Type:   field.TypeString,
			Value:  *value,
			Column: objecttype.FieldKind,
		})
	}
	if value := otu.description; value != nil {
		_spec.Fields.Set = append(_spec.Fields.Set, &sqlgraph.FieldSpec{
			Type:   field.TypeString,
			Value:  *value,
			Column: objecttype.FieldDescription,
		})
	}
	if nodes := otu.removedFields; len(nodes) > 0 {
		edge := &sqlgraph.EdgeSpec{
			Rel:     sqlgraph.O2M,
			Inverse: false,
			Table:   objecttype.FieldsTable,
			Columns: []string{objecttype.FieldsColumn},
			Bidi:    false,
			Target: &sqlgraph.EdgeTarget{
				IDSpec: &sqlgraph.FieldSpec{
					Type:   field.TypeInt,
					Column: fieldtype.FieldID,
				},
			},
		}
		for k, _ := range nodes {
			edge.Target.Nodes = append(edge.Target.Nodes, k)
		}
		_spec.Edges.Clear = append(_spec.Edges.Clear, edge)
	}
	if nodes := otu.fields; len(nodes) > 0 {
		edge := &sqlgraph.EdgeSpec{
			Rel:     sqlgraph.O2M,
			Inverse: false,
			Table:   objecttype.FieldsTable,
			Columns: []string{objecttype.FieldsColumn},
			Bidi:    false,
			Target: &sqlgraph.EdgeTarget{
				IDSpec: &sqlgraph.FieldSpec{
					Type:   field.TypeInt,
					Column: fieldtype.FieldID,
				},
			},
		}
		for k, _ := range nodes {
			edge.Target.Nodes = append(edge.Target.Nodes, k)
		}
		_spec.Edges.Add = append(_spec.Edges.Add, edge)
	}
	if n, err = sqlgraph.UpdateNodes(ctx, otu.driver, _spec); err != nil {
		if cerr, ok := isSQLConstraintError(err); ok {
			err = cerr
		}
		return 0, err
	}
	return n, nil
}

// ObjectTypeUpdateOne is the builder for updating a single ObjectType entity.
type ObjectTypeUpdateOne struct {
	config
	id            int
	name          *string
	kind          *string
	description   *string
	fields        map[int]struct{}
	removedFields map[int]struct{}
}

// SetName sets the name field.
func (otuo *ObjectTypeUpdateOne) SetName(s string) *ObjectTypeUpdateOne {
	otuo.name = &s
	return otuo
}

// SetKind sets the kind field.
func (otuo *ObjectTypeUpdateOne) SetKind(s string) *ObjectTypeUpdateOne {
	otuo.kind = &s
	return otuo
}

// SetDescription sets the description field.
func (otuo *ObjectTypeUpdateOne) SetDescription(s string) *ObjectTypeUpdateOne {
	otuo.description = &s
	return otuo
}

// AddFieldIDs adds the fields edge to FieldType by ids.
func (otuo *ObjectTypeUpdateOne) AddFieldIDs(ids ...int) *ObjectTypeUpdateOne {
	if otuo.fields == nil {
		otuo.fields = make(map[int]struct{})
	}
	for i := range ids {
		otuo.fields[ids[i]] = struct{}{}
	}
	return otuo
}

// AddFields adds the fields edges to FieldType.
func (otuo *ObjectTypeUpdateOne) AddFields(f ...*FieldType) *ObjectTypeUpdateOne {
	ids := make([]int, len(f))
	for i := range f {
		ids[i] = f[i].ID
	}
	return otuo.AddFieldIDs(ids...)
}

// RemoveFieldIDs removes the fields edge to FieldType by ids.
func (otuo *ObjectTypeUpdateOne) RemoveFieldIDs(ids ...int) *ObjectTypeUpdateOne {
	if otuo.removedFields == nil {
		otuo.removedFields = make(map[int]struct{})
	}
	for i := range ids {
		otuo.removedFields[ids[i]] = struct{}{}
	}
	return otuo
}

// RemoveFields removes fields edges to FieldType.
func (otuo *ObjectTypeUpdateOne) RemoveFields(f ...*FieldType) *ObjectTypeUpdateOne {
	ids := make([]int, len(f))
	for i := range f {
		ids[i] = f[i].ID
	}
	return otuo.RemoveFieldIDs(ids...)
}

// Save executes the query and returns the updated entity.
func (otuo *ObjectTypeUpdateOne) Save(ctx context.Context) (*ObjectType, error) {
	return otuo.sqlSave(ctx)
}

// SaveX is like Save, but panics if an error occurs.
func (otuo *ObjectTypeUpdateOne) SaveX(ctx context.Context) *ObjectType {
	ot, err := otuo.Save(ctx)
	if err != nil {
		panic(err)
	}
	return ot
}

// Exec executes the query on the entity.
func (otuo *ObjectTypeUpdateOne) Exec(ctx context.Context) error {
	_, err := otuo.Save(ctx)
	return err
}

// ExecX is like Exec, but panics if an error occurs.
func (otuo *ObjectTypeUpdateOne) ExecX(ctx context.Context) {
	if err := otuo.Exec(ctx); err != nil {
		panic(err)
	}
}

func (otuo *ObjectTypeUpdateOne) sqlSave(ctx context.Context) (ot *ObjectType, err error) {
	_spec := &sqlgraph.UpdateSpec{
		Node: &sqlgraph.NodeSpec{
			Table:   objecttype.Table,
			Columns: objecttype.Columns,
			ID: &sqlgraph.FieldSpec{
				Value:  otuo.id,
				Type:   field.TypeInt,
				Column: objecttype.FieldID,
			},
		},
	}
	if value := otuo.name; value != nil {
		_spec.Fields.Set = append(_spec.Fields.Set, &sqlgraph.FieldSpec{
			Type:   field.TypeString,
			Value:  *value,
			Column: objecttype.FieldName,
		})
	}
	if value := otuo.kind; value != nil {
		_spec.Fields.Set = append(_spec.Fields.Set, &sqlgraph.FieldSpec{
			Type:   field.TypeString,
			Value:  *value,
			Column: objecttype.FieldKind,
		})
	}
	if value := otuo.description; value != nil {
		_spec.Fields.Set = append(_spec.Fields.Set, &sqlgraph.FieldSpec{
			Type:   field.TypeString,
			Value:  *value,
			Column: objecttype.FieldDescription,
		})
	}
	if nodes := otuo.removedFields; len(nodes) > 0 {
		edge := &sqlgraph.EdgeSpec{
			Rel:     sqlgraph.O2M,
			Inverse: false,
			Table:   objecttype.FieldsTable,
			Columns: []string{objecttype.FieldsColumn},
			Bidi:    false,
			Target: &sqlgraph.EdgeTarget{
				IDSpec: &sqlgraph.FieldSpec{
					Type:   field.TypeInt,
					Column: fieldtype.FieldID,
				},
			},
		}
		for k, _ := range nodes {
			edge.Target.Nodes = append(edge.Target.Nodes, k)
		}
		_spec.Edges.Clear = append(_spec.Edges.Clear, edge)
	}
	if nodes := otuo.fields; len(nodes) > 0 {
		edge := &sqlgraph.EdgeSpec{
			Rel:     sqlgraph.O2M,
			Inverse: false,
			Table:   objecttype.FieldsTable,
			Columns: []string{objecttype.FieldsColumn},
			Bidi:    false,
			Target: &sqlgraph.EdgeTarget{
				IDSpec: &sqlgraph.FieldSpec{
					Type:   field.TypeInt,
					Column: fieldtype.FieldID,
				},
			},
		}
		for k, _ := range nodes {
			edge.Target.Nodes = append(edge.Target.Nodes, k)
		}
		_spec.Edges.Add = append(_spec.Edges.Add, edge)
	}
	ot = &ObjectType{config: otuo.config}
	_spec.Assign = ot.assignValues
	_spec.ScanValues = ot.scanValues()
	if err = sqlgraph.UpdateNode(ctx, otuo.driver, _spec); err != nil {
		if cerr, ok := isSQLConstraintError(err); ok {
			err = cerr
		}
		return nil, err
	}
	return ot, nil
}