Adding query stream codec to match query to specific partition