Move index name to config only and add default value

This commit is contained in:
Mak Muftic 2021-09-24 13:43:25 +02:00
parent 22bbb113e7
commit 4438c4bd87
4 changed files with 14 additions and 12 deletions

View File

@ -161,10 +161,6 @@ var DaemonCmd = &cli.Command{
Name: "trace-to-elasticsearch",
Usage: "starts tracer and outputs to elasticsearch, flag must contain connection string for elasticsearch",
},
&cli.StringFlag{
Name: "elasticsearch-index",
Usage: "configure elasticearch index name if elasticsearch tracer is configured",
},
&cli.StringFlag{
Name: "trace-source-auth",
Usage: "auth token for trusted source of traces",
@ -213,8 +209,6 @@ var DaemonCmd = &cli.Command{
traceToElasticsearch := cctx.String("trace-to-elasticsearch")
elasticsearchIndex := cctx.String("elasticsearch-index")
traceSourceAuth := cctx.String("trace-source-auth")
ctx, _ := tag.New(context.Background(),
@ -345,7 +339,6 @@ var DaemonCmd = &cli.Command{
node.Override(new(dtypes.ShutdownChan), shutdownChan),
node.Override(new(dtypes.JsonTracerFile), traceToJsonFile),
node.Override(new(dtypes.ElasticSearchTracer), traceToElasticsearch),
node.Override(new(dtypes.ElasticSearchTracer), elasticsearchIndex),
node.Override(new(dtypes.TracerSourceAuth), traceSourceAuth),
genesis,

View File

@ -2,5 +2,4 @@ package dtypes
type JsonTracerFile string
type ElasticSearchTracer string
type ElasticSearchIndex string
type TracerSourceAuth string

View File

@ -377,6 +377,7 @@ func GossipSub(in GossipIn) (service *pubsub.PubSub, err error) {
if in.Cfg.ElasticSearchTracer != "" {
elasticSearchTransport, err := tracer.NewElasticSearchTransport(
in.Cfg.ElasticSearchTracer,
in.Cfg.ElasticSearchIndex,
)
if err != nil {
return nil, err

View File

@ -12,10 +12,10 @@ import (
)
const (
ElasticSearch_INDEX = "lotus-pubsub"
ElasticSearch_INDEX_DEFAULT = "lotus-pubsub"
)
func NewElasticSearchTransport(connectionString string) (TracerTransport, error) {
func NewElasticSearchTransport(connectionString string, elasticsearchIndex string) (TracerTransport, error) {
conUrl, err := url.Parse(connectionString)
if err != nil {
@ -38,13 +38,22 @@ func NewElasticSearchTransport(connectionString string) (TracerTransport, error)
return nil, err
}
var esIndex string
if elasticsearchIndex != "" {
esIndex = elasticsearchIndex
} else {
esIndex = ElasticSearch_INDEX_DEFAULT
}
return &elasticSearchTransport{
cl: es,
esIndex: esIndex,
}, nil
}
type elasticSearchTransport struct {
cl *elasticsearch.Client
esIndex string
}
func (est *elasticSearchTransport) Transport(evt TracerTransportEvent) error {