forked from dolthub/vitess
-
Notifications
You must be signed in to change notification settings - Fork 0
/
client.go
73 lines (56 loc) · 2.44 KB
/
client.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
// Copyright 2012, Google Inc. All rights reserved.
// Use of this source code is governed by a BSD-style
// license that can be found in the LICENSE file.
package binlogplayer
import (
"flag"
"time"
"golang.org/x/net/context"
log "github.com/golang/glog"
binlogdatapb "github.com/youtube/vitess/go/vt/proto/binlogdata"
topodatapb "github.com/youtube/vitess/go/vt/proto/topodata"
)
/*
This file contains the API and registration mechanism for binlog player client.
*/
var binlogPlayerProtocol = flag.String("binlog_player_protocol", "grpc", "the protocol to download binlogs from a vttablet")
// StreamEventStream is the interface of the object returned by
// ServeUpdateStream
type StreamEventStream interface {
// Recv returns the next StreamEvent, or an error if the RPC was
// interrupted.
Recv() (*binlogdatapb.StreamEvent, error)
}
// BinlogTransactionStream is the interface of the object returned by
// StreamTables and StreamKeyRange
type BinlogTransactionStream interface {
// Recv returns the next BinlogTransaction, or an error if the RPC was
// interrupted.
Recv() (*binlogdatapb.BinlogTransaction, error)
}
// Client is the interface all clients must satisfy
type Client interface {
// Dial a server
Dial(endPoint *topodatapb.EndPoint, connTimeout time.Duration) error
// Close the connection
Close()
// Ask the server to stream binlog updates.
// Should return context.Canceled if the context is canceled.
ServeUpdateStream(ctx context.Context, position string) (StreamEventStream, error)
// Ask the server to stream updates related to the provided tables.
// Should return context.Canceled if the context is canceled.
StreamTables(ctx context.Context, position string, tables []string, charset *binlogdatapb.Charset) (BinlogTransactionStream, error)
// Ask the server to stream updates related to the provided keyrange.
// Should return context.Canceled if the context is canceled.
StreamKeyRange(ctx context.Context, position string, keyRange *topodatapb.KeyRange, charset *binlogdatapb.Charset) (BinlogTransactionStream, error)
}
// ClientFactory is the factory method to create a Client
type ClientFactory func() Client
var clientFactories = make(map[string]ClientFactory)
// RegisterClientFactory adds a new factory. Call during init().
func RegisterClientFactory(name string, factory ClientFactory) {
if _, ok := clientFactories[name]; ok {
log.Fatalf("ClientFactory %s already exists", name)
}
clientFactories[name] = factory
}