96 lines
1.7 KiB
Go
96 lines
1.7 KiB
Go
package imp
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"net/url"
|
|
|
|
"git.cryptic.systems/volker.raschek/flucky/pkg/config"
|
|
"git.cryptic.systems/volker.raschek/flucky/pkg/repository"
|
|
"git.cryptic.systems/volker.raschek/go-logger"
|
|
"github.com/spf13/cobra"
|
|
)
|
|
|
|
var (
|
|
importSensors bool
|
|
importHumidities bool
|
|
importPressures bool
|
|
importTemperatures bool
|
|
)
|
|
|
|
func InitCmd(cmd *cobra.Command) error {
|
|
|
|
importCmd := &cobra.Command{
|
|
Use: "import",
|
|
Args: cobra.RangeArgs(1, 2),
|
|
Short: "Import data from passed URL",
|
|
Example: `import sqlite3:///var/cache/flucky/sqlite3.db
|
|
import sqlite3:///var/cache/flucky/sqlite3.db postgres://user:password@host:port/database?sslmode=disable`,
|
|
RunE: importSources,
|
|
}
|
|
cmd.AddCommand(importCmd)
|
|
|
|
return nil
|
|
}
|
|
|
|
func importSources(cmd *cobra.Command, args []string) error {
|
|
|
|
configFile, err := cmd.Flags().GetString("config")
|
|
if err != nil {
|
|
return fmt.Errorf("No config file defined")
|
|
}
|
|
|
|
cnf, err := config.Read(configFile)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
logLevelString, err := cmd.Flags().GetString("loglevel")
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
logLevel, err := logger.ParseLogLevel(logLevelString)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
flogger := logger.NewLogger(logLevel)
|
|
|
|
var (
|
|
srcURL *url.URL
|
|
destURL *url.URL
|
|
)
|
|
|
|
srcURL, err = url.Parse(args[0])
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
switch len(args) {
|
|
case 1:
|
|
destURL, err = url.Parse(cnf.DSN)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
case 2:
|
|
destURL, err = url.Parse(args[1])
|
|
if err != nil {
|
|
return err
|
|
}
|
|
}
|
|
|
|
srcRepository, err := repository.New(srcURL, flogger)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
destRepository, err := repository.New(destURL, flogger)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
return destRepository.Import(context.Background(), srcRepository)
|
|
}
|