使用c语言和tcp写一个仓库管理系统
时间: 2023-09-12 22:07:09 浏览: 102
远程仓库湿度监测系统设计-Proteus仿真与C语言源码
5星 · 资源好评率100%
仓库管理系统需要有一个服务器和多个客户端进行交互,因此可以使用TCP协议进行通信。下面是一个使用C语言和TCP实现的简单仓库管理系统。
1. 服务器端
服务器端需要监听客户端的连接请求,并处理客户端发送的数据。
首先需要包含头文件和定义一些常量和结构体:
```c
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <unistd.h>
#include <sys/socket.h>
#include <arpa/inet.h>
#define MAX_CLIENTS 10
#define MAX_BUFFER_SIZE 1024
typedef struct {
int id;
char name[20];
int quantity;
float price;
} Product;
```
接下来定义一个函数用于处理客户端发送的数据:
```c
void handle_client(int socket_fd, Product* products, int num_products) {
char buffer[MAX_BUFFER_SIZE];
int bytes_read = read(socket_fd, buffer, MAX_BUFFER_SIZE);
if (bytes_read == -1) {
perror("read");
exit(EXIT_FAILURE);
}
buffer[bytes_read] = '\0';
// 处理客户端发送的命令
char* command = strtok(buffer, " ");
if (strcmp(command, "list") == 0) { // 列出所有商品信息
for (int i = 0; i < num_products; i++) {
printf("%d %s %d %.2f\n", products[i].id, products[i].name, products[i].quantity, products[i].price);
}
write(socket_fd, "OK\n", 3);
} else if (strcmp(command, "add") == 0) { // 添加商品
char* name = strtok(NULL, " ");
int quantity = atoi(strtok(NULL, " "));
float price = atof(strtok(NULL, " "));
for (int i = 0; i < num_products; i++) {
if (strcmp(name, products[i].name) == 0) { // 已存在同名商品,更新数量和价格
products[i].quantity += quantity;
products[i].price = price;
write(socket_fd, "OK\n", 3);
return;
}
}
// 新建商品
int id = 0;
for (int i = 0; i < num_products; i++) {
if (products[i].id > id) {
id = products[i].id;
}
}
id++;
Product new_product = {id, "", quantity, price};
strncpy(new_product.name, name, 20);
products[num_products] = new_product;
num_products++;
write(socket_fd, "OK\n", 3);
} else if (strcmp(command, "remove") == 0) { // 删除商品
char* name = strtok(NULL, " ");
for (int i = 0; i < num_products; i++) {
if (strcmp(name, products[i].name) == 0) {
for (int j = i; j < num_products - 1; j++) {
products[j] = products[j+1];
}
num_products--;
write(socket_fd, "OK\n", 3);
return;
}
}
write(socket_fd, "ERROR\n", 6);
} else {
write(socket_fd, "ERROR\n", 6);
}
}
```
该函数首先读取客户端发送的数据,然后根据命令进行相应的处理。如果命令是"list",则列出所有商品信息;如果命令是"add",则添加商品;如果命令是"remove",则删除商品;否则返回错误信息。
接下来是主函数:
```c
int main() {
// 创建套接字
int server_fd = socket(AF_INET, SOCK_STREAM, 0);
if (server_fd == -1) {
perror("socket");
exit(EXIT_FAILURE);
}
// 绑定地址和端口
struct sockaddr_in server_addr = {0};
server_addr.sin_family = AF_INET;
server_addr.sin_addr.s_addr = INADDR_ANY;
server_addr.sin_port = htons(8888);
if (bind(server_fd, (struct sockaddr*)&server_addr, sizeof(server_addr)) == -1) {
perror("bind");
exit(EXIT_FAILURE);
}
// 监听连接
if (listen(server_fd, MAX_CLIENTS) == -1) {
perror("listen");
exit(EXIT_FAILURE);
}
// 初始化商品信息
Product products[100] = {
{1, "apple", 10, 3.5},
{2, "orange", 20, 2.8},
{3, "banana", 15, 4.0},
{4, "watermelon", 5, 12.0}
};
int num_products = 4;
while (1) {
// 接受客户端连接
struct sockaddr_in client_addr = {0};
socklen_t client_addr_len = sizeof(client_addr);
int client_fd = accept(server_fd, (struct sockaddr*)&client_addr, &client_addr_len);
if (client_fd == -1) {
perror("accept");
exit(EXIT_FAILURE);
}
// 处理客户端请求
handle_client(client_fd, products, num_products);
// 关闭连接
close(client_fd);
}
// 关闭套接字
close(server_fd);
return 0;
}
```
该函数首先创建套接字,并绑定到本地地址和端口。然后初始化商品信息,进入循环等待客户端连接。每次接收到客户端连接时,调用handle_client函数进行处理,并关闭连接。
2. 客户端
客户端需要连接服务器,并发送命令和参数,然后接收服务器的响应。
首先也需要包含头文件和定义一些常量和结构体:
```c
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <unistd.h>
#include <sys/socket.h>
#include <arpa/inet.h>
#define MAX_BUFFER_SIZE 1024
```
接下来定义一个函数用于发送命令和参数,并接收服务器响应:
```c
void send_command(int socket_fd, const char* command) {
char buffer[MAX_BUFFER_SIZE];
strncpy(buffer, command, MAX_BUFFER_SIZE);
int bytes_written = write(socket_fd, buffer, strlen(buffer));
if (bytes_written == -1) {
perror("write");
exit(EXIT_FAILURE);
}
int bytes_read = read(socket_fd, buffer, MAX_BUFFER_SIZE);
if (bytes_read == -1) {
perror("read");
exit(EXIT_FAILURE);
}
buffer[bytes_read] = '\0';
printf("%s", buffer);
}
```
该函数首先将命令和参数拼接成字符串,并发送给服务器。然后接收服务器响应,并打印出来。
接下来是主函数:
```c
int main() {
// 创建套接字
int client_fd = socket(AF_INET, SOCK_STREAM, 0);
if (client_fd == -1) {
perror("socket");
exit(EXIT_FAILURE);
}
// 连接服务器
struct sockaddr_in server_addr = {0};
server_addr.sin_family = AF_INET;
server_addr.sin_addr.s_addr = inet_addr("127.0.0.1");
server_addr.sin_port = htons(8888);
if (connect(client_fd, (struct sockaddr*)&server_addr, sizeof(server_addr)) == -1) {
perror("connect");
exit(EXIT_FAILURE);
}
// 发送命令
send_command(client_fd, "list\n");
send_command(client_fd, "add apple 5 3.2\n");
send_command(client_fd, "list\n");
send_command(client_fd, "add pear 10 2.5\n");
send_command(client_fd, "list\n");
send_command(client_fd, "remove apple\n");
send_command(client_fd, "list\n");
// 关闭套接字
close(client_fd);
return 0;
}
```
该函数首先创建套接字,并连接到服务器。然后发送一系列命令,接收服务器的响应,并打印出来。最后关闭套接字。
这就是一个简单的使用C语言和TCP实现的仓库管理系统。可以根据需要进行扩展和优化。
阅读全文