databoxLogParser.go

package main

import (
	"bufio"
	"context"
	"fmt"
	"sync"

	"github.com/docker/docker/api/types"
	"github.com/docker/docker/client"
	libDatabox "github.com/me-box/lib-go-databox"
)

func ShowLogs() {

	cli, _ := client.NewEnvClient()

	libDatabox.Info("Outputting logs")

	//filters := filters.NewArgs()
	//filters.Add("label", "databox.type")
	//services, err := d.cli.ServiceList(context.Background(), types.ServiceListOptions{Filters: filters})
	services, err := cli.ServiceList(context.Background(), types.ServiceListOptions{})
	libDatabox.ChkErr(err)

	logChan := make(chan string)
	var wg sync.WaitGroup

	for _, service := range services {
		ioLogReader, err := cli.ServiceLogs(context.Background(), service.ID,
			types.ContainerLogsOptions{
				Follow:     true,
				ShowStderr: true,
				ShowStdout: true,
				Tail:       "all",
				Timestamps: false,
			},
		)

		if err != nil {
			libDatabox.Err("Error reading logs for " + service.Spec.Name)
			continue
		}
		wg.Add(1)
		go func(name string) {
			scanner := bufio.NewScanner(ioLogReader)
			for scanner.Scan() {
				message := scanner.Text()
				if err := scanner.Err(); err != nil {
					wg.Done()
					ioLogReader.Close()
					break
				}
				if len(message) > 8 {
					message = message[8:]
				}
				logChan <- fmt.Sprintf("%-25s: %s", name, message)
			}
		}(service.Spec.Name)
	}

	go func() {
		for {
			select {
			case msg := <-logChan:
				fmt.Println(msg)
			}
		}
	}()

	wg.Wait()
}