template-monorepo/cmd/mongo-index/main.go

// Command mongo-index ensures Gateway notification MongoDB indexes exist.
// Use when docker-entrypoint-initdb.d did not run (existing mongo_data volume).
package main

import (
	"context"
	"flag"
	"fmt"
	"os"
	"time"

	"gateway/internal/config"
	notifrepo "gateway/internal/model/notification/repository"

	"github.com/zeromicro/go-zero/core/conf"
)

var configFile = flag.String("f", "etc/gateway.dev.yaml", "config file (local; copy from etc/gateway.dev.example.yaml)")

func main() {
	if err := run(); err != nil {
		fmt.Fprintln(os.Stderr, err)
		os.Exit(1)
	}
}

func run() error {
	flag.Parse()

	var c config.Config
	conf.MustLoad(*configFile, &c)
	if c.Mongo.Host == "" {
		return fmt.Errorf("mongo-index: Mongo.Host is empty in config")
	}

	ctx, cancel := context.WithTimeout(context.Background(), 30*time.Second)
	defer cancel()

	notifRepo := notifrepo.NewNotificationRepository(notifrepo.NotificationRepositoryParam{Conf: &c.Mongo})
	dlqRepo := notifrepo.NewNotificationDLQRepository(notifrepo.NotificationDLQRepositoryParam{Conf: &c.Mongo})

	if err := notifRepo.Index20260520001UP(ctx); err != nil {
		return fmt.Errorf("mongo-index: notifications: %w", err)
	}
	if err := dlqRepo.Index20260520001UP(ctx); err != nil {
		return fmt.Errorf("mongo-index: notification_dlq: %w", err)
	}

	fmt.Println("mongo-index: notifications + notification_dlq indexes OK")
	return nil
}