1
2
3
4
5
6
7
8
9
10
11
12
13
14
15 package grpcproxy
16
17 import (
18 "context"
19
20 pb "go.etcd.io/etcd/api/v3/etcdserverpb"
21 "go.etcd.io/etcd/client/v3"
22 )
23
24 type AuthProxy struct {
25 client *clientv3.Client
26 }
27
28 func NewAuthProxy(c *clientv3.Client) pb.AuthServer {
29 return &AuthProxy{client: c}
30 }
31
32 func (ap *AuthProxy) AuthEnable(ctx context.Context, r *pb.AuthEnableRequest) (*pb.AuthEnableResponse, error) {
33 conn := ap.client.ActiveConnection()
34 return pb.NewAuthClient(conn).AuthEnable(ctx, r)
35 }
36
37 func (ap *AuthProxy) AuthDisable(ctx context.Context, r *pb.AuthDisableRequest) (*pb.AuthDisableResponse, error) {
38 conn := ap.client.ActiveConnection()
39 return pb.NewAuthClient(conn).AuthDisable(ctx, r)
40 }
41
42 func (ap *AuthProxy) AuthStatus(ctx context.Context, r *pb.AuthStatusRequest) (*pb.AuthStatusResponse, error) {
43 conn := ap.client.ActiveConnection()
44 return pb.NewAuthClient(conn).AuthStatus(ctx, r)
45 }
46
47 func (ap *AuthProxy) Authenticate(ctx context.Context, r *pb.AuthenticateRequest) (*pb.AuthenticateResponse, error) {
48 conn := ap.client.ActiveConnection()
49 return pb.NewAuthClient(conn).Authenticate(ctx, r)
50 }
51
52 func (ap *AuthProxy) RoleAdd(ctx context.Context, r *pb.AuthRoleAddRequest) (*pb.AuthRoleAddResponse, error) {
53 conn := ap.client.ActiveConnection()
54 return pb.NewAuthClient(conn).RoleAdd(ctx, r)
55 }
56
57 func (ap *AuthProxy) RoleDelete(ctx context.Context, r *pb.AuthRoleDeleteRequest) (*pb.AuthRoleDeleteResponse, error) {
58 conn := ap.client.ActiveConnection()
59 return pb.NewAuthClient(conn).RoleDelete(ctx, r)
60 }
61
62 func (ap *AuthProxy) RoleGet(ctx context.Context, r *pb.AuthRoleGetRequest) (*pb.AuthRoleGetResponse, error) {
63 conn := ap.client.ActiveConnection()
64 return pb.NewAuthClient(conn).RoleGet(ctx, r)
65 }
66
67 func (ap *AuthProxy) RoleList(ctx context.Context, r *pb.AuthRoleListRequest) (*pb.AuthRoleListResponse, error) {
68 conn := ap.client.ActiveConnection()
69 return pb.NewAuthClient(conn).RoleList(ctx, r)
70 }
71
72 func (ap *AuthProxy) RoleRevokePermission(ctx context.Context, r *pb.AuthRoleRevokePermissionRequest) (*pb.AuthRoleRevokePermissionResponse, error) {
73 conn := ap.client.ActiveConnection()
74 return pb.NewAuthClient(conn).RoleRevokePermission(ctx, r)
75 }
76
77 func (ap *AuthProxy) RoleGrantPermission(ctx context.Context, r *pb.AuthRoleGrantPermissionRequest) (*pb.AuthRoleGrantPermissionResponse, error) {
78 conn := ap.client.ActiveConnection()
79 return pb.NewAuthClient(conn).RoleGrantPermission(ctx, r)
80 }
81
82 func (ap *AuthProxy) UserAdd(ctx context.Context, r *pb.AuthUserAddRequest) (*pb.AuthUserAddResponse, error) {
83 conn := ap.client.ActiveConnection()
84 return pb.NewAuthClient(conn).UserAdd(ctx, r)
85 }
86
87 func (ap *AuthProxy) UserDelete(ctx context.Context, r *pb.AuthUserDeleteRequest) (*pb.AuthUserDeleteResponse, error) {
88 conn := ap.client.ActiveConnection()
89 return pb.NewAuthClient(conn).UserDelete(ctx, r)
90 }
91
92 func (ap *AuthProxy) UserGet(ctx context.Context, r *pb.AuthUserGetRequest) (*pb.AuthUserGetResponse, error) {
93 conn := ap.client.ActiveConnection()
94 return pb.NewAuthClient(conn).UserGet(ctx, r)
95 }
96
97 func (ap *AuthProxy) UserList(ctx context.Context, r *pb.AuthUserListRequest) (*pb.AuthUserListResponse, error) {
98 conn := ap.client.ActiveConnection()
99 return pb.NewAuthClient(conn).UserList(ctx, r)
100 }
101
102 func (ap *AuthProxy) UserGrantRole(ctx context.Context, r *pb.AuthUserGrantRoleRequest) (*pb.AuthUserGrantRoleResponse, error) {
103 conn := ap.client.ActiveConnection()
104 return pb.NewAuthClient(conn).UserGrantRole(ctx, r)
105 }
106
107 func (ap *AuthProxy) UserRevokeRole(ctx context.Context, r *pb.AuthUserRevokeRoleRequest) (*pb.AuthUserRevokeRoleResponse, error) {
108 conn := ap.client.ActiveConnection()
109 return pb.NewAuthClient(conn).UserRevokeRole(ctx, r)
110 }
111
112 func (ap *AuthProxy) UserChangePassword(ctx context.Context, r *pb.AuthUserChangePasswordRequest) (*pb.AuthUserChangePasswordResponse, error) {
113 conn := ap.client.ActiveConnection()
114 return pb.NewAuthClient(conn).UserChangePassword(ctx, r)
115 }
116
View as plain text