forked from cerc-io/plugeth
21acf0bc8d
This PR is replacing the metrics.influxdb.host.tag cmd-line flag with metrics.influxdb.tags - a comma-separated key/value tags, that are passed to the InfluxDB reporter, so that we can index measurements with multiple tags, and not just one host tag. This will be useful for Swarm, where we want to index measurements not just with the host tag, but also with bzzkey and git commit version (for long-running deployments).
198 lines
5.0 KiB
Go
198 lines
5.0 KiB
Go
// Copyright 2018 The go-ethereum Authors
|
|
// This file is part of go-ethereum.
|
|
//
|
|
// go-ethereum 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.
|
|
//
|
|
// go-ethereum 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 go-ethereum. If not, see <http://www.gnu.org/licenses/>.
|
|
|
|
package main
|
|
|
|
import (
|
|
"fmt"
|
|
"os"
|
|
"sort"
|
|
|
|
"github.com/ethereum/go-ethereum/cmd/utils"
|
|
gethmetrics "github.com/ethereum/go-ethereum/metrics"
|
|
"github.com/ethereum/go-ethereum/metrics/influxdb"
|
|
swarmmetrics "github.com/ethereum/go-ethereum/swarm/metrics"
|
|
"github.com/ethereum/go-ethereum/swarm/tracing"
|
|
|
|
"github.com/ethereum/go-ethereum/log"
|
|
|
|
cli "gopkg.in/urfave/cli.v1"
|
|
)
|
|
|
|
var (
|
|
gitCommit string // Git SHA1 commit hash of the release (set via linker flags)
|
|
)
|
|
|
|
var (
|
|
endpoints []string
|
|
includeLocalhost bool
|
|
cluster string
|
|
appName string
|
|
scheme string
|
|
filesize int
|
|
syncDelay int
|
|
from int
|
|
to int
|
|
verbosity int
|
|
timeout int
|
|
single bool
|
|
)
|
|
|
|
func main() {
|
|
|
|
app := cli.NewApp()
|
|
app.Name = "smoke-test"
|
|
app.Usage = ""
|
|
|
|
app.Flags = []cli.Flag{
|
|
cli.StringFlag{
|
|
Name: "cluster-endpoint",
|
|
Value: "prod",
|
|
Usage: "cluster to point to (prod or a given namespace)",
|
|
Destination: &cluster,
|
|
},
|
|
cli.StringFlag{
|
|
Name: "app",
|
|
Value: "swarm",
|
|
Usage: "application to point to (swarm or swarm-private)",
|
|
Destination: &appName,
|
|
},
|
|
cli.IntFlag{
|
|
Name: "cluster-from",
|
|
Value: 8501,
|
|
Usage: "swarm node (from)",
|
|
Destination: &from,
|
|
},
|
|
cli.IntFlag{
|
|
Name: "cluster-to",
|
|
Value: 8512,
|
|
Usage: "swarm node (to)",
|
|
Destination: &to,
|
|
},
|
|
cli.StringFlag{
|
|
Name: "cluster-scheme",
|
|
Value: "http",
|
|
Usage: "http or https",
|
|
Destination: &scheme,
|
|
},
|
|
cli.BoolFlag{
|
|
Name: "include-localhost",
|
|
Usage: "whether to include localhost:8500 as an endpoint",
|
|
Destination: &includeLocalhost,
|
|
},
|
|
cli.IntFlag{
|
|
Name: "filesize",
|
|
Value: 1024,
|
|
Usage: "file size for generated random file in KB",
|
|
Destination: &filesize,
|
|
},
|
|
cli.IntFlag{
|
|
Name: "sync-delay",
|
|
Value: 5,
|
|
Usage: "duration of delay in seconds to wait for content to be synced",
|
|
Destination: &syncDelay,
|
|
},
|
|
cli.IntFlag{
|
|
Name: "verbosity",
|
|
Value: 1,
|
|
Usage: "verbosity",
|
|
Destination: &verbosity,
|
|
},
|
|
cli.IntFlag{
|
|
Name: "timeout",
|
|
Value: 120,
|
|
Usage: "timeout in seconds after which kill the process",
|
|
Destination: &timeout,
|
|
},
|
|
cli.BoolFlag{
|
|
Name: "single",
|
|
Usage: "whether to fetch content from a single node or from all nodes",
|
|
Destination: &single,
|
|
},
|
|
}
|
|
|
|
app.Flags = append(app.Flags, []cli.Flag{
|
|
utils.MetricsEnabledFlag,
|
|
swarmmetrics.MetricsInfluxDBEndpointFlag,
|
|
swarmmetrics.MetricsInfluxDBDatabaseFlag,
|
|
swarmmetrics.MetricsInfluxDBUsernameFlag,
|
|
swarmmetrics.MetricsInfluxDBPasswordFlag,
|
|
swarmmetrics.MetricsInfluxDBTagsFlag,
|
|
}...)
|
|
|
|
app.Flags = append(app.Flags, tracing.Flags...)
|
|
|
|
app.Commands = []cli.Command{
|
|
{
|
|
Name: "upload_and_sync",
|
|
Aliases: []string{"c"},
|
|
Usage: "upload and sync",
|
|
Action: cliUploadAndSync,
|
|
},
|
|
{
|
|
Name: "feed_sync",
|
|
Aliases: []string{"f"},
|
|
Usage: "feed update generate, upload and sync",
|
|
Action: cliFeedUploadAndSync,
|
|
},
|
|
{
|
|
Name: "upload_speed",
|
|
Aliases: []string{"u"},
|
|
Usage: "measure upload speed",
|
|
Action: cliUploadSpeed,
|
|
},
|
|
}
|
|
|
|
sort.Sort(cli.FlagsByName(app.Flags))
|
|
sort.Sort(cli.CommandsByName(app.Commands))
|
|
|
|
app.Before = func(ctx *cli.Context) error {
|
|
tracing.Setup(ctx)
|
|
return nil
|
|
}
|
|
|
|
app.After = func(ctx *cli.Context) error {
|
|
return emitMetrics(ctx)
|
|
}
|
|
|
|
err := app.Run(os.Args)
|
|
if err != nil {
|
|
log.Error(err.Error())
|
|
|
|
os.Exit(1)
|
|
}
|
|
}
|
|
|
|
func emitMetrics(ctx *cli.Context) error {
|
|
if gethmetrics.Enabled {
|
|
var (
|
|
endpoint = ctx.GlobalString(swarmmetrics.MetricsInfluxDBEndpointFlag.Name)
|
|
database = ctx.GlobalString(swarmmetrics.MetricsInfluxDBDatabaseFlag.Name)
|
|
username = ctx.GlobalString(swarmmetrics.MetricsInfluxDBUsernameFlag.Name)
|
|
password = ctx.GlobalString(swarmmetrics.MetricsInfluxDBPasswordFlag.Name)
|
|
tags = ctx.GlobalString(swarmmetrics.MetricsInfluxDBTagsFlag.Name)
|
|
)
|
|
|
|
tagsMap := utils.SplitTagsFlag(tags)
|
|
tagsMap["version"] = gitCommit
|
|
tagsMap["filesize"] = fmt.Sprintf("%v", filesize)
|
|
|
|
return influxdb.InfluxDBWithTagsOnce(gethmetrics.DefaultRegistry, endpoint, database, username, password, "swarm-smoke.", tagsMap)
|
|
}
|
|
|
|
return nil
|
|
}
|