parent
2d274e7798
commit
2edd7b08b6
Binary file not shown.
@ -0,0 +1,83 @@
|
||||
#!/usr/bin/env python
|
||||
# encoding: utf-8
|
||||
# @author: 原凯峰
|
||||
# @contact: 2894340009@qq.com
|
||||
# @software: pycharm
|
||||
# @file: MachineLearningDivider.py
|
||||
# @time: 2024/6/26 8:21
|
||||
# @desc:利用随机森林法进行模型训练，能够通过平均响应时间、故障率等数据计算出服务器的健康状态
|
||||
|
||||
|
||||
from sklearn.model_selection import train_test_split
|
||||
from sklearn.ensemble import RandomForestClassifier
|
||||
from sklearn.metrics import classification_report, accuracy_score
|
||||
import pickle
|
||||
def trainmodel():
|
||||
# 假设我们有以下数据集
|
||||
X = [
|
||||
[0.3, 0.005], # 服务器特征：平均响应时间和故障率
|
||||
[2.5, 0.03],
|
||||
[0.7, 0.045],
|
||||
[1.2, 0.002],
|
||||
[3.5, 0.1],
|
||||
[1.3, 0.05],
|
||||
[0.01, 0.15], # 服务器特征：平均响应时间和故障率
|
||||
[5, 0.03],
|
||||
[0.7, 0.015],
|
||||
[1.4, 0.02],
|
||||
[0.15, 0.2],
|
||||
[1.3, 0.005],
|
||||
|
||||
]
|
||||
y = ['良好', '一般', '一般', '良好', '差', '一般', '一般', '差', '良好', '差', '差', '良好'] # 对应的健康状态标签
|
||||
|
||||
# 将健康状态标签转换为数值
|
||||
label_mapping = {'一般': 0, '良好': 1, '差': 2}
|
||||
y_encoded = [label_mapping[label] for label in y]
|
||||
|
||||
# 划分训练集和测试集
|
||||
X_train, X_test, y_train, y_test = train_test_split(X, y_encoded, test_size=0.4, random_state=42)
|
||||
|
||||
# 选择模型，这里使用随机森林分类器
|
||||
model = RandomForestClassifier(n_estimators=100, random_state=42)
|
||||
|
||||
# 训练模型
|
||||
model.fit(X_train, y_train)
|
||||
|
||||
# 预测测试集
|
||||
y_pred = model.predict(X_test)
|
||||
|
||||
# 评估模型
|
||||
print(classification_report(y_test, y_pred))
|
||||
print("Accuracy:", accuracy_score(y_test, y_pred))
|
||||
|
||||
# 保存模型
|
||||
with open('server_health_model.pkl', 'wb') as file:
|
||||
pickle.dump(model, file)
|
||||
|
||||
|
||||
# trainmodel()
|
||||
# 定义一个函数来加载模型并进行预测
|
||||
def load_model_and_predict(new_data):
|
||||
with open('../LogAnalyze/server_health_model.pkl', 'rb') as file:
|
||||
loaded_model = pickle.load(file)
|
||||
predictions = loaded_model.predict(new_data)
|
||||
return predictions
|
||||
|
||||
# 定义一个函数来将预测结果转换为健康等级
|
||||
def predict_health_status(new_data):
|
||||
label_mapping = {'一般': 0, '良好': 1, '差': 2}
|
||||
predictions = load_model_and_predict(new_data)
|
||||
# 创建逆向映射字典
|
||||
inverse_label_mapping = {value: key for key, value in label_mapping.items()}
|
||||
# 使用逆向映射字典转换预测结果
|
||||
health_status = [inverse_label_mapping[pred] for pred in predictions]
|
||||
return health_status
|
||||
|
||||
# 测试函数
|
||||
def testcase():
|
||||
new_data = [[0.4, 0.01]] # 新的服务器数据
|
||||
health_status = predict_health_status(new_data)
|
||||
print("预测的健康状态：", health_status)
|
||||
|
||||
# testcase()
|
||||
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
@ -0,0 +1,22 @@
|
||||
from datetime import datetime, timedelta
|
||||
|
||||
# 定义时间格式
|
||||
date_format = "%Y-%m-%d"
|
||||
|
||||
# 获取当前日期，并按照定义的格式转换
|
||||
current_date = datetime.now().strftime(date_format)
|
||||
# 去除前导零
|
||||
current_date = current_date.replace("-0", "-")
|
||||
# 打印当前日期
|
||||
print("当前日期:", current_date)
|
||||
|
||||
for day in range(1,16):
|
||||
|
||||
# 计算当前日期之前15天的日期
|
||||
delta = timedelta(days=-day)
|
||||
previous_date = (datetime.now() + delta).strftime(date_format)
|
||||
# 去除前导零
|
||||
previous_date = previous_date.replace("-0", "-")
|
||||
print(previous_date)
|
||||
# 打印之前15天的日期
|
||||
print("之前15天的日期:", previous_date)
|
||||
@ -1,31 +0,0 @@
|
||||
#include <iostream>
|
||||
#include <mysql/mysql.h>
|
||||
|
||||
int main() {
|
||||
MYSQL *conn;
|
||||
conn = mysql_init(NULL);
|
||||
|
||||
if (conn == NULL) {
|
||||
std::cerr << "MySQL init failed" << std::endl;
|
||||
return 1;
|
||||
}
|
||||
|
||||
const char *server = "localhost";
|
||||
const char *user = "rtsw";
|
||||
const char *password = "123456";
|
||||
const char *database = "nginxdb";
|
||||
unsigned int port = 3306; // 使用云数据库提供的端口
|
||||
|
||||
if (mysql_real_connect(conn, server, user, password, database, port, NULL, 0) == NULL) {
|
||||
std::cerr << "Connection error: " << mysql_error(conn) << std::endl;
|
||||
mysql_close(conn);
|
||||
return 1;
|
||||
}
|
||||
|
||||
std::cout << "Successfully connected to the database" << std::endl;
|
||||
|
||||
// ... 执行数据库操作 ...
|
||||
|
||||
mysql_close(conn);
|
||||
return 0;
|
||||
}
|
||||
@ -1,86 +0,0 @@
|
||||
#include <iostream>
|
||||
#include <thread>
|
||||
#include <cstring>
|
||||
#include <sys/socket.h>
|
||||
#include <netinet/in.h>
|
||||
#include <unistd.h>
|
||||
#include <arpa/inet.h>
|
||||
|
||||
#define PORT 8080
|
||||
#define MAX_CLIENTS 5
|
||||
#define BUFFER_SIZE 1024
|
||||
|
||||
// 声明 handle_client 函数
|
||||
void handle_client(int client_socket);
|
||||
|
||||
int main() {
|
||||
int server_fd, new_socket;
|
||||
struct sockaddr_in server_addr, client_addr;
|
||||
socklen_t client_len = sizeof(client_addr);
|
||||
char buffer[BUFFER_SIZE];
|
||||
int opt = 1;
|
||||
|
||||
// 创建套接字
|
||||
server_fd = socket(AF_INET, SOCK_STREAM, 0);
|
||||
if (server_fd < 0) {
|
||||
perror("socket creation failed");
|
||||
exit(EXIT_FAILURE);
|
||||
}
|
||||
|
||||
// 设置选项，允许重用地址
|
||||
if (setsockopt(server_fd, SOL_SOCKET, SO_REUSEADDR | SO_REUSEPORT, &opt, sizeof(opt))) {
|
||||
perror("setsockopt failed");
|
||||
exit(EXIT_FAILURE);
|
||||
}
|
||||
|
||||
server_addr.sin_family = AF_INET;
|
||||
server_addr.sin_addr.s_addr = INADDR_ANY;
|
||||
server_addr.sin_port = htons(PORT);
|
||||
|
||||
// 绑定
|
||||
if (bind(server_fd, (struct sockaddr *)&server_addr, sizeof(server_addr)) < 0) {
|
||||
perror("bind failed");
|
||||
exit(EXIT_FAILURE);
|
||||
}
|
||||
|
||||
// 监听
|
||||
if (listen(server_fd, MAX_CLIENTS) < 0) {
|
||||
perror("listen failed");
|
||||
exit(EXIT_FAILURE);
|
||||
}
|
||||
|
||||
std::cout << "Server listening on port " << PORT << std::endl;
|
||||
|
||||
while (true) {
|
||||
client_len = sizeof(client_addr);
|
||||
new_socket = accept(server_fd, (struct sockaddr *)&client_addr, &client_len);
|
||||
if (new_socket < 0) {
|
||||
perror("accept failed");
|
||||
exit(EXIT_FAILURE);
|
||||
}
|
||||
|
||||
std::cout << "New client connected from " << inet_ntoa(client_addr.sin_addr) << " port " << ntohs(client_addr.sin_port) << std::endl;
|
||||
|
||||
// 创建并启动线程处理客户端
|
||||
std::thread(handle_client, new_socket).detach();
|
||||
}
|
||||
|
||||
close(server_fd);
|
||||
return 0;
|
||||
}
|
||||
|
||||
// 定义 handle_client 函数
|
||||
void handle_client(int client_socket) {
|
||||
while (true) {
|
||||
char buffer[BUFFER_SIZE];
|
||||
memset(buffer, 0, BUFFER_SIZE); // 清空缓冲区
|
||||
int nbytes = read(client_socket, buffer, BUFFER_SIZE);
|
||||
if (nbytes <= 0) {
|
||||
std::cout << "Client disconnected." << std::endl;
|
||||
close(client_socket);
|
||||
break;
|
||||
}
|
||||
std::cout << "Received message from client: " << buffer << std::endl;
|
||||
// 这里可以添加更多的处理逻辑
|
||||
}
|
||||
}
|
||||
@ -1,66 +0,0 @@
|
||||
#include <iostream>
|
||||
#include <mysql/mysql.h>
|
||||
|
||||
int main() {
|
||||
MYSQL *conn;
|
||||
conn = mysql_init(NULL);
|
||||
|
||||
if (conn == NULL) {
|
||||
std::cerr << "MySQL init failed" << std::endl;
|
||||
return 1;
|
||||
}
|
||||
|
||||
const char *server = "localhost";
|
||||
const char *user = "rtsw";
|
||||
const char *password = "123456";
|
||||
const char *database = "nginxdb";
|
||||
unsigned int port = 3306;
|
||||
|
||||
if (mysql_real_connect(conn, server, user, password, database, port, NULL, 0) == NULL) {
|
||||
std::cerr << "Connection error: " << mysql_error(conn) << std::endl;
|
||||
mysql_close(conn);
|
||||
return 1;
|
||||
}
|
||||
|
||||
std::cout << "Successfully connected to the database" << std::endl;
|
||||
|
||||
// 插入数据
|
||||
const char *insert_query = "INSERT INTO users (username, email) VALUES ('newuser', 'newuser@example.com')";
|
||||
if (mysql_query(conn, insert_query)) {
|
||||
std::cerr << "Insert error: " << mysql_error(conn) << std::endl;
|
||||
} else {
|
||||
std::cout << "Insert successful" << std::endl;
|
||||
}
|
||||
|
||||
// 查询数据
|
||||
const char *select_query = "SELECT * FROM users";
|
||||
MYSQL_RES *result = mysql_store_result(conn);
|
||||
if (result) {
|
||||
MYSQL_ROW row;
|
||||
while ((row = mysql_fetch_row(result))) {
|
||||
std::cout << "id: " << row[0] << ", username: " << row[1] << ", email: " << row[2] << std::endl;
|
||||
}
|
||||
mysql_free_result(result);
|
||||
} else {
|
||||
std::cerr << "Select error: " << mysql_error(conn) << std::endl;
|
||||
}
|
||||
|
||||
// 更新数据
|
||||
const char *update_query = "UPDATE users SET email = 'newuser_updated@example.com' WHERE username = 'newuser'";
|
||||
if (mysql_query(conn, update_query)) {
|
||||
std::cerr << "Update error: " << mysql_error(conn) << std::endl;
|
||||
} else {
|
||||
std::cout << "Update successful" << std::endl;
|
||||
}
|
||||
|
||||
// 删除数据
|
||||
const char *delete_query = "DELETE FROM users WHERE username = 'newuser'";
|
||||
if (mysql_query(conn, delete_query)) {
|
||||
std::cerr << "Delete error: " << mysql_error(conn) << std::endl;
|
||||
} else {
|
||||
std::cout << "Delete successful" << std::endl;
|
||||
}
|
||||
|
||||
mysql_close(conn);
|
||||
return 0;
|
||||
}
|
||||
@ -1,22 +0,0 @@
|
||||
#ifndef CONNECTOR_CONTROLLER_H
|
||||
#define CONNECTOR_CONTROLLER_H
|
||||
|
||||
#include <iostream>
|
||||
#include <thread>
|
||||
#include <vector>
|
||||
#include <map>
|
||||
#include <mutex>
|
||||
#include <atomic>
|
||||
#include <sys/socket.h>
|
||||
#include <netinet/in.h>
|
||||
#include <unistd.h>
|
||||
#include <arpa/inet.h>
|
||||
#include <cstring>
|
||||
|
||||
// 函数声明
|
||||
void StartHttpServer();
|
||||
void ConnectToGateway(const std::string& ip, int port);
|
||||
void KeepConnection(int sock);
|
||||
void HandleHttpRequest(int client_socket);
|
||||
|
||||
#endif // CONNECTOR_CONTROLLER_H
|
||||
@ -1,10 +0,0 @@
|
||||
#ifndef CONNECTOR_DATABASE_H
|
||||
#define CONNECTOR_DATABASE_H
|
||||
|
||||
#include <iostream>
|
||||
#include <mysql/mysql.h>
|
||||
|
||||
// 函数声明
|
||||
void DatabaseOperation();
|
||||
|
||||
#endif // CONNECTOR_DATABASE_H
|
||||
@ -1,35 +0,0 @@
|
||||
#include <iostream>
|
||||
#include <string>
|
||||
#include "connector_controller.h"
|
||||
#include "connector_database.h"
|
||||
|
||||
int main() {
|
||||
std::string command;
|
||||
bool running = true;
|
||||
|
||||
while (running) {
|
||||
std::cout << "Enter command (start_http_server/connect_gateway/database_operation/exit): ";
|
||||
std::cin >> command;
|
||||
|
||||
if (command == "start_http_server") {
|
||||
StartHttpServer();
|
||||
} else if (command == "connect_gateway") {
|
||||
std::string ip;
|
||||
int port;
|
||||
std::cout << "Enter the gateway IP address: "<<std::endl;
|
||||
std::cin>>ip; // 使用getline以获取包含空格的IP地址
|
||||
std::cout << "Enter the gateway port: "<<std::endl;
|
||||
std::cin >> port;
|
||||
ConnectToGateway(ip, port);
|
||||
} else if (command == "database_operation") {
|
||||
DatabaseOperation();
|
||||
} else if (command == "exit") {
|
||||
running = false;
|
||||
} else {
|
||||
std::cout << "Unknown command" << std::endl;
|
||||
}
|
||||
}
|
||||
|
||||
std::cout << "Exiting program." << std::endl;
|
||||
return 0;
|
||||
}
|
||||
Binary file not shown.
@ -1,125 +0,0 @@
|
||||
#include <iostream>
|
||||
#include <thread>
|
||||
#include <vector>
|
||||
#include <map>
|
||||
#include <mutex>
|
||||
#include <atomic>
|
||||
#include <sys/socket.h>
|
||||
#include <netinet/in.h>
|
||||
#include <unistd.h>
|
||||
#include <arpa/inet.h>
|
||||
|
||||
#define BUFFER_SIZE 1024
|
||||
|
||||
std::vector<std::thread> threads; // 存储所有连接线程
|
||||
std::map<int, int> sockets; // 存储套接字
|
||||
std::mutex sockets_mutex; // 用于同步访问 sockets
|
||||
std::atomic<bool> exit_flag(false); // 原子退出标志
|
||||
|
||||
|
||||
// 维护连接的函数
|
||||
void KeepConnection(int sock) {
|
||||
char buffer[BUFFER_SIZE];
|
||||
int nbytes;
|
||||
while (!exit_flag) {
|
||||
nbytes = recv(sock, buffer, BUFFER_SIZE, 0);
|
||||
if (nbytes <= 0) {
|
||||
std::cout << "Gateway disconnected" << std::endl;
|
||||
break;
|
||||
}
|
||||
std::cout << "Received from gateway (Socket " << sock << "): " << buffer << std::endl;
|
||||
}
|
||||
|
||||
{
|
||||
std::lock_guard<std::mutex> lock(sockets_mutex);
|
||||
sockets.erase(sock); // 从集合中移除套接字
|
||||
}
|
||||
close(sock);
|
||||
}
|
||||
// 连接到网关的函数
|
||||
void ConnectToGateway(const std::string& ip, int port) {
|
||||
int sock = socket(AF_INET, SOCK_STREAM, 0);
|
||||
if (sock < 0) {
|
||||
perror("Could not create socket");
|
||||
return;
|
||||
}
|
||||
|
||||
struct sockaddr_in server_addr;
|
||||
server_addr.sin_family = AF_INET;
|
||||
server_addr.sin_port = htons(port);
|
||||
if (inet_pton(AF_INET, ip.c_str(), &server_addr.sin_addr) <= 0) {
|
||||
perror("Invalid address");
|
||||
close(sock);
|
||||
return;
|
||||
}
|
||||
|
||||
if (connect(sock, (struct sockaddr *)&server_addr, sizeof(server_addr)) < 0) {
|
||||
perror("Connection to the server failed");
|
||||
close(sock);
|
||||
return;
|
||||
}
|
||||
|
||||
{
|
||||
std::lock_guard<std::mutex> lock(sockets_mutex);
|
||||
sockets[sock] = 1; // 将套接字添加到集合中
|
||||
}
|
||||
|
||||
std::cout << "Connected to gateway at " << ip << ":" << port << " (Socket " << sock << ")" << std::endl;
|
||||
|
||||
// 在新线程中维护连接
|
||||
threads.emplace_back(KeepConnection, sock);
|
||||
}
|
||||
|
||||
|
||||
|
||||
int main() {
|
||||
std::string command, ip;
|
||||
int port;
|
||||
bool in_connect_mode = false;
|
||||
|
||||
while (true) {
|
||||
std::cout << "Enter command (connect/exitconnect/exit): ";
|
||||
std::cin >> command;
|
||||
|
||||
if (command == "connect") {
|
||||
if (in_connect_mode) {
|
||||
std::cout << "Already in connect mode." << std::endl;
|
||||
continue;
|
||||
}
|
||||
in_connect_mode = true; // 进入连接模式
|
||||
std::cout << "Enter the gateway IP address: ";
|
||||
std::cin >> ip;
|
||||
std::cout << "Enter the gateway port: ";
|
||||
std::cin >> port;
|
||||
ConnectToGateway(ip, port);
|
||||
} else if (command == "exitconnect") {
|
||||
if (!in_connect_mode) {
|
||||
std::cout << "Not in connect mode." << std::endl;
|
||||
continue;
|
||||
}
|
||||
in_connect_mode = false; // 退出连接模式
|
||||
std::cout << "Exiting connect mode." << std::endl;
|
||||
} else if (command == "exit") {
|
||||
exit_flag = true; // 设置退出标志
|
||||
std::cout << "Exiting program." << std::endl;
|
||||
// 关闭所有套接字
|
||||
for (auto& sock_pair : sockets) {
|
||||
shutdown(sock_pair.first, SHUT_RDWR); // 关闭套接字的发送和接收
|
||||
close(sock_pair.first);
|
||||
}
|
||||
sockets.clear();
|
||||
// 等待所有线程结束
|
||||
for (auto& thread : threads) {
|
||||
if (thread.joinable()) {
|
||||
thread.join();
|
||||
}
|
||||
}
|
||||
threads.clear();
|
||||
break; // 退出主循环
|
||||
} else {
|
||||
std::cout << "Unknown command" << std::endl;
|
||||
}
|
||||
}
|
||||
|
||||
return 0;
|
||||
}
|
||||
@ -1,74 +0,0 @@
|
||||
#!/usr/bin/env python
|
||||
# encoding: utf-8
|
||||
# @author: 原凯峰
|
||||
# @contact: 2894340009@qq.com
|
||||
# @software: pycharm
|
||||
# @file: MachineLearningDivider.py
|
||||
# @time: 2024/6/26 8:21
|
||||
# @desc:利用随机森林法进行模型训练，能够通过平均响应时间、故障率等数据计算出服务器的健康状态
|
||||
|
||||
|
||||
from sklearn.model_selection import train_test_split
|
||||
from sklearn.ensemble import RandomForestClassifier
|
||||
from sklearn.metrics import classification_report, accuracy_score
|
||||
import pickle
|
||||
# 假设我们有以下数据集
|
||||
X = [
|
||||
[0.3, 0.005], # 服务器特征：平均响应时间和故障率
|
||||
[2.5, 0.03],
|
||||
[0.7, 0.045],
|
||||
[1.2, 0.002]
|
||||
]
|
||||
y = ['良好', '差', '差', '良好'] # 对应的健康状态标签
|
||||
|
||||
# 将健康状态标签转换为数值
|
||||
label_mapping = {'一般': 0, '良好': 1, '差': 2, '极差': 3}
|
||||
y_encoded = [label_mapping[label] for label in y]
|
||||
|
||||
# 划分训练集和测试集
|
||||
X_train, X_test, y_train, y_test = train_test_split(X, y_encoded, test_size=0.25, random_state=42)
|
||||
|
||||
# 选择模型，这里使用随机森林分类器
|
||||
model = RandomForestClassifier(n_estimators=100, random_state=42)
|
||||
|
||||
# 训练模型
|
||||
model.fit(X_train, y_train)
|
||||
|
||||
# 预测测试集
|
||||
y_pred = model.predict(X_test)
|
||||
|
||||
# 评估模型
|
||||
print(classification_report(y_test, y_pred))
|
||||
print("Accuracy:", accuracy_score(y_test, y_pred))
|
||||
|
||||
with open('server_health_model.pkl', 'wb') as file:
|
||||
pickle.dump(model, file)
|
||||
|
||||
|
||||
# 保存模型
|
||||
with open('server_health_model.pkl', 'wb') as file:
|
||||
pickle.dump(model, file)
|
||||
|
||||
# 定义一个函数来加载模型并进行预测
|
||||
def load_model_and_predict(new_data):
|
||||
with open('server_health_model.pkl', 'rb') as file:
|
||||
loaded_model = pickle.load(file)
|
||||
predictions = loaded_model.predict(new_data)
|
||||
return predictions
|
||||
|
||||
# 定义一个函数来将预测结果转换为健康等级
|
||||
def predict_health_status(new_data):
|
||||
predictions = load_model_and_predict(new_data)
|
||||
# 创建逆向映射字典
|
||||
inverse_label_mapping = {value: key for key, value in label_mapping.items()}
|
||||
# 使用逆向映射字典转换预测结果
|
||||
health_status = [inverse_label_mapping[pred] for pred in predictions]
|
||||
return health_status
|
||||
|
||||
# 测试函数
|
||||
def testcase():
|
||||
new_data = [[0.4, 0.01]] # 新的服务器数据
|
||||
health_status = predict_health_status(new_data)
|
||||
print("预测的健康状态：", health_status)
|
||||
|
||||
testcase()
|
||||
Binary file not shown.
Binary file not shown.
@ -1,2 +0,0 @@
|
||||
dic = {'a':1,'b':2,'c':3}
|
||||
for i in dic.keys():
|
||||
Loading…
Reference in new issue