crowdsec/pkg/database/ent/mutation.go

6346 lines
182 KiB
Go
Raw Normal View History

// Code generated by entc, DO NOT EDIT.
package ent
import (
"context"
"fmt"
"sync"
"time"
"github.com/crowdsecurity/crowdsec/pkg/database/ent/alert"
"github.com/crowdsecurity/crowdsec/pkg/database/ent/bouncer"
"github.com/crowdsecurity/crowdsec/pkg/database/ent/decision"
"github.com/crowdsecurity/crowdsec/pkg/database/ent/event"
"github.com/crowdsecurity/crowdsec/pkg/database/ent/machine"
"github.com/crowdsecurity/crowdsec/pkg/database/ent/meta"
2021-01-14 15:27:45 +00:00
"github.com/crowdsecurity/crowdsec/pkg/database/ent/predicate"
"github.com/facebook/ent"
)
const (
// Operation types.
OpCreate = ent.OpCreate
OpDelete = ent.OpDelete
OpDeleteOne = ent.OpDeleteOne
OpUpdate = ent.OpUpdate
OpUpdateOne = ent.OpUpdateOne
// Node types.
TypeAlert = "Alert"
TypeBouncer = "Bouncer"
TypeDecision = "Decision"
TypeEvent = "Event"
TypeMachine = "Machine"
TypeMeta = "Meta"
)
2021-01-14 15:27:45 +00:00
// AlertMutation represents an operation that mutates the Alert nodes in the graph.
type AlertMutation struct {
config
op Op
typ string
id *int
created_at *time.Time
updated_at *time.Time
scenario *string
bucketId *string
message *string
eventsCount *int32
addeventsCount *int32
startedAt *time.Time
stoppedAt *time.Time
sourceIp *string
sourceRange *string
sourceAsNumber *string
sourceAsName *string
sourceCountry *string
sourceLatitude *float32
addsourceLatitude *float32
sourceLongitude *float32
addsourceLongitude *float32
sourceScope *string
sourceValue *string
capacity *int32
addcapacity *int32
leakSpeed *string
scenarioVersion *string
scenarioHash *string
simulated *bool
clearedFields map[string]struct{}
owner *int
clearedowner bool
decisions map[int]struct{}
removeddecisions map[int]struct{}
cleareddecisions bool
events map[int]struct{}
removedevents map[int]struct{}
clearedevents bool
metas map[int]struct{}
removedmetas map[int]struct{}
clearedmetas bool
done bool
oldValue func(context.Context) (*Alert, error)
2021-01-14 15:27:45 +00:00
predicates []predicate.Alert
}
var _ ent.Mutation = (*AlertMutation)(nil)
2021-01-14 15:27:45 +00:00
// alertOption allows management of the mutation configuration using functional options.
type alertOption func(*AlertMutation)
2021-01-14 15:27:45 +00:00
// newAlertMutation creates new mutation for the Alert entity.
func newAlertMutation(c config, op Op, opts ...alertOption) *AlertMutation {
m := &AlertMutation{
config: c,
op: op,
typ: TypeAlert,
clearedFields: make(map[string]struct{}),
}
for _, opt := range opts {
opt(m)
}
return m
}
2021-01-14 15:27:45 +00:00
// withAlertID sets the ID field of the mutation.
func withAlertID(id int) alertOption {
return func(m *AlertMutation) {
var (
err error
once sync.Once
value *Alert
)
m.oldValue = func(ctx context.Context) (*Alert, error) {
once.Do(func() {
if m.done {
err = fmt.Errorf("querying old values post mutation is not allowed")
} else {
value, err = m.Client().Alert.Get(ctx, id)
}
})
return value, err
}
m.id = &id
}
}
// withAlert sets the old Alert of the mutation.
func withAlert(node *Alert) alertOption {
return func(m *AlertMutation) {
m.oldValue = func(context.Context) (*Alert, error) {
return node, nil
}
m.id = &node.ID
}
}
// Client returns a new `ent.Client` from the mutation. If the mutation was
// executed in a transaction (ent.Tx), a transactional client is returned.
func (m AlertMutation) Client() *Client {
client := &Client{config: m.config}
client.init()
return client
}
// Tx returns an `ent.Tx` for mutations that were executed in transactions;
// it returns an error otherwise.
func (m AlertMutation) Tx() (*Tx, error) {
if _, ok := m.driver.(*txDriver); !ok {
return nil, fmt.Errorf("ent: mutation is not running in a transaction")
}
tx := &Tx{config: m.config}
tx.init()
return tx, nil
}
2021-01-14 15:27:45 +00:00
// ID returns the ID value in the mutation. Note that the ID
// is only available if it was provided to the builder.
func (m *AlertMutation) ID() (id int, exists bool) {
if m.id == nil {
return
}
return *m.id, true
}
2021-01-14 15:27:45 +00:00
// SetCreatedAt sets the "created_at" field.
func (m *AlertMutation) SetCreatedAt(t time.Time) {
m.created_at = &t
}
2021-01-14 15:27:45 +00:00
// CreatedAt returns the value of the "created_at" field in the mutation.
func (m *AlertMutation) CreatedAt() (r time.Time, exists bool) {
v := m.created_at
if v == nil {
return
}
return *v, true
}
2021-01-14 15:27:45 +00:00
// OldCreatedAt returns the old "created_at" field's value of the Alert entity.
// If the Alert object wasn't provided to the builder, the object is fetched from the database.
// An error is returned if the mutation operation is not UpdateOne, or the database query fails.
func (m *AlertMutation) OldCreatedAt(ctx context.Context) (v time.Time, err error) {
if !m.op.Is(OpUpdateOne) {
2021-01-14 15:27:45 +00:00
return v, fmt.Errorf("OldCreatedAt is only allowed on UpdateOne operations")
}
if m.id == nil || m.oldValue == nil {
return v, fmt.Errorf("OldCreatedAt requires an ID field in the mutation")
}
oldValue, err := m.oldValue(ctx)
if err != nil {
return v, fmt.Errorf("querying old value for OldCreatedAt: %w", err)
}
return oldValue.CreatedAt, nil
}
2021-01-14 15:27:45 +00:00
// ResetCreatedAt resets all changes to the "created_at" field.
func (m *AlertMutation) ResetCreatedAt() {
m.created_at = nil
}
2021-01-14 15:27:45 +00:00
// SetUpdatedAt sets the "updated_at" field.
func (m *AlertMutation) SetUpdatedAt(t time.Time) {
m.updated_at = &t
}
2021-01-14 15:27:45 +00:00
// UpdatedAt returns the value of the "updated_at" field in the mutation.
func (m *AlertMutation) UpdatedAt() (r time.Time, exists bool) {
v := m.updated_at
if v == nil {
return
}
return *v, true
}
2021-01-14 15:27:45 +00:00
// OldUpdatedAt returns the old "updated_at" field's value of the Alert entity.
// If the Alert object wasn't provided to the builder, the object is fetched from the database.
// An error is returned if the mutation operation is not UpdateOne, or the database query fails.
func (m *AlertMutation) OldUpdatedAt(ctx context.Context) (v time.Time, err error) {
if !m.op.Is(OpUpdateOne) {
2021-01-14 15:27:45 +00:00
return v, fmt.Errorf("OldUpdatedAt is only allowed on UpdateOne operations")
}
if m.id == nil || m.oldValue == nil {
return v, fmt.Errorf("OldUpdatedAt requires an ID field in the mutation")
}
oldValue, err := m.oldValue(ctx)
if err != nil {
return v, fmt.Errorf("querying old value for OldUpdatedAt: %w", err)
}
return oldValue.UpdatedAt, nil
}
2021-01-14 15:27:45 +00:00
// ResetUpdatedAt resets all changes to the "updated_at" field.
func (m *AlertMutation) ResetUpdatedAt() {
m.updated_at = nil
}
2021-01-14 15:27:45 +00:00
// SetScenario sets the "scenario" field.
func (m *AlertMutation) SetScenario(s string) {
m.scenario = &s
}
2021-01-14 15:27:45 +00:00
// Scenario returns the value of the "scenario" field in the mutation.
func (m *AlertMutation) Scenario() (r string, exists bool) {
v := m.scenario
if v == nil {
return
}
return *v, true
}
2021-01-14 15:27:45 +00:00
// OldScenario returns the old "scenario" field's value of the Alert entity.
// If the Alert object wasn't provided to the builder, the object is fetched from the database.
// An error is returned if the mutation operation is not UpdateOne, or the database query fails.
func (m *AlertMutation) OldScenario(ctx context.Context) (v string, err error) {
if !m.op.Is(OpUpdateOne) {
2021-01-14 15:27:45 +00:00
return v, fmt.Errorf("OldScenario is only allowed on UpdateOne operations")
}
if m.id == nil || m.oldValue == nil {
return v, fmt.Errorf("OldScenario requires an ID field in the mutation")
}
oldValue, err := m.oldValue(ctx)
if err != nil {
return v, fmt.Errorf("querying old value for OldScenario: %w", err)
}
return oldValue.Scenario, nil
}
2021-01-14 15:27:45 +00:00
// ResetScenario resets all changes to the "scenario" field.
func (m *AlertMutation) ResetScenario() {
m.scenario = nil
}
2021-01-14 15:27:45 +00:00
// SetBucketId sets the "bucketId" field.
func (m *AlertMutation) SetBucketId(s string) {
m.bucketId = &s
}
2021-01-14 15:27:45 +00:00
// BucketId returns the value of the "bucketId" field in the mutation.
func (m *AlertMutation) BucketId() (r string, exists bool) {
v := m.bucketId
if v == nil {
return
}
return *v, true
}
2021-01-14 15:27:45 +00:00
// OldBucketId returns the old "bucketId" field's value of the Alert entity.
// If the Alert object wasn't provided to the builder, the object is fetched from the database.
// An error is returned if the mutation operation is not UpdateOne, or the database query fails.
func (m *AlertMutation) OldBucketId(ctx context.Context) (v string, err error) {
if !m.op.Is(OpUpdateOne) {
2021-01-14 15:27:45 +00:00
return v, fmt.Errorf("OldBucketId is only allowed on UpdateOne operations")
}
if m.id == nil || m.oldValue == nil {
return v, fmt.Errorf("OldBucketId requires an ID field in the mutation")
}
oldValue, err := m.oldValue(ctx)
if err != nil {
return v, fmt.Errorf("querying old value for OldBucketId: %w", err)
}
return oldValue.BucketId, nil
}
2021-01-14 15:27:45 +00:00
// ClearBucketId clears the value of the "bucketId" field.
func (m *AlertMutation) ClearBucketId() {
m.bucketId = nil
m.clearedFields[alert.FieldBucketId] = struct{}{}
}
2021-01-14 15:27:45 +00:00
// BucketIdCleared returns if the "bucketId" field was cleared in this mutation.
func (m *AlertMutation) BucketIdCleared() bool {
_, ok := m.clearedFields[alert.FieldBucketId]
return ok
}
2021-01-14 15:27:45 +00:00
// ResetBucketId resets all changes to the "bucketId" field.
func (m *AlertMutation) ResetBucketId() {
m.bucketId = nil
delete(m.clearedFields, alert.FieldBucketId)
}
2021-01-14 15:27:45 +00:00
// SetMessage sets the "message" field.
func (m *AlertMutation) SetMessage(s string) {
m.message = &s
}
2021-01-14 15:27:45 +00:00
// Message returns the value of the "message" field in the mutation.
func (m *AlertMutation) Message() (r string, exists bool) {
v := m.message
if v == nil {
return
}
return *v, true
}
2021-01-14 15:27:45 +00:00
// OldMessage returns the old "message" field's value of the Alert entity.
// If the Alert object wasn't provided to the builder, the object is fetched from the database.
// An error is returned if the mutation operation is not UpdateOne, or the database query fails.
func (m *AlertMutation) OldMessage(ctx context.Context) (v string, err error) {
if !m.op.Is(OpUpdateOne) {
2021-01-14 15:27:45 +00:00
return v, fmt.Errorf("OldMessage is only allowed on UpdateOne operations")
}
if m.id == nil || m.oldValue == nil {
return v, fmt.Errorf("OldMessage requires an ID field in the mutation")
}
oldValue, err := m.oldValue(ctx)
if err != nil {
return v, fmt.Errorf("querying old value for OldMessage: %w", err)
}
return oldValue.Message, nil
}
2021-01-14 15:27:45 +00:00
// ClearMessage clears the value of the "message" field.
func (m *AlertMutation) ClearMessage() {
m.message = nil
m.clearedFields[alert.FieldMessage] = struct{}{}
}
2021-01-14 15:27:45 +00:00
// MessageCleared returns if the "message" field was cleared in this mutation.
func (m *AlertMutation) MessageCleared() bool {
_, ok := m.clearedFields[alert.FieldMessage]
return ok
}
2021-01-14 15:27:45 +00:00
// ResetMessage resets all changes to the "message" field.
func (m *AlertMutation) ResetMessage() {
m.message = nil
delete(m.clearedFields, alert.FieldMessage)
}
2021-01-14 15:27:45 +00:00
// SetEventsCount sets the "eventsCount" field.
func (m *AlertMutation) SetEventsCount(i int32) {
m.eventsCount = &i
m.addeventsCount = nil
}
2021-01-14 15:27:45 +00:00
// EventsCount returns the value of the "eventsCount" field in the mutation.
func (m *AlertMutation) EventsCount() (r int32, exists bool) {
v := m.eventsCount
if v == nil {
return
}
return *v, true
}
2021-01-14 15:27:45 +00:00
// OldEventsCount returns the old "eventsCount" field's value of the Alert entity.
// If the Alert object wasn't provided to the builder, the object is fetched from the database.
// An error is returned if the mutation operation is not UpdateOne, or the database query fails.
func (m *AlertMutation) OldEventsCount(ctx context.Context) (v int32, err error) {
if !m.op.Is(OpUpdateOne) {
2021-01-14 15:27:45 +00:00
return v, fmt.Errorf("OldEventsCount is only allowed on UpdateOne operations")
}
if m.id == nil || m.oldValue == nil {
return v, fmt.Errorf("OldEventsCount requires an ID field in the mutation")
}
oldValue, err := m.oldValue(ctx)
if err != nil {
return v, fmt.Errorf("querying old value for OldEventsCount: %w", err)
}
return oldValue.EventsCount, nil
}
2021-01-14 15:27:45 +00:00
// AddEventsCount adds i to the "eventsCount" field.
func (m *AlertMutation) AddEventsCount(i int32) {
if m.addeventsCount != nil {
*m.addeventsCount += i
} else {
m.addeventsCount = &i
}
}
2021-01-14 15:27:45 +00:00
// AddedEventsCount returns the value that was added to the "eventsCount" field in this mutation.
func (m *AlertMutation) AddedEventsCount() (r int32, exists bool) {
v := m.addeventsCount
if v == nil {
return
}
return *v, true
}
2021-01-14 15:27:45 +00:00
// ClearEventsCount clears the value of the "eventsCount" field.
func (m *AlertMutation) ClearEventsCount() {
m.eventsCount = nil
m.addeventsCount = nil
m.clearedFields[alert.FieldEventsCount] = struct{}{}
}
2021-01-14 15:27:45 +00:00
// EventsCountCleared returns if the "eventsCount" field was cleared in this mutation.
func (m *AlertMutation) EventsCountCleared() bool {
_, ok := m.clearedFields[alert.FieldEventsCount]
return ok
}
2021-01-14 15:27:45 +00:00
// ResetEventsCount resets all changes to the "eventsCount" field.
func (m *AlertMutation) ResetEventsCount() {
m.eventsCount = nil
m.addeventsCount = nil
delete(m.clearedFields, alert.FieldEventsCount)
}
2021-01-14 15:27:45 +00:00
// SetStartedAt sets the "startedAt" field.
func (m *AlertMutation) SetStartedAt(t time.Time) {
m.startedAt = &t
}
2021-01-14 15:27:45 +00:00
// StartedAt returns the value of the "startedAt" field in the mutation.
func (m *AlertMutation) StartedAt() (r time.Time, exists bool) {
v := m.startedAt
if v == nil {
return
}
return *v, true
}
2021-01-14 15:27:45 +00:00
// OldStartedAt returns the old "startedAt" field's value of the Alert entity.
// If the Alert object wasn't provided to the builder, the object is fetched from the database.
// An error is returned if the mutation operation is not UpdateOne, or the database query fails.
func (m *AlertMutation) OldStartedAt(ctx context.Context) (v time.Time, err error) {
if !m.op.Is(OpUpdateOne) {
2021-01-14 15:27:45 +00:00
return v, fmt.Errorf("OldStartedAt is only allowed on UpdateOne operations")
}
if m.id == nil || m.oldValue == nil {
return v, fmt.Errorf("OldStartedAt requires an ID field in the mutation")
}
oldValue, err := m.oldValue(ctx)
if err != nil {
return v, fmt.Errorf("querying old value for OldStartedAt: %w", err)
}
return oldValue.StartedAt, nil
}
2021-01-14 15:27:45 +00:00
// ClearStartedAt clears the value of the "startedAt" field.
func (m *AlertMutation) ClearStartedAt() {
m.startedAt = nil
m.clearedFields[alert.FieldStartedAt] = struct{}{}
}
2021-01-14 15:27:45 +00:00
// StartedAtCleared returns if the "startedAt" field was cleared in this mutation.
func (m *AlertMutation) StartedAtCleared() bool {
_, ok := m.clearedFields[alert.FieldStartedAt]
return ok
}
2021-01-14 15:27:45 +00:00
// ResetStartedAt resets all changes to the "startedAt" field.
func (m *AlertMutation) ResetStartedAt() {
m.startedAt = nil
delete(m.clearedFields, alert.FieldStartedAt)
}
2021-01-14 15:27:45 +00:00
// SetStoppedAt sets the "stoppedAt" field.
func (m *AlertMutation) SetStoppedAt(t time.Time) {
m.stoppedAt = &t
}
2021-01-14 15:27:45 +00:00
// StoppedAt returns the value of the "stoppedAt" field in the mutation.
func (m *AlertMutation) StoppedAt() (r time.Time, exists bool) {
v := m.stoppedAt
if v == nil {
return
}
return *v, true
}
2021-01-14 15:27:45 +00:00
// OldStoppedAt returns the old "stoppedAt" field's value of the Alert entity.
// If the Alert object wasn't provided to the builder, the object is fetched from the database.
// An error is returned if the mutation operation is not UpdateOne, or the database query fails.
func (m *AlertMutation) OldStoppedAt(ctx context.Context) (v time.Time, err error) {
if !m.op.Is(OpUpdateOne) {
2021-01-14 15:27:45 +00:00
return v, fmt.Errorf("OldStoppedAt is only allowed on UpdateOne operations")
}
if m.id == nil || m.oldValue == nil {
return v, fmt.Errorf("OldStoppedAt requires an ID field in the mutation")
}
oldValue, err := m.oldValue(ctx)
if err != nil {
return v, fmt.Errorf("querying old value for OldStoppedAt: %w", err)
}
return oldValue.StoppedAt, nil
}
2021-01-14 15:27:45 +00:00
// ClearStoppedAt clears the value of the "stoppedAt" field.
func (m *AlertMutation) ClearStoppedAt() {
m.stoppedAt = nil
m.clearedFields[alert.FieldStoppedAt] = struct{}{}
}
2021-01-14 15:27:45 +00:00
// StoppedAtCleared returns if the "stoppedAt" field was cleared in this mutation.
func (m *AlertMutation) StoppedAtCleared() bool {
_, ok := m.clearedFields[alert.FieldStoppedAt]
return ok
}
2021-01-14 15:27:45 +00:00
// ResetStoppedAt resets all changes to the "stoppedAt" field.
func (m *AlertMutation) ResetStoppedAt() {
m.stoppedAt = nil
delete(m.clearedFields, alert.FieldStoppedAt)
}
2021-01-14 15:27:45 +00:00
// SetSourceIp sets the "sourceIp" field.
func (m *AlertMutation) SetSourceIp(s string) {
m.sourceIp = &s
}
2021-01-14 15:27:45 +00:00
// SourceIp returns the value of the "sourceIp" field in the mutation.
func (m *AlertMutation) SourceIp() (r string, exists bool) {
v := m.sourceIp
if v == nil {
return
}
return *v, true
}
2021-01-14 15:27:45 +00:00
// OldSourceIp returns the old "sourceIp" field's value of the Alert entity.
// If the Alert object wasn't provided to the builder, the object is fetched from the database.
// An error is returned if the mutation operation is not UpdateOne, or the database query fails.
func (m *AlertMutation) OldSourceIp(ctx context.Context) (v string, err error) {
if !m.op.Is(OpUpdateOne) {
2021-01-14 15:27:45 +00:00
return v, fmt.Errorf("OldSourceIp is only allowed on UpdateOne operations")
}
if m.id == nil || m.oldValue == nil {
return v, fmt.Errorf("OldSourceIp requires an ID field in the mutation")
}
oldValue, err := m.oldValue(ctx)
if err != nil {
return v, fmt.Errorf("querying old value for OldSourceIp: %w", err)
}
return oldValue.SourceIp, nil
}
2021-01-14 15:27:45 +00:00
// ClearSourceIp clears the value of the "sourceIp" field.
func (m *AlertMutation) ClearSourceIp() {
m.sourceIp = nil
m.clearedFields[alert.FieldSourceIp] = struct{}{}
}
2021-01-14 15:27:45 +00:00
// SourceIpCleared returns if the "sourceIp" field was cleared in this mutation.
func (m *AlertMutation) SourceIpCleared() bool {
_, ok := m.clearedFields[alert.FieldSourceIp]
return ok
}
2021-01-14 15:27:45 +00:00
// ResetSourceIp resets all changes to the "sourceIp" field.
func (m *AlertMutation) ResetSourceIp() {
m.sourceIp = nil
delete(m.clearedFields, alert.FieldSourceIp)
}
2021-01-14 15:27:45 +00:00
// SetSourceRange sets the "sourceRange" field.
func (m *AlertMutation) SetSourceRange(s string) {
m.sourceRange = &s
}
2021-01-14 15:27:45 +00:00
// SourceRange returns the value of the "sourceRange" field in the mutation.
func (m *AlertMutation) SourceRange() (r string, exists bool) {
v := m.sourceRange
if v == nil {
return
}
return *v, true
}
2021-01-14 15:27:45 +00:00
// OldSourceRange returns the old "sourceRange" field's value of the Alert entity.
// If the Alert object wasn't provided to the builder, the object is fetched from the database.
// An error is returned if the mutation operation is not UpdateOne, or the database query fails.
func (m *AlertMutation) OldSourceRange(ctx context.Context) (v string, err error) {
if !m.op.Is(OpUpdateOne) {
2021-01-14 15:27:45 +00:00
return v, fmt.Errorf("OldSourceRange is only allowed on UpdateOne operations")
}
if m.id == nil || m.oldValue == nil {
return v, fmt.Errorf("OldSourceRange requires an ID field in the mutation")
}
oldValue, err := m.oldValue(ctx)
if err != nil {
return v, fmt.Errorf("querying old value for OldSourceRange: %w", err)
}
return oldValue.SourceRange, nil
}
2021-01-14 15:27:45 +00:00
// ClearSourceRange clears the value of the "sourceRange" field.
func (m *AlertMutation) ClearSourceRange() {
m.sourceRange = nil
m.clearedFields[alert.FieldSourceRange] = struct{}{}
}
2021-01-14 15:27:45 +00:00
// SourceRangeCleared returns if the "sourceRange" field was cleared in this mutation.
func (m *AlertMutation) SourceRangeCleared() bool {
_, ok := m.clearedFields[alert.FieldSourceRange]
return ok
}
2021-01-14 15:27:45 +00:00
// ResetSourceRange resets all changes to the "sourceRange" field.
func (m *AlertMutation) ResetSourceRange() {
m.sourceRange = nil
delete(m.clearedFields, alert.FieldSourceRange)
}
2021-01-14 15:27:45 +00:00
// SetSourceAsNumber sets the "sourceAsNumber" field.
func (m *AlertMutation) SetSourceAsNumber(s string) {
m.sourceAsNumber = &s
}
2021-01-14 15:27:45 +00:00
// SourceAsNumber returns the value of the "sourceAsNumber" field in the mutation.
func (m *AlertMutation) SourceAsNumber() (r string, exists bool) {
v := m.sourceAsNumber
if v == nil {
return
}
return *v, true
}
2021-01-14 15:27:45 +00:00
// OldSourceAsNumber returns the old "sourceAsNumber" field's value of the Alert entity.
// If the Alert object wasn't provided to the builder, the object is fetched from the database.
// An error is returned if the mutation operation is not UpdateOne, or the database query fails.
func (m *AlertMutation) OldSourceAsNumber(ctx context.Context) (v string, err error) {
if !m.op.Is(OpUpdateOne) {
2021-01-14 15:27:45 +00:00
return v, fmt.Errorf("OldSourceAsNumber is only allowed on UpdateOne operations")
}
if m.id == nil || m.oldValue == nil {
return v, fmt.Errorf("OldSourceAsNumber requires an ID field in the mutation")
}
oldValue, err := m.oldValue(ctx)
if err != nil {
return v, fmt.Errorf("querying old value for OldSourceAsNumber: %w", err)
}
return oldValue.SourceAsNumber, nil
}
2021-01-14 15:27:45 +00:00
// ClearSourceAsNumber clears the value of the "sourceAsNumber" field.
func (m *AlertMutation) ClearSourceAsNumber() {
m.sourceAsNumber = nil
m.clearedFields[alert.FieldSourceAsNumber] = struct{}{}
}
2021-01-14 15:27:45 +00:00
// SourceAsNumberCleared returns if the "sourceAsNumber" field was cleared in this mutation.
func (m *AlertMutation) SourceAsNumberCleared() bool {
_, ok := m.clearedFields[alert.FieldSourceAsNumber]
return ok
}
2021-01-14 15:27:45 +00:00
// ResetSourceAsNumber resets all changes to the "sourceAsNumber" field.
func (m *AlertMutation) ResetSourceAsNumber() {
m.sourceAsNumber = nil
delete(m.clearedFields, alert.FieldSourceAsNumber)
}
2021-01-14 15:27:45 +00:00
// SetSourceAsName sets the "sourceAsName" field.
func (m *AlertMutation) SetSourceAsName(s string) {
m.sourceAsName = &s
}
2021-01-14 15:27:45 +00:00
// SourceAsName returns the value of the "sourceAsName" field in the mutation.
func (m *AlertMutation) SourceAsName() (r string, exists bool) {
v := m.sourceAsName
if v == nil {
return
}
return *v, true
}
2021-01-14 15:27:45 +00:00
// OldSourceAsName returns the old "sourceAsName" field's value of the Alert entity.
// If the Alert object wasn't provided to the builder, the object is fetched from the database.
// An error is returned if the mutation operation is not UpdateOne, or the database query fails.
func (m *AlertMutation) OldSourceAsName(ctx context.Context) (v string, err error) {
if !m.op.Is(OpUpdateOne) {
2021-01-14 15:27:45 +00:00
return v, fmt.Errorf("OldSourceAsName is only allowed on UpdateOne operations")
}
if m.id == nil || m.oldValue == nil {
return v, fmt.Errorf("OldSourceAsName requires an ID field in the mutation")
}
oldValue, err := m.oldValue(ctx)
if err != nil {
return v, fmt.Errorf("querying old value for OldSourceAsName: %w", err)
}
return oldValue.SourceAsName, nil
}
2021-01-14 15:27:45 +00:00
// ClearSourceAsName clears the value of the "sourceAsName" field.
func (m *AlertMutation) ClearSourceAsName() {
m.sourceAsName = nil
m.clearedFields[alert.FieldSourceAsName] = struct{}{}
}
2021-01-14 15:27:45 +00:00
// SourceAsNameCleared returns if the "sourceAsName" field was cleared in this mutation.
func (m *AlertMutation) SourceAsNameCleared() bool {
_, ok := m.clearedFields[alert.FieldSourceAsName]
return ok
}
2021-01-14 15:27:45 +00:00
// ResetSourceAsName resets all changes to the "sourceAsName" field.
func (m *AlertMutation) ResetSourceAsName() {
m.sourceAsName = nil
delete(m.clearedFields, alert.FieldSourceAsName)
}
2021-01-14 15:27:45 +00:00
// SetSourceCountry sets the "sourceCountry" field.
func (m *AlertMutation) SetSourceCountry(s string) {
m.sourceCountry = &s
}
2021-01-14 15:27:45 +00:00
// SourceCountry returns the value of the "sourceCountry" field in the mutation.
func (m *AlertMutation) SourceCountry() (r string, exists bool) {
v := m.sourceCountry
if v == nil {
return
}
return *v, true
}
2021-01-14 15:27:45 +00:00
// OldSourceCountry returns the old "sourceCountry" field's value of the Alert entity.
// If the Alert object wasn't provided to the builder, the object is fetched from the database.
// An error is returned if the mutation operation is not UpdateOne, or the database query fails.
func (m *AlertMutation) OldSourceCountry(ctx context.Context) (v string, err error) {
if !m.op.Is(OpUpdateOne) {
2021-01-14 15:27:45 +00:00
return v, fmt.Errorf("OldSourceCountry is only allowed on UpdateOne operations")
}
if m.id == nil || m.oldValue == nil {
return v, fmt.Errorf("OldSourceCountry requires an ID field in the mutation")
}
oldValue, err := m.oldValue(ctx)
if err != nil {
return v, fmt.Errorf("querying old value for OldSourceCountry: %w", err)
}
return oldValue.SourceCountry, nil
}
2021-01-14 15:27:45 +00:00
// ClearSourceCountry clears the value of the "sourceCountry" field.
func (m *AlertMutation) ClearSourceCountry() {
m.sourceCountry = nil
m.clearedFields[alert.FieldSourceCountry] = struct{}{}
}
2021-01-14 15:27:45 +00:00
// SourceCountryCleared returns if the "sourceCountry" field was cleared in this mutation.
func (m *AlertMutation) SourceCountryCleared() bool {
_, ok := m.clearedFields[alert.FieldSourceCountry]
return ok
}
2021-01-14 15:27:45 +00:00
// ResetSourceCountry resets all changes to the "sourceCountry" field.
func (m *AlertMutation) ResetSourceCountry() {
m.sourceCountry = nil
delete(m.clearedFields, alert.FieldSourceCountry)
}
2021-01-14 15:27:45 +00:00
// SetSourceLatitude sets the "sourceLatitude" field.
func (m *AlertMutation) SetSourceLatitude(f float32) {
m.sourceLatitude = &f
m.addsourceLatitude = nil
}
2021-01-14 15:27:45 +00:00
// SourceLatitude returns the value of the "sourceLatitude" field in the mutation.
func (m *AlertMutation) SourceLatitude() (r float32, exists bool) {
v := m.sourceLatitude
if v == nil {
return
}
return *v, true
}
2021-01-14 15:27:45 +00:00
// OldSourceLatitude returns the old "sourceLatitude" field's value of the Alert entity.
// If the Alert object wasn't provided to the builder, the object is fetched from the database.
// An error is returned if the mutation operation is not UpdateOne, or the database query fails.
func (m *AlertMutation) OldSourceLatitude(ctx context.Context) (v float32, err error) {
if !m.op.Is(OpUpdateOne) {
2021-01-14 15:27:45 +00:00
return v, fmt.Errorf("OldSourceLatitude is only allowed on UpdateOne operations")
}
if m.id == nil || m.oldValue == nil {
return v, fmt.Errorf("OldSourceLatitude requires an ID field in the mutation")
}
oldValue, err := m.oldValue(ctx)
if err != nil {
return v, fmt.Errorf("querying old value for OldSourceLatitude: %w", err)
}
return oldValue.SourceLatitude, nil
}
2021-01-14 15:27:45 +00:00
// AddSourceLatitude adds f to the "sourceLatitude" field.
func (m *AlertMutation) AddSourceLatitude(f float32) {
if m.addsourceLatitude != nil {
*m.addsourceLatitude += f
} else {
m.addsourceLatitude = &f
}
}
2021-01-14 15:27:45 +00:00
// AddedSourceLatitude returns the value that was added to the "sourceLatitude" field in this mutation.
func (m *AlertMutation) AddedSourceLatitude() (r float32, exists bool) {
v := m.addsourceLatitude
if v == nil {
return
}
return *v, true
}
2021-01-14 15:27:45 +00:00
// ClearSourceLatitude clears the value of the "sourceLatitude" field.
func (m *AlertMutation) ClearSourceLatitude() {
m.sourceLatitude = nil
m.addsourceLatitude = nil
m.clearedFields[alert.FieldSourceLatitude] = struct{}{}
}
2021-01-14 15:27:45 +00:00
// SourceLatitudeCleared returns if the "sourceLatitude" field was cleared in this mutation.
func (m *AlertMutation) SourceLatitudeCleared() bool {
_, ok := m.clearedFields[alert.FieldSourceLatitude]
return ok
}
2021-01-14 15:27:45 +00:00
// ResetSourceLatitude resets all changes to the "sourceLatitude" field.
func (m *AlertMutation) ResetSourceLatitude() {
m.sourceLatitude = nil
m.addsourceLatitude = nil
delete(m.clearedFields, alert.FieldSourceLatitude)
}
2021-01-14 15:27:45 +00:00
// SetSourceLongitude sets the "sourceLongitude" field.
func (m *AlertMutation) SetSourceLongitude(f float32) {
m.sourceLongitude = &f
m.addsourceLongitude = nil
}
2021-01-14 15:27:45 +00:00
// SourceLongitude returns the value of the "sourceLongitude" field in the mutation.
func (m *AlertMutation) SourceLongitude() (r float32, exists bool) {
v := m.sourceLongitude
if v == nil {
return
}
return *v, true
}
2021-01-14 15:27:45 +00:00
// OldSourceLongitude returns the old "sourceLongitude" field's value of the Alert entity.
// If the Alert object wasn't provided to the builder, the object is fetched from the database.
// An error is returned if the mutation operation is not UpdateOne, or the database query fails.
func (m *AlertMutation) OldSourceLongitude(ctx context.Context) (v float32, err error) {
if !m.op.Is(OpUpdateOne) {
2021-01-14 15:27:45 +00:00
return v, fmt.Errorf("OldSourceLongitude is only allowed on UpdateOne operations")
}
if m.id == nil || m.oldValue == nil {
return v, fmt.Errorf("OldSourceLongitude requires an ID field in the mutation")
}
oldValue, err := m.oldValue(ctx)
if err != nil {
return v, fmt.Errorf("querying old value for OldSourceLongitude: %w", err)
}
return oldValue.SourceLongitude, nil
}
2021-01-14 15:27:45 +00:00
// AddSourceLongitude adds f to the "sourceLongitude" field.
func (m *AlertMutation) AddSourceLongitude(f float32) {
if m.addsourceLongitude != nil {
*m.addsourceLongitude += f
} else {
m.addsourceLongitude = &f
}
}
2021-01-14 15:27:45 +00:00
// AddedSourceLongitude returns the value that was added to the "sourceLongitude" field in this mutation.
func (m *AlertMutation) AddedSourceLongitude() (r float32, exists bool) {
v := m.addsourceLongitude
if v == nil {
return
}
return *v, true
}
2021-01-14 15:27:45 +00:00
// ClearSourceLongitude clears the value of the "sourceLongitude" field.
func (m *AlertMutation) ClearSourceLongitude() {
m.sourceLongitude = nil
m.addsourceLongitude = nil
m.clearedFields[alert.FieldSourceLongitude] = struct{}{}
}
2021-01-14 15:27:45 +00:00
// SourceLongitudeCleared returns if the "sourceLongitude" field was cleared in this mutation.
func (m *AlertMutation) SourceLongitudeCleared() bool {
_, ok := m.clearedFields[alert.FieldSourceLongitude]
return ok
}
2021-01-14 15:27:45 +00:00
// ResetSourceLongitude resets all changes to the "sourceLongitude" field.
func (m *AlertMutation) ResetSourceLongitude() {
m.sourceLongitude = nil
m.addsourceLongitude = nil
delete(m.clearedFields, alert.FieldSourceLongitude)
}
2021-01-14 15:27:45 +00:00
// SetSourceScope sets the "sourceScope" field.
func (m *AlertMutation) SetSourceScope(s string) {
m.sourceScope = &s
}
2021-01-14 15:27:45 +00:00
// SourceScope returns the value of the "sourceScope" field in the mutation.
func (m *AlertMutation) SourceScope() (r string, exists bool) {
v := m.sourceScope
if v == nil {
return
}
return *v, true
}
2021-01-14 15:27:45 +00:00
// OldSourceScope returns the old "sourceScope" field's value of the Alert entity.
// If the Alert object wasn't provided to the builder, the object is fetched from the database.
// An error is returned if the mutation operation is not UpdateOne, or the database query fails.
func (m *AlertMutation) OldSourceScope(ctx context.Context) (v string, err error) {
if !m.op.Is(OpUpdateOne) {
2021-01-14 15:27:45 +00:00
return v, fmt.Errorf("OldSourceScope is only allowed on UpdateOne operations")
}
if m.id == nil || m.oldValue == nil {
return v, fmt.Errorf("OldSourceScope requires an ID field in the mutation")
}
oldValue, err := m.oldValue(ctx)
if err != nil {
return v, fmt.Errorf("querying old value for OldSourceScope: %w", err)
}
return oldValue.SourceScope, nil
}
2021-01-14 15:27:45 +00:00
// ClearSourceScope clears the value of the "sourceScope" field.
func (m *AlertMutation) ClearSourceScope() {
m.sourceScope = nil
m.clearedFields[alert.FieldSourceScope] = struct{}{}
}
2021-01-14 15:27:45 +00:00
// SourceScopeCleared returns if the "sourceScope" field was cleared in this mutation.
func (m *AlertMutation) SourceScopeCleared() bool {
_, ok := m.clearedFields[alert.FieldSourceScope]
return ok
}
2021-01-14 15:27:45 +00:00
// ResetSourceScope resets all changes to the "sourceScope" field.
func (m *AlertMutation) ResetSourceScope() {
m.sourceScope = nil
delete(m.clearedFields, alert.FieldSourceScope)
}
2021-01-14 15:27:45 +00:00
// SetSourceValue sets the "sourceValue" field.
func (m *AlertMutation) SetSourceValue(s string) {
m.sourceValue = &s
}
2021-01-14 15:27:45 +00:00
// SourceValue returns the value of the "sourceValue" field in the mutation.
func (m *AlertMutation) SourceValue() (r string, exists bool) {
v := m.sourceValue
if v == nil {
return
}
return *v, true
}
2021-01-14 15:27:45 +00:00
// OldSourceValue returns the old "sourceValue" field's value of the Alert entity.
// If the Alert object wasn't provided to the builder, the object is fetched from the database.
// An error is returned if the mutation operation is not UpdateOne, or the database query fails.
func (m *AlertMutation) OldSourceValue(ctx context.Context) (v string, err error) {
if !m.op.Is(OpUpdateOne) {
2021-01-14 15:27:45 +00:00
return v, fmt.Errorf("OldSourceValue is only allowed on UpdateOne operations")
}
if m.id == nil || m.oldValue == nil {
return v, fmt.Errorf("OldSourceValue requires an ID field in the mutation")
}
oldValue, err := m.oldValue(ctx)
if err != nil {
return v, fmt.Errorf("querying old value for OldSourceValue: %w", err)
}
return oldValue.SourceValue, nil
}
2021-01-14 15:27:45 +00:00
// ClearSourceValue clears the value of the "sourceValue" field.
func (m *AlertMutation) ClearSourceValue() {
m.sourceValue = nil
m.clearedFields[alert.FieldSourceValue] = struct{}{}
}
2021-01-14 15:27:45 +00:00
// SourceValueCleared returns if the "sourceValue" field was cleared in this mutation.
func (m *AlertMutation) SourceValueCleared() bool {
_, ok := m.clearedFields[alert.FieldSourceValue]
return ok
}
2021-01-14 15:27:45 +00:00
// ResetSourceValue resets all changes to the "sourceValue" field.
func (m *AlertMutation) ResetSourceValue() {
m.sourceValue = nil
delete(m.clearedFields, alert.FieldSourceValue)
}
2021-01-14 15:27:45 +00:00
// SetCapacity sets the "capacity" field.
func (m *AlertMutation) SetCapacity(i int32) {
m.capacity = &i
m.addcapacity = nil
}
2021-01-14 15:27:45 +00:00
// Capacity returns the value of the "capacity" field in the mutation.
func (m *AlertMutation) Capacity() (r int32, exists bool) {
v := m.capacity
if v == nil {
return
}
return *v, true
}
2021-01-14 15:27:45 +00:00
// OldCapacity returns the old "capacity" field's value of the Alert entity.
// If the Alert object wasn't provided to the builder, the object is fetched from the database.
// An error is returned if the mutation operation is not UpdateOne, or the database query fails.
func (m *AlertMutation) OldCapacity(ctx context.Context) (v int32, err error) {
if !m.op.Is(OpUpdateOne) {
2021-01-14 15:27:45 +00:00
return v, fmt.Errorf("OldCapacity is only allowed on UpdateOne operations")
}
if m.id == nil || m.oldValue == nil {
return v, fmt.Errorf("OldCapacity requires an ID field in the mutation")
}
oldValue, err := m.oldValue(ctx)
if err != nil {
return v, fmt.Errorf("querying old value for OldCapacity: %w", err)
}
return oldValue.Capacity, nil
}
2021-01-14 15:27:45 +00:00
// AddCapacity adds i to the "capacity" field.
func (m *AlertMutation) AddCapacity(i int32) {
if m.addcapacity != nil {
*m.addcapacity += i
} else {
m.addcapacity = &i
}
}
2021-01-14 15:27:45 +00:00
// AddedCapacity returns the value that was added to the "capacity" field in this mutation.
func (m *AlertMutation) AddedCapacity() (r int32, exists bool) {
v := m.addcapacity
if v == nil {
return
}
return *v, true
}
2021-01-14 15:27:45 +00:00
// ClearCapacity clears the value of the "capacity" field.
func (m *AlertMutation) ClearCapacity() {
m.capacity = nil
m.addcapacity = nil
m.clearedFields[alert.FieldCapacity] = struct{}{}
}
2021-01-14 15:27:45 +00:00
// CapacityCleared returns if the "capacity" field was cleared in this mutation.
func (m *AlertMutation) CapacityCleared() bool {
_, ok := m.clearedFields[alert.FieldCapacity]
return ok
}
2021-01-14 15:27:45 +00:00
// ResetCapacity resets all changes to the "capacity" field.
func (m *AlertMutation) ResetCapacity() {
m.capacity = nil
m.addcapacity = nil
delete(m.clearedFields, alert.FieldCapacity)
}
2021-01-14 15:27:45 +00:00
// SetLeakSpeed sets the "leakSpeed" field.
func (m *AlertMutation) SetLeakSpeed(s string) {
m.leakSpeed = &s
}
2021-01-14 15:27:45 +00:00
// LeakSpeed returns the value of the "leakSpeed" field in the mutation.
func (m *AlertMutation) LeakSpeed() (r string, exists bool) {
v := m.leakSpeed
if v == nil {
return
}
return *v, true
}
2021-01-14 15:27:45 +00:00
// OldLeakSpeed returns the old "leakSpeed" field's value of the Alert entity.
// If the Alert object wasn't provided to the builder, the object is fetched from the database.
// An error is returned if the mutation operation is not UpdateOne, or the database query fails.
func (m *AlertMutation) OldLeakSpeed(ctx context.Context) (v string, err error) {
if !m.op.Is(OpUpdateOne) {
2021-01-14 15:27:45 +00:00
return v, fmt.Errorf("OldLeakSpeed is only allowed on UpdateOne operations")
}
if m.id == nil || m.oldValue == nil {
return v, fmt.Errorf("OldLeakSpeed requires an ID field in the mutation")
}
oldValue, err := m.oldValue(ctx)
if err != nil {
return v, fmt.Errorf("querying old value for OldLeakSpeed: %w", err)
}
return oldValue.LeakSpeed, nil
}
2021-01-14 15:27:45 +00:00
// ClearLeakSpeed clears the value of the "leakSpeed" field.
func (m *AlertMutation) ClearLeakSpeed() {
m.leakSpeed = nil
m.clearedFields[alert.FieldLeakSpeed] = struct{}{}
}
2021-01-14 15:27:45 +00:00
// LeakSpeedCleared returns if the "leakSpeed" field was cleared in this mutation.
func (m *AlertMutation) LeakSpeedCleared() bool {
_, ok := m.clearedFields[alert.FieldLeakSpeed]
return ok
}
2021-01-14 15:27:45 +00:00
// ResetLeakSpeed resets all changes to the "leakSpeed" field.
func (m *AlertMutation) ResetLeakSpeed() {
m.leakSpeed = nil
delete(m.clearedFields, alert.FieldLeakSpeed)
}
2021-01-14 15:27:45 +00:00
// SetScenarioVersion sets the "scenarioVersion" field.
func (m *AlertMutation) SetScenarioVersion(s string) {
m.scenarioVersion = &s
}
2021-01-14 15:27:45 +00:00
// ScenarioVersion returns the value of the "scenarioVersion" field in the mutation.
func (m *AlertMutation) ScenarioVersion() (r string, exists bool) {
v := m.scenarioVersion
if v == nil {
return
}
return *v, true
}
2021-01-14 15:27:45 +00:00
// OldScenarioVersion returns the old "scenarioVersion" field's value of the Alert entity.
// If the Alert object wasn't provided to the builder, the object is fetched from the database.
// An error is returned if the mutation operation is not UpdateOne, or the database query fails.
func (m *AlertMutation) OldScenarioVersion(ctx context.Context) (v string, err error) {
if !m.op.Is(OpUpdateOne) {
2021-01-14 15:27:45 +00:00
return v, fmt.Errorf("OldScenarioVersion is only allowed on UpdateOne operations")
}
if m.id == nil || m.oldValue == nil {
return v, fmt.Errorf("OldScenarioVersion requires an ID field in the mutation")
}
oldValue, err := m.oldValue(ctx)
if err != nil {
return v, fmt.Errorf("querying old value for OldScenarioVersion: %w", err)
}
return oldValue.ScenarioVersion, nil
}
2021-01-14 15:27:45 +00:00
// ClearScenarioVersion clears the value of the "scenarioVersion" field.
func (m *AlertMutation) ClearScenarioVersion() {
m.scenarioVersion = nil
m.clearedFields[alert.FieldScenarioVersion] = struct{}{}
}
2021-01-14 15:27:45 +00:00
// ScenarioVersionCleared returns if the "scenarioVersion" field was cleared in this mutation.
func (m *AlertMutation) ScenarioVersionCleared() bool {
_, ok := m.clearedFields[alert.FieldScenarioVersion]
return ok
}
2021-01-14 15:27:45 +00:00
// ResetScenarioVersion resets all changes to the "scenarioVersion" field.
func (m *AlertMutation) ResetScenarioVersion() {
m.scenarioVersion = nil
delete(m.clearedFields, alert.FieldScenarioVersion)
}
2021-01-14 15:27:45 +00:00
// SetScenarioHash sets the "scenarioHash" field.
func (m *AlertMutation) SetScenarioHash(s string) {
m.scenarioHash = &s
}
2021-01-14 15:27:45 +00:00
// ScenarioHash returns the value of the "scenarioHash" field in the mutation.
func (m *AlertMutation) ScenarioHash() (r string, exists bool) {
v := m.scenarioHash
if v == nil {
return
}
return *v, true
}
2021-01-14 15:27:45 +00:00
// OldScenarioHash returns the old "scenarioHash" field's value of the Alert entity.
// If the Alert object wasn't provided to the builder, the object is fetched from the database.
// An error is returned if the mutation operation is not UpdateOne, or the database query fails.
func (m *AlertMutation) OldScenarioHash(ctx context.Context) (v string, err error) {
if !m.op.Is(OpUpdateOne) {
2021-01-14 15:27:45 +00:00
return v, fmt.Errorf("OldScenarioHash is only allowed on UpdateOne operations")
}
if m.id == nil || m.oldValue == nil {
return v, fmt.Errorf("OldScenarioHash requires an ID field in the mutation")
}
oldValue, err := m.oldValue(ctx)
if err != nil {
return v, fmt.Errorf("querying old value for OldScenarioHash: %w", err)
}
return oldValue.ScenarioHash, nil
}
2021-01-14 15:27:45 +00:00
// ClearScenarioHash clears the value of the "scenarioHash" field.
func (m *AlertMutation) ClearScenarioHash() {
m.scenarioHash = nil
m.clearedFields[alert.FieldScenarioHash] = struct{}{}
}
2021-01-14 15:27:45 +00:00
// ScenarioHashCleared returns if the "scenarioHash" field was cleared in this mutation.
func (m *AlertMutation) ScenarioHashCleared() bool {
_, ok := m.clearedFields[alert.FieldScenarioHash]
return ok
}
2021-01-14 15:27:45 +00:00
// ResetScenarioHash resets all changes to the "scenarioHash" field.
func (m *AlertMutation) ResetScenarioHash() {
m.scenarioHash = nil
delete(m.clearedFields, alert.FieldScenarioHash)
}
2021-01-14 15:27:45 +00:00
// SetSimulated sets the "simulated" field.
func (m *AlertMutation) SetSimulated(b bool) {
m.simulated = &b
}
2021-01-14 15:27:45 +00:00
// Simulated returns the value of the "simulated" field in the mutation.
func (m *AlertMutation) Simulated() (r bool, exists bool) {
v := m.simulated
if v == nil {
return
}
return *v, true
}
2021-01-14 15:27:45 +00:00
// OldSimulated returns the old "simulated" field's value of the Alert entity.
// If the Alert object wasn't provided to the builder, the object is fetched from the database.
// An error is returned if the mutation operation is not UpdateOne, or the database query fails.
func (m *AlertMutation) OldSimulated(ctx context.Context) (v bool, err error) {
if !m.op.Is(OpUpdateOne) {
2021-01-14 15:27:45 +00:00
return v, fmt.Errorf("OldSimulated is only allowed on UpdateOne operations")
}
if m.id == nil || m.oldValue == nil {
return v, fmt.Errorf("OldSimulated requires an ID field in the mutation")
}
oldValue, err := m.oldValue(ctx)
if err != nil {
return v, fmt.Errorf("querying old value for OldSimulated: %w", err)
}
return oldValue.Simulated, nil
}
2021-01-14 15:27:45 +00:00
// ResetSimulated resets all changes to the "simulated" field.
func (m *AlertMutation) ResetSimulated() {
m.simulated = nil
}
2021-01-14 15:27:45 +00:00
// SetOwnerID sets the "owner" edge to the Machine entity by id.
func (m *AlertMutation) SetOwnerID(id int) {
m.owner = &id
}
2021-01-14 15:27:45 +00:00
// ClearOwner clears the "owner" edge to the Machine entity.
func (m *AlertMutation) ClearOwner() {
m.clearedowner = true
}
2021-01-14 15:27:45 +00:00
// OwnerCleared returns if the "owner" edge to the Machine entity was cleared.
func (m *AlertMutation) OwnerCleared() bool {
return m.clearedowner
}
2021-01-14 15:27:45 +00:00
// OwnerID returns the "owner" edge ID in the mutation.
func (m *AlertMutation) OwnerID() (id int, exists bool) {
if m.owner != nil {
return *m.owner, true
}
return
}
2021-01-14 15:27:45 +00:00
// OwnerIDs returns the "owner" edge IDs in the mutation.
// Note that IDs always returns len(IDs) <= 1 for unique edges, and you should use
// OwnerID instead. It exists only for internal usage by the builders.
func (m *AlertMutation) OwnerIDs() (ids []int) {
if id := m.owner; id != nil {
ids = append(ids, *id)
}
return
}
2021-01-14 15:27:45 +00:00
// ResetOwner resets all changes to the "owner" edge.
func (m *AlertMutation) ResetOwner() {
m.owner = nil
m.clearedowner = false
}
2021-01-14 15:27:45 +00:00
// AddDecisionIDs adds the "decisions" edge to the Decision entity by ids.
func (m *AlertMutation) AddDecisionIDs(ids ...int) {
if m.decisions == nil {
m.decisions = make(map[int]struct{})
}
for i := range ids {
m.decisions[ids[i]] = struct{}{}
}
}
2021-01-14 15:27:45 +00:00
// ClearDecisions clears the "decisions" edge to the Decision entity.
func (m *AlertMutation) ClearDecisions() {
m.cleareddecisions = true
}
2021-01-14 15:27:45 +00:00
// DecisionsCleared returns if the "decisions" edge to the Decision entity was cleared.
func (m *AlertMutation) DecisionsCleared() bool {
return m.cleareddecisions
}
2021-01-14 15:27:45 +00:00
// RemoveDecisionIDs removes the "decisions" edge to the Decision entity by IDs.
func (m *AlertMutation) RemoveDecisionIDs(ids ...int) {
if m.removeddecisions == nil {
m.removeddecisions = make(map[int]struct{})
}
for i := range ids {
m.removeddecisions[ids[i]] = struct{}{}
}
}
2021-01-14 15:27:45 +00:00
// RemovedDecisions returns the removed IDs of the "decisions" edge to the Decision entity.
func (m *AlertMutation) RemovedDecisionsIDs() (ids []int) {
for id := range m.removeddecisions {
ids = append(ids, id)
}
return
}
2021-01-14 15:27:45 +00:00
// DecisionsIDs returns the "decisions" edge IDs in the mutation.
func (m *AlertMutation) DecisionsIDs() (ids []int) {
for id := range m.decisions {
ids = append(ids, id)
}
return
}
2021-01-14 15:27:45 +00:00
// ResetDecisions resets all changes to the "decisions" edge.
func (m *AlertMutation) ResetDecisions() {
m.decisions = nil
m.cleareddecisions = false
m.removeddecisions = nil
}
2021-01-14 15:27:45 +00:00
// AddEventIDs adds the "events" edge to the Event entity by ids.
func (m *AlertMutation) AddEventIDs(ids ...int) {
if m.events == nil {
m.events = make(map[int]struct{})
}
for i := range ids {
m.events[ids[i]] = struct{}{}
}
}
2021-01-14 15:27:45 +00:00
// ClearEvents clears the "events" edge to the Event entity.
func (m *AlertMutation) ClearEvents() {
m.clearedevents = true
}
2021-01-14 15:27:45 +00:00
// EventsCleared returns if the "events" edge to the Event entity was cleared.
func (m *AlertMutation) EventsCleared() bool {
return m.clearedevents
}
2021-01-14 15:27:45 +00:00
// RemoveEventIDs removes the "events" edge to the Event entity by IDs.
func (m *AlertMutation) RemoveEventIDs(ids ...int) {
if m.removedevents == nil {
m.removedevents = make(map[int]struct{})
}
for i := range ids {
m.removedevents[ids[i]] = struct{}{}
}
}
2021-01-14 15:27:45 +00:00
// RemovedEvents returns the removed IDs of the "events" edge to the Event entity.
func (m *AlertMutation) RemovedEventsIDs() (ids []int) {
for id := range m.removedevents {
ids = append(ids, id)
}
return
}
2021-01-14 15:27:45 +00:00
// EventsIDs returns the "events" edge IDs in the mutation.
func (m *AlertMutation) EventsIDs() (ids []int) {
for id := range m.events {
ids = append(ids, id)
}
return
}
2021-01-14 15:27:45 +00:00
// ResetEvents resets all changes to the "events" edge.
func (m *AlertMutation) ResetEvents() {
m.events = nil
m.clearedevents = false
m.removedevents = nil
}
2021-01-14 15:27:45 +00:00
// AddMetaIDs adds the "metas" edge to the Meta entity by ids.
func (m *AlertMutation) AddMetaIDs(ids ...int) {
if m.metas == nil {
m.metas = make(map[int]struct{})
}
for i := range ids {
m.metas[ids[i]] = struct{}{}
}
}
2021-01-14 15:27:45 +00:00
// ClearMetas clears the "metas" edge to the Meta entity.
func (m *AlertMutation) ClearMetas() {
m.clearedmetas = true
}
2021-01-14 15:27:45 +00:00
// MetasCleared returns if the "metas" edge to the Meta entity was cleared.
func (m *AlertMutation) MetasCleared() bool {
return m.clearedmetas
}
2021-01-14 15:27:45 +00:00
// RemoveMetaIDs removes the "metas" edge to the Meta entity by IDs.
func (m *AlertMutation) RemoveMetaIDs(ids ...int) {
if m.removedmetas == nil {
m.removedmetas = make(map[int]struct{})
}
for i := range ids {
m.removedmetas[ids[i]] = struct{}{}
}
}
2021-01-14 15:27:45 +00:00
// RemovedMetas returns the removed IDs of the "metas" edge to the Meta entity.
func (m *AlertMutation) RemovedMetasIDs() (ids []int) {
for id := range m.removedmetas {
ids = append(ids, id)
}
return
}
2021-01-14 15:27:45 +00:00
// MetasIDs returns the "metas" edge IDs in the mutation.
func (m *AlertMutation) MetasIDs() (ids []int) {
for id := range m.metas {
ids = append(ids, id)
}
return
}
2021-01-14 15:27:45 +00:00
// ResetMetas resets all changes to the "metas" edge.
func (m *AlertMutation) ResetMetas() {
m.metas = nil
m.clearedmetas = false
m.removedmetas = nil
}
// Op returns the operation name.
func (m *AlertMutation) Op() Op {
return m.op
}
// Type returns the node type of this mutation (Alert).
func (m *AlertMutation) Type() string {
return m.typ
}
2021-01-14 15:27:45 +00:00
// Fields returns all fields that were changed during this mutation. Note that in
// order to get all numeric fields that were incremented/decremented, call
// AddedFields().
func (m *AlertMutation) Fields() []string {
fields := make([]string, 0, 22)
if m.created_at != nil {
fields = append(fields, alert.FieldCreatedAt)
}
if m.updated_at != nil {
fields = append(fields, alert.FieldUpdatedAt)
}
if m.scenario != nil {
fields = append(fields, alert.FieldScenario)
}
if m.bucketId != nil {
fields = append(fields, alert.FieldBucketId)
}
if m.message != nil {
fields = append(fields, alert.FieldMessage)
}
if m.eventsCount != nil {
fields = append(fields, alert.FieldEventsCount)
}
if m.startedAt != nil {
fields = append(fields, alert.FieldStartedAt)
}
if m.stoppedAt != nil {
fields = append(fields, alert.FieldStoppedAt)
}
if m.sourceIp != nil {
fields = append(fields, alert.FieldSourceIp)
}
if m.sourceRange != nil {
fields = append(fields, alert.FieldSourceRange)
}
if m.sourceAsNumber != nil {
fields = append(fields, alert.FieldSourceAsNumber)
}
if m.sourceAsName != nil {
fields = append(fields, alert.FieldSourceAsName)
}
if m.sourceCountry != nil {
fields = append(fields, alert.FieldSourceCountry)
}
if m.sourceLatitude != nil {
fields = append(fields, alert.FieldSourceLatitude)
}
if m.sourceLongitude != nil {
fields = append(fields, alert.FieldSourceLongitude)
}
if m.sourceScope != nil {
fields = append(fields, alert.FieldSourceScope)
}
if m.sourceValue != nil {
fields = append(fields, alert.FieldSourceValue)
}
if m.capacity != nil {
fields = append(fields, alert.FieldCapacity)
}
if m.leakSpeed != nil {
fields = append(fields, alert.FieldLeakSpeed)
}
if m.scenarioVersion != nil {
fields = append(fields, alert.FieldScenarioVersion)
}
if m.scenarioHash != nil {
fields = append(fields, alert.FieldScenarioHash)
}
if m.simulated != nil {
fields = append(fields, alert.FieldSimulated)
}
return fields
}
2021-01-14 15:27:45 +00:00
// Field returns the value of a field with the given name. The second boolean
// return value indicates that this field was not set, or was not defined in the
// schema.
func (m *AlertMutation) Field(name string) (ent.Value, bool) {
switch name {
case alert.FieldCreatedAt:
return m.CreatedAt()
case alert.FieldUpdatedAt:
return m.UpdatedAt()
case alert.FieldScenario:
return m.Scenario()
case alert.FieldBucketId:
return m.BucketId()
case alert.FieldMessage:
return m.Message()
case alert.FieldEventsCount:
return m.EventsCount()
case alert.FieldStartedAt:
return m.StartedAt()
case alert.FieldStoppedAt:
return m.StoppedAt()
case alert.FieldSourceIp:
return m.SourceIp()
case alert.FieldSourceRange:
return m.SourceRange()
case alert.FieldSourceAsNumber:
return m.SourceAsNumber()
case alert.FieldSourceAsName:
return m.SourceAsName()
case alert.FieldSourceCountry:
return m.SourceCountry()
case alert.FieldSourceLatitude:
return m.SourceLatitude()
case alert.FieldSourceLongitude:
return m.SourceLongitude()
case alert.FieldSourceScope:
return m.SourceScope()
case alert.FieldSourceValue:
return m.SourceValue()
case alert.FieldCapacity:
return m.Capacity()
case alert.FieldLeakSpeed:
return m.LeakSpeed()
case alert.FieldScenarioVersion:
return m.ScenarioVersion()
case alert.FieldScenarioHash:
return m.ScenarioHash()
case alert.FieldSimulated:
return m.Simulated()
}
return nil, false
}
2021-01-14 15:27:45 +00:00
// OldField returns the old value of the field from the database. An error is
// returned if the mutation operation is not UpdateOne, or the query to the
// database failed.
func (m *AlertMutation) OldField(ctx context.Context, name string) (ent.Value, error) {
switch name {
case alert.FieldCreatedAt:
return m.OldCreatedAt(ctx)
case alert.FieldUpdatedAt:
return m.OldUpdatedAt(ctx)
case alert.FieldScenario:
return m.OldScenario(ctx)
case alert.FieldBucketId:
return m.OldBucketId(ctx)
case alert.FieldMessage:
return m.OldMessage(ctx)
case alert.FieldEventsCount:
return m.OldEventsCount(ctx)
case alert.FieldStartedAt:
return m.OldStartedAt(ctx)
case alert.FieldStoppedAt:
return m.OldStoppedAt(ctx)
case alert.FieldSourceIp:
return m.OldSourceIp(ctx)
case alert.FieldSourceRange:
return m.OldSourceRange(ctx)
case alert.FieldSourceAsNumber:
return m.OldSourceAsNumber(ctx)
case alert.FieldSourceAsName:
return m.OldSourceAsName(ctx)
case alert.FieldSourceCountry:
return m.OldSourceCountry(ctx)
case alert.FieldSourceLatitude:
return m.OldSourceLatitude(ctx)
case alert.FieldSourceLongitude:
return m.OldSourceLongitude(ctx)
case alert.FieldSourceScope:
return m.OldSourceScope(ctx)
case alert.FieldSourceValue:
return m.OldSourceValue(ctx)
case alert.FieldCapacity:
return m.OldCapacity(ctx)
case alert.FieldLeakSpeed:
return m.OldLeakSpeed(ctx)
case alert.FieldScenarioVersion:
return m.OldScenarioVersion(ctx)
case alert.FieldScenarioHash:
return m.OldScenarioHash(ctx)
case alert.FieldSimulated:
return m.OldSimulated(ctx)
}
return nil, fmt.Errorf("unknown Alert field %s", name)
}
2021-01-14 15:27:45 +00:00
// SetField sets the value of a field with the given name. It returns an error if
// the field is not defined in the schema, or if the type mismatched the field
// type.
func (m *AlertMutation) SetField(name string, value ent.Value) error {
switch name {
case alert.FieldCreatedAt:
v, ok := value.(time.Time)
if !ok {
return fmt.Errorf("unexpected type %T for field %s", value, name)
}
m.SetCreatedAt(v)
return nil
case alert.FieldUpdatedAt:
v, ok := value.(time.Time)
if !ok {
return fmt.Errorf("unexpected type %T for field %s", value, name)
}
m.SetUpdatedAt(v)
return nil
case alert.FieldScenario:
v, ok := value.(string)
if !ok {
return fmt.Errorf("unexpected type %T for field %s", value, name)
}
m.SetScenario(v)
return nil
case alert.FieldBucketId:
v, ok := value.(string)
if !ok {
return fmt.Errorf("unexpected type %T for field %s", value, name)
}
m.SetBucketId(v)
return nil
case alert.FieldMessage:
v, ok := value.(string)
if !ok {
return fmt.Errorf("unexpected type %T for field %s", value, name)
}
m.SetMessage(v)
return nil
case alert.FieldEventsCount:
v, ok := value.(int32)
if !ok {
return fmt.Errorf("unexpected type %T for field %s", value, name)
}
m.SetEventsCount(v)
return nil
case alert.FieldStartedAt:
v, ok := value.(time.Time)
if !ok {
return fmt.Errorf("unexpected type %T for field %s", value, name)
}
m.SetStartedAt(v)
return nil
case alert.FieldStoppedAt:
v, ok := value.(time.Time)
if !ok {
return fmt.Errorf("unexpected type %T for field %s", value, name)
}
m.SetStoppedAt(v)
return nil
case alert.FieldSourceIp:
v, ok := value.(string)
if !ok {
return fmt.Errorf("unexpected type %T for field %s", value, name)
}
m.SetSourceIp(v)
return nil
case alert.FieldSourceRange:
v, ok := value.(string)
if !ok {
return fmt.Errorf("unexpected type %T for field %s", value, name)
}
m.SetSourceRange(v)
return nil
case alert.FieldSourceAsNumber:
v, ok := value.(string)
if !ok {
return fmt.Errorf("unexpected type %T for field %s", value, name)
}
m.SetSourceAsNumber(v)
return nil
case alert.FieldSourceAsName:
v, ok := value.(string)
if !ok {
return fmt.Errorf("unexpected type %T for field %s", value, name)
}
m.SetSourceAsName(v)
return nil
case alert.FieldSourceCountry:
v, ok := value.(string)
if !ok {
return fmt.Errorf("unexpected type %T for field %s", value, name)
}
m.SetSourceCountry(v)
return nil
case alert.FieldSourceLatitude:
v, ok := value.(float32)
if !ok {
return fmt.Errorf("unexpected type %T for field %s", value, name)
}
m.SetSourceLatitude(v)
return nil
case alert.FieldSourceLongitude:
v, ok := value.(float32)
if !ok {
return fmt.Errorf("unexpected type %T for field %s", value, name)
}
m.SetSourceLongitude(v)
return nil
case alert.FieldSourceScope:
v, ok := value.(string)
if !ok {
return fmt.Errorf("unexpected type %T for field %s", value, name)
}
m.SetSourceScope(v)
return nil
case alert.FieldSourceValue:
v, ok := value.(string)
if !ok {
return fmt.Errorf("unexpected type %T for field %s", value, name)
}
m.SetSourceValue(v)
return nil
case alert.FieldCapacity:
v, ok := value.(int32)
if !ok {
return fmt.Errorf("unexpected type %T for field %s", value, name)
}
m.SetCapacity(v)
return nil
case alert.FieldLeakSpeed:
v, ok := value.(string)
if !ok {
return fmt.Errorf("unexpected type %T for field %s", value, name)
}
m.SetLeakSpeed(v)
return nil
case alert.FieldScenarioVersion:
v, ok := value.(string)
if !ok {
return fmt.Errorf("unexpected type %T for field %s", value, name)
}
m.SetScenarioVersion(v)
return nil
case alert.FieldScenarioHash:
v, ok := value.(string)
if !ok {
return fmt.Errorf("unexpected type %T for field %s", value, name)
}
m.SetScenarioHash(v)
return nil
case alert.FieldSimulated:
v, ok := value.(bool)
if !ok {
return fmt.Errorf("unexpected type %T for field %s", value, name)
}
m.SetSimulated(v)
return nil
}
return fmt.Errorf("unknown Alert field %s", name)
}
2021-01-14 15:27:45 +00:00
// AddedFields returns all numeric fields that were incremented/decremented during
// this mutation.
func (m *AlertMutation) AddedFields() []string {
var fields []string
if m.addeventsCount != nil {
fields = append(fields, alert.FieldEventsCount)
}
if m.addsourceLatitude != nil {
fields = append(fields, alert.FieldSourceLatitude)
}
if m.addsourceLongitude != nil {
fields = append(fields, alert.FieldSourceLongitude)
}
if m.addcapacity != nil {
fields = append(fields, alert.FieldCapacity)
}
return fields
}
2021-01-14 15:27:45 +00:00
// AddedField returns the numeric value that was incremented/decremented on a field
// with the given name. The second boolean return value indicates that this field
// was not set, or was not defined in the schema.
func (m *AlertMutation) AddedField(name string) (ent.Value, bool) {
switch name {
case alert.FieldEventsCount:
return m.AddedEventsCount()
case alert.FieldSourceLatitude:
return m.AddedSourceLatitude()
case alert.FieldSourceLongitude:
return m.AddedSourceLongitude()
case alert.FieldCapacity:
return m.AddedCapacity()
}
return nil, false
}
2021-01-14 15:27:45 +00:00
// AddField adds the value to the field with the given name. It returns an error if
// the field is not defined in the schema, or if the type mismatched the field
// type.
func (m *AlertMutation) AddField(name string, value ent.Value) error {
switch name {
case alert.FieldEventsCount:
v, ok := value.(int32)
if !ok {
return fmt.Errorf("unexpected type %T for field %s", value, name)
}
m.AddEventsCount(v)
return nil
case alert.FieldSourceLatitude:
v, ok := value.(float32)
if !ok {
return fmt.Errorf("unexpected type %T for field %s", value, name)
}
m.AddSourceLatitude(v)
return nil
case alert.FieldSourceLongitude:
v, ok := value.(float32)
if !ok {
return fmt.Errorf("unexpected type %T for field %s", value, name)
}
m.AddSourceLongitude(v)
return nil
case alert.FieldCapacity:
v, ok := value.(int32)
if !ok {
return fmt.Errorf("unexpected type %T for field %s", value, name)
}
m.AddCapacity(v)
return nil
}
return fmt.Errorf("unknown Alert numeric field %s", name)
}
2021-01-14 15:27:45 +00:00
// ClearedFields returns all nullable fields that were cleared during this
// mutation.
func (m *AlertMutation) ClearedFields() []string {
var fields []string
if m.FieldCleared(alert.FieldBucketId) {
fields = append(fields, alert.FieldBucketId)
}
if m.FieldCleared(alert.FieldMessage) {
fields = append(fields, alert.FieldMessage)
}
if m.FieldCleared(alert.FieldEventsCount) {
fields = append(fields, alert.FieldEventsCount)
}
if m.FieldCleared(alert.FieldStartedAt) {
fields = append(fields, alert.FieldStartedAt)
}
if m.FieldCleared(alert.FieldStoppedAt) {
fields = append(fields, alert.FieldStoppedAt)
}
if m.FieldCleared(alert.FieldSourceIp) {
fields = append(fields, alert.FieldSourceIp)
}
if m.FieldCleared(alert.FieldSourceRange) {
fields = append(fields, alert.FieldSourceRange)
}
if m.FieldCleared(alert.FieldSourceAsNumber) {
fields = append(fields, alert.FieldSourceAsNumber)
}
if m.FieldCleared(alert.FieldSourceAsName) {
fields = append(fields, alert.FieldSourceAsName)
}
if m.FieldCleared(alert.FieldSourceCountry) {
fields = append(fields, alert.FieldSourceCountry)
}
if m.FieldCleared(alert.FieldSourceLatitude) {
fields = append(fields, alert.FieldSourceLatitude)
}
if m.FieldCleared(alert.FieldSourceLongitude) {
fields = append(fields, alert.FieldSourceLongitude)
}
if m.FieldCleared(alert.FieldSourceScope) {
fields = append(fields, alert.FieldSourceScope)
}
if m.FieldCleared(alert.FieldSourceValue) {
fields = append(fields, alert.FieldSourceValue)
}
if m.FieldCleared(alert.FieldCapacity) {
fields = append(fields, alert.FieldCapacity)
}
if m.FieldCleared(alert.FieldLeakSpeed) {
fields = append(fields, alert.FieldLeakSpeed)
}
if m.FieldCleared(alert.FieldScenarioVersion) {
fields = append(fields, alert.FieldScenarioVersion)
}
if m.FieldCleared(alert.FieldScenarioHash) {
fields = append(fields, alert.FieldScenarioHash)
}
return fields
}
2021-01-14 15:27:45 +00:00
// FieldCleared returns a boolean indicating if a field with the given name was
// cleared in this mutation.
func (m *AlertMutation) FieldCleared(name string) bool {
_, ok := m.clearedFields[name]
return ok
}
2021-01-14 15:27:45 +00:00
// ClearField clears the value of the field with the given name. It returns an
// error if the field is not defined in the schema.
func (m *AlertMutation) ClearField(name string) error {
switch name {
case alert.FieldBucketId:
m.ClearBucketId()
return nil
case alert.FieldMessage:
m.ClearMessage()
return nil
case alert.FieldEventsCount:
m.ClearEventsCount()
return nil
case alert.FieldStartedAt:
m.ClearStartedAt()
return nil
case alert.FieldStoppedAt:
m.ClearStoppedAt()
return nil
case alert.FieldSourceIp:
m.ClearSourceIp()
return nil
case alert.FieldSourceRange:
m.ClearSourceRange()
return nil
case alert.FieldSourceAsNumber:
m.ClearSourceAsNumber()
return nil
case alert.FieldSourceAsName:
m.ClearSourceAsName()
return nil
case alert.FieldSourceCountry:
m.ClearSourceCountry()
return nil
case alert.FieldSourceLatitude:
m.ClearSourceLatitude()
return nil
case alert.FieldSourceLongitude:
m.ClearSourceLongitude()
return nil
case alert.FieldSourceScope:
m.ClearSourceScope()
return nil
case alert.FieldSourceValue:
m.ClearSourceValue()
return nil
case alert.FieldCapacity:
m.ClearCapacity()
return nil
case alert.FieldLeakSpeed:
m.ClearLeakSpeed()
return nil
case alert.FieldScenarioVersion:
m.ClearScenarioVersion()
return nil
case alert.FieldScenarioHash:
m.ClearScenarioHash()
return nil
}
return fmt.Errorf("unknown Alert nullable field %s", name)
}
2021-01-14 15:27:45 +00:00
// ResetField resets all changes in the mutation for the field with the given name.
// It returns an error if the field is not defined in the schema.
func (m *AlertMutation) ResetField(name string) error {
switch name {
case alert.FieldCreatedAt:
m.ResetCreatedAt()
return nil
case alert.FieldUpdatedAt:
m.ResetUpdatedAt()
return nil
case alert.FieldScenario:
m.ResetScenario()
return nil
case alert.FieldBucketId:
m.ResetBucketId()
return nil
case alert.FieldMessage:
m.ResetMessage()
return nil
case alert.FieldEventsCount:
m.ResetEventsCount()
return nil
case alert.FieldStartedAt:
m.ResetStartedAt()
return nil
case alert.FieldStoppedAt:
m.ResetStoppedAt()
return nil
case alert.FieldSourceIp:
m.ResetSourceIp()
return nil
case alert.FieldSourceRange:
m.ResetSourceRange()
return nil
case alert.FieldSourceAsNumber:
m.ResetSourceAsNumber()
return nil
case alert.FieldSourceAsName:
m.ResetSourceAsName()
return nil
case alert.FieldSourceCountry:
m.ResetSourceCountry()
return nil
case alert.FieldSourceLatitude:
m.ResetSourceLatitude()
return nil
case alert.FieldSourceLongitude:
m.ResetSourceLongitude()
return nil
case alert.FieldSourceScope:
m.ResetSourceScope()
return nil
case alert.FieldSourceValue:
m.ResetSourceValue()
return nil
case alert.FieldCapacity:
m.ResetCapacity()
return nil
case alert.FieldLeakSpeed:
m.ResetLeakSpeed()
return nil
case alert.FieldScenarioVersion:
m.ResetScenarioVersion()
return nil
case alert.FieldScenarioHash:
m.ResetScenarioHash()
return nil
case alert.FieldSimulated:
m.ResetSimulated()
return nil
}
return fmt.Errorf("unknown Alert field %s", name)
}
2021-01-14 15:27:45 +00:00
// AddedEdges returns all edge names that were set/added in this mutation.
func (m *AlertMutation) AddedEdges() []string {
edges := make([]string, 0, 4)
if m.owner != nil {
edges = append(edges, alert.EdgeOwner)
}
if m.decisions != nil {
edges = append(edges, alert.EdgeDecisions)
}
if m.events != nil {
edges = append(edges, alert.EdgeEvents)
}
if m.metas != nil {
edges = append(edges, alert.EdgeMetas)
}
return edges
}
2021-01-14 15:27:45 +00:00
// AddedIDs returns all IDs (to other nodes) that were added for the given edge
// name in this mutation.
func (m *AlertMutation) AddedIDs(name string) []ent.Value {
switch name {
case alert.EdgeOwner:
if id := m.owner; id != nil {
return []ent.Value{*id}
}
case alert.EdgeDecisions:
ids := make([]ent.Value, 0, len(m.decisions))
for id := range m.decisions {
ids = append(ids, id)
}
return ids
case alert.EdgeEvents:
ids := make([]ent.Value, 0, len(m.events))
for id := range m.events {
ids = append(ids, id)
}
return ids
case alert.EdgeMetas:
ids := make([]ent.Value, 0, len(m.metas))
for id := range m.metas {
ids = append(ids, id)
}
return ids
}
return nil
}
2021-01-14 15:27:45 +00:00
// RemovedEdges returns all edge names that were removed in this mutation.
func (m *AlertMutation) RemovedEdges() []string {
edges := make([]string, 0, 4)
if m.removeddecisions != nil {
edges = append(edges, alert.EdgeDecisions)
}
if m.removedevents != nil {
edges = append(edges, alert.EdgeEvents)
}
if m.removedmetas != nil {
edges = append(edges, alert.EdgeMetas)
}
return edges
}
2021-01-14 15:27:45 +00:00
// RemovedIDs returns all IDs (to other nodes) that were removed for the edge with
// the given name in this mutation.
func (m *AlertMutation) RemovedIDs(name string) []ent.Value {
switch name {
case alert.EdgeDecisions:
ids := make([]ent.Value, 0, len(m.removeddecisions))
for id := range m.removeddecisions {
ids = append(ids, id)
}
return ids
case alert.EdgeEvents:
ids := make([]ent.Value, 0, len(m.removedevents))
for id := range m.removedevents {
ids = append(ids, id)
}
return ids
case alert.EdgeMetas:
ids := make([]ent.Value, 0, len(m.removedmetas))
for id := range m.removedmetas {
ids = append(ids, id)
}
return ids
}
return nil
}
2021-01-14 15:27:45 +00:00
// ClearedEdges returns all edge names that were cleared in this mutation.
func (m *AlertMutation) ClearedEdges() []string {
edges := make([]string, 0, 4)
if m.clearedowner {
edges = append(edges, alert.EdgeOwner)
}
if m.cleareddecisions {
edges = append(edges, alert.EdgeDecisions)
}
if m.clearedevents {
edges = append(edges, alert.EdgeEvents)
}
if m.clearedmetas {
edges = append(edges, alert.EdgeMetas)
}
return edges
}
2021-01-14 15:27:45 +00:00
// EdgeCleared returns a boolean which indicates if the edge with the given name
// was cleared in this mutation.
func (m *AlertMutation) EdgeCleared(name string) bool {
switch name {
case alert.EdgeOwner:
return m.clearedowner
case alert.EdgeDecisions:
return m.cleareddecisions
case alert.EdgeEvents:
return m.clearedevents
case alert.EdgeMetas:
return m.clearedmetas
}
return false
}
2021-01-14 15:27:45 +00:00
// ClearEdge clears the value of the edge with the given name. It returns an error
// if that edge is not defined in the schema.
func (m *AlertMutation) ClearEdge(name string) error {
switch name {
case alert.EdgeOwner:
m.ClearOwner()
return nil
}
return fmt.Errorf("unknown Alert unique edge %s", name)
}
2021-01-14 15:27:45 +00:00
// ResetEdge resets all changes to the edge with the given name in this mutation.
// It returns an error if the edge is not defined in the schema.
func (m *AlertMutation) ResetEdge(name string) error {
switch name {
case alert.EdgeOwner:
m.ResetOwner()
return nil
case alert.EdgeDecisions:
m.ResetDecisions()
return nil
case alert.EdgeEvents:
m.ResetEvents()
return nil
case alert.EdgeMetas:
m.ResetMetas()
return nil
}
return fmt.Errorf("unknown Alert edge %s", name)
}
2021-01-14 15:27:45 +00:00
// BouncerMutation represents an operation that mutates the Bouncer nodes in the graph.
type BouncerMutation struct {
config
op Op
typ string
id *int
created_at *time.Time
updated_at *time.Time
name *string
api_key *string
revoked *bool
ip_address *string
_type *string
version *string
until *time.Time
last_pull *time.Time
clearedFields map[string]struct{}
done bool
oldValue func(context.Context) (*Bouncer, error)
2021-01-14 15:27:45 +00:00
predicates []predicate.Bouncer
}
var _ ent.Mutation = (*BouncerMutation)(nil)
2021-01-14 15:27:45 +00:00
// bouncerOption allows management of the mutation configuration using functional options.
type bouncerOption func(*BouncerMutation)
2021-01-14 15:27:45 +00:00
// newBouncerMutation creates new mutation for the Bouncer entity.
func newBouncerMutation(c config, op Op, opts ...bouncerOption) *BouncerMutation {
m := &BouncerMutation{
config: c,
op: op,
typ: TypeBouncer,
clearedFields: make(map[string]struct{}),
}
for _, opt := range opts {
opt(m)
}
return m
}
2021-01-14 15:27:45 +00:00
// withBouncerID sets the ID field of the mutation.
func withBouncerID(id int) bouncerOption {
return func(m *BouncerMutation) {
var (
err error
once sync.Once
value *Bouncer
)
m.oldValue = func(ctx context.Context) (*Bouncer, error) {
once.Do(func() {
if m.done {
err = fmt.Errorf("querying old values post mutation is not allowed")
} else {
value, err = m.Client().Bouncer.Get(ctx, id)
}
})
return value, err
}
m.id = &id
}
}
// withBouncer sets the old Bouncer of the mutation.
func withBouncer(node *Bouncer) bouncerOption {
return func(m *BouncerMutation) {
m.oldValue = func(context.Context) (*Bouncer, error) {
return node, nil
}
m.id = &node.ID
}
}
// Client returns a new `ent.Client` from the mutation. If the mutation was
// executed in a transaction (ent.Tx), a transactional client is returned.
func (m BouncerMutation) Client() *Client {
client := &Client{config: m.config}
client.init()
return client
}
// Tx returns an `ent.Tx` for mutations that were executed in transactions;
// it returns an error otherwise.
func (m BouncerMutation) Tx() (*Tx, error) {
if _, ok := m.driver.(*txDriver); !ok {
return nil, fmt.Errorf("ent: mutation is not running in a transaction")
}
tx := &Tx{config: m.config}
tx.init()
return tx, nil
}
2021-01-14 15:27:45 +00:00
// ID returns the ID value in the mutation. Note that the ID
// is only available if it was provided to the builder.
func (m *BouncerMutation) ID() (id int, exists bool) {
if m.id == nil {
return
}
return *m.id, true
}
2021-01-14 15:27:45 +00:00
// SetCreatedAt sets the "created_at" field.
func (m *BouncerMutation) SetCreatedAt(t time.Time) {
m.created_at = &t
}
2021-01-14 15:27:45 +00:00
// CreatedAt returns the value of the "created_at" field in the mutation.
func (m *BouncerMutation) CreatedAt() (r time.Time, exists bool) {
v := m.created_at
if v == nil {
return
}
return *v, true
}
2021-01-14 15:27:45 +00:00
// OldCreatedAt returns the old "created_at" field's value of the Bouncer entity.
// If the Bouncer object wasn't provided to the builder, the object is fetched from the database.
// An error is returned if the mutation operation is not UpdateOne, or the database query fails.
func (m *BouncerMutation) OldCreatedAt(ctx context.Context) (v time.Time, err error) {
if !m.op.Is(OpUpdateOne) {
2021-01-14 15:27:45 +00:00
return v, fmt.Errorf("OldCreatedAt is only allowed on UpdateOne operations")
}
if m.id == nil || m.oldValue == nil {
return v, fmt.Errorf("OldCreatedAt requires an ID field in the mutation")
}
oldValue, err := m.oldValue(ctx)
if err != nil {
return v, fmt.Errorf("querying old value for OldCreatedAt: %w", err)
}
return oldValue.CreatedAt, nil
}
2021-01-14 15:27:45 +00:00
// ResetCreatedAt resets all changes to the "created_at" field.
func (m *BouncerMutation) ResetCreatedAt() {
m.created_at = nil
}
2021-01-14 15:27:45 +00:00
// SetUpdatedAt sets the "updated_at" field.
func (m *BouncerMutation) SetUpdatedAt(t time.Time) {
m.updated_at = &t
}
2021-01-14 15:27:45 +00:00
// UpdatedAt returns the value of the "updated_at" field in the mutation.
func (m *BouncerMutation) UpdatedAt() (r time.Time, exists bool) {
v := m.updated_at
if v == nil {
return
}
return *v, true
}
2021-01-14 15:27:45 +00:00
// OldUpdatedAt returns the old "updated_at" field's value of the Bouncer entity.
// If the Bouncer object wasn't provided to the builder, the object is fetched from the database.
// An error is returned if the mutation operation is not UpdateOne, or the database query fails.
func (m *BouncerMutation) OldUpdatedAt(ctx context.Context) (v time.Time, err error) {
if !m.op.Is(OpUpdateOne) {
2021-01-14 15:27:45 +00:00
return v, fmt.Errorf("OldUpdatedAt is only allowed on UpdateOne operations")
}
if m.id == nil || m.oldValue == nil {
return v, fmt.Errorf("OldUpdatedAt requires an ID field in the mutation")
}
oldValue, err := m.oldValue(ctx)
if err != nil {
return v, fmt.Errorf("querying old value for OldUpdatedAt: %w", err)
}
return oldValue.UpdatedAt, nil
}
2021-01-14 15:27:45 +00:00
// ResetUpdatedAt resets all changes to the "updated_at" field.
func (m *BouncerMutation) ResetUpdatedAt() {
m.updated_at = nil
}
2021-01-14 15:27:45 +00:00
// SetName sets the "name" field.
func (m *BouncerMutation) SetName(s string) {
m.name = &s
}
2021-01-14 15:27:45 +00:00
// Name returns the value of the "name" field in the mutation.
func (m *BouncerMutation) Name() (r string, exists bool) {
v := m.name
if v == nil {
return
}
return *v, true
}
2021-01-14 15:27:45 +00:00
// OldName returns the old "name" field's value of the Bouncer entity.
// If the Bouncer object wasn't provided to the builder, the object is fetched from the database.
// An error is returned if the mutation operation is not UpdateOne, or the database query fails.
func (m *BouncerMutation) OldName(ctx context.Context) (v string, err error) {
if !m.op.Is(OpUpdateOne) {
2021-01-14 15:27:45 +00:00
return v, fmt.Errorf("OldName is only allowed on UpdateOne operations")
}
if m.id == nil || m.oldValue == nil {
return v, fmt.Errorf("OldName requires an ID field in the mutation")
}
oldValue, err := m.oldValue(ctx)
if err != nil {
return v, fmt.Errorf("querying old value for OldName: %w", err)
}
return oldValue.Name, nil
}
2021-01-14 15:27:45 +00:00
// ResetName resets all changes to the "name" field.
func (m *BouncerMutation) ResetName() {
m.name = nil
}
2021-01-14 15:27:45 +00:00
// SetAPIKey sets the "api_key" field.
func (m *BouncerMutation) SetAPIKey(s string) {
m.api_key = &s
}
2021-01-14 15:27:45 +00:00
// APIKey returns the value of the "api_key" field in the mutation.
func (m *BouncerMutation) APIKey() (r string, exists bool) {
v := m.api_key
if v == nil {
return
}
return *v, true
}
2021-01-14 15:27:45 +00:00
// OldAPIKey returns the old "api_key" field's value of the Bouncer entity.
// If the Bouncer object wasn't provided to the builder, the object is fetched from the database.
// An error is returned if the mutation operation is not UpdateOne, or the database query fails.
func (m *BouncerMutation) OldAPIKey(ctx context.Context) (v string, err error) {
if !m.op.Is(OpUpdateOne) {
2021-01-14 15:27:45 +00:00
return v, fmt.Errorf("OldAPIKey is only allowed on UpdateOne operations")
}
if m.id == nil || m.oldValue == nil {
return v, fmt.Errorf("OldAPIKey requires an ID field in the mutation")
}
oldValue, err := m.oldValue(ctx)
if err != nil {
return v, fmt.Errorf("querying old value for OldAPIKey: %w", err)
}
return oldValue.APIKey, nil
}
2021-01-14 15:27:45 +00:00
// ResetAPIKey resets all changes to the "api_key" field.
func (m *BouncerMutation) ResetAPIKey() {
m.api_key = nil
}
2021-01-14 15:27:45 +00:00
// SetRevoked sets the "revoked" field.
func (m *BouncerMutation) SetRevoked(b bool) {
m.revoked = &b
}
2021-01-14 15:27:45 +00:00
// Revoked returns the value of the "revoked" field in the mutation.
func (m *BouncerMutation) Revoked() (r bool, exists bool) {
v := m.revoked
if v == nil {
return
}
return *v, true
}
2021-01-14 15:27:45 +00:00
// OldRevoked returns the old "revoked" field's value of the Bouncer entity.
// If the Bouncer object wasn't provided to the builder, the object is fetched from the database.
// An error is returned if the mutation operation is not UpdateOne, or the database query fails.
func (m *BouncerMutation) OldRevoked(ctx context.Context) (v bool, err error) {
if !m.op.Is(OpUpdateOne) {
2021-01-14 15:27:45 +00:00
return v, fmt.Errorf("OldRevoked is only allowed on UpdateOne operations")
}
if m.id == nil || m.oldValue == nil {
return v, fmt.Errorf("OldRevoked requires an ID field in the mutation")
}
oldValue, err := m.oldValue(ctx)
if err != nil {
return v, fmt.Errorf("querying old value for OldRevoked: %w", err)
}
return oldValue.Revoked, nil
}
2021-01-14 15:27:45 +00:00
// ResetRevoked resets all changes to the "revoked" field.
func (m *BouncerMutation) ResetRevoked() {
m.revoked = nil
}
2021-01-14 15:27:45 +00:00
// SetIPAddress sets the "ip_address" field.
func (m *BouncerMutation) SetIPAddress(s string) {
m.ip_address = &s
}
2021-01-14 15:27:45 +00:00
// IPAddress returns the value of the "ip_address" field in the mutation.
func (m *BouncerMutation) IPAddress() (r string, exists bool) {
v := m.ip_address
if v == nil {
return
}
return *v, true
}
2021-01-14 15:27:45 +00:00
// OldIPAddress returns the old "ip_address" field's value of the Bouncer entity.
// If the Bouncer object wasn't provided to the builder, the object is fetched from the database.
// An error is returned if the mutation operation is not UpdateOne, or the database query fails.
func (m *BouncerMutation) OldIPAddress(ctx context.Context) (v string, err error) {
if !m.op.Is(OpUpdateOne) {
2021-01-14 15:27:45 +00:00
return v, fmt.Errorf("OldIPAddress is only allowed on UpdateOne operations")
}
if m.id == nil || m.oldValue == nil {
return v, fmt.Errorf("OldIPAddress requires an ID field in the mutation")
}
oldValue, err := m.oldValue(ctx)
if err != nil {
return v, fmt.Errorf("querying old value for OldIPAddress: %w", err)
}
return oldValue.IPAddress, nil
}
2021-01-14 15:27:45 +00:00
// ClearIPAddress clears the value of the "ip_address" field.
func (m *BouncerMutation) ClearIPAddress() {
m.ip_address = nil
m.clearedFields[bouncer.FieldIPAddress] = struct{}{}
}
2021-01-14 15:27:45 +00:00
// IPAddressCleared returns if the "ip_address" field was cleared in this mutation.
func (m *BouncerMutation) IPAddressCleared() bool {
_, ok := m.clearedFields[bouncer.FieldIPAddress]
return ok
}
2021-01-14 15:27:45 +00:00
// ResetIPAddress resets all changes to the "ip_address" field.
func (m *BouncerMutation) ResetIPAddress() {
m.ip_address = nil
delete(m.clearedFields, bouncer.FieldIPAddress)
}
2021-01-14 15:27:45 +00:00
// SetType sets the "type" field.
func (m *BouncerMutation) SetType(s string) {
m._type = &s
}
2021-01-14 15:27:45 +00:00
// GetType returns the value of the "type" field in the mutation.
func (m *BouncerMutation) GetType() (r string, exists bool) {
v := m._type
if v == nil {
return
}
return *v, true
}
2021-01-14 15:27:45 +00:00
// OldType returns the old "type" field's value of the Bouncer entity.
// If the Bouncer object wasn't provided to the builder, the object is fetched from the database.
// An error is returned if the mutation operation is not UpdateOne, or the database query fails.
func (m *BouncerMutation) OldType(ctx context.Context) (v string, err error) {
if !m.op.Is(OpUpdateOne) {
2021-01-14 15:27:45 +00:00
return v, fmt.Errorf("OldType is only allowed on UpdateOne operations")
}
if m.id == nil || m.oldValue == nil {
return v, fmt.Errorf("OldType requires an ID field in the mutation")
}
oldValue, err := m.oldValue(ctx)
if err != nil {
return v, fmt.Errorf("querying old value for OldType: %w", err)
}
return oldValue.Type, nil
}
2021-01-14 15:27:45 +00:00
// ClearType clears the value of the "type" field.
func (m *BouncerMutation) ClearType() {
m._type = nil
m.clearedFields[bouncer.FieldType] = struct{}{}
}
2021-01-14 15:27:45 +00:00
// TypeCleared returns if the "type" field was cleared in this mutation.
func (m *BouncerMutation) TypeCleared() bool {
_, ok := m.clearedFields[bouncer.FieldType]
return ok
}
2021-01-14 15:27:45 +00:00
// ResetType resets all changes to the "type" field.
func (m *BouncerMutation) ResetType() {
m._type = nil
delete(m.clearedFields, bouncer.FieldType)
}
2021-01-14 15:27:45 +00:00
// SetVersion sets the "version" field.
func (m *BouncerMutation) SetVersion(s string) {
m.version = &s
}
2021-01-14 15:27:45 +00:00
// Version returns the value of the "version" field in the mutation.
func (m *BouncerMutation) Version() (r string, exists bool) {
v := m.version
if v == nil {
return
}
return *v, true
}
2021-01-14 15:27:45 +00:00
// OldVersion returns the old "version" field's value of the Bouncer entity.
// If the Bouncer object wasn't provided to the builder, the object is fetched from the database.
// An error is returned if the mutation operation is not UpdateOne, or the database query fails.
func (m *BouncerMutation) OldVersion(ctx context.Context) (v string, err error) {
if !m.op.Is(OpUpdateOne) {
2021-01-14 15:27:45 +00:00
return v, fmt.Errorf("OldVersion is only allowed on UpdateOne operations")
}
if m.id == nil || m.oldValue == nil {
return v, fmt.Errorf("OldVersion requires an ID field in the mutation")
}
oldValue, err := m.oldValue(ctx)
if err != nil {
return v, fmt.Errorf("querying old value for OldVersion: %w", err)
}
return oldValue.Version, nil
}
2021-01-14 15:27:45 +00:00
// ClearVersion clears the value of the "version" field.
func (m *BouncerMutation) ClearVersion() {
m.version = nil
m.clearedFields[bouncer.FieldVersion] = struct{}{}
}
2021-01-14 15:27:45 +00:00
// VersionCleared returns if the "version" field was cleared in this mutation.
func (m *BouncerMutation) VersionCleared() bool {
_, ok := m.clearedFields[bouncer.FieldVersion]
return ok
}
2021-01-14 15:27:45 +00:00
// ResetVersion resets all changes to the "version" field.
func (m *BouncerMutation) ResetVersion() {
m.version = nil
delete(m.clearedFields, bouncer.FieldVersion)
}
2021-01-14 15:27:45 +00:00
// SetUntil sets the "until" field.
func (m *BouncerMutation) SetUntil(t time.Time) {
m.until = &t
}
2021-01-14 15:27:45 +00:00
// Until returns the value of the "until" field in the mutation.
func (m *BouncerMutation) Until() (r time.Time, exists bool) {
v := m.until
if v == nil {
return
}
return *v, true
}
2021-01-14 15:27:45 +00:00
// OldUntil returns the old "until" field's value of the Bouncer entity.
// If the Bouncer object wasn't provided to the builder, the object is fetched from the database.
// An error is returned if the mutation operation is not UpdateOne, or the database query fails.
func (m *BouncerMutation) OldUntil(ctx context.Context) (v time.Time, err error) {
if !m.op.Is(OpUpdateOne) {
2021-01-14 15:27:45 +00:00
return v, fmt.Errorf("OldUntil is only allowed on UpdateOne operations")
}
if m.id == nil || m.oldValue == nil {
return v, fmt.Errorf("OldUntil requires an ID field in the mutation")
}
oldValue, err := m.oldValue(ctx)
if err != nil {
return v, fmt.Errorf("querying old value for OldUntil: %w", err)
}
return oldValue.Until, nil
}
2021-01-14 15:27:45 +00:00
// ClearUntil clears the value of the "until" field.
func (m *BouncerMutation) ClearUntil() {
m.until = nil
m.clearedFields[bouncer.FieldUntil] = struct{}{}
}
2021-01-14 15:27:45 +00:00
// UntilCleared returns if the "until" field was cleared in this mutation.
func (m *BouncerMutation) UntilCleared() bool {
_, ok := m.clearedFields[bouncer.FieldUntil]
return ok
}
2021-01-14 15:27:45 +00:00
// ResetUntil resets all changes to the "until" field.
func (m *BouncerMutation) ResetUntil() {
m.until = nil
delete(m.clearedFields, bouncer.FieldUntil)
}
2021-01-14 15:27:45 +00:00
// SetLastPull sets the "last_pull" field.
func (m *BouncerMutation) SetLastPull(t time.Time) {
m.last_pull = &t
}
2021-01-14 15:27:45 +00:00
// LastPull returns the value of the "last_pull" field in the mutation.
func (m *BouncerMutation) LastPull() (r time.Time, exists bool) {
v := m.last_pull
if v == nil {
return
}
return *v, true
}
2021-01-14 15:27:45 +00:00
// OldLastPull returns the old "last_pull" field's value of the Bouncer entity.
// If the Bouncer object wasn't provided to the builder, the object is fetched from the database.
// An error is returned if the mutation operation is not UpdateOne, or the database query fails.
func (m *BouncerMutation) OldLastPull(ctx context.Context) (v time.Time, err error) {
if !m.op.Is(OpUpdateOne) {
2021-01-14 15:27:45 +00:00
return v, fmt.Errorf("OldLastPull is only allowed on UpdateOne operations")
}
if m.id == nil || m.oldValue == nil {
return v, fmt.Errorf("OldLastPull requires an ID field in the mutation")
}
oldValue, err := m.oldValue(ctx)
if err != nil {
return v, fmt.Errorf("querying old value for OldLastPull: %w", err)
}
return oldValue.LastPull, nil
}
2021-01-14 15:27:45 +00:00
// ResetLastPull resets all changes to the "last_pull" field.
func (m *BouncerMutation) ResetLastPull() {
m.last_pull = nil
}
// Op returns the operation name.
func (m *BouncerMutation) Op() Op {
return m.op
}
// Type returns the node type of this mutation (Bouncer).
func (m *BouncerMutation) Type() string {
return m.typ
}
2021-01-14 15:27:45 +00:00
// Fields returns all fields that were changed during this mutation. Note that in
// order to get all numeric fields that were incremented/decremented, call
// AddedFields().
func (m *BouncerMutation) Fields() []string {
fields := make([]string, 0, 10)
if m.created_at != nil {
fields = append(fields, bouncer.FieldCreatedAt)
}
if m.updated_at != nil {
fields = append(fields, bouncer.FieldUpdatedAt)
}
if m.name != nil {
fields = append(fields, bouncer.FieldName)
}
if m.api_key != nil {
fields = append(fields, bouncer.FieldAPIKey)
}
if m.revoked != nil {
fields = append(fields, bouncer.FieldRevoked)
}
if m.ip_address != nil {
fields = append(fields, bouncer.FieldIPAddress)
}
if m._type != nil {
fields = append(fields, bouncer.FieldType)
}
if m.version != nil {
fields = append(fields, bouncer.FieldVersion)
}
if m.until != nil {
fields = append(fields, bouncer.FieldUntil)
}
if m.last_pull != nil {
fields = append(fields, bouncer.FieldLastPull)
}
return fields
}
2021-01-14 15:27:45 +00:00
// Field returns the value of a field with the given name. The second boolean
// return value indicates that this field was not set, or was not defined in the
// schema.
func (m *BouncerMutation) Field(name string) (ent.Value, bool) {
switch name {
case bouncer.FieldCreatedAt:
return m.CreatedAt()
case bouncer.FieldUpdatedAt:
return m.UpdatedAt()
case bouncer.FieldName:
return m.Name()
case bouncer.FieldAPIKey:
return m.APIKey()
case bouncer.FieldRevoked:
return m.Revoked()
case bouncer.FieldIPAddress:
return m.IPAddress()
case bouncer.FieldType:
return m.GetType()
case bouncer.FieldVersion:
return m.Version()
case bouncer.FieldUntil:
return m.Until()
case bouncer.FieldLastPull:
return m.LastPull()
}
return nil, false
}
2021-01-14 15:27:45 +00:00
// OldField returns the old value of the field from the database. An error is
// returned if the mutation operation is not UpdateOne, or the query to the
// database failed.
func (m *BouncerMutation) OldField(ctx context.Context, name string) (ent.Value, error) {
switch name {
case bouncer.FieldCreatedAt:
return m.OldCreatedAt(ctx)
case bouncer.FieldUpdatedAt:
return m.OldUpdatedAt(ctx)
case bouncer.FieldName:
return m.OldName(ctx)
case bouncer.FieldAPIKey:
return m.OldAPIKey(ctx)
case bouncer.FieldRevoked:
return m.OldRevoked(ctx)
case bouncer.FieldIPAddress:
return m.OldIPAddress(ctx)
case bouncer.FieldType:
return m.OldType(ctx)
case bouncer.FieldVersion:
return m.OldVersion(ctx)
case bouncer.FieldUntil:
return m.OldUntil(ctx)
case bouncer.FieldLastPull:
return m.OldLastPull(ctx)
}
return nil, fmt.Errorf("unknown Bouncer field %s", name)
}
2021-01-14 15:27:45 +00:00
// SetField sets the value of a field with the given name. It returns an error if
// the field is not defined in the schema, or if the type mismatched the field
// type.
func (m *BouncerMutation) SetField(name string, value ent.Value) error {
switch name {
case bouncer.FieldCreatedAt:
v, ok := value.(time.Time)
if !ok {
return fmt.Errorf("unexpected type %T for field %s", value, name)
}
m.SetCreatedAt(v)
return nil
case bouncer.FieldUpdatedAt:
v, ok := value.(time.Time)
if !ok {
return fmt.Errorf("unexpected type %T for field %s", value, name)
}
m.SetUpdatedAt(v)
return nil
case bouncer.FieldName:
v, ok := value.(string)
if !ok {
return fmt.Errorf("unexpected type %T for field %s", value, name)
}
m.SetName(v)
return nil
case bouncer.FieldAPIKey:
v, ok := value.(string)
if !ok {
return fmt.Errorf("unexpected type %T for field %s", value, name)
}
m.SetAPIKey(v)
return nil
case bouncer.FieldRevoked:
v, ok := value.(bool)
if !ok {
return fmt.Errorf("unexpected type %T for field %s", value, name)
}
m.SetRevoked(v)
return nil
case bouncer.FieldIPAddress:
v, ok := value.(string)
if !ok {
return fmt.Errorf("unexpected type %T for field %s", value, name)
}
m.SetIPAddress(v)
return nil
case bouncer.FieldType:
v, ok := value.(string)
if !ok {
return fmt.Errorf("unexpected type %T for field %s", value, name)
}
m.SetType(v)
return nil
case bouncer.FieldVersion:
v, ok := value.(string)
if !ok {
return fmt.Errorf("unexpected type %T for field %s", value, name)
}
m.SetVersion(v)
return nil
case bouncer.FieldUntil:
v, ok := value.(time.Time)
if !ok {
return fmt.Errorf("unexpected type %T for field %s", value, name)
}
m.SetUntil(v)
return nil
case bouncer.FieldLastPull:
v, ok := value.(time.Time)
if !ok {
return fmt.Errorf("unexpected type %T for field %s", value, name)
}
m.SetLastPull(v)
return nil
}
return fmt.Errorf("unknown Bouncer field %s", name)
}
2021-01-14 15:27:45 +00:00
// AddedFields returns all numeric fields that were incremented/decremented during
// this mutation.
func (m *BouncerMutation) AddedFields() []string {
return nil
}
2021-01-14 15:27:45 +00:00
// AddedField returns the numeric value that was incremented/decremented on a field
// with the given name. The second boolean return value indicates that this field
// was not set, or was not defined in the schema.
func (m *BouncerMutation) AddedField(name string) (ent.Value, bool) {
return nil, false
}
2021-01-14 15:27:45 +00:00
// AddField adds the value to the field with the given name. It returns an error if
// the field is not defined in the schema, or if the type mismatched the field
// type.
func (m *BouncerMutation) AddField(name string, value ent.Value) error {
switch name {
}
return fmt.Errorf("unknown Bouncer numeric field %s", name)
}
2021-01-14 15:27:45 +00:00
// ClearedFields returns all nullable fields that were cleared during this
// mutation.
func (m *BouncerMutation) ClearedFields() []string {
var fields []string
if m.FieldCleared(bouncer.FieldIPAddress) {
fields = append(fields, bouncer.FieldIPAddress)
}
if m.FieldCleared(bouncer.FieldType) {
fields = append(fields, bouncer.FieldType)
}
if m.FieldCleared(bouncer.FieldVersion) {
fields = append(fields, bouncer.FieldVersion)
}
if m.FieldCleared(bouncer.FieldUntil) {
fields = append(fields, bouncer.FieldUntil)
}
return fields
}
2021-01-14 15:27:45 +00:00
// FieldCleared returns a boolean indicating if a field with the given name was
// cleared in this mutation.
func (m *BouncerMutation) FieldCleared(name string) bool {
_, ok := m.clearedFields[name]
return ok
}
2021-01-14 15:27:45 +00:00
// ClearField clears the value of the field with the given name. It returns an
// error if the field is not defined in the schema.
func (m *BouncerMutation) ClearField(name string) error {
switch name {
case bouncer.FieldIPAddress:
m.ClearIPAddress()
return nil
case bouncer.FieldType:
m.ClearType()
return nil
case bouncer.FieldVersion:
m.ClearVersion()
return nil
case bouncer.FieldUntil:
m.ClearUntil()
return nil
}
return fmt.Errorf("unknown Bouncer nullable field %s", name)
}
2021-01-14 15:27:45 +00:00
// ResetField resets all changes in the mutation for the field with the given name.
// It returns an error if the field is not defined in the schema.
func (m *BouncerMutation) ResetField(name string) error {
switch name {
case bouncer.FieldCreatedAt:
m.ResetCreatedAt()
return nil
case bouncer.FieldUpdatedAt:
m.ResetUpdatedAt()
return nil
case bouncer.FieldName:
m.ResetName()
return nil
case bouncer.FieldAPIKey:
m.ResetAPIKey()
return nil
case bouncer.FieldRevoked:
m.ResetRevoked()
return nil
case bouncer.FieldIPAddress:
m.ResetIPAddress()
return nil
case bouncer.FieldType:
m.ResetType()
return nil
case bouncer.FieldVersion:
m.ResetVersion()
return nil
case bouncer.FieldUntil:
m.ResetUntil()
return nil
case bouncer.FieldLastPull:
m.ResetLastPull()
return nil
}
return fmt.Errorf("unknown Bouncer field %s", name)
}
2021-01-14 15:27:45 +00:00
// AddedEdges returns all edge names that were set/added in this mutation.
func (m *BouncerMutation) AddedEdges() []string {
edges := make([]string, 0, 0)
return edges
}
2021-01-14 15:27:45 +00:00
// AddedIDs returns all IDs (to other nodes) that were added for the given edge
// name in this mutation.
func (m *BouncerMutation) AddedIDs(name string) []ent.Value {
return nil
}
2021-01-14 15:27:45 +00:00
// RemovedEdges returns all edge names that were removed in this mutation.
func (m *BouncerMutation) RemovedEdges() []string {
edges := make([]string, 0, 0)
return edges
}
2021-01-14 15:27:45 +00:00
// RemovedIDs returns all IDs (to other nodes) that were removed for the edge with
// the given name in this mutation.
func (m *BouncerMutation) RemovedIDs(name string) []ent.Value {
return nil
}
2021-01-14 15:27:45 +00:00
// ClearedEdges returns all edge names that were cleared in this mutation.
func (m *BouncerMutation) ClearedEdges() []string {
edges := make([]string, 0, 0)
return edges
}
2021-01-14 15:27:45 +00:00
// EdgeCleared returns a boolean which indicates if the edge with the given name
// was cleared in this mutation.
func (m *BouncerMutation) EdgeCleared(name string) bool {
return false
}
2021-01-14 15:27:45 +00:00
// ClearEdge clears the value of the edge with the given name. It returns an error
// if that edge is not defined in the schema.
func (m *BouncerMutation) ClearEdge(name string) error {
return fmt.Errorf("unknown Bouncer unique edge %s", name)
}
2021-01-14 15:27:45 +00:00
// ResetEdge resets all changes to the edge with the given name in this mutation.
// It returns an error if the edge is not defined in the schema.
func (m *BouncerMutation) ResetEdge(name string) error {
return fmt.Errorf("unknown Bouncer edge %s", name)
}
2021-01-14 15:27:45 +00:00
// DecisionMutation represents an operation that mutates the Decision nodes in the graph.
type DecisionMutation struct {
config
2021-01-14 15:27:45 +00:00
op Op
typ string
id *int
created_at *time.Time
updated_at *time.Time
until *time.Time
scenario *string
_type *string
start_ip *int64
addstart_ip *int64
end_ip *int64
addend_ip *int64
start_suffix *int64
addstart_suffix *int64
end_suffix *int64
addend_suffix *int64
ip_size *int64
addip_size *int64
scope *string
value *string
origin *string
simulated *bool
clearedFields map[string]struct{}
owner *int
clearedowner bool
done bool
oldValue func(context.Context) (*Decision, error)
predicates []predicate.Decision
}
var _ ent.Mutation = (*DecisionMutation)(nil)
2021-01-14 15:27:45 +00:00
// decisionOption allows management of the mutation configuration using functional options.
type decisionOption func(*DecisionMutation)
2021-01-14 15:27:45 +00:00
// newDecisionMutation creates new mutation for the Decision entity.
func newDecisionMutation(c config, op Op, opts ...decisionOption) *DecisionMutation {
m := &DecisionMutation{
config: c,
op: op,
typ: TypeDecision,
clearedFields: make(map[string]struct{}),
}
for _, opt := range opts {
opt(m)
}
return m
}
2021-01-14 15:27:45 +00:00
// withDecisionID sets the ID field of the mutation.
func withDecisionID(id int) decisionOption {
return func(m *DecisionMutation) {
var (
err error
once sync.Once
value *Decision
)
m.oldValue = func(ctx context.Context) (*Decision, error) {
once.Do(func() {
if m.done {
err = fmt.Errorf("querying old values post mutation is not allowed")
} else {
value, err = m.Client().Decision.Get(ctx, id)
}
})
return value, err
}
m.id = &id
}
}
// withDecision sets the old Decision of the mutation.
func withDecision(node *Decision) decisionOption {
return func(m *DecisionMutation) {
m.oldValue = func(context.Context) (*Decision, error) {
return node, nil
}
m.id = &node.ID
}
}
// Client returns a new `ent.Client` from the mutation. If the mutation was
// executed in a transaction (ent.Tx), a transactional client is returned.
func (m DecisionMutation) Client() *Client {
client := &Client{config: m.config}
client.init()
return client
}
// Tx returns an `ent.Tx` for mutations that were executed in transactions;
// it returns an error otherwise.
func (m DecisionMutation) Tx() (*Tx, error) {
if _, ok := m.driver.(*txDriver); !ok {
return nil, fmt.Errorf("ent: mutation is not running in a transaction")
}
tx := &Tx{config: m.config}
tx.init()
return tx, nil
}
2021-01-14 15:27:45 +00:00
// ID returns the ID value in the mutation. Note that the ID
// is only available if it was provided to the builder.
func (m *DecisionMutation) ID() (id int, exists bool) {
if m.id == nil {
return
}
return *m.id, true
}
2021-01-14 15:27:45 +00:00
// SetCreatedAt sets the "created_at" field.
func (m *DecisionMutation) SetCreatedAt(t time.Time) {
m.created_at = &t
}
2021-01-14 15:27:45 +00:00
// CreatedAt returns the value of the "created_at" field in the mutation.
func (m *DecisionMutation) CreatedAt() (r time.Time, exists bool) {
v := m.created_at
if v == nil {
return
}
return *v, true
}
2021-01-14 15:27:45 +00:00
// OldCreatedAt returns the old "created_at" field's value of the Decision entity.
// If the Decision object wasn't provided to the builder, the object is fetched from the database.
// An error is returned if the mutation operation is not UpdateOne, or the database query fails.
func (m *DecisionMutation) OldCreatedAt(ctx context.Context) (v time.Time, err error) {
if !m.op.Is(OpUpdateOne) {
2021-01-14 15:27:45 +00:00
return v, fmt.Errorf("OldCreatedAt is only allowed on UpdateOne operations")
}
if m.id == nil || m.oldValue == nil {
return v, fmt.Errorf("OldCreatedAt requires an ID field in the mutation")
}
oldValue, err := m.oldValue(ctx)
if err != nil {
return v, fmt.Errorf("querying old value for OldCreatedAt: %w", err)
}
return oldValue.CreatedAt, nil
}
2021-01-14 15:27:45 +00:00
// ResetCreatedAt resets all changes to the "created_at" field.
func (m *DecisionMutation) ResetCreatedAt() {
m.created_at = nil
}
2021-01-14 15:27:45 +00:00
// SetUpdatedAt sets the "updated_at" field.
func (m *DecisionMutation) SetUpdatedAt(t time.Time) {
m.updated_at = &t
}
2021-01-14 15:27:45 +00:00
// UpdatedAt returns the value of the "updated_at" field in the mutation.
func (m *DecisionMutation) UpdatedAt() (r time.Time, exists bool) {
v := m.updated_at
if v == nil {
return
}
return *v, true
}
2021-01-14 15:27:45 +00:00
// OldUpdatedAt returns the old "updated_at" field's value of the Decision entity.
// If the Decision object wasn't provided to the builder, the object is fetched from the database.
// An error is returned if the mutation operation is not UpdateOne, or the database query fails.
func (m *DecisionMutation) OldUpdatedAt(ctx context.Context) (v time.Time, err error) {
if !m.op.Is(OpUpdateOne) {
2021-01-14 15:27:45 +00:00
return v, fmt.Errorf("OldUpdatedAt is only allowed on UpdateOne operations")
}
if m.id == nil || m.oldValue == nil {
return v, fmt.Errorf("OldUpdatedAt requires an ID field in the mutation")
}
oldValue, err := m.oldValue(ctx)
if err != nil {
return v, fmt.Errorf("querying old value for OldUpdatedAt: %w", err)
}
return oldValue.UpdatedAt, nil
}
2021-01-14 15:27:45 +00:00
// ResetUpdatedAt resets all changes to the "updated_at" field.
func (m *DecisionMutation) ResetUpdatedAt() {
m.updated_at = nil
}
2021-01-14 15:27:45 +00:00
// SetUntil sets the "until" field.
func (m *DecisionMutation) SetUntil(t time.Time) {
m.until = &t
}
2021-01-14 15:27:45 +00:00
// Until returns the value of the "until" field in the mutation.
func (m *DecisionMutation) Until() (r time.Time, exists bool) {
v := m.until
if v == nil {
return
}
return *v, true
}
2021-01-14 15:27:45 +00:00
// OldUntil returns the old "until" field's value of the Decision entity.
// If the Decision object wasn't provided to the builder, the object is fetched from the database.
// An error is returned if the mutation operation is not UpdateOne, or the database query fails.
func (m *DecisionMutation) OldUntil(ctx context.Context) (v time.Time, err error) {
if !m.op.Is(OpUpdateOne) {
2021-01-14 15:27:45 +00:00
return v, fmt.Errorf("OldUntil is only allowed on UpdateOne operations")
}
if m.id == nil || m.oldValue == nil {
return v, fmt.Errorf("OldUntil requires an ID field in the mutation")
}
oldValue, err := m.oldValue(ctx)
if err != nil {
return v, fmt.Errorf("querying old value for OldUntil: %w", err)
}
return oldValue.Until, nil
}
2021-01-14 15:27:45 +00:00
// ResetUntil resets all changes to the "until" field.
func (m *DecisionMutation) ResetUntil() {
m.until = nil
}
2021-01-14 15:27:45 +00:00
// SetScenario sets the "scenario" field.
func (m *DecisionMutation) SetScenario(s string) {
m.scenario = &s
}
2021-01-14 15:27:45 +00:00
// Scenario returns the value of the "scenario" field in the mutation.
func (m *DecisionMutation) Scenario() (r string, exists bool) {
v := m.scenario
if v == nil {
return
}
return *v, true
}
2021-01-14 15:27:45 +00:00
// OldScenario returns the old "scenario" field's value of the Decision entity.
// If the Decision object wasn't provided to the builder, the object is fetched from the database.
// An error is returned if the mutation operation is not UpdateOne, or the database query fails.
func (m *DecisionMutation) OldScenario(ctx context.Context) (v string, err error) {
if !m.op.Is(OpUpdateOne) {
2021-01-14 15:27:45 +00:00
return v, fmt.Errorf("OldScenario is only allowed on UpdateOne operations")
}
if m.id == nil || m.oldValue == nil {
return v, fmt.Errorf("OldScenario requires an ID field in the mutation")
}
oldValue, err := m.oldValue(ctx)
if err != nil {
return v, fmt.Errorf("querying old value for OldScenario: %w", err)
}
return oldValue.Scenario, nil
}
2021-01-14 15:27:45 +00:00
// ResetScenario resets all changes to the "scenario" field.
func (m *DecisionMutation) ResetScenario() {
m.scenario = nil
}
2021-01-14 15:27:45 +00:00
// SetType sets the "type" field.
func (m *DecisionMutation) SetType(s string) {
m._type = &s
}
2021-01-14 15:27:45 +00:00
// GetType returns the value of the "type" field in the mutation.
func (m *DecisionMutation) GetType() (r string, exists bool) {
v := m._type
if v == nil {
return
}
return *v, true
}
2021-01-14 15:27:45 +00:00
// OldType returns the old "type" field's value of the Decision entity.
// If the Decision object wasn't provided to the builder, the object is fetched from the database.
// An error is returned if the mutation operation is not UpdateOne, or the database query fails.
func (m *DecisionMutation) OldType(ctx context.Context) (v string, err error) {
if !m.op.Is(OpUpdateOne) {
2021-01-14 15:27:45 +00:00
return v, fmt.Errorf("OldType is only allowed on UpdateOne operations")
}
if m.id == nil || m.oldValue == nil {
return v, fmt.Errorf("OldType requires an ID field in the mutation")
}
oldValue, err := m.oldValue(ctx)
if err != nil {
return v, fmt.Errorf("querying old value for OldType: %w", err)
}
return oldValue.Type, nil
}
2021-01-14 15:27:45 +00:00
// ResetType resets all changes to the "type" field.
func (m *DecisionMutation) ResetType() {
m._type = nil
}
2021-01-14 15:27:45 +00:00
// SetStartIP sets the "start_ip" field.
func (m *DecisionMutation) SetStartIP(i int64) {
m.start_ip = &i
m.addstart_ip = nil
}
2021-01-14 15:27:45 +00:00
// StartIP returns the value of the "start_ip" field in the mutation.
func (m *DecisionMutation) StartIP() (r int64, exists bool) {
v := m.start_ip
if v == nil {
return
}
return *v, true
}
2021-01-14 15:27:45 +00:00
// OldStartIP returns the old "start_ip" field's value of the Decision entity.
// If the Decision object wasn't provided to the builder, the object is fetched from the database.
// An error is returned if the mutation operation is not UpdateOne, or the database query fails.
func (m *DecisionMutation) OldStartIP(ctx context.Context) (v int64, err error) {
if !m.op.Is(OpUpdateOne) {
2021-01-14 15:27:45 +00:00
return v, fmt.Errorf("OldStartIP is only allowed on UpdateOne operations")
}
if m.id == nil || m.oldValue == nil {
return v, fmt.Errorf("OldStartIP requires an ID field in the mutation")
}
oldValue, err := m.oldValue(ctx)
if err != nil {
return v, fmt.Errorf("querying old value for OldStartIP: %w", err)
}
return oldValue.StartIP, nil
}
2021-01-14 15:27:45 +00:00
// AddStartIP adds i to the "start_ip" field.
func (m *DecisionMutation) AddStartIP(i int64) {
if m.addstart_ip != nil {
*m.addstart_ip += i
} else {
m.addstart_ip = &i
}
}
2021-01-14 15:27:45 +00:00
// AddedStartIP returns the value that was added to the "start_ip" field in this mutation.
func (m *DecisionMutation) AddedStartIP() (r int64, exists bool) {
v := m.addstart_ip
if v == nil {
return
}
return *v, true
}
2021-01-14 15:27:45 +00:00
// ClearStartIP clears the value of the "start_ip" field.
func (m *DecisionMutation) ClearStartIP() {
m.start_ip = nil
m.addstart_ip = nil
m.clearedFields[decision.FieldStartIP] = struct{}{}
}
2021-01-14 15:27:45 +00:00
// StartIPCleared returns if the "start_ip" field was cleared in this mutation.
func (m *DecisionMutation) StartIPCleared() bool {
_, ok := m.clearedFields[decision.FieldStartIP]
return ok
}
2021-01-14 15:27:45 +00:00
// ResetStartIP resets all changes to the "start_ip" field.
func (m *DecisionMutation) ResetStartIP() {
m.start_ip = nil
m.addstart_ip = nil
delete(m.clearedFields, decision.FieldStartIP)
}
2021-01-14 15:27:45 +00:00
// SetEndIP sets the "end_ip" field.
func (m *DecisionMutation) SetEndIP(i int64) {
m.end_ip = &i
m.addend_ip = nil
}
2021-01-14 15:27:45 +00:00
// EndIP returns the value of the "end_ip" field in the mutation.
func (m *DecisionMutation) EndIP() (r int64, exists bool) {
v := m.end_ip
if v == nil {
return
}
return *v, true
}
2021-01-14 15:27:45 +00:00
// OldEndIP returns the old "end_ip" field's value of the Decision entity.
// If the Decision object wasn't provided to the builder, the object is fetched from the database.
// An error is returned if the mutation operation is not UpdateOne, or the database query fails.
func (m *DecisionMutation) OldEndIP(ctx context.Context) (v int64, err error) {
if !m.op.Is(OpUpdateOne) {
2021-01-14 15:27:45 +00:00
return v, fmt.Errorf("OldEndIP is only allowed on UpdateOne operations")
}
if m.id == nil || m.oldValue == nil {
return v, fmt.Errorf("OldEndIP requires an ID field in the mutation")
}
oldValue, err := m.oldValue(ctx)
if err != nil {
return v, fmt.Errorf("querying old value for OldEndIP: %w", err)
}
return oldValue.EndIP, nil
}
2021-01-14 15:27:45 +00:00
// AddEndIP adds i to the "end_ip" field.
func (m *DecisionMutation) AddEndIP(i int64) {
if m.addend_ip != nil {
*m.addend_ip += i
} else {
m.addend_ip = &i
}
}
2021-01-14 15:27:45 +00:00
// AddedEndIP returns the value that was added to the "end_ip" field in this mutation.
func (m *DecisionMutation) AddedEndIP() (r int64, exists bool) {
v := m.addend_ip
if v == nil {
return
}
return *v, true
}
2021-01-14 15:27:45 +00:00
// ClearEndIP clears the value of the "end_ip" field.
func (m *DecisionMutation) ClearEndIP() {
m.end_ip = nil
m.addend_ip = nil
m.clearedFields[decision.FieldEndIP] = struct{}{}
}
2021-01-14 15:27:45 +00:00
// EndIPCleared returns if the "end_ip" field was cleared in this mutation.
func (m *DecisionMutation) EndIPCleared() bool {
_, ok := m.clearedFields[decision.FieldEndIP]
return ok
}
2021-01-14 15:27:45 +00:00
// ResetEndIP resets all changes to the "end_ip" field.
func (m *DecisionMutation) ResetEndIP() {
m.end_ip = nil
m.addend_ip = nil
delete(m.clearedFields, decision.FieldEndIP)
}
2021-01-14 15:27:45 +00:00
// SetStartSuffix sets the "start_suffix" field.
func (m *DecisionMutation) SetStartSuffix(i int64) {
m.start_suffix = &i
m.addstart_suffix = nil
}
// StartSuffix returns the value of the "start_suffix" field in the mutation.
func (m *DecisionMutation) StartSuffix() (r int64, exists bool) {
v := m.start_suffix
if v == nil {
return
}
return *v, true
}
// OldStartSuffix returns the old "start_suffix" field's value of the Decision entity.
// If the Decision object wasn't provided to the builder, the object is fetched from the database.
// An error is returned if the mutation operation is not UpdateOne, or the database query fails.
func (m *DecisionMutation) OldStartSuffix(ctx context.Context) (v int64, err error) {
if !m.op.Is(OpUpdateOne) {
return v, fmt.Errorf("OldStartSuffix is only allowed on UpdateOne operations")
}
if m.id == nil || m.oldValue == nil {
return v, fmt.Errorf("OldStartSuffix requires an ID field in the mutation")
}
oldValue, err := m.oldValue(ctx)
if err != nil {
return v, fmt.Errorf("querying old value for OldStartSuffix: %w", err)
}
return oldValue.StartSuffix, nil
}
// AddStartSuffix adds i to the "start_suffix" field.
func (m *DecisionMutation) AddStartSuffix(i int64) {
if m.addstart_suffix != nil {
*m.addstart_suffix += i
} else {
m.addstart_suffix = &i
}
}
// AddedStartSuffix returns the value that was added to the "start_suffix" field in this mutation.
func (m *DecisionMutation) AddedStartSuffix() (r int64, exists bool) {
v := m.addstart_suffix
if v == nil {
return
}
return *v, true
}
// ClearStartSuffix clears the value of the "start_suffix" field.
func (m *DecisionMutation) ClearStartSuffix() {
m.start_suffix = nil
m.addstart_suffix = nil
m.clearedFields[decision.FieldStartSuffix] = struct{}{}
}
// StartSuffixCleared returns if the "start_suffix" field was cleared in this mutation.
func (m *DecisionMutation) StartSuffixCleared() bool {
_, ok := m.clearedFields[decision.FieldStartSuffix]
return ok
}
// ResetStartSuffix resets all changes to the "start_suffix" field.
func (m *DecisionMutation) ResetStartSuffix() {
m.start_suffix = nil
m.addstart_suffix = nil
delete(m.clearedFields, decision.FieldStartSuffix)
}
// SetEndSuffix sets the "end_suffix" field.
func (m *DecisionMutation) SetEndSuffix(i int64) {
m.end_suffix = &i
m.addend_suffix = nil
}
// EndSuffix returns the value of the "end_suffix" field in the mutation.
func (m *DecisionMutation) EndSuffix() (r int64, exists bool) {
v := m.end_suffix
if v == nil {
return
}
return *v, true
}
// OldEndSuffix returns the old "end_suffix" field's value of the Decision entity.
// If the Decision object wasn't provided to the builder, the object is fetched from the database.
// An error is returned if the mutation operation is not UpdateOne, or the database query fails.
func (m *DecisionMutation) OldEndSuffix(ctx context.Context) (v int64, err error) {
if !m.op.Is(OpUpdateOne) {
return v, fmt.Errorf("OldEndSuffix is only allowed on UpdateOne operations")
}
if m.id == nil || m.oldValue == nil {
return v, fmt.Errorf("OldEndSuffix requires an ID field in the mutation")
}
oldValue, err := m.oldValue(ctx)
if err != nil {
return v, fmt.Errorf("querying old value for OldEndSuffix: %w", err)
}
return oldValue.EndSuffix, nil
}
// AddEndSuffix adds i to the "end_suffix" field.
func (m *DecisionMutation) AddEndSuffix(i int64) {
if m.addend_suffix != nil {
*m.addend_suffix += i
} else {
m.addend_suffix = &i
}
}
// AddedEndSuffix returns the value that was added to the "end_suffix" field in this mutation.
func (m *DecisionMutation) AddedEndSuffix() (r int64, exists bool) {
v := m.addend_suffix
if v == nil {
return
}
return *v, true
}
// ClearEndSuffix clears the value of the "end_suffix" field.
func (m *DecisionMutation) ClearEndSuffix() {
m.end_suffix = nil
m.addend_suffix = nil
m.clearedFields[decision.FieldEndSuffix] = struct{}{}
}
// EndSuffixCleared returns if the "end_suffix" field was cleared in this mutation.
func (m *DecisionMutation) EndSuffixCleared() bool {
_, ok := m.clearedFields[decision.FieldEndSuffix]
return ok
}
// ResetEndSuffix resets all changes to the "end_suffix" field.
func (m *DecisionMutation) ResetEndSuffix() {
m.end_suffix = nil
m.addend_suffix = nil
delete(m.clearedFields, decision.FieldEndSuffix)
}
// SetIPSize sets the "ip_size" field.
func (m *DecisionMutation) SetIPSize(i int64) {
m.ip_size = &i
m.addip_size = nil
}
// IPSize returns the value of the "ip_size" field in the mutation.
func (m *DecisionMutation) IPSize() (r int64, exists bool) {
v := m.ip_size
if v == nil {
return
}
return *v, true
}
// OldIPSize returns the old "ip_size" field's value of the Decision entity.
// If the Decision object wasn't provided to the builder, the object is fetched from the database.
// An error is returned if the mutation operation is not UpdateOne, or the database query fails.
func (m *DecisionMutation) OldIPSize(ctx context.Context) (v int64, err error) {
if !m.op.Is(OpUpdateOne) {
return v, fmt.Errorf("OldIPSize is only allowed on UpdateOne operations")
}
if m.id == nil || m.oldValue == nil {
return v, fmt.Errorf("OldIPSize requires an ID field in the mutation")
}
oldValue, err := m.oldValue(ctx)
if err != nil {
return v, fmt.Errorf("querying old value for OldIPSize: %w", err)
}
return oldValue.IPSize, nil
}
// AddIPSize adds i to the "ip_size" field.
func (m *DecisionMutation) AddIPSize(i int64) {
if m.addip_size != nil {
*m.addip_size += i
} else {
m.addip_size = &i
}
}
// AddedIPSize returns the value that was added to the "ip_size" field in this mutation.
func (m *DecisionMutation) AddedIPSize() (r int64, exists bool) {
v := m.addip_size
if v == nil {
return
}
return *v, true
}
// ClearIPSize clears the value of the "ip_size" field.
func (m *DecisionMutation) ClearIPSize() {
m.ip_size = nil
m.addip_size = nil
m.clearedFields[decision.FieldIPSize] = struct{}{}
}
// IPSizeCleared returns if the "ip_size" field was cleared in this mutation.
func (m *DecisionMutation) IPSizeCleared() bool {
_, ok := m.clearedFields[decision.FieldIPSize]
return ok
}
// ResetIPSize resets all changes to the "ip_size" field.
func (m *DecisionMutation) ResetIPSize() {
m.ip_size = nil
m.addip_size = nil
delete(m.clearedFields, decision.FieldIPSize)
}
// SetScope sets the "scope" field.
func (m *DecisionMutation) SetScope(s string) {
m.scope = &s
}
2021-01-14 15:27:45 +00:00
// Scope returns the value of the "scope" field in the mutation.
func (m *DecisionMutation) Scope() (r string, exists bool) {
v := m.scope
if v == nil {
return
}
return *v, true
}
2021-01-14 15:27:45 +00:00
// OldScope returns the old "scope" field's value of the Decision entity.
// If the Decision object wasn't provided to the builder, the object is fetched from the database.
// An error is returned if the mutation operation is not UpdateOne, or the database query fails.
func (m *DecisionMutation) OldScope(ctx context.Context) (v string, err error) {
if !m.op.Is(OpUpdateOne) {
2021-01-14 15:27:45 +00:00
return v, fmt.Errorf("OldScope is only allowed on UpdateOne operations")
}
if m.id == nil || m.oldValue == nil {
return v, fmt.Errorf("OldScope requires an ID field in the mutation")
}
oldValue, err := m.oldValue(ctx)
if err != nil {
return v, fmt.Errorf("querying old value for OldScope: %w", err)
}
return oldValue.Scope, nil
}
2021-01-14 15:27:45 +00:00
// ResetScope resets all changes to the "scope" field.
func (m *DecisionMutation) ResetScope() {
m.scope = nil
}
2021-01-14 15:27:45 +00:00
// SetValue sets the "value" field.
func (m *DecisionMutation) SetValue(s string) {
m.value = &s
}
2021-01-14 15:27:45 +00:00
// Value returns the value of the "value" field in the mutation.
func (m *DecisionMutation) Value() (r string, exists bool) {
v := m.value
if v == nil {
return
}
return *v, true
}
2021-01-14 15:27:45 +00:00
// OldValue returns the old "value" field's value of the Decision entity.
// If the Decision object wasn't provided to the builder, the object is fetched from the database.
// An error is returned if the mutation operation is not UpdateOne, or the database query fails.
func (m *DecisionMutation) OldValue(ctx context.Context) (v string, err error) {
if !m.op.Is(OpUpdateOne) {
2021-01-14 15:27:45 +00:00
return v, fmt.Errorf("OldValue is only allowed on UpdateOne operations")
}
if m.id == nil || m.oldValue == nil {
return v, fmt.Errorf("OldValue requires an ID field in the mutation")
}
oldValue, err := m.oldValue(ctx)
if err != nil {
return v, fmt.Errorf("querying old value for OldValue: %w", err)
}
return oldValue.Value, nil
}
2021-01-14 15:27:45 +00:00
// ResetValue resets all changes to the "value" field.
func (m *DecisionMutation) ResetValue() {
m.value = nil
}
2021-01-14 15:27:45 +00:00
// SetOrigin sets the "origin" field.
func (m *DecisionMutation) SetOrigin(s string) {
m.origin = &s
}
2021-01-14 15:27:45 +00:00
// Origin returns the value of the "origin" field in the mutation.
func (m *DecisionMutation) Origin() (r string, exists bool) {
v := m.origin
if v == nil {
return
}
return *v, true
}
2021-01-14 15:27:45 +00:00
// OldOrigin returns the old "origin" field's value of the Decision entity.
// If the Decision object wasn't provided to the builder, the object is fetched from the database.
// An error is returned if the mutation operation is not UpdateOne, or the database query fails.
func (m *DecisionMutation) OldOrigin(ctx context.Context) (v string, err error) {
if !m.op.Is(OpUpdateOne) {
2021-01-14 15:27:45 +00:00
return v, fmt.Errorf("OldOrigin is only allowed on UpdateOne operations")
}
if m.id == nil || m.oldValue == nil {
return v, fmt.Errorf("OldOrigin requires an ID field in the mutation")
}
oldValue, err := m.oldValue(ctx)
if err != nil {
return v, fmt.Errorf("querying old value for OldOrigin: %w", err)
}
return oldValue.Origin, nil
}
2021-01-14 15:27:45 +00:00
// ResetOrigin resets all changes to the "origin" field.
func (m *DecisionMutation) ResetOrigin() {
m.origin = nil
}
2021-01-14 15:27:45 +00:00
// SetSimulated sets the "simulated" field.
func (m *DecisionMutation) SetSimulated(b bool) {
m.simulated = &b
}
2021-01-14 15:27:45 +00:00
// Simulated returns the value of the "simulated" field in the mutation.
func (m *DecisionMutation) Simulated() (r bool, exists bool) {
v := m.simulated
if v == nil {
return
}
return *v, true
}
2021-01-14 15:27:45 +00:00
// OldSimulated returns the old "simulated" field's value of the Decision entity.
// If the Decision object wasn't provided to the builder, the object is fetched from the database.
// An error is returned if the mutation operation is not UpdateOne, or the database query fails.
func (m *DecisionMutation) OldSimulated(ctx context.Context) (v bool, err error) {
if !m.op.Is(OpUpdateOne) {
2021-01-14 15:27:45 +00:00
return v, fmt.Errorf("OldSimulated is only allowed on UpdateOne operations")
}
if m.id == nil || m.oldValue == nil {
return v, fmt.Errorf("OldSimulated requires an ID field in the mutation")
}
oldValue, err := m.oldValue(ctx)
if err != nil {
return v, fmt.Errorf("querying old value for OldSimulated: %w", err)
}
return oldValue.Simulated, nil
}
2021-01-14 15:27:45 +00:00
// ResetSimulated resets all changes to the "simulated" field.
func (m *DecisionMutation) ResetSimulated() {
m.simulated = nil
}
2021-01-14 15:27:45 +00:00
// SetOwnerID sets the "owner" edge to the Alert entity by id.
func (m *DecisionMutation) SetOwnerID(id int) {
m.owner = &id
}
2021-01-14 15:27:45 +00:00
// ClearOwner clears the "owner" edge to the Alert entity.
func (m *DecisionMutation) ClearOwner() {
m.clearedowner = true
}
2021-01-14 15:27:45 +00:00
// OwnerCleared returns if the "owner" edge to the Alert entity was cleared.
func (m *DecisionMutation) OwnerCleared() bool {
return m.clearedowner
}
2021-01-14 15:27:45 +00:00
// OwnerID returns the "owner" edge ID in the mutation.
func (m *DecisionMutation) OwnerID() (id int, exists bool) {
if m.owner != nil {
return *m.owner, true
}
return
}
2021-01-14 15:27:45 +00:00
// OwnerIDs returns the "owner" edge IDs in the mutation.
// Note that IDs always returns len(IDs) <= 1 for unique edges, and you should use
// OwnerID instead. It exists only for internal usage by the builders.
func (m *DecisionMutation) OwnerIDs() (ids []int) {
if id := m.owner; id != nil {
ids = append(ids, *id)
}
return
}
2021-01-14 15:27:45 +00:00
// ResetOwner resets all changes to the "owner" edge.
func (m *DecisionMutation) ResetOwner() {
m.owner = nil
m.clearedowner = false
}
// Op returns the operation name.
func (m *DecisionMutation) Op() Op {
return m.op
}
// Type returns the node type of this mutation (Decision).
func (m *DecisionMutation) Type() string {
return m.typ
}
2021-01-14 15:27:45 +00:00
// Fields returns all fields that were changed during this mutation. Note that in
// order to get all numeric fields that were incremented/decremented, call
// AddedFields().
func (m *DecisionMutation) Fields() []string {
2021-01-14 15:27:45 +00:00
fields := make([]string, 0, 14)
if m.created_at != nil {
fields = append(fields, decision.FieldCreatedAt)
}
if m.updated_at != nil {
fields = append(fields, decision.FieldUpdatedAt)
}
if m.until != nil {
fields = append(fields, decision.FieldUntil)
}
if m.scenario != nil {
fields = append(fields, decision.FieldScenario)
}
if m._type != nil {
fields = append(fields, decision.FieldType)
}
if m.start_ip != nil {
fields = append(fields, decision.FieldStartIP)
}
if m.end_ip != nil {
fields = append(fields, decision.FieldEndIP)
}
2021-01-14 15:27:45 +00:00
if m.start_suffix != nil {
fields = append(fields, decision.FieldStartSuffix)
}
if m.end_suffix != nil {
fields = append(fields, decision.FieldEndSuffix)
}
if m.ip_size != nil {
fields = append(fields, decision.FieldIPSize)
}
if m.scope != nil {
fields = append(fields, decision.FieldScope)
}
if m.value != nil {
fields = append(fields, decision.FieldValue)
}
if m.origin != nil {
fields = append(fields, decision.FieldOrigin)
}
if m.simulated != nil {
fields = append(fields, decision.FieldSimulated)
}
return fields
}
2021-01-14 15:27:45 +00:00
// Field returns the value of a field with the given name. The second boolean
// return value indicates that this field was not set, or was not defined in the
// schema.
func (m *DecisionMutation) Field(name string) (ent.Value, bool) {
switch name {
case decision.FieldCreatedAt:
return m.CreatedAt()
case decision.FieldUpdatedAt:
return m.UpdatedAt()
case decision.FieldUntil:
return m.Until()
case decision.FieldScenario:
return m.Scenario()
case decision.FieldType:
return m.GetType()
case decision.FieldStartIP:
return m.StartIP()
case decision.FieldEndIP:
return m.EndIP()
2021-01-14 15:27:45 +00:00
case decision.FieldStartSuffix:
return m.StartSuffix()
case decision.FieldEndSuffix:
return m.EndSuffix()
case decision.FieldIPSize:
return m.IPSize()
case decision.FieldScope:
return m.Scope()
case decision.FieldValue:
return m.Value()
case decision.FieldOrigin:
return m.Origin()
case decision.FieldSimulated:
return m.Simulated()
}
return nil, false
}
2021-01-14 15:27:45 +00:00
// OldField returns the old value of the field from the database. An error is
// returned if the mutation operation is not UpdateOne, or the query to the
// database failed.
func (m *DecisionMutation) OldField(ctx context.Context, name string) (ent.Value, error) {
switch name {
case decision.FieldCreatedAt:
return m.OldCreatedAt(ctx)
case decision.FieldUpdatedAt:
return m.OldUpdatedAt(ctx)
case decision.FieldUntil:
return m.OldUntil(ctx)
case decision.FieldScenario:
return m.OldScenario(ctx)
case decision.FieldType:
return m.OldType(ctx)
case decision.FieldStartIP:
return m.OldStartIP(ctx)
case decision.FieldEndIP:
return m.OldEndIP(ctx)
2021-01-14 15:27:45 +00:00
case decision.FieldStartSuffix:
return m.OldStartSuffix(ctx)
case decision.FieldEndSuffix:
return m.OldEndSuffix(ctx)
case decision.FieldIPSize:
return m.OldIPSize(ctx)
case decision.FieldScope:
return m.OldScope(ctx)
case decision.FieldValue:
return m.OldValue(ctx)
case decision.FieldOrigin:
return m.OldOrigin(ctx)
case decision.FieldSimulated:
return m.OldSimulated(ctx)
}
return nil, fmt.Errorf("unknown Decision field %s", name)
}
2021-01-14 15:27:45 +00:00
// SetField sets the value of a field with the given name. It returns an error if
// the field is not defined in the schema, or if the type mismatched the field
// type.
func (m *DecisionMutation) SetField(name string, value ent.Value) error {
switch name {
case decision.FieldCreatedAt:
v, ok := value.(time.Time)
if !ok {
return fmt.Errorf("unexpected type %T for field %s", value, name)
}
m.SetCreatedAt(v)
return nil
case decision.FieldUpdatedAt:
v, ok := value.(time.Time)
if !ok {
return fmt.Errorf("unexpected type %T for field %s", value, name)
}
m.SetUpdatedAt(v)
return nil
case decision.FieldUntil:
v, ok := value.(time.Time)
if !ok {
return fmt.Errorf("unexpected type %T for field %s", value, name)
}
m.SetUntil(v)
return nil
case decision.FieldScenario:
v, ok := value.(string)
if !ok {
return fmt.Errorf("unexpected type %T for field %s", value, name)
}
m.SetScenario(v)
return nil
case decision.FieldType:
v, ok := value.(string)
if !ok {
return fmt.Errorf("unexpected type %T for field %s", value, name)
}
m.SetType(v)
return nil
case decision.FieldStartIP:
v, ok := value.(int64)
if !ok {
return fmt.Errorf("unexpected type %T for field %s", value, name)
}
m.SetStartIP(v)
return nil
case decision.FieldEndIP:
v, ok := value.(int64)
if !ok {
return fmt.Errorf("unexpected type %T for field %s", value, name)
}
m.SetEndIP(v)
return nil
2021-01-14 15:27:45 +00:00
case decision.FieldStartSuffix:
v, ok := value.(int64)
if !ok {
return fmt.Errorf("unexpected type %T for field %s", value, name)
}
m.SetStartSuffix(v)
return nil
case decision.FieldEndSuffix:
v, ok := value.(int64)
if !ok {
return fmt.Errorf("unexpected type %T for field %s", value, name)
}
m.SetEndSuffix(v)
return nil
case decision.FieldIPSize:
v, ok := value.(int64)
if !ok {
return fmt.Errorf("unexpected type %T for field %s", value, name)
}
m.SetIPSize(v)
return nil
case decision.FieldScope:
v, ok := value.(string)
if !ok {
return fmt.Errorf("unexpected type %T for field %s", value, name)
}
m.SetScope(v)
return nil
case decision.FieldValue:
v, ok := value.(string)
if !ok {
return fmt.Errorf("unexpected type %T for field %s", value, name)
}
m.SetValue(v)
return nil
case decision.FieldOrigin:
v, ok := value.(string)
if !ok {
return fmt.Errorf("unexpected type %T for field %s", value, name)
}
m.SetOrigin(v)
return nil
case decision.FieldSimulated:
v, ok := value.(bool)
if !ok {
return fmt.Errorf("unexpected type %T for field %s", value, name)
}
m.SetSimulated(v)
return nil
}
return fmt.Errorf("unknown Decision field %s", name)
}
2021-01-14 15:27:45 +00:00
// AddedFields returns all numeric fields that were incremented/decremented during
// this mutation.
func (m *DecisionMutation) AddedFields() []string {
var fields []string
if m.addstart_ip != nil {
fields = append(fields, decision.FieldStartIP)
}
if m.addend_ip != nil {
fields = append(fields, decision.FieldEndIP)
}
2021-01-14 15:27:45 +00:00
if m.addstart_suffix != nil {
fields = append(fields, decision.FieldStartSuffix)
}
if m.addend_suffix != nil {
fields = append(fields, decision.FieldEndSuffix)
}
if m.addip_size != nil {
fields = append(fields, decision.FieldIPSize)
}
return fields
}
2021-01-14 15:27:45 +00:00
// AddedField returns the numeric value that was incremented/decremented on a field
// with the given name. The second boolean return value indicates that this field
// was not set, or was not defined in the schema.
func (m *DecisionMutation) AddedField(name string) (ent.Value, bool) {
switch name {
case decision.FieldStartIP:
return m.AddedStartIP()
case decision.FieldEndIP:
return m.AddedEndIP()
2021-01-14 15:27:45 +00:00
case decision.FieldStartSuffix:
return m.AddedStartSuffix()
case decision.FieldEndSuffix:
return m.AddedEndSuffix()
case decision.FieldIPSize:
return m.AddedIPSize()
}
return nil, false
}
2021-01-14 15:27:45 +00:00
// AddField adds the value to the field with the given name. It returns an error if
// the field is not defined in the schema, or if the type mismatched the field
// type.
func (m *DecisionMutation) AddField(name string, value ent.Value) error {
switch name {
case decision.FieldStartIP:
v, ok := value.(int64)
if !ok {
return fmt.Errorf("unexpected type %T for field %s", value, name)
}
m.AddStartIP(v)
return nil
case decision.FieldEndIP:
v, ok := value.(int64)
if !ok {
return fmt.Errorf("unexpected type %T for field %s", value, name)
}
m.AddEndIP(v)
return nil
2021-01-14 15:27:45 +00:00
case decision.FieldStartSuffix:
v, ok := value.(int64)
if !ok {
return fmt.Errorf("unexpected type %T for field %s", value, name)
}
m.AddStartSuffix(v)
return nil
case decision.FieldEndSuffix:
v, ok := value.(int64)
if !ok {
return fmt.Errorf("unexpected type %T for field %s", value, name)
}
m.AddEndSuffix(v)
return nil
case decision.FieldIPSize:
v, ok := value.(int64)
if !ok {
return fmt.Errorf("unexpected type %T for field %s", value, name)
}
m.AddIPSize(v)
return nil
}
return fmt.Errorf("unknown Decision numeric field %s", name)
}
2021-01-14 15:27:45 +00:00
// ClearedFields returns all nullable fields that were cleared during this
// mutation.
func (m *DecisionMutation) ClearedFields() []string {
var fields []string
if m.FieldCleared(decision.FieldStartIP) {
fields = append(fields, decision.FieldStartIP)
}
if m.FieldCleared(decision.FieldEndIP) {
fields = append(fields, decision.FieldEndIP)
}
2021-01-14 15:27:45 +00:00
if m.FieldCleared(decision.FieldStartSuffix) {
fields = append(fields, decision.FieldStartSuffix)
}
if m.FieldCleared(decision.FieldEndSuffix) {
fields = append(fields, decision.FieldEndSuffix)
}
if m.FieldCleared(decision.FieldIPSize) {
fields = append(fields, decision.FieldIPSize)
}
return fields
}
2021-01-14 15:27:45 +00:00
// FieldCleared returns a boolean indicating if a field with the given name was
// cleared in this mutation.
func (m *DecisionMutation) FieldCleared(name string) bool {
_, ok := m.clearedFields[name]
return ok
}
2021-01-14 15:27:45 +00:00
// ClearField clears the value of the field with the given name. It returns an
// error if the field is not defined in the schema.
func (m *DecisionMutation) ClearField(name string) error {
switch name {
case decision.FieldStartIP:
m.ClearStartIP()
return nil
case decision.FieldEndIP:
m.ClearEndIP()
return nil
2021-01-14 15:27:45 +00:00
case decision.FieldStartSuffix:
m.ClearStartSuffix()
return nil
case decision.FieldEndSuffix:
m.ClearEndSuffix()
return nil
case decision.FieldIPSize:
m.ClearIPSize()
return nil
}
return fmt.Errorf("unknown Decision nullable field %s", name)
}
2021-01-14 15:27:45 +00:00
// ResetField resets all changes in the mutation for the field with the given name.
// It returns an error if the field is not defined in the schema.
func (m *DecisionMutation) ResetField(name string) error {
switch name {
case decision.FieldCreatedAt:
m.ResetCreatedAt()
return nil
case decision.FieldUpdatedAt:
m.ResetUpdatedAt()
return nil
case decision.FieldUntil:
m.ResetUntil()
return nil
case decision.FieldScenario:
m.ResetScenario()
return nil
case decision.FieldType:
m.ResetType()
return nil
case decision.FieldStartIP:
m.ResetStartIP()
return nil
case decision.FieldEndIP:
m.ResetEndIP()
return nil
2021-01-14 15:27:45 +00:00
case decision.FieldStartSuffix:
m.ResetStartSuffix()
return nil
case decision.FieldEndSuffix:
m.ResetEndSuffix()
return nil
case decision.FieldIPSize:
m.ResetIPSize()
return nil
case decision.FieldScope:
m.ResetScope()
return nil
case decision.FieldValue:
m.ResetValue()
return nil
case decision.FieldOrigin:
m.ResetOrigin()
return nil
case decision.FieldSimulated:
m.ResetSimulated()
return nil
}
return fmt.Errorf("unknown Decision field %s", name)
}
2021-01-14 15:27:45 +00:00
// AddedEdges returns all edge names that were set/added in this mutation.
func (m *DecisionMutation) AddedEdges() []string {
edges := make([]string, 0, 1)
if m.owner != nil {
edges = append(edges, decision.EdgeOwner)
}
return edges
}
2021-01-14 15:27:45 +00:00
// AddedIDs returns all IDs (to other nodes) that were added for the given edge
// name in this mutation.
func (m *DecisionMutation) AddedIDs(name string) []ent.Value {
switch name {
case decision.EdgeOwner:
if id := m.owner; id != nil {
return []ent.Value{*id}
}
}
return nil
}
2021-01-14 15:27:45 +00:00
// RemovedEdges returns all edge names that were removed in this mutation.
func (m *DecisionMutation) RemovedEdges() []string {
edges := make([]string, 0, 1)
return edges
}
2021-01-14 15:27:45 +00:00
// RemovedIDs returns all IDs (to other nodes) that were removed for the edge with
// the given name in this mutation.
func (m *DecisionMutation) RemovedIDs(name string) []ent.Value {
switch name {
}
return nil
}
2021-01-14 15:27:45 +00:00
// ClearedEdges returns all edge names that were cleared in this mutation.
func (m *DecisionMutation) ClearedEdges() []string {
edges := make([]string, 0, 1)
if m.clearedowner {
edges = append(edges, decision.EdgeOwner)
}
return edges
}
2021-01-14 15:27:45 +00:00
// EdgeCleared returns a boolean which indicates if the edge with the given name
// was cleared in this mutation.
func (m *DecisionMutation) EdgeCleared(name string) bool {
switch name {
case decision.EdgeOwner:
return m.clearedowner
}
return false
}
2021-01-14 15:27:45 +00:00
// ClearEdge clears the value of the edge with the given name. It returns an error
// if that edge is not defined in the schema.
func (m *DecisionMutation) ClearEdge(name string) error {
switch name {
case decision.EdgeOwner:
m.ClearOwner()
return nil
}
return fmt.Errorf("unknown Decision unique edge %s", name)
}
2021-01-14 15:27:45 +00:00
// ResetEdge resets all changes to the edge with the given name in this mutation.
// It returns an error if the edge is not defined in the schema.
func (m *DecisionMutation) ResetEdge(name string) error {
switch name {
case decision.EdgeOwner:
m.ResetOwner()
return nil
}
return fmt.Errorf("unknown Decision edge %s", name)
}
2021-01-14 15:27:45 +00:00
// EventMutation represents an operation that mutates the Event nodes in the graph.
type EventMutation struct {
config
op Op
typ string
id *int
created_at *time.Time
updated_at *time.Time
time *time.Time
serialized *string
clearedFields map[string]struct{}
owner *int
clearedowner bool
done bool
oldValue func(context.Context) (*Event, error)
2021-01-14 15:27:45 +00:00
predicates []predicate.Event
}
var _ ent.Mutation = (*EventMutation)(nil)
2021-01-14 15:27:45 +00:00
// eventOption allows management of the mutation configuration using functional options.
type eventOption func(*EventMutation)
2021-01-14 15:27:45 +00:00
// newEventMutation creates new mutation for the Event entity.
func newEventMutation(c config, op Op, opts ...eventOption) *EventMutation {
m := &EventMutation{
config: c,
op: op,
typ: TypeEvent,
clearedFields: make(map[string]struct{}),
}
for _, opt := range opts {
opt(m)
}
return m
}
2021-01-14 15:27:45 +00:00
// withEventID sets the ID field of the mutation.
func withEventID(id int) eventOption {
return func(m *EventMutation) {
var (
err error
once sync.Once
value *Event
)
m.oldValue = func(ctx context.Context) (*Event, error) {
once.Do(func() {
if m.done {
err = fmt.Errorf("querying old values post mutation is not allowed")
} else {
value, err = m.Client().Event.Get(ctx, id)
}
})
return value, err
}
m.id = &id
}
}
// withEvent sets the old Event of the mutation.
func withEvent(node *Event) eventOption {
return func(m *EventMutation) {
m.oldValue = func(context.Context) (*Event, error) {
return node, nil
}
m.id = &node.ID
}
}
// Client returns a new `ent.Client` from the mutation. If the mutation was
// executed in a transaction (ent.Tx), a transactional client is returned.
func (m EventMutation) Client() *Client {
client := &Client{config: m.config}
client.init()
return client
}
// Tx returns an `ent.Tx` for mutations that were executed in transactions;
// it returns an error otherwise.
func (m EventMutation) Tx() (*Tx, error) {
if _, ok := m.driver.(*txDriver); !ok {
return nil, fmt.Errorf("ent: mutation is not running in a transaction")
}
tx := &Tx{config: m.config}
tx.init()
return tx, nil
}
2021-01-14 15:27:45 +00:00
// ID returns the ID value in the mutation. Note that the ID
// is only available if it was provided to the builder.
func (m *EventMutation) ID() (id int, exists bool) {
if m.id == nil {
return
}
return *m.id, true
}
2021-01-14 15:27:45 +00:00
// SetCreatedAt sets the "created_at" field.
func (m *EventMutation) SetCreatedAt(t time.Time) {
m.created_at = &t
}
2021-01-14 15:27:45 +00:00
// CreatedAt returns the value of the "created_at" field in the mutation.
func (m *EventMutation) CreatedAt() (r time.Time, exists bool) {
v := m.created_at
if v == nil {
return
}
return *v, true
}
2021-01-14 15:27:45 +00:00
// OldCreatedAt returns the old "created_at" field's value of the Event entity.
// If the Event object wasn't provided to the builder, the object is fetched from the database.
// An error is returned if the mutation operation is not UpdateOne, or the database query fails.
func (m *EventMutation) OldCreatedAt(ctx context.Context) (v time.Time, err error) {
if !m.op.Is(OpUpdateOne) {
2021-01-14 15:27:45 +00:00
return v, fmt.Errorf("OldCreatedAt is only allowed on UpdateOne operations")
}
if m.id == nil || m.oldValue == nil {
return v, fmt.Errorf("OldCreatedAt requires an ID field in the mutation")
}
oldValue, err := m.oldValue(ctx)
if err != nil {
return v, fmt.Errorf("querying old value for OldCreatedAt: %w", err)
}
return oldValue.CreatedAt, nil
}
2021-01-14 15:27:45 +00:00
// ResetCreatedAt resets all changes to the "created_at" field.
func (m *EventMutation) ResetCreatedAt() {
m.created_at = nil
}
2021-01-14 15:27:45 +00:00
// SetUpdatedAt sets the "updated_at" field.
func (m *EventMutation) SetUpdatedAt(t time.Time) {
m.updated_at = &t
}
2021-01-14 15:27:45 +00:00
// UpdatedAt returns the value of the "updated_at" field in the mutation.
func (m *EventMutation) UpdatedAt() (r time.Time, exists bool) {
v := m.updated_at
if v == nil {
return
}
return *v, true
}
2021-01-14 15:27:45 +00:00
// OldUpdatedAt returns the old "updated_at" field's value of the Event entity.
// If the Event object wasn't provided to the builder, the object is fetched from the database.
// An error is returned if the mutation operation is not UpdateOne, or the database query fails.
func (m *EventMutation) OldUpdatedAt(ctx context.Context) (v time.Time, err error) {
if !m.op.Is(OpUpdateOne) {
2021-01-14 15:27:45 +00:00
return v, fmt.Errorf("OldUpdatedAt is only allowed on UpdateOne operations")
}
if m.id == nil || m.oldValue == nil {
return v, fmt.Errorf("OldUpdatedAt requires an ID field in the mutation")
}
oldValue, err := m.oldValue(ctx)
if err != nil {
return v, fmt.Errorf("querying old value for OldUpdatedAt: %w", err)
}
return oldValue.UpdatedAt, nil
}
2021-01-14 15:27:45 +00:00
// ResetUpdatedAt resets all changes to the "updated_at" field.
func (m *EventMutation) ResetUpdatedAt() {
m.updated_at = nil
}
2021-01-14 15:27:45 +00:00
// SetTime sets the "time" field.
func (m *EventMutation) SetTime(t time.Time) {
m.time = &t
}
2021-01-14 15:27:45 +00:00
// Time returns the value of the "time" field in the mutation.
func (m *EventMutation) Time() (r time.Time, exists bool) {
v := m.time
if v == nil {
return
}
return *v, true
}
2021-01-14 15:27:45 +00:00
// OldTime returns the old "time" field's value of the Event entity.
// If the Event object wasn't provided to the builder, the object is fetched from the database.
// An error is returned if the mutation operation is not UpdateOne, or the database query fails.
func (m *EventMutation) OldTime(ctx context.Context) (v time.Time, err error) {
if !m.op.Is(OpUpdateOne) {
2021-01-14 15:27:45 +00:00
return v, fmt.Errorf("OldTime is only allowed on UpdateOne operations")
}
if m.id == nil || m.oldValue == nil {
return v, fmt.Errorf("OldTime requires an ID field in the mutation")
}
oldValue, err := m.oldValue(ctx)
if err != nil {
return v, fmt.Errorf("querying old value for OldTime: %w", err)
}
return oldValue.Time, nil
}
2021-01-14 15:27:45 +00:00
// ResetTime resets all changes to the "time" field.
func (m *EventMutation) ResetTime() {
m.time = nil
}
2021-01-14 15:27:45 +00:00
// SetSerialized sets the "serialized" field.
func (m *EventMutation) SetSerialized(s string) {
m.serialized = &s
}
2021-01-14 15:27:45 +00:00
// Serialized returns the value of the "serialized" field in the mutation.
func (m *EventMutation) Serialized() (r string, exists bool) {
v := m.serialized
if v == nil {
return
}
return *v, true
}
2021-01-14 15:27:45 +00:00
// OldSerialized returns the old "serialized" field's value of the Event entity.
// If the Event object wasn't provided to the builder, the object is fetched from the database.
// An error is returned if the mutation operation is not UpdateOne, or the database query fails.
func (m *EventMutation) OldSerialized(ctx context.Context) (v string, err error) {
if !m.op.Is(OpUpdateOne) {
2021-01-14 15:27:45 +00:00
return v, fmt.Errorf("OldSerialized is only allowed on UpdateOne operations")
}
if m.id == nil || m.oldValue == nil {
return v, fmt.Errorf("OldSerialized requires an ID field in the mutation")
}
oldValue, err := m.oldValue(ctx)
if err != nil {
return v, fmt.Errorf("querying old value for OldSerialized: %w", err)
}
return oldValue.Serialized, nil
}
2021-01-14 15:27:45 +00:00
// ResetSerialized resets all changes to the "serialized" field.
func (m *EventMutation) ResetSerialized() {
m.serialized = nil
}
2021-01-14 15:27:45 +00:00
// SetOwnerID sets the "owner" edge to the Alert entity by id.
func (m *EventMutation) SetOwnerID(id int) {
m.owner = &id
}
2021-01-14 15:27:45 +00:00
// ClearOwner clears the "owner" edge to the Alert entity.
func (m *EventMutation) ClearOwner() {
m.clearedowner = true
}
2021-01-14 15:27:45 +00:00
// OwnerCleared returns if the "owner" edge to the Alert entity was cleared.
func (m *EventMutation) OwnerCleared() bool {
return m.clearedowner
}
2021-01-14 15:27:45 +00:00
// OwnerID returns the "owner" edge ID in the mutation.
func (m *EventMutation) OwnerID() (id int, exists bool) {
if m.owner != nil {
return *m.owner, true
}
return
}
2021-01-14 15:27:45 +00:00
// OwnerIDs returns the "owner" edge IDs in the mutation.
// Note that IDs always returns len(IDs) <= 1 for unique edges, and you should use
// OwnerID instead. It exists only for internal usage by the builders.
func (m *EventMutation) OwnerIDs() (ids []int) {
if id := m.owner; id != nil {
ids = append(ids, *id)
}
return
}
2021-01-14 15:27:45 +00:00
// ResetOwner resets all changes to the "owner" edge.
func (m *EventMutation) ResetOwner() {
m.owner = nil
m.clearedowner = false
}
// Op returns the operation name.
func (m *EventMutation) Op() Op {
return m.op
}
// Type returns the node type of this mutation (Event).
func (m *EventMutation) Type() string {
return m.typ
}
2021-01-14 15:27:45 +00:00
// Fields returns all fields that were changed during this mutation. Note that in
// order to get all numeric fields that were incremented/decremented, call
// AddedFields().
func (m *EventMutation) Fields() []string {
fields := make([]string, 0, 4)
if m.created_at != nil {
fields = append(fields, event.FieldCreatedAt)
}
if m.updated_at != nil {
fields = append(fields, event.FieldUpdatedAt)
}
if m.time != nil {
fields = append(fields, event.FieldTime)
}
if m.serialized != nil {
fields = append(fields, event.FieldSerialized)
}
return fields
}
2021-01-14 15:27:45 +00:00
// Field returns the value of a field with the given name. The second boolean
// return value indicates that this field was not set, or was not defined in the
// schema.
func (m *EventMutation) Field(name string) (ent.Value, bool) {
switch name {
case event.FieldCreatedAt:
return m.CreatedAt()
case event.FieldUpdatedAt:
return m.UpdatedAt()
case event.FieldTime:
return m.Time()
case event.FieldSerialized:
return m.Serialized()
}
return nil, false
}
2021-01-14 15:27:45 +00:00
// OldField returns the old value of the field from the database. An error is
// returned if the mutation operation is not UpdateOne, or the query to the
// database failed.
func (m *EventMutation) OldField(ctx context.Context, name string) (ent.Value, error) {
switch name {
case event.FieldCreatedAt:
return m.OldCreatedAt(ctx)
case event.FieldUpdatedAt:
return m.OldUpdatedAt(ctx)
case event.FieldTime:
return m.OldTime(ctx)
case event.FieldSerialized:
return m.OldSerialized(ctx)
}
return nil, fmt.Errorf("unknown Event field %s", name)
}
2021-01-14 15:27:45 +00:00
// SetField sets the value of a field with the given name. It returns an error if
// the field is not defined in the schema, or if the type mismatched the field
// type.
func (m *EventMutation) SetField(name string, value ent.Value) error {
switch name {
case event.FieldCreatedAt:
v, ok := value.(time.Time)
if !ok {
return fmt.Errorf("unexpected type %T for field %s", value, name)
}
m.SetCreatedAt(v)
return nil
case event.FieldUpdatedAt:
v, ok := value.(time.Time)
if !ok {
return fmt.Errorf("unexpected type %T for field %s", value, name)
}
m.SetUpdatedAt(v)
return nil
case event.FieldTime:
v, ok := value.(time.Time)
if !ok {
return fmt.Errorf("unexpected type %T for field %s", value, name)
}
m.SetTime(v)
return nil
case event.FieldSerialized:
v, ok := value.(string)
if !ok {
return fmt.Errorf("unexpected type %T for field %s", value, name)
}
m.SetSerialized(v)
return nil
}
return fmt.Errorf("unknown Event field %s", name)
}
2021-01-14 15:27:45 +00:00
// AddedFields returns all numeric fields that were incremented/decremented during
// this mutation.
func (m *EventMutation) AddedFields() []string {
return nil
}
2021-01-14 15:27:45 +00:00
// AddedField returns the numeric value that was incremented/decremented on a field
// with the given name. The second boolean return value indicates that this field
// was not set, or was not defined in the schema.
func (m *EventMutation) AddedField(name string) (ent.Value, bool) {
return nil, false
}
2021-01-14 15:27:45 +00:00
// AddField adds the value to the field with the given name. It returns an error if
// the field is not defined in the schema, or if the type mismatched the field
// type.
func (m *EventMutation) AddField(name string, value ent.Value) error {
switch name {
}
return fmt.Errorf("unknown Event numeric field %s", name)
}
2021-01-14 15:27:45 +00:00
// ClearedFields returns all nullable fields that were cleared during this
// mutation.
func (m *EventMutation) ClearedFields() []string {
return nil
}
2021-01-14 15:27:45 +00:00
// FieldCleared returns a boolean indicating if a field with the given name was
// cleared in this mutation.
func (m *EventMutation) FieldCleared(name string) bool {
_, ok := m.clearedFields[name]
return ok
}
2021-01-14 15:27:45 +00:00
// ClearField clears the value of the field with the given name. It returns an
// error if the field is not defined in the schema.
func (m *EventMutation) ClearField(name string) error {
return fmt.Errorf("unknown Event nullable field %s", name)
}
2021-01-14 15:27:45 +00:00
// ResetField resets all changes in the mutation for the field with the given name.
// It returns an error if the field is not defined in the schema.
func (m *EventMutation) ResetField(name string) error {
switch name {
case event.FieldCreatedAt:
m.ResetCreatedAt()
return nil
case event.FieldUpdatedAt:
m.ResetUpdatedAt()
return nil
case event.FieldTime:
m.ResetTime()
return nil
case event.FieldSerialized:
m.ResetSerialized()
return nil
}
return fmt.Errorf("unknown Event field %s", name)
}
2021-01-14 15:27:45 +00:00
// AddedEdges returns all edge names that were set/added in this mutation.
func (m *EventMutation) AddedEdges() []string {
edges := make([]string, 0, 1)
if m.owner != nil {
edges = append(edges, event.EdgeOwner)
}
return edges
}
2021-01-14 15:27:45 +00:00
// AddedIDs returns all IDs (to other nodes) that were added for the given edge
// name in this mutation.
func (m *EventMutation) AddedIDs(name string) []ent.Value {
switch name {
case event.EdgeOwner:
if id := m.owner; id != nil {
return []ent.Value{*id}
}
}
return nil
}
2021-01-14 15:27:45 +00:00
// RemovedEdges returns all edge names that were removed in this mutation.
func (m *EventMutation) RemovedEdges() []string {
edges := make([]string, 0, 1)
return edges
}
2021-01-14 15:27:45 +00:00
// RemovedIDs returns all IDs (to other nodes) that were removed for the edge with
// the given name in this mutation.
func (m *EventMutation) RemovedIDs(name string) []ent.Value {
switch name {
}
return nil
}
2021-01-14 15:27:45 +00:00
// ClearedEdges returns all edge names that were cleared in this mutation.
func (m *EventMutation) ClearedEdges() []string {
edges := make([]string, 0, 1)
if m.clearedowner {
edges = append(edges, event.EdgeOwner)
}
return edges
}
2021-01-14 15:27:45 +00:00
// EdgeCleared returns a boolean which indicates if the edge with the given name
// was cleared in this mutation.
func (m *EventMutation) EdgeCleared(name string) bool {
switch name {
case event.EdgeOwner:
return m.clearedowner
}
return false
}
2021-01-14 15:27:45 +00:00
// ClearEdge clears the value of the edge with the given name. It returns an error
// if that edge is not defined in the schema.
func (m *EventMutation) ClearEdge(name string) error {
switch name {
case event.EdgeOwner:
m.ClearOwner()
return nil
}
return fmt.Errorf("unknown Event unique edge %s", name)
}
2021-01-14 15:27:45 +00:00
// ResetEdge resets all changes to the edge with the given name in this mutation.
// It returns an error if the edge is not defined in the schema.
func (m *EventMutation) ResetEdge(name string) error {
switch name {
case event.EdgeOwner:
m.ResetOwner()
return nil
}
return fmt.Errorf("unknown Event edge %s", name)
}
2021-01-14 15:27:45 +00:00
// MachineMutation represents an operation that mutates the Machine nodes in the graph.
type MachineMutation struct {
config
op Op
typ string
id *int
created_at *time.Time
updated_at *time.Time
machineId *string
password *string
ipAddress *string
scenarios *string
version *string
isValidated *bool
status *string
clearedFields map[string]struct{}
alerts map[int]struct{}
removedalerts map[int]struct{}
clearedalerts bool
done bool
oldValue func(context.Context) (*Machine, error)
2021-01-14 15:27:45 +00:00
predicates []predicate.Machine
}
var _ ent.Mutation = (*MachineMutation)(nil)
2021-01-14 15:27:45 +00:00
// machineOption allows management of the mutation configuration using functional options.
type machineOption func(*MachineMutation)
2021-01-14 15:27:45 +00:00
// newMachineMutation creates new mutation for the Machine entity.
func newMachineMutation(c config, op Op, opts ...machineOption) *MachineMutation {
m := &MachineMutation{
config: c,
op: op,
typ: TypeMachine,
clearedFields: make(map[string]struct{}),
}
for _, opt := range opts {
opt(m)
}
return m
}
2021-01-14 15:27:45 +00:00
// withMachineID sets the ID field of the mutation.
func withMachineID(id int) machineOption {
return func(m *MachineMutation) {
var (
err error
once sync.Once
value *Machine
)
m.oldValue = func(ctx context.Context) (*Machine, error) {
once.Do(func() {
if m.done {
err = fmt.Errorf("querying old values post mutation is not allowed")
} else {
value, err = m.Client().Machine.Get(ctx, id)
}
})
return value, err
}
m.id = &id
}
}
// withMachine sets the old Machine of the mutation.
func withMachine(node *Machine) machineOption {
return func(m *MachineMutation) {
m.oldValue = func(context.Context) (*Machine, error) {
return node, nil
}
m.id = &node.ID
}
}
// Client returns a new `ent.Client` from the mutation. If the mutation was
// executed in a transaction (ent.Tx), a transactional client is returned.
func (m MachineMutation) Client() *Client {
client := &Client{config: m.config}
client.init()
return client
}
// Tx returns an `ent.Tx` for mutations that were executed in transactions;
// it returns an error otherwise.
func (m MachineMutation) Tx() (*Tx, error) {
if _, ok := m.driver.(*txDriver); !ok {
return nil, fmt.Errorf("ent: mutation is not running in a transaction")
}
tx := &Tx{config: m.config}
tx.init()
return tx, nil
}
2021-01-14 15:27:45 +00:00
// ID returns the ID value in the mutation. Note that the ID
// is only available if it was provided to the builder.
func (m *MachineMutation) ID() (id int, exists bool) {
if m.id == nil {
return
}
return *m.id, true
}
2021-01-14 15:27:45 +00:00
// SetCreatedAt sets the "created_at" field.
func (m *MachineMutation) SetCreatedAt(t time.Time) {
m.created_at = &t
}
2021-01-14 15:27:45 +00:00
// CreatedAt returns the value of the "created_at" field in the mutation.
func (m *MachineMutation) CreatedAt() (r time.Time, exists bool) {
v := m.created_at
if v == nil {
return
}
return *v, true
}
2021-01-14 15:27:45 +00:00
// OldCreatedAt returns the old "created_at" field's value of the Machine entity.
// If the Machine object wasn't provided to the builder, the object is fetched from the database.
// An error is returned if the mutation operation is not UpdateOne, or the database query fails.
func (m *MachineMutation) OldCreatedAt(ctx context.Context) (v time.Time, err error) {
if !m.op.Is(OpUpdateOne) {
2021-01-14 15:27:45 +00:00
return v, fmt.Errorf("OldCreatedAt is only allowed on UpdateOne operations")
}
if m.id == nil || m.oldValue == nil {
return v, fmt.Errorf("OldCreatedAt requires an ID field in the mutation")
}
oldValue, err := m.oldValue(ctx)
if err != nil {
return v, fmt.Errorf("querying old value for OldCreatedAt: %w", err)
}
return oldValue.CreatedAt, nil
}
2021-01-14 15:27:45 +00:00
// ResetCreatedAt resets all changes to the "created_at" field.
func (m *MachineMutation) ResetCreatedAt() {
m.created_at = nil
}
2021-01-14 15:27:45 +00:00
// SetUpdatedAt sets the "updated_at" field.
func (m *MachineMutation) SetUpdatedAt(t time.Time) {
m.updated_at = &t
}
2021-01-14 15:27:45 +00:00
// UpdatedAt returns the value of the "updated_at" field in the mutation.
func (m *MachineMutation) UpdatedAt() (r time.Time, exists bool) {
v := m.updated_at
if v == nil {
return
}
return *v, true
}
2021-01-14 15:27:45 +00:00
// OldUpdatedAt returns the old "updated_at" field's value of the Machine entity.
// If the Machine object wasn't provided to the builder, the object is fetched from the database.
// An error is returned if the mutation operation is not UpdateOne, or the database query fails.
func (m *MachineMutation) OldUpdatedAt(ctx context.Context) (v time.Time, err error) {
if !m.op.Is(OpUpdateOne) {
2021-01-14 15:27:45 +00:00
return v, fmt.Errorf("OldUpdatedAt is only allowed on UpdateOne operations")
}
if m.id == nil || m.oldValue == nil {
return v, fmt.Errorf("OldUpdatedAt requires an ID field in the mutation")
}
oldValue, err := m.oldValue(ctx)
if err != nil {
return v, fmt.Errorf("querying old value for OldUpdatedAt: %w", err)
}
return oldValue.UpdatedAt, nil
}
2021-01-14 15:27:45 +00:00
// ResetUpdatedAt resets all changes to the "updated_at" field.
func (m *MachineMutation) ResetUpdatedAt() {
m.updated_at = nil
}
2021-01-14 15:27:45 +00:00
// SetMachineId sets the "machineId" field.
func (m *MachineMutation) SetMachineId(s string) {
m.machineId = &s
}
2021-01-14 15:27:45 +00:00
// MachineId returns the value of the "machineId" field in the mutation.
func (m *MachineMutation) MachineId() (r string, exists bool) {
v := m.machineId
if v == nil {
return
}
return *v, true
}
2021-01-14 15:27:45 +00:00
// OldMachineId returns the old "machineId" field's value of the Machine entity.
// If the Machine object wasn't provided to the builder, the object is fetched from the database.
// An error is returned if the mutation operation is not UpdateOne, or the database query fails.
func (m *MachineMutation) OldMachineId(ctx context.Context) (v string, err error) {
if !m.op.Is(OpUpdateOne) {
2021-01-14 15:27:45 +00:00
return v, fmt.Errorf("OldMachineId is only allowed on UpdateOne operations")
}
if m.id == nil || m.oldValue == nil {
return v, fmt.Errorf("OldMachineId requires an ID field in the mutation")
}
oldValue, err := m.oldValue(ctx)
if err != nil {
return v, fmt.Errorf("querying old value for OldMachineId: %w", err)
}
return oldValue.MachineId, nil
}
2021-01-14 15:27:45 +00:00
// ResetMachineId resets all changes to the "machineId" field.
func (m *MachineMutation) ResetMachineId() {
m.machineId = nil
}
2021-01-14 15:27:45 +00:00
// SetPassword sets the "password" field.
func (m *MachineMutation) SetPassword(s string) {
m.password = &s
}
2021-01-14 15:27:45 +00:00
// Password returns the value of the "password" field in the mutation.
func (m *MachineMutation) Password() (r string, exists bool) {
v := m.password
if v == nil {
return
}
return *v, true
}
2021-01-14 15:27:45 +00:00
// OldPassword returns the old "password" field's value of the Machine entity.
// If the Machine object wasn't provided to the builder, the object is fetched from the database.
// An error is returned if the mutation operation is not UpdateOne, or the database query fails.
func (m *MachineMutation) OldPassword(ctx context.Context) (v string, err error) {
if !m.op.Is(OpUpdateOne) {
2021-01-14 15:27:45 +00:00
return v, fmt.Errorf("OldPassword is only allowed on UpdateOne operations")
}
if m.id == nil || m.oldValue == nil {
return v, fmt.Errorf("OldPassword requires an ID field in the mutation")
}
oldValue, err := m.oldValue(ctx)
if err != nil {
return v, fmt.Errorf("querying old value for OldPassword: %w", err)
}
return oldValue.Password, nil
}
2021-01-14 15:27:45 +00:00
// ResetPassword resets all changes to the "password" field.
func (m *MachineMutation) ResetPassword() {
m.password = nil
}
2021-01-14 15:27:45 +00:00
// SetIpAddress sets the "ipAddress" field.
func (m *MachineMutation) SetIpAddress(s string) {
m.ipAddress = &s
}
2021-01-14 15:27:45 +00:00
// IpAddress returns the value of the "ipAddress" field in the mutation.
func (m *MachineMutation) IpAddress() (r string, exists bool) {
v := m.ipAddress
if v == nil {
return
}
return *v, true
}
2021-01-14 15:27:45 +00:00
// OldIpAddress returns the old "ipAddress" field's value of the Machine entity.
// If the Machine object wasn't provided to the builder, the object is fetched from the database.
// An error is returned if the mutation operation is not UpdateOne, or the database query fails.
func (m *MachineMutation) OldIpAddress(ctx context.Context) (v string, err error) {
if !m.op.Is(OpUpdateOne) {
2021-01-14 15:27:45 +00:00
return v, fmt.Errorf("OldIpAddress is only allowed on UpdateOne operations")
}
if m.id == nil || m.oldValue == nil {
return v, fmt.Errorf("OldIpAddress requires an ID field in the mutation")
}
oldValue, err := m.oldValue(ctx)
if err != nil {
return v, fmt.Errorf("querying old value for OldIpAddress: %w", err)
}
return oldValue.IpAddress, nil
}
2021-01-14 15:27:45 +00:00
// ResetIpAddress resets all changes to the "ipAddress" field.
func (m *MachineMutation) ResetIpAddress() {
m.ipAddress = nil
}
2021-01-14 15:27:45 +00:00
// SetScenarios sets the "scenarios" field.
func (m *MachineMutation) SetScenarios(s string) {
m.scenarios = &s
}
2021-01-14 15:27:45 +00:00
// Scenarios returns the value of the "scenarios" field in the mutation.
func (m *MachineMutation) Scenarios() (r string, exists bool) {
v := m.scenarios
if v == nil {
return
}
return *v, true
}
2021-01-14 15:27:45 +00:00
// OldScenarios returns the old "scenarios" field's value of the Machine entity.
// If the Machine object wasn't provided to the builder, the object is fetched from the database.
// An error is returned if the mutation operation is not UpdateOne, or the database query fails.
func (m *MachineMutation) OldScenarios(ctx context.Context) (v string, err error) {
if !m.op.Is(OpUpdateOne) {
2021-01-14 15:27:45 +00:00
return v, fmt.Errorf("OldScenarios is only allowed on UpdateOne operations")
}
if m.id == nil || m.oldValue == nil {
return v, fmt.Errorf("OldScenarios requires an ID field in the mutation")
}
oldValue, err := m.oldValue(ctx)
if err != nil {
return v, fmt.Errorf("querying old value for OldScenarios: %w", err)
}
return oldValue.Scenarios, nil
}
2021-01-14 15:27:45 +00:00
// ClearScenarios clears the value of the "scenarios" field.
func (m *MachineMutation) ClearScenarios() {
m.scenarios = nil
m.clearedFields[machine.FieldScenarios] = struct{}{}
}
2021-01-14 15:27:45 +00:00
// ScenariosCleared returns if the "scenarios" field was cleared in this mutation.
func (m *MachineMutation) ScenariosCleared() bool {
_, ok := m.clearedFields[machine.FieldScenarios]
return ok
}
2021-01-14 15:27:45 +00:00
// ResetScenarios resets all changes to the "scenarios" field.
func (m *MachineMutation) ResetScenarios() {
m.scenarios = nil
delete(m.clearedFields, machine.FieldScenarios)
}
2021-01-14 15:27:45 +00:00
// SetVersion sets the "version" field.
func (m *MachineMutation) SetVersion(s string) {
m.version = &s
}
2021-01-14 15:27:45 +00:00
// Version returns the value of the "version" field in the mutation.
func (m *MachineMutation) Version() (r string, exists bool) {
v := m.version
if v == nil {
return
}
return *v, true
}
2021-01-14 15:27:45 +00:00
// OldVersion returns the old "version" field's value of the Machine entity.
// If the Machine object wasn't provided to the builder, the object is fetched from the database.
// An error is returned if the mutation operation is not UpdateOne, or the database query fails.
func (m *MachineMutation) OldVersion(ctx context.Context) (v string, err error) {
if !m.op.Is(OpUpdateOne) {
2021-01-14 15:27:45 +00:00
return v, fmt.Errorf("OldVersion is only allowed on UpdateOne operations")
}
if m.id == nil || m.oldValue == nil {
return v, fmt.Errorf("OldVersion requires an ID field in the mutation")
}
oldValue, err := m.oldValue(ctx)
if err != nil {
return v, fmt.Errorf("querying old value for OldVersion: %w", err)
}
return oldValue.Version, nil
}
2021-01-14 15:27:45 +00:00
// ClearVersion clears the value of the "version" field.
func (m *MachineMutation) ClearVersion() {
m.version = nil
m.clearedFields[machine.FieldVersion] = struct{}{}
}
2021-01-14 15:27:45 +00:00
// VersionCleared returns if the "version" field was cleared in this mutation.
func (m *MachineMutation) VersionCleared() bool {
_, ok := m.clearedFields[machine.FieldVersion]
return ok
}
2021-01-14 15:27:45 +00:00
// ResetVersion resets all changes to the "version" field.
func (m *MachineMutation) ResetVersion() {
m.version = nil
delete(m.clearedFields, machine.FieldVersion)
}
2021-01-14 15:27:45 +00:00
// SetIsValidated sets the "isValidated" field.
func (m *MachineMutation) SetIsValidated(b bool) {
m.isValidated = &b
}
2021-01-14 15:27:45 +00:00
// IsValidated returns the value of the "isValidated" field in the mutation.
func (m *MachineMutation) IsValidated() (r bool, exists bool) {
v := m.isValidated
if v == nil {
return
}
return *v, true
}
2021-01-14 15:27:45 +00:00
// OldIsValidated returns the old "isValidated" field's value of the Machine entity.
// If the Machine object wasn't provided to the builder, the object is fetched from the database.
// An error is returned if the mutation operation is not UpdateOne, or the database query fails.
func (m *MachineMutation) OldIsValidated(ctx context.Context) (v bool, err error) {
if !m.op.Is(OpUpdateOne) {
2021-01-14 15:27:45 +00:00
return v, fmt.Errorf("OldIsValidated is only allowed on UpdateOne operations")
}
if m.id == nil || m.oldValue == nil {
return v, fmt.Errorf("OldIsValidated requires an ID field in the mutation")
}
oldValue, err := m.oldValue(ctx)
if err != nil {
return v, fmt.Errorf("querying old value for OldIsValidated: %w", err)
}
return oldValue.IsValidated, nil
}
2021-01-14 15:27:45 +00:00
// ResetIsValidated resets all changes to the "isValidated" field.
func (m *MachineMutation) ResetIsValidated() {
m.isValidated = nil
}
2021-01-14 15:27:45 +00:00
// SetStatus sets the "status" field.
func (m *MachineMutation) SetStatus(s string) {
m.status = &s
}
2021-01-14 15:27:45 +00:00
// Status returns the value of the "status" field in the mutation.
func (m *MachineMutation) Status() (r string, exists bool) {
v := m.status
if v == nil {
return
}
return *v, true
}
2021-01-14 15:27:45 +00:00
// OldStatus returns the old "status" field's value of the Machine entity.
// If the Machine object wasn't provided to the builder, the object is fetched from the database.
// An error is returned if the mutation operation is not UpdateOne, or the database query fails.
func (m *MachineMutation) OldStatus(ctx context.Context) (v string, err error) {
if !m.op.Is(OpUpdateOne) {
2021-01-14 15:27:45 +00:00
return v, fmt.Errorf("OldStatus is only allowed on UpdateOne operations")
}
if m.id == nil || m.oldValue == nil {
return v, fmt.Errorf("OldStatus requires an ID field in the mutation")
}
oldValue, err := m.oldValue(ctx)
if err != nil {
return v, fmt.Errorf("querying old value for OldStatus: %w", err)
}
return oldValue.Status, nil
}
2021-01-14 15:27:45 +00:00
// ClearStatus clears the value of the "status" field.
func (m *MachineMutation) ClearStatus() {
m.status = nil
m.clearedFields[machine.FieldStatus] = struct{}{}
}
2021-01-14 15:27:45 +00:00
// StatusCleared returns if the "status" field was cleared in this mutation.
func (m *MachineMutation) StatusCleared() bool {
_, ok := m.clearedFields[machine.FieldStatus]
return ok
}
2021-01-14 15:27:45 +00:00
// ResetStatus resets all changes to the "status" field.
func (m *MachineMutation) ResetStatus() {
m.status = nil
delete(m.clearedFields, machine.FieldStatus)
}
2021-01-14 15:27:45 +00:00
// AddAlertIDs adds the "alerts" edge to the Alert entity by ids.
func (m *MachineMutation) AddAlertIDs(ids ...int) {
if m.alerts == nil {
m.alerts = make(map[int]struct{})
}
for i := range ids {
m.alerts[ids[i]] = struct{}{}
}
}
2021-01-14 15:27:45 +00:00
// ClearAlerts clears the "alerts" edge to the Alert entity.
func (m *MachineMutation) ClearAlerts() {
m.clearedalerts = true
}
2021-01-14 15:27:45 +00:00
// AlertsCleared returns if the "alerts" edge to the Alert entity was cleared.
func (m *MachineMutation) AlertsCleared() bool {
return m.clearedalerts
}
2021-01-14 15:27:45 +00:00
// RemoveAlertIDs removes the "alerts" edge to the Alert entity by IDs.
func (m *MachineMutation) RemoveAlertIDs(ids ...int) {
if m.removedalerts == nil {
m.removedalerts = make(map[int]struct{})
}
for i := range ids {
m.removedalerts[ids[i]] = struct{}{}
}
}
2021-01-14 15:27:45 +00:00
// RemovedAlerts returns the removed IDs of the "alerts" edge to the Alert entity.
func (m *MachineMutation) RemovedAlertsIDs() (ids []int) {
for id := range m.removedalerts {
ids = append(ids, id)
}
return
}
2021-01-14 15:27:45 +00:00
// AlertsIDs returns the "alerts" edge IDs in the mutation.
func (m *MachineMutation) AlertsIDs() (ids []int) {
for id := range m.alerts {
ids = append(ids, id)
}
return
}
2021-01-14 15:27:45 +00:00
// ResetAlerts resets all changes to the "alerts" edge.
func (m *MachineMutation) ResetAlerts() {
m.alerts = nil
m.clearedalerts = false
m.removedalerts = nil
}
// Op returns the operation name.
func (m *MachineMutation) Op() Op {
return m.op
}
// Type returns the node type of this mutation (Machine).
func (m *MachineMutation) Type() string {
return m.typ
}
2021-01-14 15:27:45 +00:00
// Fields returns all fields that were changed during this mutation. Note that in
// order to get all numeric fields that were incremented/decremented, call
// AddedFields().
func (m *MachineMutation) Fields() []string {
fields := make([]string, 0, 9)
if m.created_at != nil {
fields = append(fields, machine.FieldCreatedAt)
}
if m.updated_at != nil {
fields = append(fields, machine.FieldUpdatedAt)
}
if m.machineId != nil {
fields = append(fields, machine.FieldMachineId)
}
if m.password != nil {
fields = append(fields, machine.FieldPassword)
}
if m.ipAddress != nil {
fields = append(fields, machine.FieldIpAddress)
}
if m.scenarios != nil {
fields = append(fields, machine.FieldScenarios)
}
if m.version != nil {
fields = append(fields, machine.FieldVersion)
}
if m.isValidated != nil {
fields = append(fields, machine.FieldIsValidated)
}
if m.status != nil {
fields = append(fields, machine.FieldStatus)
}
return fields
}
2021-01-14 15:27:45 +00:00
// Field returns the value of a field with the given name. The second boolean
// return value indicates that this field was not set, or was not defined in the
// schema.
func (m *MachineMutation) Field(name string) (ent.Value, bool) {
switch name {
case machine.FieldCreatedAt:
return m.CreatedAt()
case machine.FieldUpdatedAt:
return m.UpdatedAt()
case machine.FieldMachineId:
return m.MachineId()
case machine.FieldPassword:
return m.Password()
case machine.FieldIpAddress:
return m.IpAddress()
case machine.FieldScenarios:
return m.Scenarios()
case machine.FieldVersion:
return m.Version()
case machine.FieldIsValidated:
return m.IsValidated()
case machine.FieldStatus:
return m.Status()
}
return nil, false
}
2021-01-14 15:27:45 +00:00
// OldField returns the old value of the field from the database. An error is
// returned if the mutation operation is not UpdateOne, or the query to the
// database failed.
func (m *MachineMutation) OldField(ctx context.Context, name string) (ent.Value, error) {
switch name {
case machine.FieldCreatedAt:
return m.OldCreatedAt(ctx)
case machine.FieldUpdatedAt:
return m.OldUpdatedAt(ctx)
case machine.FieldMachineId:
return m.OldMachineId(ctx)
case machine.FieldPassword:
return m.OldPassword(ctx)
case machine.FieldIpAddress:
return m.OldIpAddress(ctx)
case machine.FieldScenarios:
return m.OldScenarios(ctx)
case machine.FieldVersion:
return m.OldVersion(ctx)
case machine.FieldIsValidated:
return m.OldIsValidated(ctx)
case machine.FieldStatus:
return m.OldStatus(ctx)
}
return nil, fmt.Errorf("unknown Machine field %s", name)
}
2021-01-14 15:27:45 +00:00
// SetField sets the value of a field with the given name. It returns an error if
// the field is not defined in the schema, or if the type mismatched the field
// type.
func (m *MachineMutation) SetField(name string, value ent.Value) error {
switch name {
case machine.FieldCreatedAt:
v, ok := value.(time.Time)
if !ok {
return fmt.Errorf("unexpected type %T for field %s", value, name)
}
m.SetCreatedAt(v)
return nil
case machine.FieldUpdatedAt:
v, ok := value.(time.Time)
if !ok {
return fmt.Errorf("unexpected type %T for field %s", value, name)
}
m.SetUpdatedAt(v)
return nil
case machine.FieldMachineId:
v, ok := value.(string)
if !ok {
return fmt.Errorf("unexpected type %T for field %s", value, name)
}
m.SetMachineId(v)
return nil
case machine.FieldPassword:
v, ok := value.(string)
if !ok {
return fmt.Errorf("unexpected type %T for field %s", value, name)
}
m.SetPassword(v)
return nil
case machine.FieldIpAddress:
v, ok := value.(string)
if !ok {
return fmt.Errorf("unexpected type %T for field %s", value, name)
}
m.SetIpAddress(v)
return nil
case machine.FieldScenarios:
v, ok := value.(string)
if !ok {
return fmt.Errorf("unexpected type %T for field %s", value, name)
}
m.SetScenarios(v)
return nil
case machine.FieldVersion:
v, ok := value.(string)
if !ok {
return fmt.Errorf("unexpected type %T for field %s", value, name)
}
m.SetVersion(v)
return nil
case machine.FieldIsValidated:
v, ok := value.(bool)
if !ok {
return fmt.Errorf("unexpected type %T for field %s", value, name)
}
m.SetIsValidated(v)
return nil
case machine.FieldStatus:
v, ok := value.(string)
if !ok {
return fmt.Errorf("unexpected type %T for field %s", value, name)
}
m.SetStatus(v)
return nil
}
return fmt.Errorf("unknown Machine field %s", name)
}
2021-01-14 15:27:45 +00:00
// AddedFields returns all numeric fields that were incremented/decremented during
// this mutation.
func (m *MachineMutation) AddedFields() []string {
return nil
}
2021-01-14 15:27:45 +00:00
// AddedField returns the numeric value that was incremented/decremented on a field
// with the given name. The second boolean return value indicates that this field
// was not set, or was not defined in the schema.
func (m *MachineMutation) AddedField(name string) (ent.Value, bool) {
return nil, false
}
2021-01-14 15:27:45 +00:00
// AddField adds the value to the field with the given name. It returns an error if
// the field is not defined in the schema, or if the type mismatched the field
// type.
func (m *MachineMutation) AddField(name string, value ent.Value) error {
switch name {
}
return fmt.Errorf("unknown Machine numeric field %s", name)
}
2021-01-14 15:27:45 +00:00
// ClearedFields returns all nullable fields that were cleared during this
// mutation.
func (m *MachineMutation) ClearedFields() []string {
var fields []string
if m.FieldCleared(machine.FieldScenarios) {
fields = append(fields, machine.FieldScenarios)
}
if m.FieldCleared(machine.FieldVersion) {
fields = append(fields, machine.FieldVersion)
}
if m.FieldCleared(machine.FieldStatus) {
fields = append(fields, machine.FieldStatus)
}
return fields
}
2021-01-14 15:27:45 +00:00
// FieldCleared returns a boolean indicating if a field with the given name was
// cleared in this mutation.
func (m *MachineMutation) FieldCleared(name string) bool {
_, ok := m.clearedFields[name]
return ok
}
2021-01-14 15:27:45 +00:00
// ClearField clears the value of the field with the given name. It returns an
// error if the field is not defined in the schema.
func (m *MachineMutation) ClearField(name string) error {
switch name {
case machine.FieldScenarios:
m.ClearScenarios()
return nil
case machine.FieldVersion:
m.ClearVersion()
return nil
case machine.FieldStatus:
m.ClearStatus()
return nil
}
return fmt.Errorf("unknown Machine nullable field %s", name)
}
2021-01-14 15:27:45 +00:00
// ResetField resets all changes in the mutation for the field with the given name.
// It returns an error if the field is not defined in the schema.
func (m *MachineMutation) ResetField(name string) error {
switch name {
case machine.FieldCreatedAt:
m.ResetCreatedAt()
return nil
case machine.FieldUpdatedAt:
m.ResetUpdatedAt()
return nil
case machine.FieldMachineId:
m.ResetMachineId()
return nil
case machine.FieldPassword:
m.ResetPassword()
return nil
case machine.FieldIpAddress:
m.ResetIpAddress()
return nil
case machine.FieldScenarios:
m.ResetScenarios()
return nil
case machine.FieldVersion:
m.ResetVersion()
return nil
case machine.FieldIsValidated:
m.ResetIsValidated()
return nil
case machine.FieldStatus:
m.ResetStatus()
return nil
}
return fmt.Errorf("unknown Machine field %s", name)
}
2021-01-14 15:27:45 +00:00
// AddedEdges returns all edge names that were set/added in this mutation.
func (m *MachineMutation) AddedEdges() []string {
edges := make([]string, 0, 1)
if m.alerts != nil {
edges = append(edges, machine.EdgeAlerts)
}
return edges
}
2021-01-14 15:27:45 +00:00
// AddedIDs returns all IDs (to other nodes) that were added for the given edge
// name in this mutation.
func (m *MachineMutation) AddedIDs(name string) []ent.Value {
switch name {
case machine.EdgeAlerts:
ids := make([]ent.Value, 0, len(m.alerts))
for id := range m.alerts {
ids = append(ids, id)
}
return ids
}
return nil
}
2021-01-14 15:27:45 +00:00
// RemovedEdges returns all edge names that were removed in this mutation.
func (m *MachineMutation) RemovedEdges() []string {
edges := make([]string, 0, 1)
if m.removedalerts != nil {
edges = append(edges, machine.EdgeAlerts)
}
return edges
}
2021-01-14 15:27:45 +00:00
// RemovedIDs returns all IDs (to other nodes) that were removed for the edge with
// the given name in this mutation.
func (m *MachineMutation) RemovedIDs(name string) []ent.Value {
switch name {
case machine.EdgeAlerts:
ids := make([]ent.Value, 0, len(m.removedalerts))
for id := range m.removedalerts {
ids = append(ids, id)
}
return ids
}
return nil
}
2021-01-14 15:27:45 +00:00
// ClearedEdges returns all edge names that were cleared in this mutation.
func (m *MachineMutation) ClearedEdges() []string {
edges := make([]string, 0, 1)
if m.clearedalerts {
edges = append(edges, machine.EdgeAlerts)
}
return edges
}
2021-01-14 15:27:45 +00:00
// EdgeCleared returns a boolean which indicates if the edge with the given name
// was cleared in this mutation.
func (m *MachineMutation) EdgeCleared(name string) bool {
switch name {
case machine.EdgeAlerts:
return m.clearedalerts
}
return false
}
2021-01-14 15:27:45 +00:00
// ClearEdge clears the value of the edge with the given name. It returns an error
// if that edge is not defined in the schema.
func (m *MachineMutation) ClearEdge(name string) error {
switch name {
}
return fmt.Errorf("unknown Machine unique edge %s", name)
}
2021-01-14 15:27:45 +00:00
// ResetEdge resets all changes to the edge with the given name in this mutation.
// It returns an error if the edge is not defined in the schema.
func (m *MachineMutation) ResetEdge(name string) error {
switch name {
case machine.EdgeAlerts:
m.ResetAlerts()
return nil
}
return fmt.Errorf("unknown Machine edge %s", name)
}
2021-01-14 15:27:45 +00:00
// MetaMutation represents an operation that mutates the Meta nodes in the graph.
type MetaMutation struct {
config
op Op
typ string
id *int
created_at *time.Time
updated_at *time.Time
key *string
value *string
clearedFields map[string]struct{}
owner *int
clearedowner bool
done bool
oldValue func(context.Context) (*Meta, error)
2021-01-14 15:27:45 +00:00
predicates []predicate.Meta
}
var _ ent.Mutation = (*MetaMutation)(nil)
2021-01-14 15:27:45 +00:00
// metaOption allows management of the mutation configuration using functional options.
type metaOption func(*MetaMutation)
2021-01-14 15:27:45 +00:00
// newMetaMutation creates new mutation for the Meta entity.
func newMetaMutation(c config, op Op, opts ...metaOption) *MetaMutation {
m := &MetaMutation{
config: c,
op: op,
typ: TypeMeta,
clearedFields: make(map[string]struct{}),
}
for _, opt := range opts {
opt(m)
}
return m
}
2021-01-14 15:27:45 +00:00
// withMetaID sets the ID field of the mutation.
func withMetaID(id int) metaOption {
return func(m *MetaMutation) {
var (
err error
once sync.Once
value *Meta
)
m.oldValue = func(ctx context.Context) (*Meta, error) {
once.Do(func() {
if m.done {
err = fmt.Errorf("querying old values post mutation is not allowed")
} else {
value, err = m.Client().Meta.Get(ctx, id)
}
})
return value, err
}
m.id = &id
}
}
// withMeta sets the old Meta of the mutation.
func withMeta(node *Meta) metaOption {
return func(m *MetaMutation) {
m.oldValue = func(context.Context) (*Meta, error) {
return node, nil
}
m.id = &node.ID
}
}
// Client returns a new `ent.Client` from the mutation. If the mutation was
// executed in a transaction (ent.Tx), a transactional client is returned.
func (m MetaMutation) Client() *Client {
client := &Client{config: m.config}
client.init()
return client
}
// Tx returns an `ent.Tx` for mutations that were executed in transactions;
// it returns an error otherwise.
func (m MetaMutation) Tx() (*Tx, error) {
if _, ok := m.driver.(*txDriver); !ok {
return nil, fmt.Errorf("ent: mutation is not running in a transaction")
}
tx := &Tx{config: m.config}
tx.init()
return tx, nil
}
2021-01-14 15:27:45 +00:00
// ID returns the ID value in the mutation. Note that the ID
// is only available if it was provided to the builder.
func (m *MetaMutation) ID() (id int, exists bool) {
if m.id == nil {
return
}
return *m.id, true
}
2021-01-14 15:27:45 +00:00
// SetCreatedAt sets the "created_at" field.
func (m *MetaMutation) SetCreatedAt(t time.Time) {
m.created_at = &t
}
2021-01-14 15:27:45 +00:00
// CreatedAt returns the value of the "created_at" field in the mutation.
func (m *MetaMutation) CreatedAt() (r time.Time, exists bool) {
v := m.created_at
if v == nil {
return
}
return *v, true
}
2021-01-14 15:27:45 +00:00
// OldCreatedAt returns the old "created_at" field's value of the Meta entity.
// If the Meta object wasn't provided to the builder, the object is fetched from the database.
// An error is returned if the mutation operation is not UpdateOne, or the database query fails.
func (m *MetaMutation) OldCreatedAt(ctx context.Context) (v time.Time, err error) {
if !m.op.Is(OpUpdateOne) {
2021-01-14 15:27:45 +00:00
return v, fmt.Errorf("OldCreatedAt is only allowed on UpdateOne operations")
}
if m.id == nil || m.oldValue == nil {
return v, fmt.Errorf("OldCreatedAt requires an ID field in the mutation")
}
oldValue, err := m.oldValue(ctx)
if err != nil {
return v, fmt.Errorf("querying old value for OldCreatedAt: %w", err)
}
return oldValue.CreatedAt, nil
}
2021-01-14 15:27:45 +00:00
// ResetCreatedAt resets all changes to the "created_at" field.
func (m *MetaMutation) ResetCreatedAt() {
m.created_at = nil
}
2021-01-14 15:27:45 +00:00
// SetUpdatedAt sets the "updated_at" field.
func (m *MetaMutation) SetUpdatedAt(t time.Time) {
m.updated_at = &t
}
2021-01-14 15:27:45 +00:00
// UpdatedAt returns the value of the "updated_at" field in the mutation.
func (m *MetaMutation) UpdatedAt() (r time.Time, exists bool) {
v := m.updated_at
if v == nil {
return
}
return *v, true
}
2021-01-14 15:27:45 +00:00
// OldUpdatedAt returns the old "updated_at" field's value of the Meta entity.
// If the Meta object wasn't provided to the builder, the object is fetched from the database.
// An error is returned if the mutation operation is not UpdateOne, or the database query fails.
func (m *MetaMutation) OldUpdatedAt(ctx context.Context) (v time.Time, err error) {
if !m.op.Is(OpUpdateOne) {
2021-01-14 15:27:45 +00:00
return v, fmt.Errorf("OldUpdatedAt is only allowed on UpdateOne operations")
}
if m.id == nil || m.oldValue == nil {
return v, fmt.Errorf("OldUpdatedAt requires an ID field in the mutation")
}
oldValue, err := m.oldValue(ctx)
if err != nil {
return v, fmt.Errorf("querying old value for OldUpdatedAt: %w", err)
}
return oldValue.UpdatedAt, nil
}
2021-01-14 15:27:45 +00:00
// ResetUpdatedAt resets all changes to the "updated_at" field.
func (m *MetaMutation) ResetUpdatedAt() {
m.updated_at = nil
}
2021-01-14 15:27:45 +00:00
// SetKey sets the "key" field.
func (m *MetaMutation) SetKey(s string) {
m.key = &s
}
2021-01-14 15:27:45 +00:00
// Key returns the value of the "key" field in the mutation.
func (m *MetaMutation) Key() (r string, exists bool) {
v := m.key
if v == nil {
return
}
return *v, true
}
2021-01-14 15:27:45 +00:00
// OldKey returns the old "key" field's value of the Meta entity.
// If the Meta object wasn't provided to the builder, the object is fetched from the database.
// An error is returned if the mutation operation is not UpdateOne, or the database query fails.
func (m *MetaMutation) OldKey(ctx context.Context) (v string, err error) {
if !m.op.Is(OpUpdateOne) {
2021-01-14 15:27:45 +00:00
return v, fmt.Errorf("OldKey is only allowed on UpdateOne operations")
}
if m.id == nil || m.oldValue == nil {
return v, fmt.Errorf("OldKey requires an ID field in the mutation")
}
oldValue, err := m.oldValue(ctx)
if err != nil {
return v, fmt.Errorf("querying old value for OldKey: %w", err)
}
return oldValue.Key, nil
}
2021-01-14 15:27:45 +00:00
// ResetKey resets all changes to the "key" field.
func (m *MetaMutation) ResetKey() {
m.key = nil
}
2021-01-14 15:27:45 +00:00
// SetValue sets the "value" field.
func (m *MetaMutation) SetValue(s string) {
m.value = &s
}
2021-01-14 15:27:45 +00:00
// Value returns the value of the "value" field in the mutation.
func (m *MetaMutation) Value() (r string, exists bool) {
v := m.value
if v == nil {
return
}
return *v, true
}
2021-01-14 15:27:45 +00:00
// OldValue returns the old "value" field's value of the Meta entity.
// If the Meta object wasn't provided to the builder, the object is fetched from the database.
// An error is returned if the mutation operation is not UpdateOne, or the database query fails.
func (m *MetaMutation) OldValue(ctx context.Context) (v string, err error) {
if !m.op.Is(OpUpdateOne) {
2021-01-14 15:27:45 +00:00
return v, fmt.Errorf("OldValue is only allowed on UpdateOne operations")
}
if m.id == nil || m.oldValue == nil {
return v, fmt.Errorf("OldValue requires an ID field in the mutation")
}
oldValue, err := m.oldValue(ctx)
if err != nil {
return v, fmt.Errorf("querying old value for OldValue: %w", err)
}
return oldValue.Value, nil
}
2021-01-14 15:27:45 +00:00
// ResetValue resets all changes to the "value" field.
func (m *MetaMutation) ResetValue() {
m.value = nil
}
2021-01-14 15:27:45 +00:00
// SetOwnerID sets the "owner" edge to the Alert entity by id.
func (m *MetaMutation) SetOwnerID(id int) {
m.owner = &id
}
2021-01-14 15:27:45 +00:00
// ClearOwner clears the "owner" edge to the Alert entity.
func (m *MetaMutation) ClearOwner() {
m.clearedowner = true
}
2021-01-14 15:27:45 +00:00
// OwnerCleared returns if the "owner" edge to the Alert entity was cleared.
func (m *MetaMutation) OwnerCleared() bool {
return m.clearedowner
}
2021-01-14 15:27:45 +00:00
// OwnerID returns the "owner" edge ID in the mutation.
func (m *MetaMutation) OwnerID() (id int, exists bool) {
if m.owner != nil {
return *m.owner, true
}
return
}
2021-01-14 15:27:45 +00:00
// OwnerIDs returns the "owner" edge IDs in the mutation.
// Note that IDs always returns len(IDs) <= 1 for unique edges, and you should use
// OwnerID instead. It exists only for internal usage by the builders.
func (m *MetaMutation) OwnerIDs() (ids []int) {
if id := m.owner; id != nil {
ids = append(ids, *id)
}
return
}
2021-01-14 15:27:45 +00:00
// ResetOwner resets all changes to the "owner" edge.
func (m *MetaMutation) ResetOwner() {
m.owner = nil
m.clearedowner = false
}
// Op returns the operation name.
func (m *MetaMutation) Op() Op {
return m.op
}
// Type returns the node type of this mutation (Meta).
func (m *MetaMutation) Type() string {
return m.typ
}
2021-01-14 15:27:45 +00:00
// Fields returns all fields that were changed during this mutation. Note that in
// order to get all numeric fields that were incremented/decremented, call
// AddedFields().
func (m *MetaMutation) Fields() []string {
fields := make([]string, 0, 4)
if m.created_at != nil {
fields = append(fields, meta.FieldCreatedAt)
}
if m.updated_at != nil {
fields = append(fields, meta.FieldUpdatedAt)
}
if m.key != nil {
fields = append(fields, meta.FieldKey)
}
if m.value != nil {
fields = append(fields, meta.FieldValue)
}
return fields
}
2021-01-14 15:27:45 +00:00
// Field returns the value of a field with the given name. The second boolean
// return value indicates that this field was not set, or was not defined in the
// schema.
func (m *MetaMutation) Field(name string) (ent.Value, bool) {
switch name {
case meta.FieldCreatedAt:
return m.CreatedAt()
case meta.FieldUpdatedAt:
return m.UpdatedAt()
case meta.FieldKey:
return m.Key()
case meta.FieldValue:
return m.Value()
}
return nil, false
}
2021-01-14 15:27:45 +00:00
// OldField returns the old value of the field from the database. An error is
// returned if the mutation operation is not UpdateOne, or the query to the
// database failed.
func (m *MetaMutation) OldField(ctx context.Context, name string) (ent.Value, error) {
switch name {
case meta.FieldCreatedAt:
return m.OldCreatedAt(ctx)
case meta.FieldUpdatedAt:
return m.OldUpdatedAt(ctx)
case meta.FieldKey:
return m.OldKey(ctx)
case meta.FieldValue:
return m.OldValue(ctx)
}
return nil, fmt.Errorf("unknown Meta field %s", name)
}
2021-01-14 15:27:45 +00:00
// SetField sets the value of a field with the given name. It returns an error if
// the field is not defined in the schema, or if the type mismatched the field
// type.
func (m *MetaMutation) SetField(name string, value ent.Value) error {
switch name {
case meta.FieldCreatedAt:
v, ok := value.(time.Time)
if !ok {
return fmt.Errorf("unexpected type %T for field %s", value, name)
}
m.SetCreatedAt(v)
return nil
case meta.FieldUpdatedAt:
v, ok := value.(time.Time)
if !ok {
return fmt.Errorf("unexpected type %T for field %s", value, name)
}
m.SetUpdatedAt(v)
return nil
case meta.FieldKey:
v, ok := value.(string)
if !ok {
return fmt.Errorf("unexpected type %T for field %s", value, name)
}
m.SetKey(v)
return nil
case meta.FieldValue:
v, ok := value.(string)
if !ok {
return fmt.Errorf("unexpected type %T for field %s", value, name)
}
m.SetValue(v)
return nil
}
return fmt.Errorf("unknown Meta field %s", name)
}
2021-01-14 15:27:45 +00:00
// AddedFields returns all numeric fields that were incremented/decremented during
// this mutation.
func (m *MetaMutation) AddedFields() []string {
return nil
}
2021-01-14 15:27:45 +00:00
// AddedField returns the numeric value that was incremented/decremented on a field
// with the given name. The second boolean return value indicates that this field
// was not set, or was not defined in the schema.
func (m *MetaMutation) AddedField(name string) (ent.Value, bool) {
return nil, false
}
2021-01-14 15:27:45 +00:00
// AddField adds the value to the field with the given name. It returns an error if
// the field is not defined in the schema, or if the type mismatched the field
// type.
func (m *MetaMutation) AddField(name string, value ent.Value) error {
switch name {
}
return fmt.Errorf("unknown Meta numeric field %s", name)
}
2021-01-14 15:27:45 +00:00
// ClearedFields returns all nullable fields that were cleared during this
// mutation.
func (m *MetaMutation) ClearedFields() []string {
return nil
}
2021-01-14 15:27:45 +00:00
// FieldCleared returns a boolean indicating if a field with the given name was
// cleared in this mutation.
func (m *MetaMutation) FieldCleared(name string) bool {
_, ok := m.clearedFields[name]
return ok
}
2021-01-14 15:27:45 +00:00
// ClearField clears the value of the field with the given name. It returns an
// error if the field is not defined in the schema.
func (m *MetaMutation) ClearField(name string) error {
return fmt.Errorf("unknown Meta nullable field %s", name)
}
2021-01-14 15:27:45 +00:00
// ResetField resets all changes in the mutation for the field with the given name.
// It returns an error if the field is not defined in the schema.
func (m *MetaMutation) ResetField(name string) error {
switch name {
case meta.FieldCreatedAt:
m.ResetCreatedAt()
return nil
case meta.FieldUpdatedAt:
m.ResetUpdatedAt()
return nil
case meta.FieldKey:
m.ResetKey()
return nil
case meta.FieldValue:
m.ResetValue()
return nil
}
return fmt.Errorf("unknown Meta field %s", name)
}
2021-01-14 15:27:45 +00:00
// AddedEdges returns all edge names that were set/added in this mutation.
func (m *MetaMutation) AddedEdges() []string {
edges := make([]string, 0, 1)
if m.owner != nil {
edges = append(edges, meta.EdgeOwner)
}
return edges
}
2021-01-14 15:27:45 +00:00
// AddedIDs returns all IDs (to other nodes) that were added for the given edge
// name in this mutation.
func (m *MetaMutation) AddedIDs(name string) []ent.Value {
switch name {
case meta.EdgeOwner:
if id := m.owner; id != nil {
return []ent.Value{*id}
}
}
return nil
}
2021-01-14 15:27:45 +00:00
// RemovedEdges returns all edge names that were removed in this mutation.
func (m *MetaMutation) RemovedEdges() []string {
edges := make([]string, 0, 1)
return edges
}
2021-01-14 15:27:45 +00:00
// RemovedIDs returns all IDs (to other nodes) that were removed for the edge with
// the given name in this mutation.
func (m *MetaMutation) RemovedIDs(name string) []ent.Value {
switch name {
}
return nil
}
2021-01-14 15:27:45 +00:00
// ClearedEdges returns all edge names that were cleared in this mutation.
func (m *MetaMutation) ClearedEdges() []string {
edges := make([]string, 0, 1)
if m.clearedowner {
edges = append(edges, meta.EdgeOwner)
}
return edges
}
2021-01-14 15:27:45 +00:00
// EdgeCleared returns a boolean which indicates if the edge with the given name
// was cleared in this mutation.
func (m *MetaMutation) EdgeCleared(name string) bool {
switch name {
case meta.EdgeOwner:
return m.clearedowner
}
return false
}
2021-01-14 15:27:45 +00:00
// ClearEdge clears the value of the edge with the given name. It returns an error
// if that edge is not defined in the schema.
func (m *MetaMutation) ClearEdge(name string) error {
switch name {
case meta.EdgeOwner:
m.ClearOwner()
return nil
}
return fmt.Errorf("unknown Meta unique edge %s", name)
}
2021-01-14 15:27:45 +00:00
// ResetEdge resets all changes to the edge with the given name in this mutation.
// It returns an error if the edge is not defined in the schema.
func (m *MetaMutation) ResetEdge(name string) error {
switch name {
case meta.EdgeOwner:
m.ResetOwner()
return nil
}
return fmt.Errorf("unknown Meta edge %s", name)
}