Как использовать boost, чтобы поместить поток в буфер - PullRequest
Новая
       14

Как использовать boost, чтобы поместить поток в буфер

0 голосов
/ 30 сентября 2018

Я новичок в реализации потоков и буферов, и я не знаю, как это исправить: 

#include <boost/beast/core.hpp>
#include <boost/beast/websocket.hpp>
#include <boost/asio/ip/tcp.hpp>
#include <cstdlib>
#include <functional>
#include <iostream>
#include <string>
#include <thread>
#include <boost/property_tree/json_parser.hpp>
#include <boost/iostreams/device/array.hpp>
#include <boost/iostreams/stream.hpp>

using tcp = boost::asio::ip::tcp;
namespace websocket = boost::beast::websocket;
namespace pt = boost::property_tree;

struct DefaultMessage {
    std::string command;
    int value;
};

void defaultMessage(DefaultMessage *dm ,pt::ptree *root) {
    root->put("command", dm->command);
    root->put("value", dm->value);
}

// Echoes back all received WebSocket messages
void do_session(tcp::socket &socket) {
    try {
        // Construct the stream by moving in the socket
        websocket::stream<tcp::socket> ws{std::move(socket)};

        // Accept the websocket handshake
        ws.accept();

        for (;;) {
            // Read a message into the buffer
            boost::beast::multi_buffer buffer;
            ws.read(buffer);

            // Make string from buffer
            auto s = boost::beast::buffers_to_string(buffer.data());

            // Create ptree root
            pt::ptree root;

            // Create array source from s
            boost::iostreams::array_source array_source(&s[0], s.size());

            // Create input stream from array source
            boost::iostreams::stream<boost::iostreams::array_source> input_stream(array_source);

            // Read the json an populate ptree root
            pt::read_json(input_stream, root);

            // Discard all in buffer
            buffer.consume(buffer.size());

            // Construct a default message
            auto message = DefaultMessage{
                root.get<std::string>("command"),
                root.get<int>("value")
            };
            defaultMessage(&message, &root);

             // **This won't compile.**
            pt::write_json(buffer, root);

            // Echo the message back
            ws.text(ws.got_text());
            ws.write(buffer.data());
        }
    }
    catch (boost::system::system_error const &se) {
        // This indicates that the session was closed
        if (se.code() != websocket::error::closed) {
            std::cerr << "Error: " << se.code().message() << std::endl;
        }
    }
    catch (std::exception const &e) {
        std::cerr << "Error: " << e.what() << std::endl;
    }
}

Я хочу ответить структурой message.Так что, думаю, мне нужно положить его обратно в буфер.Эта часть является то, что не знаю, как это сделать.

1 Ответ

0 голосов
/ 30 сентября 2018

Если эффективность не является вашей главной задачей, я бы предложил 

std::ostringstream oss;
pt::write_json(oss, root);

для записи в строку и 

// Echo the message back
ws.text(ws.got_text());
ws.write(boost::asio::buffer(oss.str()));

для записи этого буфера.

Предостережения:

  • это делает больше выделений, чем строго требуется, вероятно
  • Boost PropertyTree НЕ является библиотекой JSON
...