feat(config): add auth-key CLI and server cache settings

1 files changed, 8 insertions, 6 deletions

diff --git a/cmd/dmap/main.go b/cmd/dmap/main.go
index ea5f020..498a09e 100644
--- a/cmd/dmap/main.go
+++ b/cmd/dmap/main.go
@@ -28,12 +28,13 @@ func main() {
 	var profileFlags profiling.Flags
 
 	args := config.Args{
-		Mode:              omode.MapClient,
-		SSHAgentKeyIndex:  -1,
+		Mode:             omode.MapClient,
+		SSHAgentKeyIndex: -1,
 	}
 	userName := user.Name()
 
 	flag.BoolVar(&args.NoColor, "noColor", false, "Disable ANSII terminal colors")
+	flag.BoolVar(&args.NoAuthKey, "no-auth-key", false, "Disable auth-key fast reconnect feature")
 	flag.BoolVar(&args.Quiet, "quiet", false, "Quiet output mode")
 	flag.BoolVar(&args.Plain, "plain", false, "Plain output mode")
 	flag.BoolVar(&args.TrustAllHosts, "trustAllHosts", false, "Trust all unknown host keys")
@@ -49,12 +50,13 @@ func main() {
 	flag.StringVar(&args.Logger, "logger", config.DefaultClientLogger, "Logger name")
 	flag.StringVar(&args.LogLevel, "logLevel", config.DefaultLogLevel, "Log level")
 	flag.StringVar(&args.SSHPrivateKeyFilePath, "key", "", "Path to private key")
+	flag.StringVar(&args.SSHPrivateKeyFilePath, "auth-key-path", "", "Path to auth key/private key (default ~/.ssh/id_rsa)")
 	flag.StringVar(&args.QueryStr, "query", "", "Map reduce query")
 	flag.StringVar(&args.ServersStr, "servers", "", "Remote servers to connect")
 	flag.StringVar(&args.UserName, "user", userName, "Your system user name")
 	flag.StringVar(&args.What, "files", "", "File(s) to read")
 	flag.StringVar(&pprof, "pprof", "", "Start PProf server this address")
-	
+
 	// Add profiling flags
 	profiling.AddFlags(&profileFlags)
 
@@ -92,15 +94,15 @@ func main() {
 	}
 
 	status := client.Start(ctx, signal.InterruptChWithCancel(ctx, cancel))
-	
+
 	// Log final metrics if profiling is enabled
 	if profileFlags.Enabled() {
 		profiler.LogMetrics("shutdown")
 	}
-	
+
 	// Stop profiler before exit
 	profiler.Stop()
-	
+
 	cancel()
 
 	wg.Wait()