Merge pull request #9 from drone/FFM-778_go_test_wrapper

(FFM-778) Adds go test wrapper

Author: Andrew-Hayes

test_wrapper/Makefile

@@ -0,0 +1,10 @@
+install:
+	go get github.com/deepmap/oapi-codegen/cmd/[email protected]
+
+generate:
+	# generate admin
+	oapi-codegen -generate server,spec -package=restapi ./test_wrapper_openapi/sdk_wrapper-v1.yaml > ./restapi/test_wrapper_service.gen.go
+	oapi-codegen -generate types -package=restapi ./test_wrapper_openapi/sdk_wrapper-v1.yaml > ./restapi/test_wrapper_types.gen.go
+
+build:
+	docker build -t go_sdk_wrapper:latest -f ./docker/Dockerfile .

test_wrapper/docker/Dockerfile

@@ -0,0 +1,20 @@
+FROM golang:1.14-alpine as builder
+RUN apk update && apk add --no-cache make gcc musl-dev git ca-certificates && update-ca-certificates
+WORKDIR /app
+
+# Fetch dependencies.
+COPY ./go.mod .
+RUN go mod download
+
+COPY . .
+RUN go mod verify
+RUN make install
+RUN make generate
+
+RUN CGO_ENABLED=0 GOOS=linux GOARCH=amd64 go build -o /app/go_wrapper .
+
+FROM alpine:latest
+RUN apk update && apk add --no-cache bash
+COPY --from=builder /app/go_wrapper /app/go_wrapper
+
+CMD ["/app/go_wrapper"]

test_wrapper/ffgosdk/ff_go_sdk.go

@@ -0,0 +1,83 @@
+package ffgosdk
+
+import (
+	"encoding/json"
+	"fmt"
+
+	harness "github.com/drone/ff-golang-server-sdk/client"
+	"github.com/drone/ff-golang-server-sdk/evaluation"
+	"github.com/drone/ff-golang-server-sdk/test_wrapper/wrapperconfig"
+	"github.com/drone/ff-golang-server-sdk/types"
+)
+
+// SDK .
+type SDK struct {
+	sdkClient *harness.CfClient
+}
+
+// NewSDK .
+func NewSDK(config wrapperconfig.Config) (SDK, error) {
+	client, err := harness.NewCfClient(config.SdkKey,
+		harness.WithURL(config.BaseURL),
+		harness.WithStreamEnabled(config.EnableStreaming),
+	)
+
+	if err != nil {
+		return SDK{}, err
+	}
+
+	return SDK{
+		sdkClient: client,
+	}, nil
+}
+
+// Close .
+func (s SDK) Close() error {
+	return s.sdkClient.Close()
+}
+
+// GetVariant .
+func (s SDK) GetVariant(kind string, flagKey string, targetMap *map[string]string) (string, error) {
+	out := ""
+	t := &evaluation.Target{}
+	if targetMap != nil {
+		t.Attributes = make(map[string]interface{})
+		for k, v := range *targetMap {
+			t.Attributes[k] = v
+		}
+	}
+
+	var err error
+	switch kind {
+	case "boolean":
+		var variation bool
+		variation, err = s.sdkClient.BoolVariation(flagKey, t, false)
+		out = fmt.Sprintf("%t", variation)
+	case "string":
+		var variation string
+		variation, err = s.sdkClient.StringVariation(flagKey, t, "")
+		out = fmt.Sprintf("%s", variation)
+	case "int":
+		var variation int64
+		variation, err = s.sdkClient.IntVariation(flagKey, t, -1)
+		out = fmt.Sprintf("%d", variation)
+	case "number":
+		var variation float64
+		variation, err = s.sdkClient.NumberVariation(flagKey, t, -1)
+		out = fmt.Sprintf("%g", variation)
+	case "json":
+		var variation types.JSON
+		variation, err = s.sdkClient.JSONVariation(flagKey, t, types.JSON{})
+		data, err := json.Marshal(variation)
+		if err != nil {
+			out = fmt.Sprintf("{}")
+		} else {
+			out = fmt.Sprintf("%s", string(data))
+		}
+	}
+
+	if err != nil {
+		return out, err
+	}
+	return out, nil
+}

test_wrapper/go.mod

@@ -0,0 +1,12 @@
+module github.com/drone/ff-golang-server-sdk/test_wrapper
+
+go 1.16
+
+require (
+	github.com/deepmap/oapi-codegen v1.6.1 // indirect
+	github.com/drone/ff-golang-server-sdk v0.0.4
+	github.com/getkin/kin-openapi v0.53.0
+	github.com/labstack/echo/v4 v4.2.2
+	github.com/sirupsen/logrus v1.8.1
+	github.com/spf13/viper v1.7.1
+)

test_wrapper/go.sum

