Method Run

cmd/litestream/replicate.go:178–390

Run loads all databases specified in the configuration.

(ctx context.Context)

// Run loads all databases specified in the configuration.
func (c *ReplicateCommand) Run(ctx context.Context) (err error) {
	// Display version information.
	slog.Info("litestream", "version", Version, "level", c.Config.Logging.Level)

	// Start MCP server if enabled
	if c.Config.MCPAddr != "" {
		c.MCP, err = NewMCP(ctx, c.Config.ConfigPath)
		if err != nil {
			return err
		}
		go c.MCP.Start(c.Config.MCPAddr)
	}

	// Setup databases.
	if len(c.Config.DBs) == 0 {
		slog.Error("no databases specified in configuration")
	}

	// Attempt restore for databases that need it (before creating DB objects)
	for _, dbConfig := range c.Config.DBs {
		if dbConfig.RestoreIfDBNotExists && dbConfig.Path != "" {
			if err := c.restoreIfNeeded(ctx, dbConfig); err != nil {
				return err
			}
		}
	}

	var dbs []*litestream.DB
	var watchables []struct {
		config *DBConfig
		dbs    []*litestream.DB
	}
	for _, dbConfig := range c.Config.DBs {
		// Handle directory configuration
		if dbConfig.Dir != "" {
			dirDbs, err := NewDBsFromDirectoryConfig(dbConfig)
			if err != nil {
				return err
			}
			dbs = append(dbs, dirDbs...)
			slog.Info("found databases in directory", "dir", dbConfig.Dir, "count", len(dirDbs), "watch", dbConfig.Watch)
			if dbConfig.Watch {
				watchables = append(watchables, struct {
					config *DBConfig
					dbs    []*litestream.DB
				}{config: dbConfig, dbs: dirDbs})
			}
		} else {
			// Handle single database configuration
			db, err := NewDBFromConfig(dbConfig)
			if err != nil {
				return err
			}
			dbs = append(dbs, db)
		}
	}

	levels := c.Config.CompactionLevels()