FeedLoki routine.
This commit is contained in:
parent
6bf80ee083
commit
a81d52a1ff
|
@ -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) {
|
func TestOneShotAcquisition(t *testing.T) {
|
||||||
log.SetOutput(os.Stdout)
|
log.SetOutput(os.Stdout)
|
||||||
log.SetLevel(log.InfoLevel)
|
log.SetLevel(log.InfoLevel)
|
||||||
|
@ -196,38 +232,10 @@ since: 1h
|
||||||
t.Fatalf("Unexpected error : %s", err)
|
t.Fatalf("Unexpected error : %s", err)
|
||||||
}
|
}
|
||||||
|
|
||||||
streams := LogStreams{
|
err = feedLoki(subLogger, 20, title)
|
||||||
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)
|
|
||||||
if err != nil {
|
if err != nil {
|
||||||
t.Fatalf("Unexpected error : %s", err)
|
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)
|
out := make(chan types.Event)
|
||||||
go func() {
|
go func() {
|
||||||
|
@ -327,38 +335,7 @@ query: >
|
||||||
|
|
||||||
writerTomb := tomb.Tomb{}
|
writerTomb := tomb.Tomb{}
|
||||||
writerTomb.Go(func() error {
|
writerTomb.Go(func() error {
|
||||||
streams := LogStreams{
|
return feedLoki(subLogger, 20, title)
|
||||||
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
|
|
||||||
})
|
})
|
||||||
err = writerTomb.Wait()
|
err = writerTomb.Wait()
|
||||||
if err != nil {
|
if err != nil {
|
||||||
|
|
Loading…
Reference in a new issue