-
Notifications
You must be signed in to change notification settings - Fork 4
/
audit.go
123 lines (118 loc) · 3.64 KB
/
audit.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
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
package migrations
import (
"github.com/iancoleman/strcase"
"github.com/samber/lo"
"github.com/teamkeel/keel/auditing"
"github.com/teamkeel/keel/proto"
"google.golang.org/protobuf/types/known/wrapperspb"
)
// pushAuditModel adds a (hard coded) KeelAudit model to the given schema.
func pushAuditModel(schema *proto.Schema) {
schema.Models = append(schema.Models, auditModel())
}
// popAuditModel removes any instances present of a KeelAudit model from the given
// schema.
func popAuditModel(schema *proto.Schema) {
schema.Models = lo.Reject(schema.Models, func(m *proto.Model, _ int) bool {
return m.Name == strcase.ToCamel(auditing.TableName)
})
}
// auditModel provides a hard-coded Model to represent the audit table.
//
// Migrations will fire as we would wish, if you edit the fields in this model definition.
func auditModel() *proto.Model {
modelName := strcase.ToCamel(auditing.TableName)
mdl := proto.Model{
Name: modelName,
Fields: []*proto.Field{
{
ModelName: modelName,
Name: strcase.ToLowerCamel(auditing.ColumnId),
PrimaryKey: true,
DefaultValue: &proto.DefaultValue{
UseZeroValue: true,
},
Type: &proto.TypeInfo{
Type: proto.Type_TYPE_ID,
ModelName: wrapperspb.String(modelName),
FieldName: wrapperspb.String(strcase.ToLowerCamel(auditing.ColumnId)),
},
Optional: false,
},
{
ModelName: modelName,
Name: strcase.ToLowerCamel(auditing.ColumnTableName),
Type: &proto.TypeInfo{
Type: proto.Type_TYPE_STRING,
ModelName: wrapperspb.String(modelName),
FieldName: wrapperspb.String(strcase.ToLowerCamel(auditing.ColumnTableName)),
},
Optional: false,
},
{
ModelName: modelName,
Name: strcase.ToLowerCamel(auditing.ColumnOp),
Type: &proto.TypeInfo{
Type: proto.Type_TYPE_STRING,
ModelName: wrapperspb.String(modelName),
FieldName: wrapperspb.String(strcase.ToLowerCamel(auditing.ColumnOp)),
},
Optional: false,
},
{
ModelName: modelName,
Name: strcase.ToLowerCamel(auditing.ColumnData),
Type: &proto.TypeInfo{
Type: proto.Type_TYPE_STRING,
ModelName: wrapperspb.String(modelName),
FieldName: wrapperspb.String(strcase.ToLowerCamel(auditing.ColumnData)),
},
Optional: false,
},
{
ModelName: modelName,
Name: strcase.ToLowerCamel(auditing.ColumnCreatedAt),
DefaultValue: &proto.DefaultValue{
UseZeroValue: true,
},
Type: &proto.TypeInfo{
Type: proto.Type_TYPE_TIMESTAMP,
ModelName: wrapperspb.String(modelName),
FieldName: wrapperspb.String(strcase.ToLowerCamel(auditing.ColumnCreatedAt)),
},
Optional: false,
},
{
ModelName: modelName,
Name: strcase.ToLowerCamel(auditing.ColumnIdentityId),
Type: &proto.TypeInfo{
Type: proto.Type_TYPE_ID,
ModelName: wrapperspb.String(modelName),
FieldName: wrapperspb.String(strcase.ToLowerCamel(auditing.ColumnIdentityId)),
},
Optional: true,
},
{
ModelName: modelName,
Name: strcase.ToLowerCamel(auditing.ColumnTraceId),
Type: &proto.TypeInfo{
Type: proto.Type_TYPE_STRING,
ModelName: wrapperspb.String(modelName),
FieldName: wrapperspb.String(strcase.ToLowerCamel(auditing.ColumnTraceId)),
},
Optional: true,
},
{
ModelName: modelName,
Name: strcase.ToLowerCamel(auditing.ColumnEventProcessedAt),
Type: &proto.TypeInfo{
Type: proto.Type_TYPE_TIMESTAMP,
ModelName: wrapperspb.String(modelName),
FieldName: wrapperspb.String(strcase.ToLowerCamel(auditing.ColumnEventProcessedAt)),
},
Optional: true,
},
},
}
return &mdl
}