Initial commit

internal/db/db.go

@@ -0,0 +1,47 @@
+package db
+
+import (
+	"context"
+	"time"
+
+	"go.mongodb.org/mongo-driver/v2/mongo"
+	"go.mongodb.org/mongo-driver/v2/mongo/options"
+)
+
+// Client wraps the mongo client and exposes typed collection accessors.
+type Client struct {
+	client *mongo.Client
+	db     *mongo.Database
+}
+
+func Connect(ctx context.Context, uri string) (*Client, error) {
+	opts := options.Client().ApplyURI(uri)
+	c, err := mongo.Connect(opts)
+	if err != nil {
+		return nil, err
+	}
+
+	pingCtx, cancel := context.WithTimeout(ctx, 5*time.Second)
+	defer cancel()
+	if err := c.Ping(pingCtx, nil); err != nil {
+		return nil, err
+	}
+
+	return &Client{client: c, db: c.Database("logjensticks")}, nil
+}
+
+func (c *Client) Disconnect(ctx context.Context) error {
+	return c.client.Disconnect(ctx)
+}
+
+func (c *Client) Users() *mongo.Collection {
+	return c.db.Collection("users")
+}
+
+func (c *Client) Sessions() *mongo.Collection {
+	return c.db.Collection("sessions")
+}
+
+func (c *Client) Lanes() *mongo.Collection {
+	return c.db.Collection("lanes")
+}