/
audit_log.go
60 lines (50 loc) · 1.3 KB
/
audit_log.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
package loader
import (
"github.com/authgear/authgear-server/pkg/lib/audit"
"github.com/authgear/authgear-server/pkg/lib/infra/db/auditdb"
"github.com/authgear/authgear-server/pkg/util/graphqlutil"
)
type AuditLogQuery interface {
GetByIDs(ids []string) ([]*audit.Log, error)
}
type AuditLogLoader struct {
*graphqlutil.DataLoader `wire:"-"`
AuditDatabase *auditdb.ReadHandle
Query AuditLogQuery
}
func NewAuditLogLoader(query AuditLogQuery, handle *auditdb.ReadHandle) *AuditLogLoader {
l := &AuditLogLoader{
Query: query,
AuditDatabase: handle,
}
l.DataLoader = graphqlutil.NewDataLoader(l.LoadFunc)
return l
}
func (l *AuditLogLoader) LoadFunc(keys []interface{}) ([]interface{}, error) {
// Prepare IDs.
ids := make([]string, len(keys))
for i, key := range keys {
ids[i] = key.(string)
}
var entities []*audit.Log
// Get entities.
err := l.AuditDatabase.ReadOnly(func() (err error) {
entities, err = l.Query.GetByIDs(ids)
return
})
if err != nil {
return nil, err
}
// Create map.
entityMap := make(map[string]*audit.Log)
for _, entity := range entities {
entityMap[entity.ID] = entity
}
// Ensure output is in correct order.
out := make([]interface{}, len(keys))
for i, id := range ids {
entity := entityMap[id]
out[i] = entity
}
return out, nil
}