基于 SQLite 的设备状态监控与建议生成服务器代码
#include <stdio.h> #include <stdlib.h> #include <string.h> #include <unistd.h> #include <sys/socket.h> #include <arpa/inet.h> #include <sqlite3.h>
#define MAX_BUFFER_SIZE 1024
typedef struct { int uid; char device_name[10]; char device_state[10]; char value[10]; char mode[10]; } Status;
typedef struct { int sockfd; struct sockaddr_in client_addr; socklen_t client_addr_len; sqlite3 *db; } ServerContext;
void handleClientRequest(ServerContext *context) { char buffer[MAX_BUFFER_SIZE]; int userid;
// 接收客户端发送的userid
ssize_t recvSize = recv(context->sockfd, &userid, sizeof(int), 0);
if (recvSize == -1) {
perror("recv");
return;
}
// 查询数据库获取设备状态信息
char sql[100];
snprintf(sql, sizeof(sql), "SELECT * FROM Status WHERE uid = %d", userid);
sqlite3_stmt *stmt;
int rc = sqlite3_prepare_v2(context->db, sql, -1, &stmt, NULL);
if (rc != SQLITE_OK) {
fprintf(stderr, "Failed to execute statement: %s\n", sqlite3_errmsg(context->db));
return;
}
Status status;
while (sqlite3_step(stmt) == SQLITE_ROW) {
status.uid = sqlite3_column_int(stmt, 1);
strncpy(status.device_name, sqlite3_column_text(stmt, 2), sizeof(status.device_name));
strncpy(status.device_state, sqlite3_column_text(stmt, 3), sizeof(status.device_state));
strncpy(status.value, sqlite3_column_text(stmt, 4), sizeof(status.value));
strncpy(status.mode, sqlite3_column_text(stmt, 5), sizeof(status.mode));
}
sqlite3_finalize(stmt);
// 分析设备状态并生成建议
char suggestion[MAX_BUFFER_SIZE];
if (strcmp(status.device_name, "空调") == 0) {
int temperature = atoi(status.value);
if (temperature < 24) {
snprintf(suggestion, sizeof(suggestion), "空调温度过低,建议提高温度至26℃");
} else {
snprintf(suggestion, sizeof(suggestion), "空调温度正常");
}
} else if (strcmp(status.device_name, "加湿器") == 0) {
int humidity = atoi(status.value);
if (humidity < 40 || humidity > 70) {
snprintf(suggestion, sizeof(suggestion), "加湿器湿度过高或过低,建议调整加湿器湿度");
} else {
snprintf(suggestion, sizeof(suggestion), "加湿器湿度正常");
}
} else {
snprintf(suggestion, sizeof(suggestion), "设备状态未知");
}
// 向客户端发送建议
ssize_t sendSize = send(context->sockfd, suggestion, strlen(suggestion), 0);
if (sendSize == -1) {
perror("send");
return;
}
}
void handleClientConnection(ServerContext *context) { int clientSockfd = accept(context->sockfd, (struct sockaddr *)&context->client_addr, &context->client_addr_len); if (clientSockfd == -1) { perror("accept"); return; }
context->sockfd = clientSockfd;
handleClientRequest(context);
close(clientSockfd);
}
int main() { int sockfd = socket(AF_INET, SOCK_STREAM, 0); if (sockfd == -1) { perror("socket"); return 1; }
struct sockaddr_in server_addr;
server_addr.sin_family = AF_INET;
server_addr.sin_addr.s_addr = INADDR_ANY;
server_addr.sin_port = htons(12345);
if (bind(sockfd, (struct sockaddr *)&server_addr, sizeof(server_addr)) == -1) {
perror("bind");
close(sockfd);
return 1;
}
if (listen(sockfd, 5) == -1) {
perror("listen");
close(sockfd);
return 1;
}
sqlite3 *db;
int rc = sqlite3_open("database.db", &db);
if (rc != SQLITE_OK) {
fprintf(stderr, "Failed to open database: %s\n", sqlite3_errmsg(db));
return 1;
}
ServerContext context;
context.sockfd = sockfd;
context.client_addr_len = sizeof(context.client_addr);
context.db = db;
while (1) {
handleClientConnection(&context);
}
sqlite3_close(db);
close(sockfd);
return 0;
}
这段代码是一个简单的服务器程序,用于接收客户端发送的用户ID,然后根据用户ID从数据库中查询设备状态信息,并根据设备状态生成建议,最后将建议发送给客户端。
从代码的结构和逻辑上来看,代码是合理的。它使用了合适的函数和数据结构来实现服务器的功能,包括创建套接字、绑定地址、监听连接、接收数据、查询数据库、生成建议和发送数据等操作。
然而,代码中还存在一些需要改进的地方。例如,没有对函数调用的返回值进行错误检查,没有处理数据库查询结果为空的情况,没有处理接收和发送数据的错误情况等。此外,代码中也没有进行错误处理和异常情况处理。
另外,代码中使用了硬编码的数据库文件名和服务器端口号,这样不够灵活。可以考虑将这些配置信息作为参数传递给程序,或者通过配置文件来读取。
总的来说,代码的基本结构和逻辑是合理的,但还需要进行一些改进和完善,以提高代码的健壮性和可靠性。
原文地址: http://www.cveoy.top/t/topic/fCcl 著作权归作者所有。请勿转载和采集!