cerc-io/lotus
16
0
Fork 0
Code Issues Pull Requests Packages Projects Releases 27 Wiki Activity

Code cleanup

Browse Source
This commit is contained in:
Mak Muftic 2021-09-15 15:11:34 +02:00
parent e20cd0ef27
commit 707faf57b5
2 changed files with 7 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
node/modules/lp2p/pubsub.go
View File

@ -374,7 +374,9 @@ func GossipSub(in GossipIn) (service *pubsub.PubSub, err error) {
}
jsonTransport := tracer.NewJsonTracerTransport(out)
transports = append(transports, jsonTransport)
} else if in.Cfg.ElasticSearchTracer != "" {
}
if in.Cfg.ElasticSearchTracer != "" {
elasticSearchTransport, err := tracer.NewElasticSearchTransport()
if err != nil {
return nil, err

9
node/modules/tracer/elasticsearch_transport.go
View File

@ -8,10 +8,11 @@ import (
"github.com/elastic/go-elasticsearch/v7"
"github.com/elastic/go-elasticsearch/v7/esapi"
logging "github.com/ipfs/go-log/v2"
)
var rpclog = logging.Logger("elasticsearch")
const (
ElasticSearch_INDEX = "pubsub"
)
func NewElasticSearchTransport() (TracerTransport, error) {
es, err := elasticsearch.NewDefaultClient()
@ -31,7 +32,7 @@ type elasticSearchTransport struct {
func (est *elasticSearchTransport) Transport(jsonEvent []byte) error {
req := esapi.IndexRequest{
Index: "PeerScore",
Index: ElasticSearch_INDEX,
DocumentID: "1", // todo
Body: strings.NewReader(string(jsonEvent)),
Refresh: "true",
@ -51,8 +52,6 @@ func (est *elasticSearchTransport) Transport(jsonEvent []byte) error {
var r map[string]interface{}
if err := json.NewDecoder(res.Body).Decode(&r); err != nil {
return err
} else {
rpclog.Infof("[%s] %s; version=%d", res.Status(), r["result"], int(r["_version"].(float64)))
}
}
return nil