From 2d3ae322e0c80830cd8f43c23769c62e87458782 Mon Sep 17 00:00:00 2001
From: Denis Zheleztsov <d_zheleztsov@wargaming.net>
Date: Mon, 12 Nov 2018 15:06:39 +0300
Subject: [PATCH] Assign topic id for point message

---
 main.go         |  2 +-
 node/api.go     |  2 +-
 node/auth.go    |  2 +-
 node/elastic.go |  2 +-
 node/helpers.go |  2 +-
 node/logger.go  |  2 +-
 node/point.go   | 44 +++++++++++++++++++++++++++++++++++++++++---
 7 files changed, 47 insertions(+), 9 deletions(-)

diff --git a/main.go b/main.go
index f21ccce..df4cd80 100644
--- a/main.go
+++ b/main.go
@@ -5,7 +5,7 @@ import (
 	"os"
 
 	"gitea.difrex.ru/Umbrella/lessmore/node"
-	log "github.com/Sirupsen/logrus"
+	log "github.com/sirupsen/logrus"
 )
 
 var (
diff --git a/node/api.go b/node/api.go
index 30ff7f6..244e08d 100644
--- a/node/api.go
+++ b/node/api.go
@@ -7,8 +7,8 @@ import (
 	"strings"
 	"time"
 
-	log "github.com/Sirupsen/logrus"
 	"github.com/gorilla/mux"
+	log "github.com/sirupsen/logrus"
 )
 
 // ListTXTHandler ...
diff --git a/node/auth.go b/node/auth.go
index 910f040..9d649b8 100644
--- a/node/auth.go
+++ b/node/auth.go
@@ -17,7 +17,7 @@ import (
 
 	"time"
 
-	log "github.com/Sirupsen/logrus"
+	log "github.com/sirupsen/logrus"
 )
 
 const (
diff --git a/node/elastic.go b/node/elastic.go
index e9e2ca1..25f608a 100644
--- a/node/elastic.go
+++ b/node/elastic.go
@@ -13,7 +13,7 @@ import (
 	"encoding/base64"
 
 	"gitea.difrex.ru/Umbrella/fetcher/i2es"
-	log "github.com/Sirupsen/logrus"
+	log "github.com/sirupsen/logrus"
 )
 
 const (
diff --git a/node/helpers.go b/node/helpers.go
index c3f547d..fc5424f 100644
--- a/node/helpers.go
+++ b/node/helpers.go
@@ -7,7 +7,7 @@ import (
 
 	"strings"
 
-	log "github.com/Sirupsen/logrus"
+	log "github.com/sirupsen/logrus"
 	"golang.org/x/crypto/bcrypt"
 )
 
diff --git a/node/logger.go b/node/logger.go
index 9145259..6e4a29e 100644
--- a/node/logger.go
+++ b/node/logger.go
@@ -4,7 +4,7 @@ import (
 	"fmt"
 	"net/http"
 
-	log "github.com/Sirupsen/logrus"
+	log "github.com/sirupsen/logrus"
 )
 
 // LogRequest ...
diff --git a/node/point.go b/node/point.go
index 71b8241..1890551 100644
--- a/node/point.go
+++ b/node/point.go
@@ -5,11 +5,12 @@ import (
 	"fmt"
 	"io/ioutil"
 	"net/http"
+	"strings"
 
 	"bytes"
 
-	log "github.com/Sirupsen/logrus"
 	idec "github.com/idec-net/go-idec"
+	log "github.com/sirupsen/logrus"
 )
 
 type ESDoc struct {
@@ -23,6 +24,7 @@ type ESDoc struct {
 	Tags    string `json:"tags"`
 	Repto   string `json:"repto"`
 	Address string `json:"address"`
+	TopicID string `json:"topicid"`
 }
 
 // PointMessage add point message into DB
@@ -41,8 +43,7 @@ func (es ESConf) PointMessage(req PointRequest, user User) error {
 	}
 
 	// Make bundle ID
-	// Prevent collission via adding Timestamp
-	id := idec.MakeMsgID(fmt.Sprintf("%s\n%d", pmsg.String(), bmsg.Timestamp))
+	id := idec.MakeMsgID(pmsg.String())
 	bmsg.ID = id
 	bmsg.From = user.Name
 	bmsg.Address = fmt.Sprintf("%s,%d", user.Address, user.UserID)
@@ -53,6 +54,42 @@ func (es ESConf) PointMessage(req PointRequest, user User) error {
 	return nil
 }
 
+func (es ESConf) getTopicID(msgid string) string {
+	var topicid string
+	if msgid == "" {
+		return topicid
+	}
+	reqURL := fmt.Sprintf("%s/%s/%s/%s", es.Host, es.Index, es.Type, msgid)
+	req, err := http.NewRequest("GET", reqURL, strings.NewReader(""))
+	if err != nil {
+		log.Error(err)
+		return topicid
+	}
+	client := &http.Client{}
+	resp, err := client.Do(req)
+	if err != nil {
+		log.Error(err)
+		return topicid
+	}
+
+	defer resp.Body.Close()
+
+	var hit Hit
+	err = json.NewDecoder(resp.Body).Decode(&hit)
+	if err != nil {
+		log.Error(err)
+		return topicid
+	}
+
+	if hit.Source.TopicID != "" {
+		topicid = hit.Source.TopicID
+	} else if hit.Source.Repto != "" {
+		return es.getTopicID(hit.Source.Repto)
+	}
+
+	return topicid
+}
+
 func (es ESConf) IndexMessage(msg idec.Message) error {
 	tags, _ := msg.Tags.CollectTags()
 	doc := ESDoc{
@@ -66,6 +103,7 @@ func (es ESConf) IndexMessage(msg idec.Message) error {
 		Repto:   msg.Repto,
 		Address: msg.Address,
 		MsgID:   msg.ID,
+		TopicID: es.getTopicID(msg.Repto),
 	}
 	reqURL := fmt.Sprintf("%s/%s/%s/%s", es.Host, es.Index, es.Type, msg.ID)
 	bdoc, err := json.Marshal(doc)