final routerHandler = shelf_router.Router()
      ..get('/message', (request) => handleWs(request));

Cascade cascade = Cascade()
      .add(routerHandler)
var handler = const Pipeline().addHandler(cascade.handler);

await shelf_io.serve(handler, ipAddress, port)

FutureOr<Response> handleWs(Request request) {
final channels = <WebSocketChannel>[];

return ws.webSocketHandler((WebSocketChannel channel) {
  channels.add(channel);
  channel.stream.listen((message) {
    try {
      final receivedMessage = Message.fromJson(json.decode(message));
      final updatedMessage = receivedMessage.copyWith(messageState: MessageState.sent);

      final response = json.encode(updatedMessage.toJson());
      for (final c in channels) {
        c.sink.add(response);
      }
    } catch (e) {
      debugPrint('Error while handling incoming message in socket: ');
    }
  }, onDone: () => channels.remove(channel)
  );
})(request);
}

@override
void initState() {
super.initState();
    _handleSocket();
}

void _handleSocket() {
    _webSocketChannel = WebSocketChannel.connect(Uri.parse(endpoint));
    _webSocketChannel.stream.listen((response) {
      print('response from ws: $response');
    });
}

void _sendMessage() async {
    final data = json.encode(message.toJson());
    _webSocketChannel.sink.add(data);
}