36 lines
764 B
Go
36 lines
764 B
Go
package cli
|
|
|
|
import (
|
|
"fmt"
|
|
|
|
"github.com/urfave/cli/v2"
|
|
|
|
"github.com/haunt98/populatedb-go/internal/populatedb"
|
|
)
|
|
|
|
func (a *action) RunGenerate(c *cli.Context) error {
|
|
a.getFlags(c)
|
|
|
|
populator, err := populatedb.NewPopulator(
|
|
a.flags.dialect,
|
|
a.flags.url,
|
|
a.flags.verbose,
|
|
a.flags.dryRun,
|
|
)
|
|
if err != nil {
|
|
return fmt.Errorf("populatedb: failed to new populator: %w", err)
|
|
}
|
|
|
|
if a.flags.batchMode {
|
|
if err := populator.InsertBatch(c.Context, a.flags.table, a.flags.numberRecord); err != nil {
|
|
return fmt.Errorf("populatedb: failed to batch: %w", err)
|
|
}
|
|
} else {
|
|
if err := populator.Insert(c.Context, a.flags.table, a.flags.numberRecord); err != nil {
|
|
return fmt.Errorf("populatedb: failed to insert: %w", err)
|
|
}
|
|
}
|
|
|
|
return nil
|
|
}
|