diff --git a/websocket_utils/websocket_utils.go b/websocket_utils/websocket_utils.go index dc27a36db516bd3405685d96b3302725c9fd96a3..7c508a79a3e1f7ce47ff0e8807678cd70dd1428f 100644 --- a/websocket_utils/websocket_utils.go +++ b/websocket_utils/websocket_utils.go @@ -59,16 +59,22 @@ func NewClient(req *events.APIGatewayWebsocketProxyRequest, config aws.Config) * } } -func (s APIGateWayClientWithHelpers) PostToConnectionIDs(data []byte, connectionIDs []string) error { +func (s APIGateWayClientWithHelpers) PostToConnectionID(data []byte, connectionID string) error { + _, err := s.APIGatewayClient.PostToConnection(context.TODO(), &apigatewaymanagementapi.PostToConnectionInput{ + ConnectionId: &connectionID, + Data: data, + }) + return err +} + +func (s APIGateWayClientWithHelpers) PostToConnectionIDs(data []byte, connectionIDs []string) { for _, connectionID := range connectionIDs { _, err := s.APIGatewayClient.PostToConnection(context.TODO(), &apigatewaymanagementapi.PostToConnectionInput{ ConnectionId: &connectionID, Data: data, }) if err != nil { - return err + logs.Info("failed to send data to connection ID: %s", connectionID) } } - - return nil }