We did a write-up last year about combining our WebSocket gateway with a FaaS backend: http://blog.fanout.io/2016/09/29/serverless-websocket-chat/

The FaaS used in the article is Microcule, but the same approach should work with Lambda now that API Gateway supports binary.