123 lines
2.7 KiB
Go
123 lines
2.7 KiB
Go
// Copyright 2019-present Facebook Inc. All rights reserved.
|
|
// This source code is licensed under the Apache 2.0 license found
|
|
// in the LICENSE file in the root directory of this source tree.
|
|
|
|
package schema
|
|
|
|
import (
|
|
"context"
|
|
"errors"
|
|
"fmt"
|
|
|
|
"entgo.io/ent/entc/integration/hooks/ent/user"
|
|
|
|
"entgo.io/ent"
|
|
"entgo.io/ent/entc/integration/hooks/ent/hook"
|
|
"entgo.io/ent/schema/edge"
|
|
"entgo.io/ent/schema/field"
|
|
"entgo.io/ent/schema/mixin"
|
|
)
|
|
|
|
// User holds the schema definition for the User entity.
|
|
type User struct {
|
|
ent.Schema
|
|
}
|
|
|
|
// Mixin of the User.
|
|
func (User) Mixin() []ent.Mixin {
|
|
return []ent.Mixin{
|
|
VersionMixin{},
|
|
}
|
|
}
|
|
|
|
// Fields of the User.
|
|
func (User) Fields() []ent.Field {
|
|
return []ent.Field{
|
|
field.String("name"),
|
|
field.Uint("worth").
|
|
Optional(),
|
|
field.String("password").
|
|
Optional().
|
|
Sensitive(),
|
|
field.Bool("active").
|
|
Default(true),
|
|
}
|
|
}
|
|
|
|
// Edges of the User.
|
|
func (User) Edges() []ent.Edge {
|
|
return []ent.Edge{
|
|
edge.To("cards", Card.Type),
|
|
edge.To("pets", Pet.Type),
|
|
edge.To("friends", User.Type),
|
|
edge.To("best_friend", User.Type).
|
|
Unique(),
|
|
}
|
|
}
|
|
|
|
// Hooks of the User.
|
|
func (User) Hooks() []ent.Hook {
|
|
return []ent.Hook{
|
|
hook.If(
|
|
hook.FixedError(errors.New("password cannot be edited on update-many")),
|
|
hook.And(
|
|
hook.HasOp(ent.OpUpdate),
|
|
hook.Or(
|
|
hook.HasFields(user.FieldPassword),
|
|
hook.HasClearedFields(user.FieldPassword),
|
|
),
|
|
),
|
|
),
|
|
}
|
|
}
|
|
|
|
type VersionMixin struct {
|
|
mixin.Schema
|
|
}
|
|
|
|
func (VersionMixin) Fields() []ent.Field {
|
|
return []ent.Field{
|
|
field.Int("version").
|
|
Default(0),
|
|
}
|
|
}
|
|
|
|
func (VersionMixin) Hooks() []ent.Hook {
|
|
return []ent.Hook{
|
|
hook.On(VersionHook(), ent.OpUpdateOne),
|
|
}
|
|
}
|
|
|
|
func VersionHook() ent.Hook {
|
|
type OldSetVersion interface {
|
|
SetVersion(int)
|
|
Version() (int, bool)
|
|
OldVersion(context.Context) (int, error)
|
|
}
|
|
return func(next ent.Mutator) ent.Mutator {
|
|
// A hook that validates the "version" field is incremented by 1 on each update.
|
|
// Note that this is just a dummy example, and it doesn't promise consistency in
|
|
// the database.
|
|
return ent.MutateFunc(func(ctx context.Context, m ent.Mutation) (ent.Value, error) {
|
|
ver, ok := m.(OldSetVersion)
|
|
if !ok {
|
|
return next.Mutate(ctx, m)
|
|
}
|
|
oldV, err := ver.OldVersion(ctx)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
curV, exists := ver.Version()
|
|
if !exists {
|
|
return nil, fmt.Errorf("version field is required in update mutation")
|
|
}
|
|
if curV != oldV+1 {
|
|
return nil, fmt.Errorf("version field must be incremented by 1")
|
|
}
|
|
// Add an SQL predicate that validates the "version" column is equal
|
|
// to "oldV" (it wasn't changed during the mutation by other process).
|
|
return next.Mutate(ctx, m)
|
|
})
|
|
}
|
|
}
|