TcpServer_test.cpp 1.2 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546
  1. /*
  2. * TcpServer_test.cpp
  3. *
  4. * @build: make evpp
  5. *
  6. */
  7. #include "TcpServer.h"
  8. using namespace hv;
  9. int main(int argc, char* argv[]) {
  10. if (argc < 2) {
  11. printf("Usage: %s port\n", argv[0]);
  12. return -10;
  13. }
  14. int port = atoi(argv[1]);
  15. TcpServer srv;
  16. int listenfd = srv.createsocket(port);
  17. if (listenfd < 0) {
  18. return -20;
  19. }
  20. printf("server listen on port %d, listenfd=%d ...\n", port, listenfd);
  21. srv.onConnection = [](const SocketChannelPtr& channel) {
  22. std::string peeraddr = channel->peeraddr();
  23. if (channel->isConnected()) {
  24. printf("%s connected! connfd=%d\n", peeraddr.c_str(), channel->fd());
  25. } else {
  26. printf("%s disconnected! connfd=%d\n", peeraddr.c_str(), channel->fd());
  27. }
  28. };
  29. srv.onMessage = [](const SocketChannelPtr& channel, Buffer* buf) {
  30. // echo
  31. printf("< %.*s\n", (int)buf->size(), (char*)buf->data());
  32. channel->write(buf);
  33. };
  34. srv.onWriteComplete = [](const SocketChannelPtr& channel, Buffer* buf) {
  35. printf("> %.*s\n", (int)buf->size(), (char*)buf->data());
  36. };
  37. srv.setThreadNum(4);
  38. srv.start();
  39. while (1) hv_sleep(1);
  40. return 0;
  41. }