netsampler-goflow2/transport/transport.go

package transport

import (
	"context"
	"fmt"
	"sync"
)

var (
	transportDrivers = make(map[string]TransportDriver)
	lock             = &sync.RWMutex{}
)

type TransportDriver interface {
	Prepare() error              // Prepare driver (eg: flag registration)
	Init(context.Context) error  // Initialize driver (eg: start connections, open files...)
	Close(context.Context) error // Close driver (eg: close connections and files...)
	Send(key, data []byte) error // Send a formatted message
}

type TransportInterface interface {
	Send(key, data []byte) error
}

type Transport struct {
	driver TransportDriver
}

func (t *Transport) Close(ctx context.Context) {
	t.driver.Close(ctx)
}
func (t *Transport) Send(key, data []byte) error {
	return t.driver.Send(key, data)
}

func RegisterTransportDriver(name string, t TransportDriver) {
	lock.Lock()
	transportDrivers[name] = t
	lock.Unlock()

	if err := t.Prepare(); err != nil {
		panic(err)
	}
}

func FindTransport(ctx context.Context, name string) (*Transport, error) {
	lock.RLock()
	t, ok := transportDrivers[name]
	lock.RUnlock()
	if !ok {
		return nil, fmt.Errorf("Transport %s not found", name)
	}

	err := t.Init(ctx)
	return &Transport{t}, err
}

func GetTransports() []string {
	lock.RLock()
	t := make([]string, len(transportDrivers))
	var i int
	for k, _ := range transportDrivers {
		t[i] = k
		i++
	}
	lock.RUnlock()
	return t
}