1bf130b25a
Sync CLI module imports across all command packages. Refactor daemon command with expanded functionality. Update go.mod and go.work dependencies. Co-Authored-By: Virgil <virgil@lethean.io>
49 lines
1.1 KiB
Go
49 lines
1.1 KiB
Go
package ml
|
|
|
|
import (
|
|
"fmt"
|
|
"os"
|
|
|
|
"forge.lthn.ai/core/go/pkg/cli"
|
|
"forge.lthn.ai/core/go-ai/ml"
|
|
)
|
|
|
|
var seedInfluxCmd = &cli.Command{
|
|
Use: "seed-influx",
|
|
Short: "Seed InfluxDB golden_gen from DuckDB golden_set",
|
|
Long: "One-time migration: batch-loads DuckDB golden_set records into InfluxDB golden_gen measurement.",
|
|
RunE: runSeedInflux,
|
|
}
|
|
|
|
var (
|
|
seedInfluxForce bool
|
|
seedInfluxBatchSize int
|
|
)
|
|
|
|
func init() {
|
|
seedInfluxCmd.Flags().BoolVar(&seedInfluxForce, "force", false, "Re-seed even if InfluxDB already has data")
|
|
seedInfluxCmd.Flags().IntVar(&seedInfluxBatchSize, "batch-size", 500, "Lines per InfluxDB write batch")
|
|
}
|
|
|
|
func runSeedInflux(cmd *cli.Command, args []string) error {
|
|
path := dbPath
|
|
if path == "" {
|
|
path = os.Getenv("LEM_DB")
|
|
}
|
|
if path == "" {
|
|
return fmt.Errorf("--db or LEM_DB required")
|
|
}
|
|
|
|
db, err := ml.OpenDB(path)
|
|
if err != nil {
|
|
return fmt.Errorf("open db: %w", err)
|
|
}
|
|
defer db.Close()
|
|
|
|
influx := ml.NewInfluxClient(influxURL, influxDB)
|
|
|
|
return ml.SeedInflux(db, influx, ml.SeedInfluxConfig{
|
|
Force: seedInfluxForce,
|
|
BatchSize: seedInfluxBatchSize,
|
|
}, os.Stdout)
|
|
}
|