-
Notifications
You must be signed in to change notification settings - Fork 1
/
context_adapter.go
81 lines (68 loc) · 2.49 KB
/
context_adapter.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
// this implementation refers to the following link:
// https://github.com/casbin/gorm-adapter/blob/master/context_adapter.go
package casbinbunadapter
import (
"context"
"github.com/casbin/casbin/v2/model"
"github.com/casbin/casbin/v2/persist"
)
var (
// check if the ctxBunAdapter implements the ContextAdapter interface
_ persist.ContextAdapter = (*ctxBunAdapter)(nil) // Ensure ctxBunAdapter
)
type ctxBunAdapter struct {
*bunAdapter
}
func NewCtxAdapter(driverName string, dataSourceName string, opts ...adapterOption) (*ctxBunAdapter, error) {
adapter, err := NewAdapter(driverName, dataSourceName, opts...)
if err != nil {
return nil, err
}
return &ctxBunAdapter{adapter}, nil
}
// executeWithContext is a helper function to execute a function with context and return the result or error.
func executeWithContext(ctx context.Context, fn func() error) error {
done := make(chan error)
go func() {
done <- fn()
}()
select {
case <-ctx.Done():
return ctx.Err()
case err := <-done:
return err
}
}
// LoadPolicyCtx loads all policy rules from the storage with context.
func (a *ctxBunAdapter) LoadPolicyCtx(ctx context.Context, model model.Model) error {
return executeWithContext(ctx, func() error {
return a.LoadPolicy(model)
})
}
// SavePolicyCtx saves all policy rules to the storage with context.
func (a *ctxBunAdapter) SavePolicyCtx(ctx context.Context, model model.Model) error {
return executeWithContext(ctx, func() error {
return a.SavePolicy(model)
})
}
// AddPolicyCtx adds a policy rule to the storage with context.
// This is part of the Auto-Save feature.
func (a *ctxBunAdapter) AddPolicyCtx(ctx context.Context, sec string, ptype string, rule []string) error {
return executeWithContext(ctx, func() error {
return a.AddPolicy(sec, ptype, rule)
})
}
// RemovePolicyCtx removes a policy rule from the storage with context.
// This is part of the Auto-Save feature.
func (a *ctxBunAdapter) RemovePolicyCtx(ctx context.Context, sec string, ptype string, rule []string) error {
return executeWithContext(ctx, func() error {
return a.RemovePolicy(sec, ptype, rule)
})
}
// RemoveFilteredPolicyCtx removes policy rules that match the filter from the storage with context.
// This is part of the Auto-Save feature.
func (a *ctxBunAdapter) RemoveFilteredPolicyCtx(ctx context.Context, sec string, ptype string, fieldIndex int, fieldValues ...string) error {
return executeWithContext(ctx, func() error {
return a.RemoveFilteredPolicy(sec, ptype, fieldIndex, fieldValues...)
})
}