You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
194 lines
4.8 KiB
194 lines
4.8 KiB
package main
|
|
|
|
import (
|
|
"context"
|
|
"flag"
|
|
"fmt"
|
|
"os"
|
|
"strings"
|
|
)
|
|
|
|
func handleAliasCommand() {
|
|
if len(os.Args) < 3 {
|
|
fmt.Fprintf(os.Stderr, "%s alias: subcommand missing\n", os.Args[0])
|
|
printAliasUsage()
|
|
os.Exit(1)
|
|
}
|
|
|
|
subcmd := strings.ToLower(os.Args[2])
|
|
switch subcmd {
|
|
case "ls":
|
|
fallthrough
|
|
case "list":
|
|
cmd := flag.NewFlagSet("alias list", flag.ExitOnError)
|
|
cmd.SetOutput(os.Stdout)
|
|
cfgpath := cmd.String("config", "", "path to configuration")
|
|
server := cmd.String("server", "", "server name to use")
|
|
aname := cmd.String("name", "", "filter list to named alias")
|
|
cmd.Parse(os.Args[3:])
|
|
|
|
cfg := LoadConfig(*cfgpath)
|
|
es := cfg.GetClient(*server)
|
|
|
|
aliases, err := es.CatAliases().Do(context.Background())
|
|
if err != nil {
|
|
fmt.Fprintf(os.Stderr, "elastic error: %s\n", err.Error())
|
|
os.Exit(1)
|
|
}
|
|
|
|
for _, alias := range aliases {
|
|
if *aname != "" {
|
|
if alias.Alias == *aname {
|
|
fmt.Printf("%s\n", alias.Index)
|
|
}
|
|
} else {
|
|
fmt.Printf("%s %s\n", alias.Alias, alias.Index)
|
|
}
|
|
}
|
|
|
|
case "add":
|
|
cmd := flag.NewFlagSet("alias add", flag.ExitOnError)
|
|
cmd.SetOutput(os.Stdout)
|
|
cmd.Usage = func() {
|
|
fmt.Printf("Usage: %s alias %s [args] <alias> <index>...\n",
|
|
os.Args[0], subcmd)
|
|
cmd.PrintDefaults()
|
|
}
|
|
cfgpath := cmd.String("config", "", "path to configuration")
|
|
server := cmd.String("server", "", "server name to use")
|
|
cmd.Parse(os.Args[3:])
|
|
|
|
if len(cmd.Args()) < 2 {
|
|
fmt.Fprintf(os.Stderr, "alias and index required for alias add\n")
|
|
cmd.Usage()
|
|
os.Exit(1)
|
|
}
|
|
aname := cmd.Arg(0)
|
|
indexes := cmd.Args()[1:]
|
|
|
|
cfg := LoadConfig(*cfgpath)
|
|
es := cfg.GetClient(*server)
|
|
|
|
alias := es.Alias()
|
|
for _, index := range indexes {
|
|
alias.Add(index, aname)
|
|
}
|
|
_, err := alias.Do(context.Background())
|
|
if err != nil {
|
|
fmt.Fprintf(os.Stderr, "elastic error: %s\n", err.Error())
|
|
os.Exit(1)
|
|
}
|
|
|
|
case "del":
|
|
fallthrough
|
|
case "delete":
|
|
cmd := flag.NewFlagSet("alias delete", flag.ExitOnError)
|
|
cmd.SetOutput(os.Stdout)
|
|
cmd.Usage = func() {
|
|
fmt.Printf("Usage: %s alias %s [args] <alias> <index>...\n",
|
|
os.Args[0], subcmd)
|
|
cmd.PrintDefaults()
|
|
}
|
|
cfgpath := cmd.String("config", "", "path to configuration")
|
|
server := cmd.String("server", "", "server name to use")
|
|
cmd.Parse(os.Args[3:])
|
|
|
|
if len(cmd.Args()) < 2 {
|
|
fmt.Fprintf(os.Stderr, "alias and index required for alias delete\n")
|
|
cmd.Usage()
|
|
os.Exit(1)
|
|
}
|
|
|
|
aname := cmd.Arg(0)
|
|
indexes := cmd.Args()[1:]
|
|
|
|
cfg := LoadConfig(*cfgpath)
|
|
es := cfg.GetClient(*server)
|
|
|
|
alias := es.Alias()
|
|
for _, index := range indexes {
|
|
alias.Remove(index, aname)
|
|
}
|
|
_, err := alias.Do(context.Background())
|
|
if err != nil {
|
|
fmt.Fprintf(os.Stderr, "elastic error: %s\n", err.Error())
|
|
os.Exit(1)
|
|
}
|
|
|
|
case "replace":
|
|
cmd := flag.NewFlagSet("alias replace", flag.ExitOnError)
|
|
cmd.SetOutput(os.Stdout)
|
|
cfgpath := cmd.String("config", "", "path to configuration")
|
|
server := cmd.String("server", "", "server name to use")
|
|
aname := cmd.String("name", "", "(required) name of alias")
|
|
oiname := cmd.String("oldindex", "", "name of the old index")
|
|
niname := cmd.String("newindex", "",
|
|
"(required) name of the new index")
|
|
cmd.Parse(os.Args[3:])
|
|
|
|
if *aname == "" {
|
|
fmt.Fprintf(os.Stderr, "name parameter is required for alias replace\n")
|
|
cmd.PrintDefaults()
|
|
os.Exit(1)
|
|
}
|
|
|
|
if *niname == "" {
|
|
fmt.Fprintf(os.Stderr,
|
|
"newindex parameter is required for alias replace\n")
|
|
cmd.PrintDefaults()
|
|
os.Exit(1)
|
|
}
|
|
|
|
cfg := LoadConfig(*cfgpath)
|
|
es := cfg.GetClient(*server)
|
|
|
|
oinames := make([]string, 0, 10)
|
|
if *oiname != "" {
|
|
oinames = append(oinames, *oiname)
|
|
} else {
|
|
aliases, err := es.CatAliases().Do(context.Background())
|
|
if err != nil {
|
|
fmt.Fprintf(os.Stderr, "elastic error: %s\n", err.Error())
|
|
os.Exit(1)
|
|
}
|
|
|
|
for _, alias := range aliases {
|
|
if alias.Alias == *aname {
|
|
oinames = append(oinames, alias.Index)
|
|
}
|
|
}
|
|
}
|
|
|
|
as := es.Alias()
|
|
for _, idx := range oinames {
|
|
as.Remove(idx, *aname)
|
|
}
|
|
as.Add(*niname, *aname)
|
|
_, err := as.Do(context.Background())
|
|
if err != nil {
|
|
fmt.Fprintf(os.Stderr, "elastic error: %s\n", err.Error())
|
|
os.Exit(1)
|
|
}
|
|
|
|
case "--help":
|
|
printAliasUsage()
|
|
os.Exit(0)
|
|
|
|
default:
|
|
fmt.Fprintf(os.Stderr, "%s: alias '%s' is not a recognized subcommand\n",
|
|
os.Args[0], os.Args[2])
|
|
printAliasUsage()
|
|
os.Exit(1)
|
|
}
|
|
}
|
|
|
|
func printAliasUsage() {
|
|
fmt.Printf("Usage: %s alias <subcommand> ... \n", os.Args[0])
|
|
fmt.Printf("See '%s alias <subcommand> --help' for information", os.Args[0])
|
|
fmt.Printf(" on a specific command\n")
|
|
fmt.Printf("valid subcommands:\n")
|
|
fmt.Printf(" list list aliases\n")
|
|
fmt.Printf(" add add alias\n")
|
|
fmt.Printf(" delete delete alias\n")
|
|
fmt.Printf(" replace replace alias (single step add/delete)\n")
|
|
}
|