Revert "Add support websocket return response"

This reverts commit 7ee1944.

Author: yacchi

aws/api_gateway_websocket.go

@@ -166,14 +166,6 @@ func WebsocketResponse(w *WebsocketResponseWriter, multiValue bool) (r *events.A
 		r.Headers = utils.SemicolonSeparatedHeaderMap(w.Header())
 	}
 
-	if 0 < w.buf.Len() {
-		if r.IsBase64Encoded {
-			r.Body = base64.StdEncoding.EncodeToString(w.buf.Bytes())
-		} else {
-			r.Body = w.buf.String()
-		}
-	}
-
 	w.Done()
 	return
 }

aws/websocket_response_writer.go

@@ -1,7 +1,6 @@
 package aws
 
 import (
-	"bytes"
 	"context"
 	"github.com/aws/aws-lambda-go/events"
 	"net/http"
@@ -15,7 +14,6 @@ type WebsocketResponseWriter struct {
 	headers     http.Header
 	wroteHeader bool
 	closeCh     chan bool
-	buf         *bytes.Buffer
 }
 
 func NewWebsocketResponseWriter(ctx context.Context, client APIGatewayManagementAPI, request *events.APIGatewayWebsocketProxyRequest) *WebsocketResponseWriter {
@@ -25,7 +23,6 @@ func NewWebsocketResponseWriter(ctx context.Context, client APIGatewayManagement
 		req:     request,
 		headers: map[string][]string{},
 		closeCh: make(chan bool, 1),
-		buf:     bytes.NewBuffer(nil),
 	}
 }
 
@@ -38,8 +35,6 @@ func (w *WebsocketResponseWriter) Write(i []byte) (int, error) {
 		w.WriteHeader(http.StatusOK)
 	}
 
-	w.buf.Write(i)
-
 	err := w.client.PostToConnection(w.ctx, w.req.RequestContext.ConnectionID, i)
 	if err != nil {
 		return 0, err