-
Notifications
You must be signed in to change notification settings - Fork 0
/
relations.go
68 lines (56 loc) · 1.62 KB
/
relations.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
package sqlite
import (
"context"
"fmt"
"sort"
"github.com/pkg/errors"
"admini.dev/admini/app/lib/database"
"admini.dev/admini/app/lib/schema/model"
"admini.dev/admini/app/util"
"admini.dev/admini/queries/qsqlite"
)
type foreignKeyResult struct {
Ordinal int `db:"idx"`
Table string `db:"src"`
Field string `db:"src_col"`
TargetTable string `db:"tgt"`
TargetField string `db:"tgt_col"`
}
func (r *foreignKeyResult) String() string {
return fmt.Sprintf("rel-%s-%s", r.Table, r.TargetTable)
}
type foreignKeyResults []*foreignKeyResult
func (r foreignKeyResults) Sort() {
sort.Slice(r, func(i int, j int) bool {
if r[i].Table == r[j].Table {
return r[i].Ordinal < r[j].Ordinal
}
return r[i].Table < r[j].Table
})
}
func loadForeignKeys(ctx context.Context, models model.Models, db *database.Service, logger util.Logger) error {
keys := foreignKeyResults{}
err := db.Select(ctx, &keys, qsqlite.ListForeignKeys(db.SchemaName), nil, logger)
if err != nil {
return errors.Wrap(err, "can't list foreign keys")
}
for _, k := range keys {
mod := models.Get(nil, k.Table)
if mod == nil {
return errors.Errorf("no model [%s] found among [%d] candidates", k.Table, len(models))
}
curr := mod.Relationships.Get(k.String())
if curr == nil {
curr = &model.Relationship{
Key: k.String(),
SourceFields: []string{},
TargetModel: k.TargetTable,
TargetFields: []string{},
}
mod.Relationships = append(mod.Relationships, curr)
}
curr.SourceFields = append(curr.SourceFields, k.Field)
curr.TargetFields = append(curr.TargetFields, k.TargetField)
}
return nil
}