/
alias.go
70 lines (59 loc) · 1.36 KB
/
alias.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
package main
import (
"fmt"
"io"
"os"
"github.com/kaneshin/esindex/elasticsearch"
)
type Alias struct {
}
func NewAlias() *Alias {
return &Alias{}
}
func (cmd *Alias) Run(args []string) error {
only := cmdFlag.Bool("only", true, "")
if len(args) < 1 {
cmd.Usage(os.Stdout)
return nil
}
indexName := args[0]
aliasName := aliasNameFromIndexName(indexName)
cmdFlag.Parse(args[1:])
client := elasticsearch.NewClient(
elasticsearch.NewConfig().WithURL(*urlStr),
)
_, aliased := indexNamesAndAliasedIndexByAliasName(client, aliasName)
if len(aliased) == 0 {
if err := makeAlias(client, aliasName, indexName); err != nil {
return err
}
fmt.Printf("%s <- %s\n", indexName, aliasName)
return nil
}
if *only {
addActions := []Action{
Action{aliasName, indexName},
}
removeActions := []Action{}
for name, _ := range aliased {
if name != indexName {
removeActions = append(removeActions, Action{aliasName, name})
}
}
if err := updateAliases(client, addActions, removeActions); err != nil {
return err
}
} else {
if err := makeAlias(client, aliasName, indexName); err != nil {
return err
}
}
fmt.Printf("%s <- %s\n", indexName, aliasName)
return nil
}
func (cmd *Alias) Name() string {
return "alias"
}
func (cmd *Alias) Usage(w io.Writer) {
fmt.Fprintf(w, "%-8s%10s %s", cmd.Name(), "<index-name>", "[options]")
}