实现一个简单的TCP网络程序(多进程,多线程,线程池版本的服务器)

这里只放了三个版本的服务器代码,完整代码请前往我的GitHub仓库—>戳我,戳我

tcp_process_server.hpp

#pragma once
#include"tcp_socket.hpp"
#include<signal.h>

typedef void (*Handler)(const string& buf,string& ret);

class TcpProcessServer{
  public:
    TcpProcessServer()
    {}
    ~TcpProcessServer(){
      _sock.Close();
    }
    bool Start(const string& ip,const int port,Handler handler){
      signal(SIGCLD,SIG_IGN);
      if(_sock.Bind(ip,port)==false)
        return false;
      if(_sock.Listen()==false)
        return false;
      while(1){
        TcpSocket NewSock;
        string ClientIp;
        int ClientPort;
        if(_sock.Accept(&NewSock,ClientIp,ClientPort)==false)
          continue;
        printf("[%s:%d]客户端已连接!~\n",ClientIp.c_str(),ClientPort);
        ProcessConnect(NewSock,ClientIp,ClientPort,handler);
      }
      return true;
    }
  private:
    void ProcessConnect(TcpSocket& NewSock,const string& ip,const int port,Handler handler){
      int fid=fork();
      if(fid>0){
        NewSock.Close();
        return;
      }
      while(1){
        string msg;
        int n=NewSock.Recv(msg);
        if(n<0){
          continue;
        }
        else if(n==0){
          printf("[%s:%d]客户端已断开!\n",ip.c_str(),port);
          break;
        }
        printf("[%s:%d]客户端发送了:%s\n",ip.c_str(),port,msg.c_str());
        string ret;
        handler(msg,ret);
        if(NewSock.Send(ret)==false)
          continue;
      }
      NewSock.Close();
      exit(0);
    }
    TcpSocket _sock;
};

tcp_thread_server.hpp

#pragma once
#include"tcp_socket.hpp"
#include<pthread.h>
typedef void (*Handler)(const string& buf,string& ret);

class TcpThreadServer{
  public:
    TcpThreadServer(){

    }
    ~TcpThreadServer(){
      _listen_sock.Close();
    }
    bool Start(const string& ip,const int port,Handler handler){
      int n=_listen_sock.Bind(ip,port);
      if(n<0){
        return false;
      }
      n=_listen_sock.Listen();
      if(n<0){
        return false;
      }
      while(1){
        TcpSocket client_sock;
        string peer_ip;
        int peer_port;
        if(_listen_sock.Accept(&client_sock,peer_ip,peer_port)==false)
          continue;
        printf("[%s:%d]客户端已连接!\n",peer_ip.c_str(),peer_port);
        ProcessClient(client_sock,ip,port,handler);
      }
    }
  private:
    TcpSocket _listen_sock;
    struct ThreadEntryArg{
      TcpSocket client_sock;
      string ip;
      int port;
      Handler handler;
    };
    void ProcessClient(TcpSocket& client_sock,const string& ip,const int port,Handler handler){
      pthread_t tid;
      ThreadEntryArg* arg=new ThreadEntryArg;
      arg->client_sock=client_sock;
      arg->ip=ip;
      arg->handler=handler;
      arg->port=port;
      pthread_create(&tid,NULL,ThreadEntry,(void*)arg);
      pthread_detach(tid);
    }
    static void* ThreadEntry(void* arg){
      ThreadEntryArg* argument=(ThreadEntryArg*)arg;
      string ip=argument->ip;
      int port=argument->port;
      Handler handler=argument->handler;
      TcpSocket client_sock=argument->client_sock;

      while(1){
        string req;
        int ret=client_sock.Recv(req);
        if(ret<0){
          continue;
        }
        else if(ret==0){
          printf("[%s:%d]客户端断连接!\n",ip.c_str(),port);
          break;
        }
        printf("[%s:%d]客户端输入:%s\n",ip.c_str(),port,req.c_str());
        string resp;
        handler(req,resp);
        client_sock.Send(resp);
      }
      client_sock.Close();
      delete argument;
      return NULL;
    }
};

tcp_threadpool_server.hpp

#pragma once
#include"tcp_socket.hpp"
#include<pthread.h>
#include"threadpool.hpp"
#include<sys/syscall.h>
#include<unistd.h>
#include<stdio.h>
  
typedef void (*Handler)(const string& ip,string& ret);
struct Arg{
  TcpSocket _sock;
  string _ip;
  int _port;
  Handler _handler;
};

class MyStack:public Stack{
  public:
    MyStack(void* arg)
      :Stack(arg)
      {}
    virtual void Entry()override
    {
      Arg *ret=(Arg*)_arg;
      TcpSocket ArgSock=ret->_sock;
      string ArgIp=ret->_ip;
      int ArgPort=ret->_port;
      Handler ArgHandler=ret->_handler;
      while(1){
        string msg;
        int n=ArgSock.Recv(msg);
        if(n<0){
          continue;
        }
        else if(n==0){
          printf("[%s:%d]客户端已经关闭!\n",ArgIp.c_str(),ArgPort);
          break;
        }
        printf("[%s:%d]客户端输入:%s\n",ArgIp.c_str(),ArgPort,msg.c_str());
        string ret;
        ArgHandler(msg,ret);
        if(ArgSock.Send(ret)==false)
          continue;
      }
      ArgSock.Close();
      delete ret;
    }
};


class TcpThreadServer{
  public:
    TcpThreadServer()
      :pool(10)
    {}
    ~TcpThreadServer(){
      _sock.Close();
    }
    bool Start(const string& ip,const int port,Handler handler){
      if(_sock.Bind(ip,port)==false)
        return false;
      if(_sock.Listen()==false)
        return false;
      while(1){
        Arg *NewArg=new Arg;
        NewArg->_handler=handler;
        if(_sock.Accept(&(NewArg->_sock),NewArg->_ip,NewArg->_port)==false)
          continue;
        printf("[%s:%d]客户端已连接!\n",NewArg->_ip.c_str(),NewArg->_port);
        pool.AddStack(new MyStack((void*)NewArg));
      }
    }

  private:
    TcpSocket _sock;
    ThreadPool pool;
};
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
多线程/线程池TCP服务器端程序设计需要考虑以下几个方面: 1. 网络通信:使用socket进行TCP通信,需要考虑如何接收和发送数据。 2. 多线程/线程池:使用多线程线程池来处理客户端连接请求,可以提高服务器的并发性能。 3. 数据处理:对接收到的数据进行处理,根据业务逻辑进行相应的处理操作。 4. 数据存储:将处理好的数据进行存储,可以使用数据库或者文件等方式。 下面是一个简单多线程TCP服务器端程序设计: ```python import socket import threading def handle_client(conn, addr): print("[*] Accepted connection from: %s:%d" % (addr[0], addr[1])) data = conn.recv(1024) response = "Hello client!" conn.send(response.encode()) conn.close() def main(): host = "127.0.0.1" port = 9999 server_socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM) server_socket.bind((host, port)) server_socket.listen(5) print("[*] Listening on %s:%d" % (host, port)) while True: conn, addr = server_socket.accept() client_handler = threading.Thread(target=handle_client, args=(conn, addr)) client_handler.start() if __name__ == "__main__": main() ``` 上面的程序使用了多线程来处理客户端连接请求,每个线程都会调用handle_client函数来处理连接。在handle_client函数中,首先接收客户端发送的数据,然后给客户端发送一个简单的响应,最后关闭连接。 需要注意的是,上面的程序只是一个简单的示例,实际应用中可能需要考虑更多的细节问题,比如线程池实现、异常处理、数据加密等。

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值