Add crud variants

pkg/handler/crud.go

@@ -10,6 +10,7 @@ import (
 	"github.com/checkr/flagr/swagger_gen/restapi/operations/distribution"
 	"github.com/checkr/flagr/swagger_gen/restapi/operations/flag"
 	"github.com/checkr/flagr/swagger_gen/restapi/operations/segment"
+	"github.com/checkr/flagr/swagger_gen/restapi/operations/variant"
 	"github.com/go-openapi/runtime/middleware"
 )
 
@@ -33,6 +34,10 @@ type CRUD interface {
 	// Distributions
 	FindDistributions(distribution.FindDistributionsParams) middleware.Responder
 	PutDistributions(distribution.PutDistributionsParams) middleware.Responder
+
+	// Variants
+	CreateVariant(variant.CreateVariantParams) middleware.Responder
+	FindVariants(variant.FindVariantsParams) middleware.Responder
 }
 
 // NewCRUD creates a new CRUD instance
@@ -121,7 +126,7 @@ func (c *crud) CreateSegment(params segment.CreateSegmentParams) middleware.Resp
 
 	err := s.Create(repo.GetDB())
 	if err != nil {
-		return segment.NewFindSegmentsDefault(500)
+		return segment.NewCreateSegmentDefault(500)
 	}
 
 	resp := segment.NewCreateSegmentOK()
@@ -218,3 +223,32 @@ func (c *crud) FindDistributions(params distribution.FindDistributionsParams) mi
 	resp.SetPayload(e2r.MapDistributions(ds))
 	return resp
 }
+
+func (c *crud) CreateVariant(params variant.CreateVariantParams) middleware.Responder {
+	v := &entity.Variant{}
+	v.FlagID = uint(params.FlagID)
+	v.Key = util.SafeString(params.Body.Key)
+
+	err := v.Create(repo.GetDB())
+	if err != nil {
+		return variant.NewCreateVariantDefault(500)
+	}
+
+	resp := variant.NewCreateVariantOK()
+	resp.SetPayload(e2r.MapVariant(v))
+	return resp
+}
+
+func (c *crud) FindVariants(params variant.FindVariantsParams) middleware.Responder {
+	vs := []entity.Variant{}
+
+	q := entity.NewVariantQuerySet(repo.GetDB())
+	err := q.FlagIDEq(uint(params.FlagID)).OrderAscByID().All(&vs)
+	if err != nil {
+		return variant.NewFindVariantsDefault(500)
+	}
+
+	resp := variant.NewFindVariantsOK()
+	resp.SetPayload(e2r.MapVariants(vs))
+	return resp
+}

pkg/handler/handler.go

@@ -7,6 +7,7 @@ import (
 	"github.com/checkr/flagr/swagger_gen/restapi/operations/evaluation"
 	"github.com/checkr/flagr/swagger_gen/restapi/operations/flag"
 	"github.com/checkr/flagr/swagger_gen/restapi/operations/segment"
+	"github.com/checkr/flagr/swagger_gen/restapi/operations/variant"
 )
 
 // Setup initialize all the hanlder functions
@@ -31,6 +32,10 @@ func Setup(api *operations.FlagrAPI) {
 	api.DistributionFindDistributionsHandler = distribution.FindDistributionsHandlerFunc(c.FindDistributions)
 	api.DistributionPutDistributionsHandler = distribution.PutDistributionsHandlerFunc(c.PutDistributions)
 
+	// variants
+	api.VariantCreateVariantHandler = variant.CreateVariantHandlerFunc(c.CreateVariant)
+	api.VariantFindVariantsHandler = variant.FindVariantsHandlerFunc(c.FindVariants)
+
 	// evaluation
 	e := NewEval()
 	api.EvaluationPostEvaluationHandler = evaluation.PostEvaluationHandlerFunc(e.PostEvaluation)

pkg/mapper/entity_restapi/e2r/e2r.go

@@ -81,3 +81,21 @@ func MapDistributions(e []entity.Distribution) []*models.Distribution {
 	}
 	return ret
 }
+
+// MapVariant maps variant
+func MapVariant(e *entity.Variant) *models.Variant {
+	r := &models.Variant{
+		ID:  int64(e.ID),
+		Key: util.StringPtr(e.Key),
+	}
+	return r
+}
+
+// MapVariants maps variant
+func MapVariants(e []entity.Variant) []*models.Variant {
+	ret := make([]*models.Variant, len(e), len(e))
+	for i, v := range e {
+		ret[i] = MapVariant(&v)
+	}
+	return ret
+}

swagger.yml

@@ -140,6 +140,33 @@ paths:
           description: generic error response
           schema:
             $ref: "#/definitions/error"
+    post:
+      tags:
+        - variant
+      operationId: createVariant
+      parameters:
+        - in: path
+          name: flagID
+          description: numeric ID of the flag
+          required: true
+          type: integer
+          format: int32
+          minimum: 1
+        - in: body
+          name: body
+          description: create a variant
+          required: true
+          schema:
+            $ref: "#/definitions/createVariantRequest"
+      responses:
+        200:
+          description: variant just created
+          schema:
+            $ref: "#/definitions/variant"
+        default:
+          description: generic error response
+          schema:
+            $ref: "#/definitions/error"
   /flags/{flagID}/segments:
     get:
       tags:
@@ -376,6 +403,14 @@ definitions:
         format: int32
         minimum: 0
         maximum: 100
+  createVariantRequest:
+    type: object
+    required:
+      - key
+    properties:
+      key:
+        type: string
+        minLength: 1
   createConstraintRequest:
     type: object
     required: