feat: Initial commit

main.go

@@ -0,0 +1,51 @@
+package main
+
+import (
+	"context"
+	"log"
+	"net/http"
+	"os"
+
+	"github.com/joho/godotenv"
+	"github.com/mrhid6/go-mongoose/mongoose"
+)
+
+const (
+	collectionName = "tasks"
+)
+
+// Task statuses
+const (
+	StatusPending = "pending"
+	StatusLeased  = "leased"
+	StatusDone    = "done"
+)
+
+func main() {
+
+	godotenv.Load()
+
+	ctx := context.Background()
+
+	mc, err := mongoose.NewMongoClient(ctx, mongoose.GetConnectionOptionsFromEnv())
+	if err != nil {
+		log.Fatalf("mongo connect: %v", err)
+	}
+
+	col := mc.GetCollection(collectionName)
+	s := &Server{mc: mc, col: col}
+
+	if err := s.ensureIndexes(ctx); err != nil {
+		log.Fatalf("ensure indexes: %v", err)
+	}
+
+	http.HandleFunc("/enqueue", s.handleEnqueue)
+	http.HandleFunc("/pop", s.handlePop)
+	http.HandleFunc("/complete", s.handleComplete)
+	http.HandleFunc("/extend", s.handleExtend)
+	http.HandleFunc("/requeue", s.handleRequeue)
+
+	addr := os.Getenv("HTTP_BIND")
+	log.Printf("task queue API listening on %s", addr)
+	log.Fatal(http.ListenAndServe(addr, nil))
+}