blob: 5ff9b96c0c4283cbc67000f4111aa6a2bf65cca4 [file] [log] [blame]
khenaidoo59ce9dd2019-11-11 13:05:32 -05001// Copyright 2017 The etcd Authors
2//
3// Licensed under the Apache License, Version 2.0 (the "License");
4// you may not use this file except in compliance with the License.
5// You may obtain a copy of the License at
6//
7// http://www.apache.org/licenses/LICENSE-2.0
8//
9// Unless required by applicable law or agreed to in writing, software
10// distributed under the License is distributed on an "AS IS" BASIS,
11// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12// See the License for the specific language governing permissions and
13// limitations under the License.
14
15package v2v3
16
17import (
18 "context"
19 "net/http"
20 "time"
21
22 "go.etcd.io/etcd/clientv3"
23 "go.etcd.io/etcd/etcdserver"
24 "go.etcd.io/etcd/etcdserver/api"
25 "go.etcd.io/etcd/etcdserver/api/membership"
26 pb "go.etcd.io/etcd/etcdserver/etcdserverpb"
27 "go.etcd.io/etcd/pkg/types"
28
29 "github.com/coreos/go-semver/semver"
30 "go.uber.org/zap"
31)
32
33type fakeStats struct{}
34
35func (s *fakeStats) SelfStats() []byte { return nil }
36func (s *fakeStats) LeaderStats() []byte { return nil }
37func (s *fakeStats) StoreStats() []byte { return nil }
38
39type v2v3Server struct {
40 lg *zap.Logger
41 c *clientv3.Client
42 store *v2v3Store
43 fakeStats
44}
45
46func NewServer(lg *zap.Logger, c *clientv3.Client, pfx string) etcdserver.ServerPeer {
47 return &v2v3Server{lg: lg, c: c, store: newStore(c, pfx)}
48}
49
50func (s *v2v3Server) ClientCertAuthEnabled() bool { return false }
51
52func (s *v2v3Server) LeaseHandler() http.Handler { panic("STUB: lease handler") }
53func (s *v2v3Server) RaftHandler() http.Handler { panic("STUB: raft handler") }
54
55func (s *v2v3Server) Leader() types.ID {
56 ctx, cancel := context.WithTimeout(context.TODO(), 5*time.Second)
57 defer cancel()
58 resp, err := s.c.Status(ctx, s.c.Endpoints()[0])
59 if err != nil {
60 return 0
61 }
62 return types.ID(resp.Leader)
63}
64
65func (s *v2v3Server) AddMember(ctx context.Context, memb membership.Member) ([]*membership.Member, error) {
66 // adding member as learner is not supported by V2 Server.
67 resp, err := s.c.MemberAdd(ctx, memb.PeerURLs)
68 if err != nil {
69 return nil, err
70 }
71 return v3MembersToMembership(resp.Members), nil
72}
73
74func (s *v2v3Server) RemoveMember(ctx context.Context, id uint64) ([]*membership.Member, error) {
75 resp, err := s.c.MemberRemove(ctx, id)
76 if err != nil {
77 return nil, err
78 }
79 return v3MembersToMembership(resp.Members), nil
80}
81
82func (s *v2v3Server) PromoteMember(ctx context.Context, id uint64) ([]*membership.Member, error) {
83 resp, err := s.c.MemberPromote(ctx, id)
84 if err != nil {
85 return nil, err
86 }
87 return v3MembersToMembership(resp.Members), nil
88}
89
90func (s *v2v3Server) UpdateMember(ctx context.Context, m membership.Member) ([]*membership.Member, error) {
91 resp, err := s.c.MemberUpdate(ctx, uint64(m.ID), m.PeerURLs)
92 if err != nil {
93 return nil, err
94 }
95 return v3MembersToMembership(resp.Members), nil
96}
97
98func v3MembersToMembership(v3membs []*pb.Member) []*membership.Member {
99 membs := make([]*membership.Member, len(v3membs))
100 for i, m := range v3membs {
101 membs[i] = &membership.Member{
102 ID: types.ID(m.ID),
103 RaftAttributes: membership.RaftAttributes{
104 PeerURLs: m.PeerURLs,
105 IsLearner: m.IsLearner,
106 },
107 Attributes: membership.Attributes{
108 Name: m.Name,
109 ClientURLs: m.ClientURLs,
110 },
111 }
112 }
113 return membs
114}
115
116func (s *v2v3Server) ClusterVersion() *semver.Version { return s.Version() }
117func (s *v2v3Server) Cluster() api.Cluster { return s }
118func (s *v2v3Server) Alarms() []*pb.AlarmMember { return nil }
119
120func (s *v2v3Server) Do(ctx context.Context, r pb.Request) (etcdserver.Response, error) {
121 applier := etcdserver.NewApplierV2(s.lg, s.store, nil)
122 reqHandler := etcdserver.NewStoreRequestV2Handler(s.store, applier)
123 req := (*etcdserver.RequestV2)(&r)
124 resp, err := req.Handle(ctx, reqHandler)
125 if resp.Err != nil {
126 return resp, resp.Err
127 }
128 return resp, err
129}