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
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
|
/*
Copyright (c) 2022 Sogou, Inc.
Licensed under the Apache License, Version 2.0 (the "License");
you may not use this file except in compliance with the License.
You may obtain a copy of the License at
http://www.apache.org/licenses/LICENSE-2.0
Unless required by applicable law or agreed to in writing, software
distributed under the License is distributed on an "AS IS" BASIS,
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
See the License for the specific language governing permissions and
limitations under the License.
*/
#ifndef __SRPC_CONFIG__
#define __SRPC_CONFIG__
#include <stdint.h>
#include <generator/generator.h>
#ifndef _WIN32
#include <unistd.h>
#include <sys/param.h>
#else
#define MAXPATHLEN 4096
#include <direct.h>
#endif
#define COLOR_OFF "\033[0m"
#define COLOR_WHITE "\033[37;1m"
#define COLOR_RED "\033[31;1m"
#define COLOR_GREEN "\033[32;1m"
#define COLOR_YELLOW "\033[33;1m"
#define COLOR_PURPLE "\033[34;1m"
#define COLOR_PINK "\033[35;1m"
#define COLOR_BLUE "\033[36;1m"
#define COLOR_LPURPLE "\033[94;1m"
#define COLOR_COMMAND COLOR_BLUE
#define COLOR_INFO COLOR_YELLOW
#define COLOR_FLAG COLOR_PURPLE
struct srpc_config
{
uint8_t type;
const char *project_name;
const char *service_name;
char template_path[MAXPATHLEN];
char depend_path[MAXPATHLEN];
char output_path[MAXPATHLEN];
// rpc
uint8_t rpc_type;
uint8_t idl_type;
uint8_t data_type;
uint8_t compress_type;
bool specified_depend_path;
const char *specified_idl_file;
const char *specified_idl_path;
// proxy
uint8_t proxy_server_type;
uint8_t proxy_client_type;
srpc_config();
bool prepare_args();
bool prepare_project_path();
bool prepare_dependencies() const;
bool prepare_specified_idl_file();
const char *rpc_type_string() const;
const char *rpc_compress_string() const;
const char *rpc_data_string() const;
void set_rpc_type(const char *optarg);
void set_idl_type(const char *optarg);
void set_data_type(const char *optarg);
void set_compress_type(const char *optarg);
const char *proxy_server_type_string() const;
const char *proxy_client_type_string() const;
};
class ControlGenerator : public Generator
{
public:
bool generate_client_cpp_file(const idl_info& cur_info,
const std::string& idl_file_name);
bool generate_server_cpp_file(const idl_info& cur_info,
const std::string& idl_file_name);
ControlGenerator(const struct srpc_config *config);
private:
class ControlPrinter : public Printer
{
public:
ControlPrinter(bool is_thrift) : Printer(is_thrift) { }
void print_clt_include();
void print_server_load_config();
void print_client_load_config();
void print_client_params();
void print_client_done_method_ctl(const std::vector<std::string>& pkg,
const std::string& service,
const std::string& method,
const std::string& response);
void print_client_main_service_ctl(const std::string& type,
const std::vector<std::string>& pkg,
const std::string& service,
const std::string& suffix);
void print_server_construct(const char *rpc_type);
void print_server_main_end_ctl(const char *project_name,
const char *rpc_type);
};
ControlPrinter ctl_printer;
const struct srpc_config *config;
};
enum
{
COMMAND_HTTP,
COMMAND_REDIS,
COMMAND_MYSQL,
COMMAND_KAFKA,
COMMAND_RPC,
COMMAND_API,
COMMAND_PROXY,
COMMAND_FILE,
COMMAND_COMPUTE
};
enum
{
PROTOCOL_TYPE_HTTP = COMMAND_HTTP,
PROTOCOL_TYPE_REDIS = COMMAND_REDIS,
PROTOCOL_TYPE_MYSQL = COMMAND_MYSQL,
PROTOCOL_TYPE_KAFKA = COMMAND_KAFKA,
PROTOCOL_TYPE_SRPC = 22,
PROTOCOL_TYPE_SRPC_HTTP,
PROTOCOL_TYPE_BRPC,
PROTOCOL_TYPE_THRIFT,
PROTOCOL_TYPE_THRIFT_HTTP,
PROTOCOL_TYPE_TRPC,
PROTOCOL_TYPE_TRPC_HTTP,
PROTOCOL_TYPE_MAX
};
enum
{
IDL_TYPE_DEFAULT,
IDL_TYPE_PROTOBUF,
IDL_TYPE_THRIFT,
IDL_TYPE_MAX
};
enum
{
DATA_TYPE_DEFAULT,
DATA_TYPE_PROTOBUF,
DATA_TYPE_THRIFT,
DATA_TYPE_JSON,
DATA_TYPE_MAX
};
enum
{
COMPRESS_TYPE_NONE,
COMPRESS_TYPE_SNAPPY,
COMPRESS_TYPE_GZIP,
COMPRESS_TYPE_ZLIB,
COMPRESS_TYPE_LZ4,
COMPRESS_TYPE_MAX
};
enum
{
PROXY_BASIC_TYPE = 1,
PROXY_PROTOBUF_TYPE = 2,
PROXY_THRIFT_TYPE = 3
};
void usage(int argc, const char *argv[]);
void usage_http(int argc, const char *argv[]);
void usage_db(int argc, const char *argv[], const struct srpc_config *config);
void usage_kafka(int argc, const char *argv[]);
void usage_rpc(int argc, const char *argv[], const struct srpc_config *config);
const char *get_type_string(uint8_t type);
int check_proxy_type(uint8_t type);
#endif
|
