Files
rippled/examples/iostream_server/iostream_server.cpp
2013-04-24 06:56:59 -05:00

126 lines
3.4 KiB
C++

#include <websocketpp/config/core.hpp>
#include <websocketpp/server.hpp>
#include <iostream>
#include <fstream>
#include <unistd.h>
typedef websocketpp::server<websocketpp::config::core> server;
using websocketpp::lib::placeholders::_1;
using websocketpp::lib::placeholders::_2;
using websocketpp::lib::bind;
// pull out the type of messages sent by our config
typedef server::message_ptr message_ptr;
// Define a callback to handle incoming messages
void on_message(server* s, websocketpp::connection_hdl hdl, message_ptr msg) {
/*std::cout << "on_message called with hdl: " << hdl.lock().get()
<< " and message: " << msg->get_payload()
<< std::endl;
*/
try {
s->send(hdl, msg->get_payload(), msg->get_opcode());
} catch (const websocketpp::lib::error_code& e) {
/* std::cout << "Echo failed because: " << e
<< "(" << e.message() << ")" << std::endl;*/
}
}
int main() {
server s;
std::ofstream log;
try {
// Clear logging because we are using std out for data
// TODO: fix when we can log to files
s.set_error_channels(websocketpp::log::elevel::all);
s.set_access_channels(websocketpp::log::alevel::all);
log.open("output.log");
s.get_alog().set_ostream(&log);
s.get_elog().set_ostream(&log);
// print all output to stdout
s.register_ostream(&std::cout);
// Register our message handler
s.set_message_handler(bind(&on_message,&s,::_1,::_2));
server::connection_ptr con = s.get_connection();
con->start();
//std::cin >> *con;
//log << "ready done" << std::endl;
std::string temp;
while(std::cin >> temp) {
con->readsome(temp.data(),temp.size());
std::cout << temp;
std::cout.flush();
}
/*char buf[512];
size_t bytes_read;
size_t bytes_processed;
while(std::cin) {
bytes_read = std::cin.readsome(buf,512);
if (bytes_read > 0) {
std::cout << "read " << bytes_read << " bytes "
<< websocketpp::utility::to_hex(buf,bytes_read)
<< std::endl;
}
bytes_processed = 0;
while (bytes_processed < bytes_read) {
std::cout << "foo" << std::endl;
bytes_processed += con->readsome(buf+bytes_processed,
bytes_read-bytes_processed);
std::cout << "bar" << std::endl;
sleep(1);
}
sleep(1);
}*/
std::cout << "end" << std::endl;
} catch (const std::exception & e) {
std::cout << e.what() << std::endl;
} catch (websocketpp::lib::error_code e) {
std::cout << e.message() << std::endl;
} catch (...) {
std::cout << "other exception" << std::endl;
}
log.close();
}
/*server test_server;
server::connection_ptr con;
test_server.set_message_handler(bind(&echo_func,&test_server,::_1,::_2));
std::stringstream output;
test_server.register_ostream(&output);
con = test_server.get_connection();
con->start();
std::stringstream channel;
channel << input;
channel >> *con;
return output.str();*/