mirror of
https://codeberg.org/scip/esctl.git
synced 2026-08-24 19:44:18 +02:00
add datastream support, refactor custom completion (#32)
This commit is contained in:
@@ -59,7 +59,7 @@ func CcrStatus(conf *cfg.Config) *cli.Command {
|
|||||||
},
|
},
|
||||||
|
|
||||||
ShellComplete: func(ctx context.Context, cmd *cli.Command) {
|
ShellComplete: func(ctx context.Context, cmd *cli.Command) {
|
||||||
completeCluster(cmd)
|
complete(cmd, Ccluster)
|
||||||
},
|
},
|
||||||
|
|
||||||
Action: func(ctx context.Context, cmd *cli.Command) error {
|
Action: func(ctx context.Context, cmd *cli.Command) error {
|
||||||
@@ -115,7 +115,7 @@ func CcrRemoteInfo(conf *cfg.Config) *cli.Command {
|
|||||||
UsageText: "info [options] [<index>]",
|
UsageText: "info [options] [<index>]",
|
||||||
|
|
||||||
ShellComplete: func(ctx context.Context, cmd *cli.Command) {
|
ShellComplete: func(ctx context.Context, cmd *cli.Command) {
|
||||||
completeIndex(cmd)
|
complete(cmd, Cindex)
|
||||||
},
|
},
|
||||||
|
|
||||||
Action: func(ctx context.Context, cmd *cli.Command) error {
|
Action: func(ctx context.Context, cmd *cli.Command) error {
|
||||||
|
|||||||
@@ -60,7 +60,7 @@ func CcrFollowerRenew(conf *cfg.Config) *cli.Command {
|
|||||||
},
|
},
|
||||||
|
|
||||||
ShellComplete: func(ctx context.Context, cmd *cli.Command) {
|
ShellComplete: func(ctx context.Context, cmd *cli.Command) {
|
||||||
completeIndex(cmd)
|
complete(cmd, Cindex)
|
||||||
},
|
},
|
||||||
|
|
||||||
Action: func(ctx context.Context, cmd *cli.Command) error {
|
Action: func(ctx context.Context, cmd *cli.Command) error {
|
||||||
@@ -82,7 +82,7 @@ func CcrFollowerResume(conf *cfg.Config) *cli.Command {
|
|||||||
UsageText: "resume [options] <index>",
|
UsageText: "resume [options] <index>",
|
||||||
|
|
||||||
ShellComplete: func(ctx context.Context, cmd *cli.Command) {
|
ShellComplete: func(ctx context.Context, cmd *cli.Command) {
|
||||||
completeIndex(cmd)
|
complete(cmd, Cindex)
|
||||||
},
|
},
|
||||||
|
|
||||||
Action: func(ctx context.Context, cmd *cli.Command) error {
|
Action: func(ctx context.Context, cmd *cli.Command) error {
|
||||||
@@ -104,7 +104,7 @@ func CcrFollowerPause(conf *cfg.Config) *cli.Command {
|
|||||||
UsageText: "pause [options] <index>",
|
UsageText: "pause [options] <index>",
|
||||||
|
|
||||||
ShellComplete: func(ctx context.Context, cmd *cli.Command) {
|
ShellComplete: func(ctx context.Context, cmd *cli.Command) {
|
||||||
completeIndex(cmd)
|
complete(cmd, Cindex)
|
||||||
},
|
},
|
||||||
|
|
||||||
Action: func(ctx context.Context, cmd *cli.Command) error {
|
Action: func(ctx context.Context, cmd *cli.Command) error {
|
||||||
@@ -126,7 +126,7 @@ func CcrFollowerUnfollow(conf *cfg.Config) *cli.Command {
|
|||||||
UsageText: "unfollow [options] <index>",
|
UsageText: "unfollow [options] <index>",
|
||||||
|
|
||||||
ShellComplete: func(ctx context.Context, cmd *cli.Command) {
|
ShellComplete: func(ctx context.Context, cmd *cli.Command) {
|
||||||
completeIndex(cmd)
|
complete(cmd, Cindex)
|
||||||
},
|
},
|
||||||
|
|
||||||
Action: func(ctx context.Context, cmd *cli.Command) error {
|
Action: func(ctx context.Context, cmd *cli.Command) error {
|
||||||
@@ -158,7 +158,7 @@ func CcrFollowerAdd(conf *cfg.Config) *cli.Command {
|
|||||||
},
|
},
|
||||||
|
|
||||||
ShellComplete: func(ctx context.Context, cmd *cli.Command) {
|
ShellComplete: func(ctx context.Context, cmd *cli.Command) {
|
||||||
completeIndex(cmd)
|
complete(cmd, Cindex)
|
||||||
},
|
},
|
||||||
|
|
||||||
Action: func(ctx context.Context, cmd *cli.Command) error {
|
Action: func(ctx context.Context, cmd *cli.Command) error {
|
||||||
@@ -181,7 +181,7 @@ func CcrFollowerDelete(conf *cfg.Config) *cli.Command {
|
|||||||
UsageText: "delete <index>",
|
UsageText: "delete <index>",
|
||||||
|
|
||||||
ShellComplete: func(ctx context.Context, cmd *cli.Command) {
|
ShellComplete: func(ctx context.Context, cmd *cli.Command) {
|
||||||
completeIndex(cmd)
|
complete(cmd, Cindex)
|
||||||
},
|
},
|
||||||
|
|
||||||
Action: func(ctx context.Context, cmd *cli.Command) error {
|
Action: func(ctx context.Context, cmd *cli.Command) error {
|
||||||
@@ -205,7 +205,7 @@ func CcrFollowerShow(conf *cfg.Config) *cli.Command {
|
|||||||
UsageText: "show <index>",
|
UsageText: "show <index>",
|
||||||
|
|
||||||
ShellComplete: func(ctx context.Context, cmd *cli.Command) {
|
ShellComplete: func(ctx context.Context, cmd *cli.Command) {
|
||||||
completeIndex(cmd)
|
complete(cmd, Cindex)
|
||||||
},
|
},
|
||||||
|
|
||||||
Action: func(ctx context.Context, cmd *cli.Command) error {
|
Action: func(ctx context.Context, cmd *cli.Command) error {
|
||||||
|
|||||||
@@ -25,7 +25,14 @@ import (
|
|||||||
"github.com/urfave/cli/v3"
|
"github.com/urfave/cli/v3"
|
||||||
)
|
)
|
||||||
|
|
||||||
func completeIndex(cmd *cli.Command) {
|
const (
|
||||||
|
Cindex = iota
|
||||||
|
Cdatastream
|
||||||
|
Crole
|
||||||
|
Ccluster
|
||||||
|
)
|
||||||
|
|
||||||
|
func complete(cmd *cli.Command, what int) {
|
||||||
if cmd.NArg() > 0 {
|
if cmd.NArg() > 0 {
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
@@ -37,53 +44,27 @@ func completeIndex(cmd *cli.Command) {
|
|||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
indices, err := es.IndexNames(conf)
|
var list []string
|
||||||
|
var err error
|
||||||
|
|
||||||
|
switch what {
|
||||||
|
case Cindex:
|
||||||
|
list, err = es.IndexNames(conf)
|
||||||
|
case Crole:
|
||||||
|
list, err = es.RoleNames(conf)
|
||||||
|
case Ccluster:
|
||||||
|
for cluster := range conf.Clusters {
|
||||||
|
list = append(list, cluster)
|
||||||
|
}
|
||||||
|
case Cdatastream:
|
||||||
|
list, err = es.DatastreamNames(conf)
|
||||||
|
}
|
||||||
|
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
for _, index := range indices {
|
for _, item := range list {
|
||||||
fmt.Println(index)
|
fmt.Println(item)
|
||||||
}
|
|
||||||
|
|
||||||
}
|
|
||||||
|
|
||||||
func completeRole(cmd *cli.Command) {
|
|
||||||
if cmd.NArg() > 0 {
|
|
||||||
return
|
|
||||||
}
|
|
||||||
|
|
||||||
// FIXME: config should load from root.Before(), see https://github.com/urfave/cli/issues/2348
|
|
||||||
// workaround: load it directly here
|
|
||||||
conf := cfg.NewConfig()
|
|
||||||
if err := conf.Init(); err != nil {
|
|
||||||
return
|
|
||||||
}
|
|
||||||
|
|
||||||
roles, err := es.RoleNames(conf)
|
|
||||||
if err != nil {
|
|
||||||
return
|
|
||||||
}
|
|
||||||
|
|
||||||
for _, role := range roles {
|
|
||||||
fmt.Println(role)
|
|
||||||
}
|
|
||||||
|
|
||||||
}
|
|
||||||
|
|
||||||
func completeCluster(cmd *cli.Command) {
|
|
||||||
if cmd.NArg() > 0 {
|
|
||||||
return
|
|
||||||
}
|
|
||||||
|
|
||||||
// FIXME: config should load from root.Before(), see https://github.com/urfave/cli/issues/2348
|
|
||||||
// workaround: load it directly here
|
|
||||||
conf := cfg.NewConfig()
|
|
||||||
if err := conf.Init(); err != nil {
|
|
||||||
return
|
|
||||||
}
|
|
||||||
|
|
||||||
for cluster := range conf.Clusters {
|
|
||||||
fmt.Println(cluster)
|
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|||||||
140
cmd/datastream.go
Normal file
140
cmd/datastream.go
Normal file
@@ -0,0 +1,140 @@
|
|||||||
|
/*
|
||||||
|
Copyright © 2026 Thomas von Dein
|
||||||
|
|
||||||
|
This program is free software: you can redistribute it and/or modify
|
||||||
|
it under the terms of the GNU General Public License as published by
|
||||||
|
the Free Software Foundation, either version 3 of the License, or
|
||||||
|
(at your option) any later version.
|
||||||
|
|
||||||
|
This program is distributed in the hope that it will be useful,
|
||||||
|
but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||||||
|
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
||||||
|
GNU General Public License for more details.
|
||||||
|
|
||||||
|
You should have received a copy of the GNU General Public License
|
||||||
|
along with this program. If not, see <http://www.gnu.org/licenses/>.
|
||||||
|
*/
|
||||||
|
package cmd
|
||||||
|
|
||||||
|
import (
|
||||||
|
"context"
|
||||||
|
"errors"
|
||||||
|
|
||||||
|
"codeberg.org/scip/esctl/pkg/cfg"
|
||||||
|
"codeberg.org/scip/esctl/pkg/es"
|
||||||
|
|
||||||
|
"github.com/urfave/cli/v3"
|
||||||
|
)
|
||||||
|
|
||||||
|
func Datastream(conf *cfg.Config) *cli.Command {
|
||||||
|
return &cli.Command{
|
||||||
|
Name: "datastream",
|
||||||
|
Aliases: []string{"ds"},
|
||||||
|
Usage: "manage data streams",
|
||||||
|
|
||||||
|
Commands: []*cli.Command{
|
||||||
|
DatastreamList(conf),
|
||||||
|
DatastreamShow(conf),
|
||||||
|
DatastreamCreate(conf),
|
||||||
|
DatastreamDelete(conf),
|
||||||
|
},
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func DatastreamList(conf *cfg.Config) *cli.Command {
|
||||||
|
return &cli.Command{
|
||||||
|
Name: "list",
|
||||||
|
Aliases: []string{"ls"},
|
||||||
|
Usage: "list indicies",
|
||||||
|
|
||||||
|
Flags: []cli.Flag{
|
||||||
|
&cli.IntFlag{
|
||||||
|
Name: "max",
|
||||||
|
Usage: "max number of items to process",
|
||||||
|
Destination: &conf.MaxItems,
|
||||||
|
Aliases: []string{"m"},
|
||||||
|
},
|
||||||
|
&cli.BoolFlag{
|
||||||
|
Name: "hidden",
|
||||||
|
Usage: "include hidden data streams",
|
||||||
|
Destination: &conf.Hidden,
|
||||||
|
Aliases: []string{"H"},
|
||||||
|
},
|
||||||
|
&cli.StringSliceFlag{
|
||||||
|
Name: "filter",
|
||||||
|
Usage: "show only data streams matching the filter",
|
||||||
|
Destination: &conf.Filter,
|
||||||
|
Aliases: []string{"F"},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
|
||||||
|
Action: func(ctx context.Context, cmd *cli.Command) error {
|
||||||
|
return es.DatastreamList(conf)
|
||||||
|
},
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func DatastreamShow(conf *cfg.Config) *cli.Command {
|
||||||
|
return &cli.Command{
|
||||||
|
Name: "show",
|
||||||
|
Aliases: []string{"sh"},
|
||||||
|
Usage: "show details about an data stream",
|
||||||
|
|
||||||
|
Action: func(ctx context.Context, cmd *cli.Command) error {
|
||||||
|
index := cmd.Args().Get(0)
|
||||||
|
if index == "" {
|
||||||
|
return errors.New("no data stream specified")
|
||||||
|
}
|
||||||
|
|
||||||
|
return es.DatastreamShow(conf, cmd.Args().Get(0))
|
||||||
|
},
|
||||||
|
|
||||||
|
ShellComplete: func(ctx context.Context, cmd *cli.Command) {
|
||||||
|
complete(cmd, Cdatastream)
|
||||||
|
},
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func DatastreamCreate(conf *cfg.Config) *cli.Command {
|
||||||
|
return &cli.Command{
|
||||||
|
Name: "create",
|
||||||
|
Aliases: []string{"+"},
|
||||||
|
Usage: "create a new data stream",
|
||||||
|
UsageText: "create [options] <name>",
|
||||||
|
|
||||||
|
Action: func(ctx context.Context, cmd *cli.Command) error {
|
||||||
|
index := cmd.Args().Get(0)
|
||||||
|
if index == "" {
|
||||||
|
return errors.New("no data stream specified")
|
||||||
|
}
|
||||||
|
|
||||||
|
return es.DatastreamCreate(conf, cmd.Args().Get(0))
|
||||||
|
},
|
||||||
|
|
||||||
|
ShellComplete: func(ctx context.Context, cmd *cli.Command) {
|
||||||
|
complete(cmd, Cdatastream)
|
||||||
|
},
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func DatastreamDelete(conf *cfg.Config) *cli.Command {
|
||||||
|
return &cli.Command{
|
||||||
|
Name: "delete",
|
||||||
|
Aliases: []string{"rm"},
|
||||||
|
Usage: "delete a data stream",
|
||||||
|
UsageText: "delete [options] <name>",
|
||||||
|
|
||||||
|
Action: func(ctx context.Context, cmd *cli.Command) error {
|
||||||
|
index := cmd.Args().Get(0)
|
||||||
|
if index == "" {
|
||||||
|
return errors.New("no data stream specified")
|
||||||
|
}
|
||||||
|
|
||||||
|
return es.DatastreamDelete(conf, cmd.Args().Get(0))
|
||||||
|
},
|
||||||
|
|
||||||
|
ShellComplete: func(ctx context.Context, cmd *cli.Command) {
|
||||||
|
complete(cmd, Cdatastream)
|
||||||
|
},
|
||||||
|
}
|
||||||
|
}
|
||||||
10
cmd/index.go
10
cmd/index.go
@@ -134,7 +134,7 @@ func IndexShow(conf *cfg.Config) *cli.Command {
|
|||||||
},
|
},
|
||||||
|
|
||||||
ShellComplete: func(ctx context.Context, cmd *cli.Command) {
|
ShellComplete: func(ctx context.Context, cmd *cli.Command) {
|
||||||
completeIndex(cmd)
|
complete(cmd, Cindex)
|
||||||
},
|
},
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
@@ -189,7 +189,7 @@ func IndexDelete(conf *cfg.Config) *cli.Command {
|
|||||||
UsageText: "delete <index>",
|
UsageText: "delete <index>",
|
||||||
|
|
||||||
ShellComplete: func(ctx context.Context, cmd *cli.Command) {
|
ShellComplete: func(ctx context.Context, cmd *cli.Command) {
|
||||||
completeIndex(cmd)
|
complete(cmd, Cindex)
|
||||||
},
|
},
|
||||||
|
|
||||||
Action: func(ctx context.Context, cmd *cli.Command) error {
|
Action: func(ctx context.Context, cmd *cli.Command) error {
|
||||||
@@ -210,7 +210,7 @@ func IndexClose(conf *cfg.Config) *cli.Command {
|
|||||||
UsageText: "close <index>",
|
UsageText: "close <index>",
|
||||||
|
|
||||||
ShellComplete: func(ctx context.Context, cmd *cli.Command) {
|
ShellComplete: func(ctx context.Context, cmd *cli.Command) {
|
||||||
completeIndex(cmd)
|
complete(cmd, Cindex)
|
||||||
},
|
},
|
||||||
|
|
||||||
Action: func(ctx context.Context, cmd *cli.Command) error {
|
Action: func(ctx context.Context, cmd *cli.Command) error {
|
||||||
@@ -231,7 +231,7 @@ func IndexModify(conf *cfg.Config) *cli.Command {
|
|||||||
UsageText: "modify <index[,index,...]|_all>",
|
UsageText: "modify <index[,index,...]|_all>",
|
||||||
|
|
||||||
ShellComplete: func(ctx context.Context, cmd *cli.Command) {
|
ShellComplete: func(ctx context.Context, cmd *cli.Command) {
|
||||||
completeIndex(cmd)
|
complete(cmd, Cindex)
|
||||||
},
|
},
|
||||||
|
|
||||||
Flags: []cli.Flag{
|
Flags: []cli.Flag{
|
||||||
@@ -282,7 +282,7 @@ func IndexFields(conf *cfg.Config) *cli.Command {
|
|||||||
},
|
},
|
||||||
|
|
||||||
ShellComplete: func(ctx context.Context, cmd *cli.Command) {
|
ShellComplete: func(ctx context.Context, cmd *cli.Command) {
|
||||||
completeIndex(cmd)
|
complete(cmd, Cindex)
|
||||||
},
|
},
|
||||||
|
|
||||||
Action: func(ctx context.Context, cmd *cli.Command) error {
|
Action: func(ctx context.Context, cmd *cli.Command) error {
|
||||||
|
|||||||
@@ -51,7 +51,7 @@ func IndexAliasCreate(conf *cfg.Config) *cli.Command {
|
|||||||
UsageText: "create <index> <alias>",
|
UsageText: "create <index> <alias>",
|
||||||
|
|
||||||
ShellComplete: func(ctx context.Context, cmd *cli.Command) {
|
ShellComplete: func(ctx context.Context, cmd *cli.Command) {
|
||||||
completeIndex(cmd)
|
complete(cmd, Cindex)
|
||||||
},
|
},
|
||||||
|
|
||||||
Action: func(ctx context.Context, cmd *cli.Command) error {
|
Action: func(ctx context.Context, cmd *cli.Command) error {
|
||||||
@@ -75,7 +75,7 @@ func IndexAliasDelete(conf *cfg.Config) *cli.Command {
|
|||||||
UsageText: "delete <index> <alias>",
|
UsageText: "delete <index> <alias>",
|
||||||
|
|
||||||
ShellComplete: func(ctx context.Context, cmd *cli.Command) {
|
ShellComplete: func(ctx context.Context, cmd *cli.Command) {
|
||||||
completeIndex(cmd)
|
complete(cmd, Cindex)
|
||||||
},
|
},
|
||||||
|
|
||||||
Action: func(ctx context.Context, cmd *cli.Command) error {
|
Action: func(ctx context.Context, cmd *cli.Command) error {
|
||||||
|
|||||||
@@ -68,7 +68,7 @@ func RoleShow(conf *cfg.Config) *cli.Command {
|
|||||||
UsageText: "show [options] <role>",
|
UsageText: "show [options] <role>",
|
||||||
|
|
||||||
ShellComplete: func(ctx context.Context, cmd *cli.Command) {
|
ShellComplete: func(ctx context.Context, cmd *cli.Command) {
|
||||||
completeRole(cmd)
|
complete(cmd, Crole)
|
||||||
},
|
},
|
||||||
|
|
||||||
Action: func(ctx context.Context, cmd *cli.Command) error {
|
Action: func(ctx context.Context, cmd *cli.Command) error {
|
||||||
|
|||||||
15
cmd/root.go
15
cmd/root.go
@@ -90,6 +90,7 @@ func Main() int {
|
|||||||
Cluster(conf),
|
Cluster(conf),
|
||||||
Ccr(conf),
|
Ccr(conf),
|
||||||
Index(conf),
|
Index(conf),
|
||||||
|
Datastream(conf),
|
||||||
Shard(conf),
|
Shard(conf),
|
||||||
Snapshot(conf),
|
Snapshot(conf),
|
||||||
Node(conf),
|
Node(conf),
|
||||||
@@ -146,6 +147,20 @@ friends.#(first%"D*").last >> "Murphy"
|
|||||||
friends.#(first!%"D*").last >> "Craig"
|
friends.#(first!%"D*").last >> "Craig"
|
||||||
friends.#(nets.#(=="fb"))#.first >> ["Dale","Roger"]
|
friends.#(nets.#(=="fb"))#.first >> ["Dale","Roger"]
|
||||||
|
|
||||||
|
To extract more than one field, use:
|
||||||
|
|
||||||
|
{"ns":source.namespace_name,"time":source.@timestamp} >>
|
||||||
|
|
||||||
|
{
|
||||||
|
"ns": "3f80316965c64405-275a75782e984bcc8d82",
|
||||||
|
"time": "2026-06-10T04:23:38.914080361+00:00"
|
||||||
|
}
|
||||||
|
{
|
||||||
|
"ns": "3f80316965c64405-f0e1dcef668e45bb9afd",
|
||||||
|
"time": "2026-06-10T09:39:00.554892542+00:00"
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
Documentation: https://github.com/tidwall/gjson/blob/master/SYNTAX.md`
|
Documentation: https://github.com/tidwall/gjson/blob/master/SYNTAX.md`
|
||||||
|
|
||||||
return &cli.Command{
|
return &cli.Command{
|
||||||
|
|||||||
@@ -34,7 +34,7 @@ import (
|
|||||||
)
|
)
|
||||||
|
|
||||||
const (
|
const (
|
||||||
Version string = `v0.0.18`
|
Version string = `v0.0.19`
|
||||||
)
|
)
|
||||||
|
|
||||||
var (
|
var (
|
||||||
@@ -83,6 +83,7 @@ type Config struct {
|
|||||||
NotDeployed bool // role diff: -n
|
NotDeployed bool // role diff: -n
|
||||||
Undefined bool // role diff: -u
|
Undefined bool // role diff: -u
|
||||||
Diff bool // role diff: -D
|
Diff bool // role diff: -D
|
||||||
|
Hidden bool // ds ls: -H
|
||||||
}
|
}
|
||||||
|
|
||||||
func NewConfig() *Config {
|
func NewConfig() *Config {
|
||||||
|
|||||||
239
pkg/es/datastream.go
Normal file
239
pkg/es/datastream.go
Normal file
@@ -0,0 +1,239 @@
|
|||||||
|
/*
|
||||||
|
Copyright © 2026 Thomas von Dein
|
||||||
|
|
||||||
|
This program is free software: you can redistribute it and/or modify
|
||||||
|
it under the terms of the GNU General Public License as published by
|
||||||
|
the Free Software Foundation, either version 3 of the License, or
|
||||||
|
(at your option) any later version.
|
||||||
|
|
||||||
|
This program is distributed in the hope that it will be useful,
|
||||||
|
but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||||||
|
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
||||||
|
GNU General Public License for more details.
|
||||||
|
|
||||||
|
You should have received a copy of the GNU General Public License
|
||||||
|
along with this program. If not, see <http://www.gnu.org/licenses/>.
|
||||||
|
*/
|
||||||
|
package es
|
||||||
|
|
||||||
|
import (
|
||||||
|
"context"
|
||||||
|
"errors"
|
||||||
|
"fmt"
|
||||||
|
"log/slog"
|
||||||
|
"regexp"
|
||||||
|
|
||||||
|
"codeberg.org/scip/esctl/pkg/cfg"
|
||||||
|
"codeberg.org/scip/esctl/pkg/printer"
|
||||||
|
"github.com/dustin/go-humanize"
|
||||||
|
"github.com/elastic/go-elasticsearch/v9/typedapi/types"
|
||||||
|
)
|
||||||
|
|
||||||
|
func DatastreamNames(conf *cfg.Config) ([]string, error) {
|
||||||
|
res, err := conf.DefaultCluster.ES.Indices.GetDataStream().
|
||||||
|
Header("content-type", "application/json").
|
||||||
|
Header("accept", "application/json").
|
||||||
|
Do(context.Background())
|
||||||
|
if err != nil {
|
||||||
|
return nil, fmt.Errorf("failed to get data streams: %s", esErrorString(err))
|
||||||
|
}
|
||||||
|
|
||||||
|
dss := make([]string, len(res.DataStreams))
|
||||||
|
for idx, ds := range res.DataStreams {
|
||||||
|
dss[idx] = ds.Name
|
||||||
|
}
|
||||||
|
|
||||||
|
return dss, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func DatastreamList(conf *cfg.Config) error {
|
||||||
|
res, err := conf.DefaultCluster.ES.Indices.GetDataStream().
|
||||||
|
Header("content-type", "application/json").
|
||||||
|
Header("accept", "application/json").
|
||||||
|
Do(context.Background())
|
||||||
|
if err != nil {
|
||||||
|
return fmt.Errorf("failed to get data streams: %s", esErrorString(err))
|
||||||
|
}
|
||||||
|
|
||||||
|
slog.Debug("ES result", "data streams", res)
|
||||||
|
|
||||||
|
list := filterDatastreams(conf, res.DataStreams)
|
||||||
|
|
||||||
|
size := len(list)
|
||||||
|
|
||||||
|
if conf.MaxItems > 0 {
|
||||||
|
if size > conf.MaxItems {
|
||||||
|
size = conf.MaxItems
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
table := printer.NewTable(conf, 7, size)
|
||||||
|
table.Addheaders("name", "ilm policy", "hidden", "system", "replicated", "generation", "timestamp field")
|
||||||
|
|
||||||
|
for idx, ds := range list {
|
||||||
|
name := printer.Colorize(conf, ds.Status.String(), ds.Name)
|
||||||
|
policy := ""
|
||||||
|
if ds.IlmPolicy != nil {
|
||||||
|
policy = *ds.IlmPolicy
|
||||||
|
}
|
||||||
|
|
||||||
|
table.Entries[idx] = []string{
|
||||||
|
name,
|
||||||
|
policy,
|
||||||
|
fmt.Sprintf("%t", ds.Hidden),
|
||||||
|
fmt.Sprintf("%t", *ds.System),
|
||||||
|
fmt.Sprintf("%t", *ds.Replicated),
|
||||||
|
fmt.Sprintf("%d", ds.Generation),
|
||||||
|
ds.TimestampField.Name,
|
||||||
|
}
|
||||||
|
|
||||||
|
if idx == size-1 {
|
||||||
|
break
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
table.Sort()
|
||||||
|
if err := table.Print(); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func filterDatastreams(conf *cfg.Config, list []types.DataStream) []types.DataStream {
|
||||||
|
selected := []types.DataStream{}
|
||||||
|
|
||||||
|
for _, ds := range list {
|
||||||
|
if !conf.Hidden && ds.Hidden {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
|
||||||
|
selected = append(selected, ds)
|
||||||
|
}
|
||||||
|
|
||||||
|
if len(conf.Filter) == 0 {
|
||||||
|
return selected
|
||||||
|
}
|
||||||
|
|
||||||
|
// we support just one filter here, for now
|
||||||
|
filter := *regexp.MustCompile(conf.Filter[0])
|
||||||
|
newlist := []types.DataStream{}
|
||||||
|
|
||||||
|
for _, ds := range selected {
|
||||||
|
if filter.MatchString(ds.Name) {
|
||||||
|
newlist = append(newlist, ds)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
return newlist
|
||||||
|
}
|
||||||
|
|
||||||
|
func DatastreamShow(conf *cfg.Config, dsname string) error {
|
||||||
|
res, err := conf.DefaultCluster.ES.Indices.GetDataStream().
|
||||||
|
Name(dsname).
|
||||||
|
Header("content-type", "application/json").
|
||||||
|
Header("accept", "application/json").
|
||||||
|
Do(context.Background())
|
||||||
|
if err != nil {
|
||||||
|
return fmt.Errorf("failed to get data stream: %s", esErrorString(err))
|
||||||
|
}
|
||||||
|
|
||||||
|
slog.Debug("ES result", "data stream", res)
|
||||||
|
|
||||||
|
if len(res.DataStreams) == 0 {
|
||||||
|
return errors.New("no data stream found with that name")
|
||||||
|
}
|
||||||
|
|
||||||
|
stats, err := conf.DefaultCluster.ES.Indices.DataStreamsStats().
|
||||||
|
Name(dsname).
|
||||||
|
Header("content-type", "application/json").
|
||||||
|
Header("accept", "application/json").
|
||||||
|
Do(context.Background())
|
||||||
|
if err != nil {
|
||||||
|
return fmt.Errorf("failed to get data stream stats: %s", esErrorString(err))
|
||||||
|
}
|
||||||
|
|
||||||
|
table := printer.NewTable(conf, 11, 2)
|
||||||
|
table.Addheaders("data stream property", "value")
|
||||||
|
|
||||||
|
ds := res.DataStreams[0]
|
||||||
|
|
||||||
|
name := printer.Colorize(conf, ds.Status.String(), ds.Name)
|
||||||
|
policy := ""
|
||||||
|
if ds.IlmPolicy != nil {
|
||||||
|
policy = *ds.IlmPolicy
|
||||||
|
}
|
||||||
|
|
||||||
|
table.Entries = [][]string{
|
||||||
|
{"name", name},
|
||||||
|
{"ilm policy", policy},
|
||||||
|
{"hidden", fmt.Sprintf("%t", ds.Hidden)},
|
||||||
|
{"system", fmt.Sprintf("%t", *ds.System)},
|
||||||
|
{"replicated", fmt.Sprintf("%t", *ds.Replicated)},
|
||||||
|
{"generation", fmt.Sprintf("%d", ds.Generation)},
|
||||||
|
{"timestamp field", ds.TimestampField.Name},
|
||||||
|
{"backing indices", fmt.Sprintf("%d", stats.BackingIndices)},
|
||||||
|
{"size", humanize.Bytes(uint64(stats.TotalStoreSizeBytes))},
|
||||||
|
{"shards-failed", fmt.Sprintf("%d", stats.Shards_.Failed)},
|
||||||
|
{"shards-successful", fmt.Sprintf("%d", stats.Shards_.Successful)},
|
||||||
|
}
|
||||||
|
|
||||||
|
if err := table.Print(); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
table = printer.NewTable(conf, 5, len(ds.Indices))
|
||||||
|
table.Addheaders("backing index name", "uuid", "prefer ilm", "ilm policy", "managed by")
|
||||||
|
|
||||||
|
for idx, index := range ds.Indices {
|
||||||
|
policy := ""
|
||||||
|
if ds.IlmPolicy != nil {
|
||||||
|
policy = *ds.IlmPolicy
|
||||||
|
}
|
||||||
|
|
||||||
|
table.Entries[idx] = []string{
|
||||||
|
index.IndexName,
|
||||||
|
index.IndexUuid,
|
||||||
|
fmt.Sprintf("%t", *index.PreferIlm),
|
||||||
|
policy,
|
||||||
|
index.ManagedBy.Name,
|
||||||
|
}
|
||||||
|
|
||||||
|
if idx < len(ds.Indices) {
|
||||||
|
fmt.Println()
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
table.Sort()
|
||||||
|
if err := table.Print(); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func DatastreamCreate(conf *cfg.Config, dsname string) error {
|
||||||
|
_, err := conf.DefaultCluster.ES.Indices.CreateDataStream(dsname).
|
||||||
|
Header("content-type", "application/json").
|
||||||
|
Header("accept", "application/json").
|
||||||
|
Do(context.Background())
|
||||||
|
|
||||||
|
if err != nil {
|
||||||
|
return fmt.Errorf("failed to create datastream: %s", esErrorString(err))
|
||||||
|
}
|
||||||
|
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func DatastreamDelete(conf *cfg.Config, dsname string) error {
|
||||||
|
_, err := conf.DefaultCluster.ES.Indices.DeleteDataStream(dsname).
|
||||||
|
Header("content-type", "application/json").
|
||||||
|
Header("accept", "application/json").
|
||||||
|
Do(context.Background())
|
||||||
|
|
||||||
|
if err != nil {
|
||||||
|
return fmt.Errorf("failed to delete datastream: %s", esErrorString(err))
|
||||||
|
}
|
||||||
|
|
||||||
|
return nil
|
||||||
|
}
|
||||||
@@ -59,6 +59,12 @@ func filterIndices(conf *cfg.Config, list indices.Response) indices.Response {
|
|||||||
if !conf.Partials && strings.HasPrefix(*index.Index, "partial-") {
|
if !conf.Partials && strings.HasPrefix(*index.Index, "partial-") {
|
||||||
continue
|
continue
|
||||||
}
|
}
|
||||||
|
|
||||||
|
if strings.HasPrefix(*index.Index, ".ds-") {
|
||||||
|
// ignore data stream backing indicies
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
|
||||||
selectedlist = append(selectedlist, index)
|
selectedlist = append(selectedlist, index)
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|||||||
Reference in New Issue
Block a user