-
Notifications
You must be signed in to change notification settings - Fork 0
/
hook.go
105 lines (86 loc) · 2.3 KB
/
hook.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
package enthistory
import (
"context"
"fmt"
"entgo.io/ent"
)
type Mutation interface {
Op() ent.Op
CreateHistoryFromCreate(ctx context.Context) error
CreateHistoryFromUpdate(ctx context.Context) error
CreateHistoryFromDelete(ctx context.Context) error
}
type Mutator interface {
Mutate(context.Context, Mutation) (ent.Value, error)
}
func On(hk ent.Hook, op ent.Op) ent.Hook {
return func(next ent.Mutator) ent.Mutator {
return ent.MutateFunc(func(ctx context.Context, m ent.Mutation) (ent.Value, error) {
if m.Op().Is(op) {
return hk(next).Mutate(ctx, m)
}
return next.Mutate(ctx, m)
})
}
}
func HistoryHooks[T Mutation]() []ent.Hook {
return []ent.Hook{
On(historyHookCreate[T](), ent.OpCreate),
On(historyHookUpdate[T](), ent.OpUpdate|ent.OpUpdateOne),
On(historyHookDelete[T](), ent.OpDelete|ent.OpDeleteOne),
}
}
func getTypedMutation[T Mutation](m ent.Mutation) (T, error) {
f, ok := any(m).(T)
if !ok {
return f, fmt.Errorf("expected appropriately typed mutation in schema hook, got: %+v", m) //nolint:goerr113
}
return f, nil
}
func historyHookCreate[T Mutation]() ent.Hook {
return func(next ent.Mutator) ent.Mutator {
return ent.MutateFunc(func(ctx context.Context, m ent.Mutation) (ent.Value, error) {
mutation, err := getTypedMutation[T](m)
if err != nil {
return nil, err
}
value, err := next.Mutate(ctx, m)
if err != nil {
return nil, err
}
err = mutation.CreateHistoryFromCreate(ctx)
if err != nil {
return nil, err
}
return value, nil
})
}
}
func historyHookUpdate[T Mutation]() ent.Hook {
return func(next ent.Mutator) ent.Mutator {
return ent.MutateFunc(func(ctx context.Context, m ent.Mutation) (ent.Value, error) {
mutation, err := getTypedMutation[T](m)
if err != nil {
return nil, err
}
if err = mutation.CreateHistoryFromUpdate(ctx); err != nil {
return nil, err
}
return next.Mutate(ctx, m)
})
}
}
func historyHookDelete[T Mutation]() ent.Hook {
return func(next ent.Mutator) ent.Mutator {
return ent.MutateFunc(func(ctx context.Context, m ent.Mutation) (ent.Value, error) {
mutation, err := getTypedMutation[T](m)
if err != nil {
return nil, err
}
if err = mutation.CreateHistoryFromDelete(ctx); err != nil {
return nil, err
}
return next.Mutate(ctx, m)
})
}
}