-
Notifications
You must be signed in to change notification settings - Fork 1
/
server_1.cc
executable file
·113 lines (101 loc) · 3.22 KB
/
server_1.cc
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
// g++ main.cc -lpthread -lgsky -o gsky
// ./gsky -c ../conf/gsky.conf
//
#include <iostream>
#include <signal.h>
#include <unistd.h>
#include <gsky/server.hh>
#define UNUSED(var) do { (void)(var); } while (false)
using namespace gsky;
class gsky_server : public gsky::server {
protected:
void request(sp_request r, sp_writer w) override;
void offline(int fd) override;
};
gsky_server ser; // 创建服务器
void gsky_exit(int s) {
UNUSED(s);
ser.stop(); // 停止服务
}
void help() {
std::cout << "Usage: ./gsky [OPTION...] [SECTION] PAGE...\n"
"-c load configure file\n"
"-h help of gsky server\n"
"-v check version of gsky server\n"
;
}
enum class RouteRoot {
TestError = 0x00,
TestEcho = 0x10,
TestPush = 0x20,
TestMultiPush = 0x30,
};
void gsky_server::request(sp_request r, sp_writer w) {
log::info() << "client: " << r->fd << " 收到数据: " << r->content() << '\n';
switch((RouteRoot)r->route(0)) {
case RouteRoot::TestError: {
log::info() << "TestError";
w->send_data("TestError");
} break;
case RouteRoot::TestEcho: {
log::info() << "TestEcho";
w->send_data(r->content());
} break;
case RouteRoot::TestPush: {
log::info() << "TestPush";
std::string data;
data.resize(0x100000);
w->push_data(data);
w->push_data("Push data 1111 to you" + data);
} break;
case RouteRoot::TestMultiPush: {
log::info() << "TestMultiPush";
w->push_data("Push data 1111 to you");
w->push_data("Push data 2222 to you");
w->push_data("Push data 1111 to you");
std::string data;
data.resize(0x8000000);
w->send_data(data); // push big data
} break;
default: {
std::cout << "None\n";
w->send_data("None");
} break;
}
}
// 断开连接处理函数,fd为客户端文件描述符
void gsky_server::offline(int fd) {
std::cout << "client: " << fd << " offline ...\n";
}
int main(int argc, char **argv) {
::signal(SIGINT, gsky_exit); // Ctrl + c 退出服务器
int opt = 0;
ser.set_logger_path("./gsky.log"); // 设置服务日志路径 logger path, Default "./gksy.log"
ser.set_listen("0.0.0.0", 4096); // 设置服务日志路径, Defualt "0.0.0.0" 4096
ser.set_threads(8); // 设置服务线程数量, Default 4
//ser.set_protocol("pp"); // 设置协议, Default "pp"
// 获取参数
while((opt = getopt(argc, argv,"h::v::a::c:"))!=-1) {
switch (opt) {
case 'h': { // 帮助
help();
return 0;
} break;
case 'c': {
// 设置服务器配置文件路径
ser.load_config(optarg);
} break;
case 'v': {
// 显示 gsky lib 的版本号
std::cout << "gsky version: " << gsky::version() << '\n';
return 0;
} break;
default: {
std::cout << "-h get more info" << std::endl;
return -1;
}
}
}
ser.run(); // 启动gsky服务器
return 0;
}