-
Notifications
You must be signed in to change notification settings - Fork 83
/
server.go
100 lines (88 loc) · 2.91 KB
/
server.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
// Licensed to Apache Software Foundation (ASF) under one or more contributor
// license agreements. See the NOTICE file distributed with
// this work for additional information regarding copyright
// ownership. Apache Software Foundation (ASF) licenses this file to you under
// the Apache License, Version 2.0 (the "License"); you may
// not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
// http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing,
// software distributed under the License is distributed on an
// "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
// KIND, either express or implied. See the License for the
// specific language governing permissions and limitations
// under the License.
package metadata
import (
"context"
"errors"
"strings"
databasev1 "github.com/apache/skywalking-banyandb/api/proto/banyandb/database/v1"
"github.com/apache/skywalking-banyandb/banyand/metadata/embeddedetcd"
"github.com/apache/skywalking-banyandb/pkg/run"
)
type server struct {
Service
metaServer embeddedetcd.Server
rootDir string
listenClientURL []string
listenPeerURL []string
}
func (s *server) Name() string {
return "metadata"
}
func (s *server) Role() databasev1.Role {
return databasev1.Role_ROLE_META
}
func (s *server) FlagSet() *run.FlagSet {
fs := run.NewFlagSet("metadata")
fs.StringVar(&s.rootDir, "metadata-root-path", "/tmp", "the root path of metadata")
fs.StringArrayVar(&s.listenClientURL, "etcd-listen-client-url", []string{"http://localhost:2379"}, "A URL to listen on for client traffic")
fs.StringArrayVar(&s.listenPeerURL, "etcd-listen-peer-url", []string{"http://localhost:2380"}, "A URL to listen on for peer traffic")
return fs
}
func (s *server) Validate() error {
if s.rootDir == "" {
return errors.New("rootDir is empty")
}
if s.listenClientURL == nil {
return errors.New("listenClientURL is empty")
}
if s.listenPeerURL == nil {
return errors.New("listenPeerURL is empty")
}
if err := s.Service.FlagSet().Set(flagEtcdEndpointsName,
strings.Join(s.listenClientURL, ",")); err != nil {
return err
}
return s.Service.Validate()
}
func (s *server) PreRun(ctx context.Context) error {
var err error
s.metaServer, err = embeddedetcd.NewServer(embeddedetcd.RootDir(s.rootDir), embeddedetcd.ConfigureListener(s.listenClientURL, s.listenPeerURL))
if err != nil {
return err
}
<-s.metaServer.ReadyNotify()
return s.Service.PreRun(ctx)
}
func (s *server) Serve() run.StopNotify {
return s.metaServer.StoppingNotify()
}
func (s *server) GracefulStop() {
s.Service.GracefulStop()
s.metaServer.Close()
<-s.metaServer.StopNotify()
}
// NewService returns a new metadata repository Service.
func NewService(_ context.Context) (Service, error) {
s := &server{}
var err error
s.Service, err = NewClient(true)
if err != nil {
return nil, err
}
return s, nil
}