#include"../source/socket.hpp"

void test()
{
    Socket listensock;
    listensock.CreateServer("0.0.0.0",8888,true);
    std::cout<<"创建服务器成功"<<std::endl;
    int retsock=listensock.Accept();
    //阻塞发送数据
    std::string msg="hello world";
    ssize_t ret=listensock.Send(retsock,msg);
    if(ret>0)
    {
        std::cout<<"阻塞发送数据成功"<<std::endl;
        std::cout<<"发送的数据为: "<<msg<<std::endl;
    }
    else
    {
        std::cout<<"阻塞发送数据失败"<<std::endl;
        return;
    }
    //非阻塞发送数据
    ret=listensock.SendNoBlock(retsock,msg);
    if(ret>0)
    {
        std::cout<<"非阻塞发送数据成功"<<std::endl;
        std::cout<<"发送的数据为: "<<msg<<std::endl;
    }
    else
    {
        std::cout<<"非阻塞发送数据失败"<<std::endl;
        return;
    }
    //阻塞接收数据
    char buffer[1024];
    ret=listensock.Recv(retsock,buffer,sizeof(buffer));
    if(ret>0)
    {
        std::cout<<"阻塞接收数据成功"<<std::endl;
        std::cout<<"接收的数据为: "<<buffer<<std::endl;
    }
    else
    {
        std::cout<<"阻塞接收数据失败"<<std::endl;
        return;
    }
    //非阻塞接受数据
    while(1)
    {
        ret=listensock.RecvNoBlock(retsock,buffer,sizeof(buffer));
        if(ret>0)
        {
            std::cout<<"非阻塞接收数据成功"<<std::endl;
            std::cout<<"接收的数据为:" <<buffer<<std::endl;
            break;
        }
        else if(ret==0)
        {
            std::cout<<"数据未准备就绪"<<std::endl;
        }
        else
        {
            std::cout<<"数据接收失败"<<std::endl;
            return;
        }
        sleep(1);
    }
    listensock.Close();
}


int main()
{
    test();
    return 0;
}