breeze/CppNet
1
0
Fork 0
mirror of https://github.com/caozhiyi/CppNet.git synced 2026-01-26 16:30:05 +08:00
Code Issues Packages Projects Releases Wiki Activity

support windows iocp.

Browse Source
This commit is contained in:
caozhiyi
2021-03-24 22:42:26 +08:00
parent 41efa68459
commit c6ea003094
21 changed files with 794 additions and 221 deletions
Download Patch File Download Diff File Expand all files Collapse all files

125
cppnet/socket/rw_socket.cpp
View File

@@ -1,7 +1,6 @@
#include <errno.h>
#include "rw_socket.h"
#include "include/cppnet_type.h"
#include "cppnet/dispatcher.h"
#include "cppnet/cppnet_base.h"
@@ -10,8 +9,6 @@
#include "cppnet/event/action_interface.h"
#include "common/log/log.h"
#include "common/network/address.h"
#include "common/network/io_handle.h"
#include "common/alloter/pool_block.h"
#include "common/buffer/buffer_queue.h"
#include "common/alloter/alloter_interface.h"
@@ -65,31 +62,6 @@ void RWSocket::Read() {
}
}
bool RWSocket::Write(const char* src, uint32_t len) {
if (!_event) {
_event = _alloter->PoolNewSharePtr<Event>();
_event->SetSocket(shared_from_this());
}
//can't send now
if (_write_buffer->GetCanReadLength() > 0) {
if (_write_buffer->GetCanReadLength() > __max_write_cache) {
return false;
}
_write_buffer->Write(src, len);
auto actions = GetEventActions();
if (actions) {
return actions->AddSendEvent(_event);
}
return false;
} else {
_write_buffer->Write(src, len);
return Send();
}
}
void RWSocket::Connect(const std::string& ip, uint16_t port) {
if (!_event) {
_event = _alloter->PoolNewSharePtr<Event>();
@@ -154,14 +126,6 @@ void RWSocket::StopTimer(uint64_t timer_id) {
}
}
void RWSocket::OnRead(uint32_t len) {
Recv();
}
void RWSocket::OnWrite(uint32_t len) {
Send();
}
void RWSocket::OnConnect(uint16_t err) {
auto sock = shared_from_this();
if (err == CEC_SUCCESS) {
@@ -190,93 +154,4 @@ void RWSocket::OnDisConnect(uint16_t err) {
}
}
bool RWSocket::Recv() {
auto cppnet_base = _cppnet_base.lock();
if (!cppnet_base) {
return false;
}
uint32_t off_set = 0;
//read all data.
uint32_t expand_buff_len = __linux_read_buff_expand_len;
while (true) {
uint32_t expand = 0;
if (_read_buffer->GetCanWriteLength() == 0) {
expand = expand_buff_len;
if (expand_buff_len < __linux_read_buff_expand_max) {
expand_buff_len *= 2;
}
}
std::vector<Iovec> io_vec;
uint32_t buff_len = _read_buffer->GetFreeMemoryBlock(io_vec, expand);
auto ret = OsHandle::Readv(_sock, &*io_vec.begin(), io_vec.size());
if (ret._return_value < 0) {
if (errno == EWOULDBLOCK || errno == EAGAIN || errno == EINTR) {
_read_buffer->MoveWritePt(ret._return_value);
off_set += ret._return_value;
break;
} else if (errno == EBADMSG) {
OnDisConnect(CEC_CONNECT_BREAK);
return false;
} else {
}
} else if (ret._return_value == 0) {
OnDisConnect(CEC_CLOSED);
return false;
} else {
_read_buffer->MoveWritePt(ret._return_value);
off_set += ret._return_value;
// read all
if (ret._return_value < buff_len) {
break;
}
}
}
cppnet_base->OnRead(shared_from_this(), off_set);
return true;
}
bool RWSocket::Send() {
auto cppnet_base = _cppnet_base.lock();
if (!cppnet_base) {
return false;
}
uint32_t off_set = 0;
while(_write_buffer && _write_buffer->GetCanReadLength() > 0) {
std::vector<Iovec> io_vec;
uint32_t data_len = _write_buffer->GetUseMemoryBlock(io_vec, __linux_write_buff_get);
auto ret = OsHandle::Writev(_sock, &*io_vec.begin(), io_vec.size());
if (ret._return_value >= 0) {
_write_buffer->MoveReadPt(ret._return_value);
off_set += ret._return_value;
} else {
if (errno == EWOULDBLOCK || errno == EAGAIN || errno == EINTR) {
//can't send complete
auto actions = GetEventActions();
if (actions) {
return actions->AddSendEvent(_event);
}
return false;
} else if (errno == EBADMSG) {
OnDisConnect(CEC_CONNECT_BREAK);
return false;
} else {
OnDisConnect(CEC_CLOSED);
return false;
}
}
}
cppnet_base->OnWrite(shared_from_this(), off_set);
return true;
}
}