diff --git a/cmd/abc/appbase_import.go b/cmd/abc/appbase_import.go index 3deac0be200bcf582b9f8abe4636d9ea4541d1bc..6a30c49d963ab80c9974898dff1fd9adc7ff3934 100644 --- a/cmd/abc/appbase_import.go +++ b/cmd/abc/appbase_import.go @@ -31,6 +31,7 @@ var srcParamMap = map[string]string{ "sac_path": "sacPath", // "timeout": "timeout", "transform_file": "_transform_", + "log_dir": "xlog_dir", } var destParamMap = map[string]string{ @@ -64,6 +65,8 @@ func runImport(args []string) error { requestSize := flagset.Int64("request_size", 2<<19, "Http request size in bytes, specifically for bulk requests to ES.") bulkRequests := flagset.Int("bulk_requests", 1000, "Number of bulk requests to send during a network request to ES.") + logDir := flagset.String("log_dir", "", "used for storing commit logs") + transformFile := flagset.String("transform_file", "", "transform file to use") // use external config @@ -104,6 +107,7 @@ func runImport(args []string) error { "sacPath": *sacPath, "ssl": *ssl, "_transform_": *transformFile, + "xlog_dir": *logDir, } var destConfig = map[string]interface{}{ @@ -219,10 +223,20 @@ func writeConfigFile(srcConfig map[string]interface{}, destConfig map[string]int appFileHandle.WriteString(string(dat)) } else { // no transform file - appFileHandle.WriteString( - fmt.Sprintf(`t.Source("source", source, "/%s/").Save("sink", sink, "/.*/")`, - srcConfig["srcRegex"]), - ) + + // set Config({xlog_dir}) + if srcConfig["xlog_dir"] != "" { + confStr := fmt.Sprintf(`t.Config({"xlog_dir":"%s"}).Source("source", source, "/%s/").Save("sink", sink, "/.*/")`, srcConfig["xlog_dir"], srcConfig["srcRegex"]) + + fmt.Println(confStr) + + appFileHandle.WriteString(confStr) + } else { + appFileHandle.WriteString( + fmt.Sprintf(`t.Source("source", source, "/%s/").Save("sink", sink, "/.*/")`, + srcConfig["srcRegex"]), + ) + } } appFileHandle.WriteString("\n")