beenull/crowdsec
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

FeedLoki routine.

Browse source
This commit is contained in:
Mathieu Lecarme 2022-06-16 16:10:33 +02:00 committed by lperdereau
parent 6bf80ee083
commit a81d52a1ff
1 changed files with 38 additions and 61 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

99
pkg/acquisition/modules/loki/loki_test.go
View file

@ -164,6 +164,42 @@ func TestConfigureDSN(t *testing.T) {
}
}
func feedLoki(logger *log.Entry, n int, title string) error {
streams := LogStreams{
Streams: []LogStream{
{
Stream: map[string]string{
"server": "demo",
"domain": "cw.example.com",
"key": title,
},
Values: make([]LogValue, n),
},
},
}
for i := 0; i < n; i++ {
streams.Streams[0].Values[i] = LogValue{
Time: time.Now(),
Line: fmt.Sprintf("Log line #%d %v", i, title),
}
}
buff, err := json.Marshal(streams)
if err != nil {
return err
}
resp, err := http.Post("http://127.0.0.1:3100/loki/api/v1/push", "application/json", bytes.NewBuffer(buff))
if err != nil {
return err
}
if resp.StatusCode != 204 {
b, _ := ioutil.ReadAll(resp.Body)
logger.Error(string(b))
return fmt.Errorf("Bad post status %d", resp.StatusCode)
}
logger.Info(n, " Events sent")
return nil
}
func TestOneShotAcquisition(t *testing.T) {
log.SetOutput(os.Stdout)
log.SetLevel(log.InfoLevel)
@ -196,38 +232,10 @@ since: 1h
t.Fatalf("Unexpected error : %s", err)
}
streams := LogStreams{
Streams: []LogStream{
{
Stream: map[string]string{
"server": "demo",
"domain": "cw.example.com",
"key": title,
},
Values: make([]LogValue, 20),
},
},
}
for i := 0; i < 20; i++ {
streams.Streams[0].Values[i] = LogValue{
Time: time.Now(),
Line: fmt.Sprintf("Log line #%d %v", i, title),
}
}
buff, err := json.Marshal(streams)
err = feedLoki(subLogger, 20, title)
if err != nil {
t.Fatalf("Unexpected error : %s", err)
}
resp, err := http.Post("http://127.0.0.1:3100/loki/api/v1/push", "application/json", bytes.NewBuffer(buff))
if err != nil {
t.Fatalf("Unexpected error : %s", err)
}
if resp.StatusCode != 204 {
b, _ := ioutil.ReadAll(resp.Body)
log.Error(string(b))
t.Fatalf("Bad post status %d", resp.StatusCode)
}
subLogger.Info("20 Events sent")
out := make(chan types.Event)
go func() {
@ -327,38 +335,7 @@ query: >
writerTomb := tomb.Tomb{}
writerTomb.Go(func() error {
streams := LogStreams{
Streams: []LogStream{
{
Stream: map[string]string{
"server": "demo",
"domain": "cw.example.com",
},
Values: make([]LogValue, 20),
},
},
}
for i := 0; i < 20; i++ {
streams.Streams[0].Values[i] = LogValue{
Time: time.Now(),
Line: fmt.Sprintf("Log line #%d %v", i, title),
}
}
buff, err := json.Marshal(streams)
if err != nil {
return err
}
resp, err := http.Post("http://127.0.0.1:3100/loki/api/v1/push", "application/json", bytes.NewBuffer(buff))
if err != nil {
return err
}
if resp.StatusCode != 204 {
b, _ := ioutil.ReadAll(resp.Body)
log.Error(string(b))
return fmt.Errorf("Bad post status %d", resp.StatusCode)
}
subLogger.Info("20 Events sent")
return nil
return feedLoki(subLogger, 20, title)
})
err = writerTomb.Wait()
if err != nil {