69 lines
1.3 KiB
Go
69 lines
1.3 KiB
Go
package main
|
|
|
|
import (
|
|
"context"
|
|
"flag"
|
|
"net/url"
|
|
"os"
|
|
"os/signal"
|
|
"replicator/replicator"
|
|
"syscall"
|
|
"time"
|
|
|
|
"gopkg.in/yaml.v3"
|
|
)
|
|
|
|
func main() {
|
|
ctx, can := signal.NotifyContext(context.Background(), syscall.SIGINT)
|
|
defer can()
|
|
|
|
config, err := getConfig()
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
|
|
if err := replicator.Main(ctx, config); err != nil && ctx.Err() == nil {
|
|
panic(err)
|
|
}
|
|
}
|
|
|
|
func getConfig() (replicator.Config, error) {
|
|
var config replicator.Config
|
|
if p := os.Getenv("CONFIG"); p != "" {
|
|
f, err := os.Open(p)
|
|
if err != nil {
|
|
return config, err
|
|
}
|
|
defer f.Close()
|
|
if err := yaml.NewDecoder(f).Decode(&config); err != nil {
|
|
return config, err
|
|
}
|
|
}
|
|
|
|
var srcDriver, destDriver, interval string
|
|
flag.StringVar(&srcDriver, "src", "map://", "source db")
|
|
flag.StringVar(&destDriver, "dest", "map://", "dest db")
|
|
flag.StringVar(&interval, "interval", "1s", "interval")
|
|
flag.Parse()
|
|
|
|
if srcDriver, err := url.Parse(srcDriver); err != nil {
|
|
return config, err
|
|
} else {
|
|
config.Src.Driver = *srcDriver
|
|
}
|
|
|
|
if destDriver, err := url.Parse(destDriver); err != nil {
|
|
return config, err
|
|
} else {
|
|
config.Dest.Driver = *destDriver
|
|
}
|
|
|
|
if interval, err := time.ParseDuration(interval); err != nil {
|
|
return config, err
|
|
} else {
|
|
config.Interval = replicator.Duration(interval)
|
|
}
|
|
|
|
return config, nil
|
|
}
|