forked from qor/transition
-
Notifications
You must be signed in to change notification settings - Fork 0
/
state_change_log.go
65 lines (55 loc) · 1.82 KB
/
state_change_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
61
62
63
64
65
package transition
import (
"fmt"
"strings"
"time"
"github.com/jinzhu/gorm"
"github.com/qor/admin"
"github.com/qor/audited"
"github.com/qor/qor/resource"
"github.com/qor/roles"
)
// StateChangeLog a model that is used to keep state change logs
type StateChangeLog struct {
ID string `gorm:"type:varchar(36);primary_key"`
CreatedAt time.Time `gorm:"precision:6"`
UpdatedAt time.Time `gorm:"precision:6"`
DeletedAt *time.Time `gorm:"precision:6" sql:"index"`
ReferTable string
ReferID string `gorm:"type:varchar(56)" sql:"index"`
From string
To string
Snapshot *string `gorm:"type:text"`
Note string `gorm:"type:varchar(1024)"`
audited.AuditedModel
}
// GenerateReferenceKey generate reference key used for change log
func GenerateReferenceKey(model interface{}, db *gorm.DB) string {
var (
scope = db.NewScope(model)
primaryValues []string
)
for _, field := range scope.PrimaryFields() {
primaryValues = append(primaryValues, fmt.Sprint(field.Field.Interface()))
}
return strings.Join(primaryValues, "::")
}
// GetStateChangeLogs get state change logs
func GetStateChangeLogs(model interface{}, db *gorm.DB) []StateChangeLog {
var (
changelogs []StateChangeLog
scope = db.NewScope(model)
)
db.Where("refer_table = ? AND refer_id = ?", scope.TableName(), GenerateReferenceKey(model, db)).Find(&changelogs)
return changelogs
}
// ConfigureQorResource used to configure transition for qor admin
func (stageChangeLog *StateChangeLog) ConfigureQorResource(res resource.Resourcer) {
if res, ok := res.(*admin.Resource); ok {
if res.Permission == nil {
res.Permission = roles.Deny(roles.Update, roles.Anyone).Deny(roles.Create, roles.Anyone)
} else {
res.Permission = res.Permission.Deny(roles.Update, roles.Anyone).Deny(roles.Create, roles.Anyone)
}
}
}