mirror of
https://codeberg.org/scip/esctl.git
synced 2026-08-24 07:54:17 +02:00
107 lines
3.0 KiB
Go
107 lines
3.0 KiB
Go
/*
|
|
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"
|
|
"fmt"
|
|
"log/slog"
|
|
|
|
"codeberg.org/scip/esctl/pkg/cfg"
|
|
)
|
|
|
|
func CcrFollowerAdd(conf *cfg.Config, index string) error {
|
|
res, err := conf.DefaultCluster.ES.Cluster.RemoteInfo().
|
|
Header("content-type", "application/json").
|
|
Header("accept", "application/json").
|
|
Do(context.Background())
|
|
if err != nil {
|
|
return fmt.Errorf("failed to retrieve follower info: %s", err)
|
|
}
|
|
|
|
remote := ""
|
|
for name := range res {
|
|
remote = name
|
|
break
|
|
}
|
|
|
|
if remote == "" {
|
|
return fmt.Errorf("cluster doesn't have a follower: %s", err)
|
|
}
|
|
|
|
create := conf.DefaultCluster.ES.Ccr.Follow(index).
|
|
LeaderIndex(index).
|
|
RemoteCluster(remote).
|
|
Header("content-type", "application/json").
|
|
Header("accept", "application/json")
|
|
|
|
if conf.Wait {
|
|
create.WaitForActiveShards("all")
|
|
}
|
|
|
|
_, err = create.Do(context.Background())
|
|
|
|
if err != nil {
|
|
return fmt.Errorf("failed to create follower index: %s", err)
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func CcrFollowerShow(conf *cfg.Config, index string) error {
|
|
res, err := conf.DefaultCluster.ES.Ccr.FollowStats(index).
|
|
Header("content-type", "application/json").
|
|
Header("accept", "application/json").
|
|
Do(context.Background())
|
|
if err != nil {
|
|
return fmt.Errorf("failed to retrieve follower index info: %s", err)
|
|
}
|
|
|
|
slog.Debug("ES result", "follower stats", res.Indices)
|
|
|
|
if len(res.Indices) == 0 {
|
|
return fmt.Errorf("cluster did not return any follower stats for index %s", index)
|
|
}
|
|
|
|
if len(res.Indices[0].Shards) == 0 {
|
|
return fmt.Errorf("cluster did not return any shard stats on follower index %s", index)
|
|
}
|
|
|
|
follower := res.Indices[0].Shards[0]
|
|
|
|
table := NewTable(2, 9)
|
|
table.Addheaders("field", "value")
|
|
|
|
table.entries = [][]string{
|
|
{"name", index},
|
|
{"remote_cluster", follower.RemoteCluster},
|
|
{"leader_checkpoint", fmt.Sprintf("%d", follower.LeaderGlobalCheckpoint)},
|
|
{"follower_checkpoint", fmt.Sprintf("%d", follower.FollowerGlobalCheckpoint)},
|
|
{"bytes_read", fmt.Sprintf("%d", follower.BytesRead)},
|
|
{"failed_read_requests", fmt.Sprintf("%d", follower.FailedReadRequests)},
|
|
{"failed_write_requests", fmt.Sprintf("%d", follower.FailedWriteRequests)},
|
|
{"successful_read_requests", fmt.Sprintf("%d", follower.SuccessfulReadRequests)},
|
|
{"successful_write_requests", fmt.Sprintf("%d", follower.SuccessfulWriteRequests)},
|
|
}
|
|
|
|
if err := table.PrintMarkdown(); err != nil {
|
|
return err
|
|
}
|
|
|
|
return nil
|
|
}
|