kafka io attack: add username and password flags (#231)

* add username and password flags

Signed-off-by: cwen0 <[email protected]>

* fix ci

Signed-off-by: cwen0 <[email protected]>

---------

Signed-off-by: cwen0 <[email protected]>

Author: cwen0

.github/workflows/ci.yml

@@ -50,6 +50,7 @@ jobs:
           elif [[ "$job" == "unit-test" ]]; then
             make unit-test
           elif [[ "$job" == "integration-test" ]]; then
+            curl -sS https://dl.yarnpkg.com/debian/pubkey.gpg | sudo apt-key add -
             sudo apt-get update
             sudo apt-get install -y stress-ng
             make integration-test

cmd/attack/kafka.go

@@ -103,6 +103,10 @@ func NewKafkaIOCommand(dep fx.Option, options *core.KafkaCommand) *cobra.Command
 		},
 	}
 
+	cmd.Flags().StringVarP(&options.Host, "host", "H", "localhost", "the host of kafka server")
+	cmd.Flags().Uint16VarP(&options.Port, "port", "P", 9092, "the port of kafka server")
+	cmd.Flags().StringVarP(&options.Username, "username", "u", "", "the username of kafka client")
+	cmd.Flags().StringVarP(&options.Password, "password", "p", "", "the password of kafka client")
 	cmd.Flags().StringVarP(&options.ConfigFile, "config", "c", "/etc/kafka/server.properties", "the path of server config")
 	cmd.Flags().BoolVarP(&options.NonReadable, "non-readable", "r", false, "make kafka cluster non-readable")
 	cmd.Flags().BoolVarP(&options.NonWritable, "non-writable", "w", false, "make kafka cluster non-writable")