first function runtime implementation close #10

Author: dstpierre

db/base.go

@@ -40,10 +40,10 @@ func (b *Base) Add(auth internal.Auth, db *mongo.Database, col string, doc map[s
 }
 
 type PagedResult struct {
-	Page    int64         `json:"page"`
-	Size    int64         `json:"size"`
-	Total   int64         `json:"total"`
-	Results []interface{} `json:"results"`
+	Page    int64    `json:"page"`
+	Size    int64    `json:"size"`
+	Total   int64    `json:"total"`
+	Results []bson.M `json:"results"`
 }
 
 type ListParams struct {
@@ -100,7 +100,7 @@ func (b *Base) List(auth internal.Auth, db *mongo.Database, col string, params L
 	}
 	defer cur.Close(ctx)
 
-	var results []interface{}
+	var results []bson.M
 
 	for cur.Next(ctx) {
 		var v bson.M
@@ -120,7 +120,7 @@ func (b *Base) List(auth internal.Auth, db *mongo.Database, col string, params L
 	}
 
 	if len(results) == 0 {
-		results = make([]interface{}, 1)
+		results = make([]bson.M, 1)
 	}
 
 	result.Results = results
@@ -155,7 +155,7 @@ func (b *Base) Query(auth internal.Auth, db *mongo.Database, col string, filter
 	result.Total = count
 
 	if count == 0 {
-		result.Results = make([]interface{}, 0)
+		result.Results = make([]bson.M, 0)
 		return result, nil
 	}
 
@@ -180,7 +180,7 @@ func (b *Base) Query(auth internal.Auth, db *mongo.Database, col string, filter
 	}
 	defer cur.Close(ctx)
 
-	var results []interface{}
+	var results []bson.M
 
 	for cur.Next(ctx) {
 		var v bson.M
@@ -200,7 +200,7 @@ func (b *Base) Query(auth internal.Auth, db *mongo.Database, col string, filter
 	}
 
 	if len(results) == 0 {
-		results = make([]interface{}, 1)
+		results = make([]bson.M, 1)
 	}
 
 	result.Results = results

db_test.go

@@ -13,7 +13,16 @@ import (
 	"time"
 )
 
-func dbPost(t *testing.T, hf func(http.ResponseWriter, *http.Request), repo string, v interface{}) *http.Response {
+// dbPost post on behalf of adminToken by default (use params[0] true for root)
+func dbPost(t *testing.T, hf func(http.ResponseWriter, *http.Request), repo string, v interface{}, params ...bool) *http.Response {
+	if params == nil {
+		params = make([]bool, 0)
+	}
+
+	if len(params) == 0 {
+		params = append(params, false)
+	}
+
 	b, err := json.Marshal(v)
 	if err != nil {
 		t.Fatal("error marshaling post data:", err)
@@ -23,12 +32,23 @@ func dbPost(t *testing.T, hf func(http.ResponseWriter, *http.Request), repo stri
 	w := httptest.NewRecorder()
 
 	req.Header.Set("SB-PUBLIC-KEY", pubKey)
-	req.Header.Set("Authorization", fmt.Sprintf("Bearer %s", adminToken))
+
+	tok := adminToken
+	if params[0] {
+		tok = rootToken
+	}
+	req.Header.Set("Authorization", fmt.Sprintf("Bearer %s", tok))
 
 	stdAuth := []middleware.Middleware{
 		middleware.WithDB(database.client),
 		middleware.RequireAuth(database.client),
 	}
+	if params[0] {
+		stdAuth = []middleware.Middleware{
+			middleware.WithDB(client),
+			middleware.RequireRoot(client),
+		}
+	}
 	h := middleware.Chain(http.HandlerFunc(hf), stdAuth...)
 
 	h.ServeHTTP(w, req)

function/management.go

@@ -0,0 +1,143 @@
+package function
+
+import (
+	"context"
+	"staticbackend/internal"
+	"time"
+
+	"go.mongodb.org/mongo-driver/bson"
+	"go.mongodb.org/mongo-driver/bson/primitive"
+	"go.mongodb.org/mongo-driver/mongo"
+	"go.mongodb.org/mongo-driver/mongo/options"
+)
+
+// ExecData represents a server-side function with its name, code and execution
+// history
+type ExecData struct {
+	ID           primitive.ObjectID `bson:"_id" json:"id"`
+	FunctionName string             `bson:"name" json:"name"`
+	TriggerTopic string             `bson:"tr" json:"trigger"`
+	Code         string             `bson:"code" json:"code"`
+	Version      int                `bson:"v" json:"version"`
+	LastUpdated  time.Time          `bson:"lu" json:"lastUpdated"`
+	History      []ExecHistory      `bson:"h" json:"history"`
+}
+
+// ExecHistory represents a function run ending result
+type ExecHistory struct {
+	Version   int       `bson:"v" json:"version"`
+	Started   time.Time `bson:"s" json:"started"`
+	Completed time.Time `bson:"c" json:"completed"`
+	Success   bool      `bson:"ok" json:"success"`
+	Output    []string  `bson:"out" json:"output"`
+}
+
+func Add(db *mongo.Database, data ExecData) (string, error) {
+	data.ID = primitive.NewObjectID()
+	data.Version = 1
+	data.LastUpdated = time.Now()
+
+	ctx := context.Background()
+	if _, err := db.Collection("sb_functions").InsertOne(ctx, data); err != nil {
+		return "", err
+	}
+
+	return data.ID.Hex(), nil
+}
+
+func Update(db *mongo.Database, id, code, trigger string) error {
+	oid, err := primitive.ObjectIDFromHex(id)
+	if err != nil {
+		return err
+	}
+
+	update := bson.M{
+		"$set": bson.M{"code": code, "lu": time.Now(), "tr": trigger},
+		"$inc": bson.M{"v": 1},
+	}
+	filter := bson.M{internal.FieldID: oid}
+
+	ctx := context.Background()
+	res := db.Collection("sb_functions").FindOneAndUpdate(ctx, filter, update)
+	if err := res.Err(); err != nil {
+		return err
+	}
+
+	return nil
+}
+
+func GetForExecution(db *mongo.Database, name string) (ExecData, error) {
+	var result ExecData
+
+	filter := bson.M{"name": name}
+
+	ctx := context.Background()
+	opt := &options.FindOneOptions{}
+	opt.SetProjection(bson.M{"h": -1})
+
+	sr := db.Collection("sb_functions").FindOne(ctx, filter, opt)
+	if err := sr.Decode(&result); err != nil {
+		return result, err
+	} else if err := sr.Err(); err != nil {
+		return result, err
+	}
+
+	return result, nil
+}
+
+func GetByID(db *mongo.Database, id string) (ExecData, error) {
+	var result ExecData
+
+	oid, err := primitive.ObjectIDFromHex(id)
+	if err != nil {
+		return result, err
+	}
+
+	filter := bson.M{internal.FieldID: oid}
+
+	ctx := context.Background()
+	sr := db.Collection("sb_functions").FindOne(ctx, filter)
+	if err := sr.Decode(&result); err != nil {
+		return result, err
+	} else if err := sr.Err(); err != nil {
+		return result, err
+	}
+
+	return result, nil
+}
+
+func List(db *mongo.Database) ([]ExecData, error) {
+	opt := &options.FindOptions{}
+	opt.SetProjection(bson.M{"h": -1})
+
+	ctx := context.Background()
+	cur, err := db.Collection("sb_functions").Find(ctx, bson.M{}, opt)
+	if err != nil {
+		return nil, err
+	}
+	defer cur.Close(ctx)
+
+	var results []ExecData
+	for cur.Next(ctx) {
+		var ed ExecData
+		err := cur.Decode(&ed)
+		if err != nil {
+			return nil, err
+		}
+
+		results = append(results, ed)
+	}
+
+	return results, nil
+}
+
+func Delete(db *mongo.Database, name string) error {
+	filter := bson.M{"name": name}
+
+	ctx := context.Background()
+	if _, err := db.Collection("sb_functions").DeleteOne(ctx, filter); err != nil {
+		return err
+	}
+
+	return nil
+}