Fix dependencies

vendor/github.com/testcontainers/testcontainers-go/reaper.go

@@ -0,0 +1,127 @@
+package testcontainers
+
+import (
+	"bufio"
+	"context"
+	"fmt"
+	"net"
+	"strings"
+	"time"
+
+	"github.com/pkg/errors"
+)
+
+const (
+	TestcontainerLabel          = "org.testcontainers.golang"
+	TestcontainerLabelSessionID = TestcontainerLabel + ".sessionId"
+	TestcontainerLabelIsReaper  = TestcontainerLabel + ".reaper"
+
+	ReaperDefaultImage = "quay.io/testcontainers/ryuk:0.2.3"
+)
+
+// ReaperProvider represents a provider for the reaper to run itself with
+// The ContainerProvider interface should usually satisfy this as well, so it is pluggable
+type ReaperProvider interface {
+	RunContainer(ctx context.Context, req ContainerRequest) (Container, error)
+}
+
+// Reaper is used to start a sidecar container that cleans up resources
+type Reaper struct {
+	Provider  ReaperProvider
+	SessionID string
+	Endpoint  string
+}
+
+// NewReaper creates a Reaper with a sessionID to identify containers and a provider to use
+func NewReaper(ctx context.Context, sessionID string, provider ReaperProvider, reaperImageName string) (*Reaper, error) {
+	r := &Reaper{
+		Provider:  provider,
+		SessionID: sessionID,
+	}
+
+	// TODO: reuse reaper if there already is one
+
+	req := ContainerRequest{
+		Image:        reaperImage(reaperImageName),
+		ExposedPorts: []string{"8080"},
+		Labels: map[string]string{
+			TestcontainerLabel:         "true",
+			TestcontainerLabelIsReaper: "true",
+		},
+		SkipReaper: true,
+		BindMounts: map[string]string{
+			"/var/run/docker.sock": "/var/run/docker.sock",
+		},
+		AutoRemove: true,
+	}
+
+	c, err := provider.RunContainer(ctx, req)
+	if err != nil {
+		return nil, err
+	}
+
+	endpoint, err := c.PortEndpoint(ctx, "8080", "")
+	if err != nil {
+		return nil, err
+	}
+	r.Endpoint = endpoint
+
+	return r, nil
+}
+
+func reaperImage(reaperImageName string) string {
+	if reaperImageName == "" {
+		return ReaperDefaultImage
+	} else {
+		return reaperImageName
+	}
+}
+
+// Connect runs a goroutine which can be terminated by sending true into the returned channel
+func (r *Reaper) Connect() (chan bool, error) {
+	conn, err := net.DialTimeout("tcp", r.Endpoint, 10*time.Second)
+	if err != nil {
+		return nil, errors.Wrap(err, "Connecting to Ryuk on "+r.Endpoint+" failed")
+	}
+
+	terminationSignal := make(chan bool)
+	go func(conn net.Conn) {
+		sock := bufio.NewReadWriter(bufio.NewReader(conn), bufio.NewWriter(conn))
+		defer conn.Close()
+
+		labelFilters := []string{}
+		for l, v := range r.Labels() {
+			labelFilters = append(labelFilters, fmt.Sprintf("label=%s=%s", l, v))
+		}
+
+		retryLimit := 3
+		for retryLimit > 0 {
+			retryLimit--
+
+			sock.WriteString(strings.Join(labelFilters, "&"))
+			sock.WriteString("\n")
+			if err := sock.Flush(); err != nil {
+				continue
+			}
+
+			resp, err := sock.ReadString('\n')
+			if err != nil {
+				continue
+			}
+			if resp == "ACK\n" {
+				break
+			}
+		}
+
+		<-terminationSignal
+	}(conn)
+	return terminationSignal, nil
+}
+
+// Labels returns the container labels to use so that this Reaper cleans them up
+func (r *Reaper) Labels() map[string]string {
+	return map[string]string{
+		TestcontainerLabel:          "true",
+		TestcontainerLabelSessionID: r.SessionID,
+	}
+}