@@ -0,0 +1,65 @@
+package handlers
+
+import (
+	"net/http"
+
+	"github.com/drone/ff-golang-server-sdk/test_wrapper/ffgosdk"
+	restapi "github.com/drone/ff-golang-server-sdk/test_wrapper/restapi"
+	"github.com/drone/ff-golang-server-sdk/test_wrapper/wrapperconfig"
+	"github.com/labstack/echo/v4"
+	log "github.com/sirupsen/logrus"
+)
+
+// ServerImpl .
+type ServerImpl struct {
+	sdk ffgosdk.SDK
+}
+
+// NewServer .
+func NewServer(config wrapperconfig.Config) ServerImpl {
+	sdkClient, err := ffgosdk.NewSDK(config)
+	if err != nil {
+		panic(err)
+	}
+	return ServerImpl{
+		sdk: sdkClient,
+	}
+}
+
+// Ping .
+func (s *ServerImpl) Ping(ctx echo.Context) error {
+	return ctx.JSON(http.StatusOK, restapi.PongResponse{Pong: true})
+}
+
+// GetFlagValue .
+func (s *ServerImpl) GetFlagValue(ctx echo.Context) error {
+	log.Infof("GetFlagValue %+v", ctx.Request())
+	flagBody := new(restapi.FlagCheckBody)
+	err := ctx.Bind(flagBody)
+	if err != nil {
+		log.Error(err)
+		return ctx.JSON(http.StatusInternalServerError, restapi.ErrorResponse{ErrorMessage: err.Error()})
+	}
+
+	var targetMap map[string]string
+	if flagBody.Target != nil {
+		if flagBody.Target.Name != nil {
+			targetMap["name"] = *flagBody.Target.Name
+		}
+		if flagBody.Target.Email != nil {
+			targetMap["email"] = *flagBody.Target.Email
+		}
+		if flagBody.Target.Region != nil {
+			targetMap["region"] = *flagBody.Target.Region
+		}
+	}
+
+	variantion, err := s.sdk.GetVariant(flagBody.FlagKind, flagBody.FlagKey, &targetMap)
+	if err != nil {
+		log.Error(err)
+		return ctx.JSON(http.StatusInternalServerError, restapi.ErrorResponse{ErrorMessage: err.Error()})
+	}
+
+	flagValue := restapi.FlagCheckResponse{FlagKey: flagBody.FlagKey, FlagValue: variantion}
+	return ctx.JSON(http.StatusOK, flagValue)
+}

test_wrapper/handlers/handlers.go

@@ -0,0 +1,88 @@
+package main
+
+import (
+	"context"
+	"fmt"
+	"net/http"
+	"os"
+	"os/signal"
+	"syscall"
+	"time"
+
+	"github.com/drone/ff-golang-server-sdk/test_wrapper/handlers"
+	"github.com/drone/ff-golang-server-sdk/test_wrapper/restapi"
+	"github.com/drone/ff-golang-server-sdk/test_wrapper/wrapperconfig"
+	"github.com/labstack/echo/v4"
+	"github.com/labstack/echo/v4/middleware"
+	log "github.com/sirupsen/logrus"
+	"github.com/spf13/viper"
+)
+
+func main() {
+	log.SetFormatter(&log.JSONFormatter{})
+
+	config := wrapperconfig.Config{}
+
+	viper.SetDefault("WrapperHostname", "")
+	viper.SetDefault("WrapperPort", "4000")
+	viper.SetDefault("BaseURL", "http://localhost/api/1.0")
+
+	viper.SetDefault("SdkKey", "1bca46aa-0abe-4b22-a5f0-904422db288b")
+	viper.SetDefault("EnableStreaming", false)
+	viper.BindEnv("WrapperHostname", "WRAPPER_HOSTNAME")
+	viper.BindEnv("WrapperPort", "WRAPPER_PORT")
+	viper.BindEnv("BaseURL", "SDK_BASE_URL")
+	viper.BindEnv("SdkKey", "SDK_KEY")
+	viper.BindEnv("EnableStreaming", "ENABLE_STREAMING")
+
+	err := viper.Unmarshal(&config)
+	if err != nil {
+		log.Fatalf("unable to viper decode into struct, %v", err)
+	}
+
+	log.Infof("Starting up with config: \n%+v\n", config)
+	setupHandlers(config)
+}
+
+func setupHandlers(config wrapperconfig.Config) {
+	myServer := handlers.NewServer(config)
+	e := echo.New()
+
+	apiGroup := e.Group("/api/1.0")
+	//apiGroup.Use(oapimiddleware.OapiRequestValidatorWithOptions(swagger, &oapimiddleware.Options{}))
+	e.Use(middleware.Recover())
+	e.Use(middleware.CORSWithConfig(middleware.CORSConfig{
+		AllowOrigins: []string{"*"},
+		AllowHeaders: []string{echo.HeaderOrigin, echo.HeaderContentType, echo.HeaderAccept, "Cache-Control"},
+	}))
+
+	restapi.RegisterHandlers(apiGroup, &myServer)
+
+	go func() {
+		serverAddress := fmt.Sprintf("%s:%s", config.WrapperHostname, config.WrapperPort)
+		fmt.Printf("Starting api wrapper on: %s", serverAddress)
+		err := e.Start(serverAddress)
+		if err != nil && err != http.ErrServerClosed {
+			log.Fatal("shutting down http server")
+		}
+	}()
+
+	waitForShutdown(e)
+}
+
+func waitForShutdown(e *echo.Echo) {
+	// Handle sigterm and shutdown gracefully
+	termChan := make(chan os.Signal)
+	signal.Notify(termChan, syscall.SIGINT, syscall.SIGTERM)
+	log.Debug("Waiting for Shutdown")
+
+	<-termChan // Blocks here until interrupted
+
+	// Handle shutdown
+	log.Info("Shutdown signal received")
+	ctx, cancel := context.WithTimeout(context.Background(), 10*time.Second)
+	defer cancel()
+	if err := e.Shutdown(ctx); err != nil {
+		log.Fatal(err)
+	}
+}