-
Notifications
You must be signed in to change notification settings - Fork 0
/
endpoint.go
93 lines (79 loc) · 2.5 KB
/
endpoint.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
/**
* @Time : 2019-06-26 14:48
* @Author : ygqbasic@gmail.com
* @File : endpoint
* @Software: VS Code
*/
package persistentvolumeclaim
import (
"context"
"github.com/go-kit/kit/endpoint"
"github.com/fuhsicloud/fuhsicloud/src/util/encode"
)
type ResourceUnit string
const (
ResourceMi ResourceUnit = "Mi"
ResourceGi ResourceUnit = "Gi"
)
func (c ResourceUnit) String() string {
return string(c)
}
type getRequest struct {
Namespace string `json:"namespace"`
Name string `json:"name"`
}
type pvcRequest struct {
Namespace string `json:"namespace"`
Name string `json:"name"`
AccessModes []string `json:"access_modes"`
Storage string `json:"storage"`
Unit ResourceUnit `json:"unit"`
StorageClassName string `json:"storage_class_name"`
}
type listRequest struct {
getRequest
Page int `json:"page"`
Limit int `json:"limit"`
}
func makeSyncEndpoint(s Service) endpoint.Endpoint {
return func(ctx context.Context, request interface{}) (interface{}, error) {
req := request.(pvcRequest)
err := s.Sync(ctx, req.Namespace)
return encode.Response{Err: err}, err
}
}
func makeGetEndpoint(s Service) endpoint.Endpoint {
return func(ctx context.Context, request interface{}) (interface{}, error) {
req := request.(pvcRequest)
rs, err := s.Get(ctx, req.Namespace, req.Name)
return encode.Response{Err: err, Data: rs}, err
}
}
func makeDeleteEndpoint(s Service) endpoint.Endpoint {
return func(ctx context.Context, request interface{}) (interface{}, error) {
req := request.(pvcRequest)
err := s.Delete(ctx, req.Namespace, req.Name)
return encode.Response{Err: err}, err
}
}
func makePostEndpoint(s Service) endpoint.Endpoint {
return func(ctx context.Context, request interface{}) (interface{}, error) {
req := request.(pvcRequest)
err := s.Post(ctx, req.Namespace, req.Name, req.Storage, req.StorageClassName, req.AccessModes)
return encode.Response{Err: err}, err
}
}
func makeListEndpoint(s Service) endpoint.Endpoint {
return func(ctx context.Context, request interface{}) (interface{}, error) {
req := request.(listRequest)
res, err := s.List(ctx, req.Namespace, req.Page, req.Limit)
return encode.Response{Err: err, Data: res}, err
}
}
func makeAllEndpoint(s Service) endpoint.Endpoint {
return func(ctx context.Context, request interface{}) (interface{}, error) {
req := request.(listRequest)
res, err := s.List(ctx, req.Namespace, req.Page, req.Limit)
return encode.Response{Err: err, Data: res}, err
}
}