-
Notifications
You must be signed in to change notification settings - Fork 175
/
acl-cache.go
117 lines (106 loc) · 3.47 KB
/
acl-cache.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
/*
* Copyright (c) 2019-2021. Abstrium SAS <team (at) pydio.com>
* This file is part of Pydio Cells.
*
* Pydio Cells is free software: you can redistribute it and/or modify
* it under the terms of the GNU Affero General Public License as published by
* the Free Software Foundation, either version 3 of the License, or
* (at your option) any later version.
*
* Pydio Cells is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU Affero General Public License for more details.
*
* You should have received a copy of the GNU Affero General Public License
* along with Pydio Cells. If not, see <http://www.gnu.org/licenses/>.
*
* The latest code can be found at <https://pydio.com>.
*/
package permissions
import (
"context"
"github.com/pydio/cells/v4/common/utils/std"
"sync"
"github.com/pydio/cells/v4/common"
"github.com/pydio/cells/v4/common/broker"
"github.com/pydio/cells/v4/common/proto/idm"
"github.com/pydio/cells/v4/common/runtime"
"github.com/pydio/cells/v4/common/utils/cache"
)
var (
aclCache cache.Cache
)
func initAclCache() {
aclCache, _ = cache.OpenCache(context.TODO(), runtime.ShortCacheURL("evictionTime", "500ms", "cleanWindow", "30s"))
_, _ = broker.Subscribe(context.TODO(), common.TopicIdmEvent, func(message broker.Message) error {
event := &idm.ChangeEvent{}
if _, e := message.Unmarshal(event); e != nil {
return e
}
switch event.Type {
case idm.ChangeEventType_CREATE, idm.ChangeEventType_UPDATE, idm.ChangeEventType_DELETE:
return aclCache.Reset()
}
return nil
})
}
func getAclCache() cache.Cache {
if aclCache == nil {
initAclCache()
}
return aclCache
}
type CachedAccessList struct {
Wss map[string]*idm.Workspace
WssRootsMasks map[string]map[string]Bitmask
OrderedRoles []*idm.Role
WsACLs []*idm.ACL
FrontACLs []*idm.ACL
MasksByUUIDs map[string]Bitmask
MasksByPaths map[string]Bitmask
ClaimsScopes map[string]Bitmask
HasClaimsScopes bool
}
// cache uses the CachedAccessList struct with public fields for cache serialization
func (a *AccessList) cache(key string) error {
a.maskBPLock.RLock()
a.maskBULock.RLock()
defer a.maskBPLock.RUnlock()
defer a.maskBULock.RUnlock()
m := &CachedAccessList{
OrderedRoles: a.orderedRoles,
WsACLs: a.wsACLs,
FrontACLs: a.frontACLs,
HasClaimsScopes: a.hasClaimsScopes,
Wss: std.CloneMap(a.wss),
WssRootsMasks: std.CloneMap(a.wssRootsMasks),
MasksByUUIDs: std.CloneMap(a.masksByUUIDs),
MasksByPaths: std.CloneMap(a.masksByPaths),
ClaimsScopes: std.CloneMap(a.claimsScopes),
}
return getAclCache().Set(key, m)
}
// newFromCache looks up for a CachedAccessList struct in the cache and init an AccessList with the values.
func newFromCache(key string) (*AccessList, bool) {
m := &CachedAccessList{}
if b := getAclCache().Get(key, &m); !b {
return nil, b
}
a := &AccessList{
// Use cached value directly
orderedRoles: m.OrderedRoles,
wsACLs: m.WsACLs,
frontACLs: m.FrontACLs,
hasClaimsScopes: m.HasClaimsScopes,
// Re-init these
maskBPLock: &sync.RWMutex{},
maskBULock: &sync.RWMutex{},
wss: std.CloneMap(m.Wss),
wssRootsMasks: std.CloneMap(m.WssRootsMasks),
masksByUUIDs: std.CloneMap(m.MasksByUUIDs),
masksByPaths: std.CloneMap(m.MasksByPaths),
claimsScopes: std.CloneMap(m.ClaimsScopes),
}
return a, true
}