#40 Fix sending data via websocket

websocket_utils/websocket_utils.go

@@ -53,7 +53,7 @@ func NewSession(session *session.Session) *APIGateWaySessionWithHelpers {
 	}
 }
 
-func (s APIGateWaySessionWithHelpers) PostToConnectionIDs(req *events.APIGatewayWebsocketProxyRequest, connectionIDs []string) error {
+func (s APIGateWaySessionWithHelpers) PostToConnectionIDs(data []byte, req *events.APIGatewayWebsocketProxyRequest, connectionIDs []string) error {
 	if req == nil {
 		return errors.Error("websocket request is nil")
 	}
@@ -62,7 +62,7 @@ func (s APIGateWaySessionWithHelpers) PostToConnectionIDs(req *events.APIGateway
 		s.APIGatewaySession.Endpoint = fmt.Sprintf("https://%s/%s", req.RequestContext.DomainName, req.RequestContext.Stage)
 		_, err := s.APIGatewaySession.PostToConnection(&apigatewaymanagementapi.PostToConnectionInput{
 			ConnectionId: &connectionID,
-			Data:         []byte("This is a message sent to the requesting connection ID via websocket."),
+			Data:         data,
 		})
 		if err != nil {
 			return err