internal/client: add initial body data

README.md

@@ -18,7 +18,7 @@ go run cmd/h2-stream/main.go serve --addr=localhost:8080 --tls=false
 
 launch client:
 ```
-go run cmd/h2-stream/main.go cli --addr=http://localhost:8080 --method=POST
+go run cmd/h2-stream/main.go cli --addr=http://localhost:8080 --method=POST --data="BODY DATA"
 ```
 
 ### with TLS
@@ -29,5 +29,5 @@ go run cmd/h2-stream/main.go serve --addr=localhost:8080 --tls=true --cert=certs
 
 launch client:
 ```
-go run cmd/h2-stream/main.go cli --addr=https://localhost:8080 --method=POST --insecure
+go run cmd/h2-stream/main.go cli --addr=https://localhost:8080 --method=POST --data="BODY DATA" --insecure
 ```

internal/cli/cli.go

@@ -58,6 +58,7 @@ func newClientCommand() *cobra.Command {
 	cmd.Flags().StringVar(&cfg.Addr, "addr", "https://localhost:8080", "http server address")
 	cmd.Flags().StringVar(&cfg.Method, "method", "POST", "http method (GET, POST, PUT, DELETE...)")
 	cmd.Flags().BoolVar(&cfg.Insecure, "insecure", false, "skip tls validation")
+	cmd.Flags().StringVar(&cfg.Data, "data", "", "initial body data to be sent")
 	cmd.Flags().StringArrayVar(&cfg.Headers, "header", nil, "custom headers, ex.: \"Content-Type: application/json\"")
 	return cmd
 }

internal/client/client.go

@@ -21,6 +21,7 @@ type Config struct {
 	Method   string
 	Insecure bool
 	Headers  []string
+	Data     string
 }
 
 // Client is a http2 client
@@ -65,8 +66,16 @@ func New(cfg Config) (*Client, error) {
 
 // Run launches a client
 func (c *Client) Run() error {
+	// Set body reader
+	var body io.Reader
+	if c.Data != "" {
+		body = io.MultiReader(strings.NewReader(c.Data), os.Stdin)
+	} else {
+		body = os.Stdin
+	}
+
 	// Establish http connection
-	req, err := http.NewRequest(c.Method, c.Addr, os.Stdin)
+	req, err := http.NewRequest(c.Method, c.Addr, body)
 	if c.Headers != nil {
 		for _, h := range c.Headers {
 			kv := strings.Split(h, ":")