feat: Initial commit

server.go

@@ -0,0 +1,46 @@
+package main
+
+import (
+	"context"
+
+	"github.com/mrhid6/go-mongoose/mongoose"
+	"go.mongodb.org/mongo-driver/bson"
+	"go.mongodb.org/mongo-driver/mongo"
+	"go.mongodb.org/mongo-driver/mongo/options"
+)
+
+type Server struct {
+	mc  *mongoose.MongooseClient
+	col *mongo.Collection
+}
+
+// ---------- Indexes ----------
+func (s *Server) ensureIndexes(ctx context.Context) error {
+	idxs := s.col.Indexes()
+	models := []mongo.IndexModel{
+		// frequently query by status + availableAt + priority
+		{
+			Keys: bson.D{
+				{"status", 1},
+				{"availableAt", 1},
+				{"priority", -1},
+			},
+			Options: options.Index().SetName("status_available_priority"),
+		},
+		// lease expiration useful for queries
+		{
+			Keys: bson.D{
+				{"leasedUntil", 1},
+			},
+			Options: options.Index().SetName("leased_until_idx"),
+		},
+		// TTL index on done tasks if you want automatic removal after X seconds.
+		// Uncomment or adjust the seconds as you need. Here we do not create by default.
+		// {
+		// 	Keys: bson.D{{"updatedAt", 1}},
+		// 	Options: options.Index().SetExpireAfterSeconds(60 * 60 * 24), // e.g. 24h
+		// },
+	}
+	_, err := idxs.CreateMany(ctx, models)
+	return err
+}