62 lines
1.4 KiB
Go
62 lines
1.4 KiB
Go
package main
|
|
|
|
import (
|
|
"context"
|
|
"encoding/json"
|
|
"flag"
|
|
"fmt"
|
|
"os"
|
|
"time"
|
|
|
|
pb "git.ipng.ch/ipng/nginx-logtail/proto/logtailpb"
|
|
)
|
|
|
|
func runTargets(args []string) {
|
|
fs := flag.NewFlagSet("targets", flag.ExitOnError)
|
|
sf, target := bindShared(fs)
|
|
fs.Usage = func() {
|
|
fmt.Fprintln(os.Stderr, "usage: logtail-cli targets [--target host:port] [--json]")
|
|
fs.PrintDefaults()
|
|
}
|
|
fs.Parse(args)
|
|
sf.resolve(*target)
|
|
|
|
for _, addr := range sf.targets {
|
|
conn, client, err := dial(addr)
|
|
if err != nil {
|
|
fmt.Fprintf(os.Stderr, "targets: cannot connect to %s: %v\n", addr, err)
|
|
continue
|
|
}
|
|
ctx, cancel := context.WithTimeout(context.Background(), 10*time.Second)
|
|
resp, err := client.ListTargets(ctx, &pb.ListTargetsRequest{})
|
|
cancel()
|
|
conn.Close()
|
|
if err != nil {
|
|
fmt.Fprintf(os.Stderr, "targets: %s: %v\n", addr, err)
|
|
continue
|
|
}
|
|
|
|
if sf.jsonOut {
|
|
type row struct {
|
|
QueryTarget string `json:"query_target"`
|
|
Name string `json:"name"`
|
|
Addr string `json:"addr"`
|
|
}
|
|
for _, t := range resp.Targets {
|
|
json.NewEncoder(os.Stdout).Encode(row{QueryTarget: addr, Name: t.Name, Addr: t.Addr})
|
|
}
|
|
} else {
|
|
if len(sf.targets) > 1 {
|
|
fmt.Println(targetHeader(addr, "", len(sf.targets)))
|
|
}
|
|
for _, t := range resp.Targets {
|
|
addrCol := t.Addr
|
|
if addrCol == "" {
|
|
addrCol = "(self)"
|
|
}
|
|
fmt.Printf("%-40s %s\n", t.Name, addrCol)
|
|
}
|
|
}
|
|
}
|
|
}
|