-
Notifications
You must be signed in to change notification settings - Fork 1k
/
fake.go
121 lines (101 loc) · 2.65 KB
/
fake.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
118
119
120
121
// Copyright 2020 Lingfei Kong <colin404@foxmail.com>. All rights reserved.
// Use of this source code is governed by a MIT style
// license that can be found in the LICENSE file.
package fake
import (
"fmt"
"sync"
v1 "github.com/marmotedu/api/apiserver/v1"
metav1 "github.com/marmotedu/component-base/pkg/meta/v1"
"github.com/marmotedu/component-base/pkg/util/idutil"
"github.com/ory/ladon"
"github.com/marmotedu/iam/internal/apiserver/store"
)
// ResourceCount defines the number of fake resources.
const ResourceCount = 1000
type datastore struct {
sync.RWMutex
users []*v1.User
secrets []*v1.Secret
policies []*v1.Policy
}
func (ds *datastore) Users() store.UserStore {
return newUsers(ds)
}
func (ds *datastore) Secrets() store.SecretStore {
return newSecrets(ds)
}
func (ds *datastore) Policies() store.PolicyStore {
return newPolicies(ds)
}
func (ds *datastore) Close() error {
return nil
}
var (
fakeFactory store.Factory
once sync.Once
)
// GetFakeFactoryOr create fake store.
func GetFakeFactoryOr() (store.Factory, error) {
once.Do(func() {
fakeFactory = &datastore{
users: FakeUsers(ResourceCount),
secrets: FakeSecrets(ResourceCount),
policies: FakePolicies(ResourceCount),
}
})
if fakeFactory == nil {
return nil, fmt.Errorf("failed to get mysql store fatory, mysqlFactory: %+v", fakeFactory)
}
return fakeFactory, nil
}
// FakeUsers returns fake user data.
func FakeUsers(count int) []*v1.User {
// init some user records
users := make([]*v1.User, 0)
for i := 1; i <= count; i++ {
users = append(users, &v1.User{
ObjectMeta: metav1.ObjectMeta{
Name: fmt.Sprintf("user%d", i),
ID: uint64(i),
},
Nickname: fmt.Sprintf("user%d", i),
Password: fmt.Sprintf("User%d@2020", i),
Email: fmt.Sprintf("user%d@qq.com", i),
})
}
return users
}
// FakeSecrets returns fake secret data.
func FakeSecrets(count int) []*v1.Secret {
secrets := make([]*v1.Secret, 0)
for i := 1; i <= count; i++ {
secrets = append(secrets, &v1.Secret{
ObjectMeta: metav1.ObjectMeta{
Name: fmt.Sprintf("secret%d", i),
ID: uint64(i),
},
Username: fmt.Sprintf("user%d", i),
SecretID: idutil.NewSecretID(),
SecretKey: idutil.NewSecretKey(),
})
}
return secrets
}
// FakePolicies returns fake policy data.
func FakePolicies(count int) []*v1.Policy {
policies := make([]*v1.Policy, 0)
for i := 1; i <= count; i++ {
policies = append(policies, &v1.Policy{
ObjectMeta: metav1.ObjectMeta{
Name: fmt.Sprintf("policy%d", i),
ID: uint64(i),
},
Username: fmt.Sprintf("user%d", i),
Policy: v1.AuthzPolicy{
DefaultPolicy: ladon.DefaultPolicy{},
},
})
}
return policies
}