forked from dolthub/vitess
-
Notifications
You must be signed in to change notification settings - Fork 0
/
shard.go
61 lines (53 loc) · 1.91 KB
/
shard.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
// Copyright 2015, 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 topoproto
import (
"encoding/hex"
"fmt"
"html/template"
"strings"
"github.com/youtube/vitess/go/vt/key"
topodatapb "github.com/youtube/vitess/go/vt/proto/topodata"
)
// KeyspaceShardString returns a "keyspace/shard" string taking
// keyspace and shard as separate inputs.
func KeyspaceShardString(keyspace, shard string) string {
return fmt.Sprintf("%v/%v", keyspace, shard)
}
// ParseKeyspaceShard parse a "keyspace/shard" string and extract
// both keyspace and shard
func ParseKeyspaceShard(param string) (string, string, error) {
keySpaceShard := strings.Split(param, "/")
if len(keySpaceShard) != 2 {
return "", "", fmt.Errorf("Invalid shard path: %v", param)
}
return keySpaceShard[0], keySpaceShard[1], nil
}
// SourceShardString returns a printable view of a SourceShard.
func SourceShardString(source *topodatapb.Shard_SourceShard) string {
return fmt.Sprintf("SourceShard(%v,%v/%v)", source.Uid, source.Keyspace, source.Shard)
}
// SourceShardAsHTML returns a HTML version of the object.
func SourceShardAsHTML(source *topodatapb.Shard_SourceShard) template.HTML {
result := fmt.Sprintf("<b>Uid</b>: %v</br>\n<b>Source</b>: %v/%v</br>\n", source.Uid, source.Keyspace, source.Shard)
if key.KeyRangeIsPartial(source.KeyRange) {
result += fmt.Sprintf("<b>KeyRange</b>: %v-%v</br>\n",
hex.EncodeToString(source.KeyRange.Start),
hex.EncodeToString(source.KeyRange.End))
}
if len(source.Tables) > 0 {
result += fmt.Sprintf("<b>Tables</b>: %v</br>\n",
strings.Join(source.Tables, " "))
}
return template.HTML(result)
}
// ShardHasCell returns true if the cell is listed in the Cells for the shard.
func ShardHasCell(shard *topodatapb.Shard, cell string) bool {
for _, c := range shard.Cells {
if c == cell {
return true
}
}
return false
}