設計模式
文章目錄
- 設計模式
- 一、`socket.h` 接口（interface）
- 二、`socket.cpp` 實現（implementation）
- 三、`server.cpp` 使用封裝（main 函數）
- 四、`client.cpp` 使用封裝（main 函數）
- 五、退出方法
- 六、終端輸出
一、socket.h 接口（interface）
#ifndef SOCKET_H
#define SOCKET_H#include <string>
using std::string;class Socket
{
public:Socket();~Socket();bool bind(int port);bool listen(int backlog = 5);int accept();bool connect(const string &ip, int port); // ? 客戶端連接方法int fd() const { return m_sockfd; } // 可選：暴露 sockfd 方便客戶端用private:int m_sockfd;
};#endif // SOCKET_H
二、socket.cpp 實現（implementation）
// socket.cpp
#include "socket.h"#include <sys/socket.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#include <unistd.h>
#include <cstring>
#include <cstdio>
#include <cerrno>using std::string;Socket::Socket()
{m_sockfd = ::socket(AF_INET, SOCK_STREAM, 0);if (m_sockfd < 0){printf("Socket creation failed: errno=%d, errmsg=%s\n", errno, strerror(errno));}
}Socket::~Socket()
{if (m_sockfd >= 0){::close(m_sockfd);}
}bool Socket::connect(const string &ip, int port)
{sockaddr_in addr;std::memset(&addr, 0, sizeof(addr));addr.sin_family = AF_INET;addr.sin_port = htons(port);if (::inet_pton(AF_INET, ip.c_str(), &addr.sin_addr) <= 0){printf("Invalid IP address: %s\n", ip.c_str());return false;}int ret = ::connect(m_sockfd, (struct sockaddr*)&addr, sizeof(addr));if (ret < 0){printf("Connect failed: errno=%d, errmsg=%s\n", errno, strerror(errno));return false;}return true;
}void Socket::close()
{if (m_sockfd >= 0){::close(m_sockfd);m_sockfd = -1;}
}
三、server.cpp 使用封裝（main 函數）
#include "socket.h"#include <iostream>
#include <unistd.h> // read(), write()
#include <cstring> // memsetint main()
{Socket server;if (!server.bind(8888)){std::cerr << "Failed to bind port 8888" << std::endl;return 1;}if (!server.listen()){std::cerr << "Failed to listen on port 8888" << std::endl;return 1;}std::cout << "Server listening on port 8888..." << std::endl;while (true){int client_fd = server.accept();if (client_fd < 0){continue; // accept failed}char buffer[1024];ssize_t n = ::read(client_fd, buffer, sizeof(buffer) - 1);if (n > 0){buffer[n] = '\0';std::cout << "Received: " << buffer << std::endl;std::string response = "Echo: ";response += buffer;::write(client_fd, response.c_str(), response.size());}::close(client_fd);}return 0;
}
四、client.cpp 使用封裝（main 函數）
#include "socket.h"#include <iostream>
#include <unistd.h> // read(), write()
#include <cstring> // strlen()int main()
{Socket client;if (!client.connect("127.0.0.1", 8888)){std::cerr << "Failed to connect to server." << std::endl;return 1;}std::cout << "Connected to server." << std::endl;std::string input;while (true){std::cout << "Enter message: ";std::getline(std::cin, input);if (input == "exit")break;::write(client.fd(), input.c_str(), input.size());char buffer[1024];ssize_t n = ::read(client.fd(), buffer, sizeof(buffer) - 1);if (n > 0){buffer[n] = '\0';std::cout << "Server response: " << buffer << std::endl;}else{std::cout << "Server closed connection." << std::endl;break;}}return 0;
}
五、退出方法
- 客戶端：
當你在客戶端輸入：
Enter message: exit
客戶端程序會退出（break 出循環）。
- 服務器：
- 當前版本是永不退出的死循環
- 你可以使用
Ctrl+C來中斷服務器進程
六、終端輸出
- 終端 1：服務器
$ ./server
Server listening on port 8888...
Received: hello
Received: world
- 終端 2：客戶端
$ ./client
Connected to server.
Enter message: hello
Server response: Echo: helloEnter message: world
Server response: Echo: worldEnter message: exit
