event sender shared
This commit is contained in:
parent
407e1e5f70
commit
100f14cbdb
|
@ -1,92 +0,0 @@
|
|||
package event_sender
|
||||
|
||||
import (
|
||||
"context"
|
||||
"google.golang.org/grpc"
|
||||
"loafle.com/overflow/agent_api/observer/messages"
|
||||
pb "loafle.com/overflow/crawler_go/grpc"
|
||||
q "loafle.com/overflow/queue_go"
|
||||
"log"
|
||||
"reflect"
|
||||
)
|
||||
|
||||
const (
|
||||
CENTRAL_ADDR = "192.168.1.105:50052"
|
||||
SENDER_ID = "OVERFLOW_EVENT_SENDER"
|
||||
DEFAULT_INTERVAL = 1
|
||||
)
|
||||
|
||||
type Data struct {
|
||||
AgentId string
|
||||
SensorId string
|
||||
Data map[string]string
|
||||
StartedAt uint64
|
||||
FinishedAt uint64
|
||||
}
|
||||
|
||||
type EventSender struct {
|
||||
lq *q.LoafleQueue
|
||||
sc chan interface{}
|
||||
}
|
||||
|
||||
func (es *EventSender) Start() {
|
||||
es.sc = make(chan interface{}, 0)
|
||||
|
||||
es.lq = q.NewQueue(messages.QUEUE_EVENT, DEFAULT_INTERVAL, es.sc)
|
||||
|
||||
go es.checkQueue()
|
||||
}
|
||||
|
||||
func (es *EventSender) Stop() {
|
||||
if es.sc != nil {
|
||||
close(es.sc)
|
||||
}
|
||||
}
|
||||
|
||||
func (es *EventSender) checkQueue() {
|
||||
|
||||
result := make([]*Data, 0)
|
||||
|
||||
for sc := range es.sc {
|
||||
items := reflect.ValueOf(sc)
|
||||
if items.Kind() != reflect.Slice {
|
||||
log.Println("ddddd")
|
||||
}
|
||||
|
||||
for i := 0; i < items.Len(); i++ {
|
||||
item := items.Index(i).Elem().Interface()
|
||||
tempCollectedData := item.(q.Item)
|
||||
collectedData := tempCollectedData.Value.(*Data)
|
||||
collectedData.AgentId = agentIdentifier()
|
||||
result = append(result, collectedData)
|
||||
log.Println("Result Len: ", len(result))
|
||||
}
|
||||
es.send(result)
|
||||
result = nil
|
||||
}
|
||||
|
||||
}
|
||||
|
||||
func (es *EventSender) send(data []*Data) {
|
||||
log.Println("Send Started")
|
||||
conn, err := grpc.Dial(CENTRAL_ADDR, grpc.WithInsecure())
|
||||
|
||||
if err != nil {
|
||||
log.Fatal("Connection Error :", err.Error())
|
||||
return
|
||||
}
|
||||
defer conn.Close()
|
||||
|
||||
client := pb.NewStatusClient(conn)
|
||||
out, err := client.Status(context.Background(), &pb.Empty{})
|
||||
|
||||
if err != nil {
|
||||
log.Fatal("client Error:", err.Error())
|
||||
return
|
||||
}
|
||||
log.Print(out)
|
||||
}
|
||||
|
||||
func agentIdentifier() string {
|
||||
return "agentID_000000001"
|
||||
}
|
Loading…
Reference in New Issue
Block a user