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
|
#include <stdio.h>
#include <signal.h>
#include "workflow/WFTaskFactory.h"
#include "workflow/WF%sServer.h"
#include "workflow/WFFacilities.h"
#include "config/config.h"
#include "config/util.h"
static WFFacilities::WaitGroup wait_group(1);
static srpc::RPCConfig config;
void sig_handler(int signo)
{
wait_group.done();
}
void init()
{
if (config.load("./proxy.conf") == false)
{
perror("Load config failed");
exit(1);
}
signal(SIGINT, sig_handler);
signal(SIGTERM, sig_handler);
}
void callback(WF%sTask *client_task)
{
int state = client_task->get_state();
int error = client_task->get_error();
SeriesWork *series = series_of(client_task);
protocol::%sResponse *resp = client_task->get_resp();
protocol::%sResponse *proxy_resp = (protocol::%sResponse *)series->get_context();
// Copy the remote server's response, to proxy response.
if (state == WFT_STATE_SUCCESS)%s
fprintf(stderr, "backend server state = %%d error = %%d. response client.\n",
state, error);
}
void process(WF%sTask *server_task)
{
protocol::%sRequest *req = server_task->get_req();
std::string backend_server = config.client_host();
unsigned short backend_server_port = config.client_port();
std::string url = std::string("%s://") + backend_server +
std::string(":") + std::to_string(backend_server_port);
WF%sTask *client_task = WFTaskFactory::create_%s_task(url,%s
config.retry_max(),
callback);
// Copy user's request to the new task's request using std::move()
%s
SeriesWork *series = series_of(server_task);
series->set_context(server_task->get_resp());
series->push_back(client_task);
fprintf(stderr, "proxy get request from client: ");
print_peer_address(server_task);
}
int main()
{
init();
WF%sServer proxy_server(process);
if (proxy_server.start(config.server_port()) == 0)
{
fprintf(stderr, "[%s]-[%s] proxy started, port %%u\n", config.server_port());
wait_group.wait();
proxy_server.stop();
}
else
perror("server start");
return 0;
}
|
