-
Notifications
You must be signed in to change notification settings - Fork 2
/
registry.go
executable file
·79 lines (61 loc) · 2.54 KB
/
registry.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
/*
Copyright 2018 This Project Authors.
Author: seanchann <seanchann@foxmail.com>
See docs/ for more information about the project.
*/
package user
import (
"context"
metainternalversion "k8s.io/apimachinery/pkg/apis/meta/internalversion"
metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
"k8s.io/apimachinery/pkg/watch"
"k8s.io/apiserver/pkg/registry/rest"
api "github.com/seanchann/sample-apiserver/pkg/apis/sample"
)
// Registry is an interface for things that know how to store node.
type Registry interface {
ListUsers(ctx context.Context, options *metainternalversion.ListOptions) (*api.UserList, error)
WatchUsers(ctx context.Context, options *metainternalversion.ListOptions) (watch.Interface, error)
GetUser(ctx context.Context, name string, options *metav1.GetOptions) (*api.User, error)
CreateUser(ctx context.Context, user *api.User, createValidation rest.ValidateObjectFunc) error
UpdateUser(ctx context.Context, svc *api.User, createValidation rest.ValidateObjectFunc, updateValidation rest.ValidateObjectUpdateFunc) error
DeleteUser(ctx context.Context, name string) error
}
// storage puts strong typing around storage calls
type storage struct {
rest.StandardStorage
}
// NewRegistry returns a new Registry interface for the given Storage. Any mismatched
// types will panic.
func NewRegistry(s rest.StandardStorage) Registry {
return &storage{s}
}
func (s *storage) ListUsers(ctx context.Context, options *metainternalversion.ListOptions) (*api.UserList, error) {
obj, err := s.List(ctx, options)
if err != nil {
return nil, err
}
return obj.(*api.UserList), nil
}
func (s *storage) WatchUsers(ctx context.Context, options *metainternalversion.ListOptions) (watch.Interface, error) {
return s.Watch(ctx, options)
}
func (s *storage) GetUser(ctx context.Context, name string, options *metav1.GetOptions) (*api.User, error) {
obj, err := s.Get(ctx, name, options)
if err != nil {
return nil, err
}
return obj.(*api.User), nil
}
func (s *storage) CreateUser(ctx context.Context, user *api.User, createValidation rest.ValidateObjectFunc) error {
_, err := s.Create(ctx, user, createValidation, nil)
return err
}
func (s *storage) UpdateUser(ctx context.Context, user *api.User, createValidation rest.ValidateObjectFunc, updateValidation rest.ValidateObjectUpdateFunc) error {
_, _, err := s.Update(ctx, user.Name, rest.DefaultUpdatedObjectInfo(user), createValidation, updateValidation, false, nil)
return err
}
func (s *storage) DeleteUser(ctx context.Context, name string) error {
_, _, err := s.Delete(ctx, name, nil)
return err
}