Files
CppNet/test/simple/CppNetClient.cpp
2019-09-29 21:47:55 +08:00

79 lines
2.1 KiB
C++

#include <iostream>
#include <string>
#include "CppNet.h"
#include "Runnable.h"
#include "Socket.h"
using namespace cppnet;
int index = 0;
std::string msg = "test msg => ";
std::string GetMsg() {
index++;
return (msg + std::to_string(index));
}
void WriteFunc(const Handle& handle, uint32_t len, uint32_t error) {
if (error != CEC_SUCCESS) {
std::cout << " [WriteFunc] something error : " << error << std::endl;
}
}
void ReadFunc(const Handle& handle, base::CBuffer* data, uint32_t len, uint32_t error) {
if (error != CEC_CLOSED && error != CEC_CONNECT_BREAK) {
std::cout << " [ReadFunc]" << std::endl;
std::cout << *(data) << std::endl;
data->Clear();
std::cout << " Thread ID : " << std::this_thread::get_id() << std::endl;
std::cout << " Read size : " << len << std::endl << std::endl;
base::CRunnable::Sleep(1000);
if (index > 5) {
Close(handle);
return;
}
auto msg = GetMsg();
Write(handle, msg.c_str(), msg.length());
} else {
std::cout << "Close" << std::endl;
}
}
void ConnectFunc(const Handle& handle, uint32_t err) {
if (err == CEC_SUCCESS) {
std::string ip;
uint16_t port;
GetIpAddress(handle, ip, port);
std::cout << " [ConnectFunc] : ip : " << ip << "port : " << port << std::endl;
auto msg = GetMsg();
Write(handle, msg.c_str(), msg.length());
} else {
std::cout << " [ConnectFunc] some thing error : " << err << std::endl;
}
}
void DisConnectionFunc(const Handle& handle, uint32_t err) {
std::cout << " [DisConnectionFunc] : " << err << std::endl;
}
int main() {
cppnet::Init(1);
cppnet::SetConnectionCallback(ConnectFunc);
cppnet::SetWriteCallback(WriteFunc);
cppnet::SetReadCallback(ReadFunc);
cppnet::SetDisconnectionCallback(DisConnectionFunc);
auto msg = GetMsg();
#ifndef __linux__
cppnet::Connection("127.0.0.1", 8921, msg.c_str(), msg.length());
#else
cppnet::Connection("127.0.0.1", 8921);
#endif // !__linux__
cppnet::Join();
}