63 lines
1.8 KiB
Go
63 lines
1.8 KiB
Go
// Command mongo-index ensures Gateway MongoDB indexes exist.
|
|
package main
|
|
|
|
import (
|
|
"context"
|
|
"flag"
|
|
"fmt"
|
|
"os"
|
|
"time"
|
|
|
|
"gateway/internal/config"
|
|
authrepo "gateway/internal/model/auth/repository"
|
|
memberrepo "gateway/internal/model/member/repository"
|
|
notifrepo "gateway/internal/model/notification/repository"
|
|
permrepo "gateway/internal/model/permission/repository"
|
|
|
|
"github.com/zeromicro/go-zero/core/conf"
|
|
)
|
|
|
|
var configFile = flag.String("f", "etc/gateway.dev.yaml", "config file")
|
|
|
|
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)
|
|
}
|
|
if err := memberrepo.EnsureMongoIndexes(ctx, &c.Mongo); err != nil {
|
|
return fmt.Errorf("mongo-index: member: %w", err)
|
|
}
|
|
if err := authrepo.EnsureMongoIndexes(ctx, &c.Mongo); err != nil {
|
|
return fmt.Errorf("mongo-index: auth: %w", err)
|
|
}
|
|
if err := permrepo.EnsureMongoIndexes(ctx, &c.Mongo); err != nil {
|
|
return fmt.Errorf("mongo-index: permission: %w", err)
|
|
}
|
|
|
|
fmt.Println("mongo-index: notifications + notification_dlq + member + auth + permission indexes OK")
|
|
return nil
|
|
}
|