Skip to content
GitLab
Explore
Sign in
Primary navigation
Search or go to…
Project
G
Grpc
Manage
Activity
Members
Labels
Plan
Issues
Issue boards
Milestones
Wiki
Code
Merge requests
Repository
Branches
Commits
Tags
Repository graph
Compare revisions
Snippets
Build
Pipelines
Jobs
Pipeline schedules
Artifacts
Deploy
Releases
Container registry
Model registry
Operate
Environments
Monitor
Incidents
Service Desk
Analyze
Value stream analytics
Contributor analytics
CI/CD analytics
Repository analytics
Model experiments
Help
Help
Support
GitLab documentation
Compare GitLab plans
Community forum
Contribute to GitLab
Provide feedback
Keyboard shortcuts
?
Snippets
Groups
Projects
Show more breadcrumbs
tci-gateway-module
Grpc
Commits
0084179d
Commit
0084179d
authored
10 years ago
by
Nicolas "Pixel" Noble
Browse files
Options
Downloads
Patches
Plain Diff
Removing the `using std::string` in the python plugin code.
parent
51c92625
No related branches found
No related tags found
No related merge requests found
Changes
2
Hide whitespace changes
Inline
Side-by-side
Showing
2 changed files
src/compiler/python_generator.cc
+39
-41
39 additions, 41 deletions
src/compiler/python_generator.cc
src/compiler/python_plugin.cc
+4
-6
4 additions, 6 deletions
src/compiler/python_plugin.cc
with
43 additions
and
47 deletions
src/compiler/python_generator.cc
+
39
−
41
View file @
0084179d
...
@@ -60,8 +60,6 @@ using std::make_pair;
...
@@ -60,8 +60,6 @@ using std::make_pair;
using
std
::
map
;
using
std
::
map
;
using
std
::
pair
;
using
std
::
pair
;
using
std
::
replace
;
using
std
::
replace
;
using
std
::
string
;
using
std
::
strlen
;
using
std
::
vector
;
using
std
::
vector
;
namespace
grpc_python_generator
{
namespace
grpc_python_generator
{
...
@@ -72,14 +70,14 @@ namespace {
...
@@ -72,14 +70,14 @@ namespace {
// Converts an initializer list of the form { key0, value0, key1, value1, ... }
// Converts an initializer list of the form { key0, value0, key1, value1, ... }
// into a map of key* to value*. Is merely a readability helper for later code.
// into a map of key* to value*. Is merely a readability helper for later code.
map
<
string
,
string
>
ListToDict
(
const
initializer_list
<
string
>&
values
)
{
map
<
std
::
string
,
std
::
string
>
ListToDict
(
const
initializer_list
<
std
::
string
>&
values
)
{
assert
(
values
.
size
()
%
2
==
0
);
assert
(
values
.
size
()
%
2
==
0
);
map
<
string
,
string
>
value_map
;
map
<
std
::
string
,
std
::
string
>
value_map
;
auto
value_iter
=
values
.
begin
();
auto
value_iter
=
values
.
begin
();
for
(
unsigned
i
=
0
;
i
<
values
.
size
()
/
2
;
++
i
)
{
for
(
unsigned
i
=
0
;
i
<
values
.
size
()
/
2
;
++
i
)
{
string
key
=
*
value_iter
;
std
::
string
key
=
*
value_iter
;
++
value_iter
;
++
value_iter
;
string
value
=
*
value_iter
;
std
::
string
value
=
*
value_iter
;
value_map
[
key
]
=
value
;
value_map
[
key
]
=
value
;
++
value_iter
;
++
value_iter
;
}
}
...
@@ -113,8 +111,8 @@ class IndentScope {
...
@@ -113,8 +111,8 @@ class IndentScope {
bool
PrintServicer
(
const
ServiceDescriptor
*
service
,
bool
PrintServicer
(
const
ServiceDescriptor
*
service
,
Printer
*
out
)
{
Printer
*
out
)
{
string
doc
=
"<fill me in later!>"
;
std
::
string
doc
=
"<fill me in later!>"
;
map
<
string
,
string
>
dict
=
ListToDict
({
map
<
std
::
string
,
std
::
string
>
dict
=
ListToDict
({
"Service"
,
service
->
name
(),
"Service"
,
service
->
name
(),
"Documentation"
,
doc
,
"Documentation"
,
doc
,
});
});
...
@@ -125,7 +123,7 @@ bool PrintServicer(const ServiceDescriptor* service,
...
@@ -125,7 +123,7 @@ bool PrintServicer(const ServiceDescriptor* service,
out
->
Print
(
"__metaclass__ = abc.ABCMeta
\n
"
);
out
->
Print
(
"__metaclass__ = abc.ABCMeta
\n
"
);
for
(
int
i
=
0
;
i
<
service
->
method_count
();
++
i
)
{
for
(
int
i
=
0
;
i
<
service
->
method_count
();
++
i
)
{
auto
meth
=
service
->
method
(
i
);
auto
meth
=
service
->
method
(
i
);
string
arg_name
=
meth
->
client_streaming
()
?
std
::
string
arg_name
=
meth
->
client_streaming
()
?
"request_iterator"
:
"request"
;
"request_iterator"
:
"request"
;
out
->
Print
(
"@abc.abstractmethod
\n
"
);
out
->
Print
(
"@abc.abstractmethod
\n
"
);
out
->
Print
(
"def $Method$(self, $ArgName$, context):
\n
"
,
out
->
Print
(
"def $Method$(self, $ArgName$, context):
\n
"
,
...
@@ -140,8 +138,8 @@ bool PrintServicer(const ServiceDescriptor* service,
...
@@ -140,8 +138,8 @@ bool PrintServicer(const ServiceDescriptor* service,
}
}
bool
PrintServer
(
const
ServiceDescriptor
*
service
,
Printer
*
out
)
{
bool
PrintServer
(
const
ServiceDescriptor
*
service
,
Printer
*
out
)
{
string
doc
=
"<fill me in later!>"
;
std
::
string
doc
=
"<fill me in later!>"
;
map
<
string
,
string
>
dict
=
ListToDict
({
map
<
std
::
string
,
std
::
string
>
dict
=
ListToDict
({
"Service"
,
service
->
name
(),
"Service"
,
service
->
name
(),
"Documentation"
,
doc
,
"Documentation"
,
doc
,
});
});
...
@@ -169,8 +167,8 @@ bool PrintServer(const ServiceDescriptor* service, Printer* out) {
...
@@ -169,8 +167,8 @@ bool PrintServer(const ServiceDescriptor* service, Printer* out) {
bool
PrintStub
(
const
ServiceDescriptor
*
service
,
bool
PrintStub
(
const
ServiceDescriptor
*
service
,
Printer
*
out
)
{
Printer
*
out
)
{
string
doc
=
"<fill me in later!>"
;
std
::
string
doc
=
"<fill me in later!>"
;
map
<
string
,
string
>
dict
=
ListToDict
({
map
<
std
::
string
,
std
::
string
>
dict
=
ListToDict
({
"Service"
,
service
->
name
(),
"Service"
,
service
->
name
(),
"Documentation"
,
doc
,
"Documentation"
,
doc
,
});
});
...
@@ -181,7 +179,7 @@ bool PrintStub(const ServiceDescriptor* service,
...
@@ -181,7 +179,7 @@ bool PrintStub(const ServiceDescriptor* service,
out
->
Print
(
"__metaclass__ = abc.ABCMeta
\n
"
);
out
->
Print
(
"__metaclass__ = abc.ABCMeta
\n
"
);
for
(
int
i
=
0
;
i
<
service
->
method_count
();
++
i
)
{
for
(
int
i
=
0
;
i
<
service
->
method_count
();
++
i
)
{
const
MethodDescriptor
*
meth
=
service
->
method
(
i
);
const
MethodDescriptor
*
meth
=
service
->
method
(
i
);
string
arg_name
=
meth
->
client_streaming
()
?
std
::
string
arg_name
=
meth
->
client_streaming
()
?
"request_iterator"
:
"request"
;
"request_iterator"
:
"request"
;
auto
methdict
=
ListToDict
({
"Method"
,
meth
->
name
(),
"ArgName"
,
arg_name
});
auto
methdict
=
ListToDict
({
"Method"
,
meth
->
name
(),
"ArgName"
,
arg_name
});
out
->
Print
(
"@abc.abstractmethod
\n
"
);
out
->
Print
(
"@abc.abstractmethod
\n
"
);
...
@@ -198,29 +196,29 @@ bool PrintStub(const ServiceDescriptor* service,
...
@@ -198,29 +196,29 @@ bool PrintStub(const ServiceDescriptor* service,
// TODO(protobuf team): Export `ModuleName` from protobuf's
// TODO(protobuf team): Export `ModuleName` from protobuf's
// `src/google/protobuf/compiler/python/python_generator.cc` file.
// `src/google/protobuf/compiler/python/python_generator.cc` file.
string
ModuleName
(
const
string
&
filename
)
{
std
::
string
ModuleName
(
const
std
::
string
&
filename
)
{
string
basename
=
StripProto
(
filename
);
std
::
string
basename
=
StripProto
(
filename
);
basename
=
StringReplace
(
basename
,
"-"
,
"_"
);
basename
=
StringReplace
(
basename
,
"-"
,
"_"
);
basename
=
StringReplace
(
basename
,
"/"
,
"."
);
basename
=
StringReplace
(
basename
,
"/"
,
"."
);
return
basename
+
"_pb2"
;
return
basename
+
"_pb2"
;
}
}
bool
GetModuleAndMessagePath
(
const
Descriptor
*
type
,
bool
GetModuleAndMessagePath
(
const
Descriptor
*
type
,
pair
<
string
,
string
>*
out
)
{
pair
<
std
::
string
,
std
::
string
>*
out
)
{
const
Descriptor
*
path_elem_type
=
type
;
const
Descriptor
*
path_elem_type
=
type
;
vector
<
const
Descriptor
*>
message_path
;
vector
<
const
Descriptor
*>
message_path
;
do
{
do
{
message_path
.
push_back
(
path_elem_type
);
message_path
.
push_back
(
path_elem_type
);
path_elem_type
=
path_elem_type
->
containing_type
();
path_elem_type
=
path_elem_type
->
containing_type
();
}
while
(
path_elem_type
!=
nullptr
);
}
while
(
path_elem_type
!=
nullptr
);
string
file_name
=
type
->
file
()
->
name
();
std
::
string
file_name
=
type
->
file
()
->
name
();
static
const
int
proto_suffix_length
=
strlen
(
".proto"
);
static
const
int
proto_suffix_length
=
strlen
(
".proto"
);
if
(
!
(
file_name
.
size
()
>
static_cast
<
size_t
>
(
proto_suffix_length
)
&&
if
(
!
(
file_name
.
size
()
>
static_cast
<
size_t
>
(
proto_suffix_length
)
&&
file_name
.
find_last_of
(
".proto"
)
==
file_name
.
size
()
-
1
))
{
file_name
.
find_last_of
(
".proto"
)
==
file_name
.
size
()
-
1
))
{
return
false
;
return
false
;
}
}
string
module
=
ModuleName
(
file_name
);
std
::
string
module
=
ModuleName
(
file_name
);
string
message_type
;
std
::
string
message_type
;
for
(
auto
path_iter
=
message_path
.
rbegin
();
for
(
auto
path_iter
=
message_path
.
rbegin
();
path_iter
!=
message_path
.
rend
();
++
path_iter
)
{
path_iter
!=
message_path
.
rend
();
++
path_iter
)
{
message_type
+=
(
*
path_iter
)
->
name
()
+
"."
;
message_type
+=
(
*
path_iter
)
->
name
()
+
"."
;
...
@@ -237,21 +235,21 @@ bool PrintServerFactory(const ServiceDescriptor* service, Printer* out) {
...
@@ -237,21 +235,21 @@ bool PrintServerFactory(const ServiceDescriptor* service, Printer* out) {
"Service"
,
service
->
name
());
"Service"
,
service
->
name
());
{
{
IndentScope
raii_create_server_indent
(
out
);
IndentScope
raii_create_server_indent
(
out
);
map
<
string
,
string
>
method_description_constructors
;
map
<
std
::
string
,
std
::
string
>
method_description_constructors
;
map
<
string
,
pair
<
string
,
string
>>
input_message_modules_and_classes
;
map
<
std
::
string
,
pair
<
std
::
string
,
std
::
string
>>
input_message_modules_and_classes
;
map
<
string
,
pair
<
string
,
string
>>
output_message_modules_and_classes
;
map
<
std
::
string
,
pair
<
std
::
string
,
std
::
string
>>
output_message_modules_and_classes
;
for
(
int
i
=
0
;
i
<
service
->
method_count
();
++
i
)
{
for
(
int
i
=
0
;
i
<
service
->
method_count
();
++
i
)
{
const
MethodDescriptor
*
method
=
service
->
method
(
i
);
const
MethodDescriptor
*
method
=
service
->
method
(
i
);
const
string
method_description_constructor
=
const
std
::
string
method_description_constructor
=
string
(
method
->
client_streaming
()
?
"stream_"
:
"unary_"
)
+
std
::
string
(
method
->
client_streaming
()
?
"stream_"
:
"unary_"
)
+
string
(
method
->
server_streaming
()
?
"stream_"
:
"unary_"
)
+
std
::
string
(
method
->
server_streaming
()
?
"stream_"
:
"unary_"
)
+
"service_description"
;
"service_description"
;
pair
<
string
,
string
>
input_message_module_and_class
;
pair
<
std
::
string
,
std
::
string
>
input_message_module_and_class
;
if
(
!
GetModuleAndMessagePath
(
method
->
input_type
(),
if
(
!
GetModuleAndMessagePath
(
method
->
input_type
(),
&
input_message_module_and_class
))
{
&
input_message_module_and_class
))
{
return
false
;
return
false
;
}
}
pair
<
string
,
string
>
output_message_module_and_class
;
pair
<
std
::
string
,
std
::
string
>
output_message_module_and_class
;
if
(
!
GetModuleAndMessagePath
(
method
->
output_type
(),
if
(
!
GetModuleAndMessagePath
(
method
->
output_type
(),
&
output_message_module_and_class
))
{
&
output_message_module_and_class
))
{
return
false
;
return
false
;
...
@@ -272,7 +270,7 @@ bool PrintServerFactory(const ServiceDescriptor* service, Printer* out) {
...
@@ -272,7 +270,7 @@ bool PrintServerFactory(const ServiceDescriptor* service, Printer* out) {
for
(
auto
&
name_and_description_constructor
:
for
(
auto
&
name_and_description_constructor
:
method_description_constructors
)
{
method_description_constructors
)
{
IndentScope
raii_descriptions_indent
(
out
);
IndentScope
raii_descriptions_indent
(
out
);
const
string
method_name
=
name_and_description_constructor
.
first
;
const
std
::
string
method_name
=
name_and_description_constructor
.
first
;
auto
input_message_module_and_class
=
auto
input_message_module_and_class
=
input_message_modules_and_classes
.
find
(
method_name
);
input_message_modules_and_classes
.
find
(
method_name
);
auto
output_message_module_and_class
=
auto
output_message_module_and_class
=
...
@@ -306,27 +304,27 @@ bool PrintServerFactory(const ServiceDescriptor* service, Printer* out) {
...
@@ -306,27 +304,27 @@ bool PrintServerFactory(const ServiceDescriptor* service, Printer* out) {
}
}
bool
PrintStubFactory
(
const
ServiceDescriptor
*
service
,
Printer
*
out
)
{
bool
PrintStubFactory
(
const
ServiceDescriptor
*
service
,
Printer
*
out
)
{
map
<
string
,
string
>
dict
=
ListToDict
({
map
<
std
::
string
,
std
::
string
>
dict
=
ListToDict
({
"Service"
,
service
->
name
(),
"Service"
,
service
->
name
(),
});
});
out
->
Print
(
dict
,
"def early_adopter_create_$Service$_stub(host, port):
\n
"
);
out
->
Print
(
dict
,
"def early_adopter_create_$Service$_stub(host, port):
\n
"
);
{
{
IndentScope
raii_create_server_indent
(
out
);
IndentScope
raii_create_server_indent
(
out
);
map
<
string
,
string
>
method_description_constructors
;
map
<
std
::
string
,
std
::
string
>
method_description_constructors
;
map
<
string
,
pair
<
string
,
string
>>
input_message_modules_and_classes
;
map
<
std
::
string
,
pair
<
std
::
string
,
std
::
string
>>
input_message_modules_and_classes
;
map
<
string
,
pair
<
string
,
string
>>
output_message_modules_and_classes
;
map
<
std
::
string
,
pair
<
std
::
string
,
std
::
string
>>
output_message_modules_and_classes
;
for
(
int
i
=
0
;
i
<
service
->
method_count
();
++
i
)
{
for
(
int
i
=
0
;
i
<
service
->
method_count
();
++
i
)
{
const
MethodDescriptor
*
method
=
service
->
method
(
i
);
const
MethodDescriptor
*
method
=
service
->
method
(
i
);
const
string
method_description_constructor
=
const
std
::
string
method_description_constructor
=
string
(
method
->
client_streaming
()
?
"stream_"
:
"unary_"
)
+
std
::
string
(
method
->
client_streaming
()
?
"stream_"
:
"unary_"
)
+
string
(
method
->
server_streaming
()
?
"stream_"
:
"unary_"
)
+
std
::
string
(
method
->
server_streaming
()
?
"stream_"
:
"unary_"
)
+
"invocation_description"
;
"invocation_description"
;
pair
<
string
,
string
>
input_message_module_and_class
;
pair
<
std
::
string
,
std
::
string
>
input_message_module_and_class
;
if
(
!
GetModuleAndMessagePath
(
method
->
input_type
(),
if
(
!
GetModuleAndMessagePath
(
method
->
input_type
(),
&
input_message_module_and_class
))
{
&
input_message_module_and_class
))
{
return
false
;
return
false
;
}
}
pair
<
string
,
string
>
output_message_module_and_class
;
pair
<
std
::
string
,
std
::
string
>
output_message_module_and_class
;
if
(
!
GetModuleAndMessagePath
(
method
->
output_type
(),
if
(
!
GetModuleAndMessagePath
(
method
->
output_type
(),
&
output_message_module_and_class
))
{
&
output_message_module_and_class
))
{
return
false
;
return
false
;
...
@@ -347,7 +345,7 @@ bool PrintStubFactory(const ServiceDescriptor* service, Printer* out) {
...
@@ -347,7 +345,7 @@ bool PrintStubFactory(const ServiceDescriptor* service, Printer* out) {
for
(
auto
&
name_and_description_constructor
:
for
(
auto
&
name_and_description_constructor
:
method_description_constructors
)
{
method_description_constructors
)
{
IndentScope
raii_descriptions_indent
(
out
);
IndentScope
raii_descriptions_indent
(
out
);
const
string
method_name
=
name_and_description_constructor
.
first
;
const
std
::
string
method_name
=
name_and_description_constructor
.
first
;
auto
input_message_module_and_class
=
auto
input_message_module_and_class
=
input_message_modules_and_classes
.
find
(
method_name
);
input_message_modules_and_classes
.
find
(
method_name
);
auto
output_message_module_and_class
=
auto
output_message_module_and_class
=
...
@@ -385,8 +383,8 @@ bool PrintPreamble(const FileDescriptor* file, Printer* out) {
...
@@ -385,8 +383,8 @@ bool PrintPreamble(const FileDescriptor* file, Printer* out) {
}
// namespace
}
// namespace
pair
<
bool
,
string
>
GetServices
(
const
FileDescriptor
*
file
)
{
pair
<
bool
,
std
::
string
>
GetServices
(
const
FileDescriptor
*
file
)
{
string
output
;
std
::
string
output
;
{
{
// Scope the output stream so it closes and finalizes output to the string.
// Scope the output stream so it closes and finalizes output to the string.
StringOutputStream
output_stream
(
&
output
);
StringOutputStream
output_stream
(
&
output
);
...
...
This diff is collapsed.
Click to expand it.
src/compiler/python_plugin.cc
+
4
−
6
View file @
0084179d
...
@@ -51,18 +51,16 @@ using google::protobuf::compiler::GeneratorContext;
...
@@ -51,18 +51,16 @@ using google::protobuf::compiler::GeneratorContext;
using
google
::
protobuf
::
compiler
::
PluginMain
;
using
google
::
protobuf
::
compiler
::
PluginMain
;
using
google
::
protobuf
::
io
::
CodedOutputStream
;
using
google
::
protobuf
::
io
::
CodedOutputStream
;
using
google
::
protobuf
::
io
::
ZeroCopyOutputStream
;
using
google
::
protobuf
::
io
::
ZeroCopyOutputStream
;
using
std
::
string
;
using
std
::
strlen
;
class
PythonGrpcGenerator
:
public
CodeGenerator
{
class
PythonGrpcGenerator
:
public
CodeGenerator
{
public:
public:
PythonGrpcGenerator
()
{}
PythonGrpcGenerator
()
{}
~
PythonGrpcGenerator
()
{}
~
PythonGrpcGenerator
()
{}
bool
Generate
(
const
FileDescriptor
*
file
,
const
string
&
parameter
,
bool
Generate
(
const
FileDescriptor
*
file
,
const
std
::
string
&
parameter
,
GeneratorContext
*
context
,
string
*
error
)
const
{
GeneratorContext
*
context
,
std
::
string
*
error
)
const
{
// Get output file name.
// Get output file name.
string
file_name
;
std
::
string
file_name
;
static
const
int
proto_suffix_length
=
strlen
(
".proto"
);
static
const
int
proto_suffix_length
=
strlen
(
".proto"
);
if
(
file
->
name
().
size
()
>
static_cast
<
size_t
>
(
proto_suffix_length
)
&&
if
(
file
->
name
().
size
()
>
static_cast
<
size_t
>
(
proto_suffix_length
)
&&
file
->
name
().
find_last_of
(
".proto"
)
==
file
->
name
().
size
()
-
1
)
{
file
->
name
().
find_last_of
(
".proto"
)
==
file
->
name
().
size
()
-
1
)
{
...
@@ -77,7 +75,7 @@ class PythonGrpcGenerator : public CodeGenerator {
...
@@ -77,7 +75,7 @@ class PythonGrpcGenerator : public CodeGenerator {
context
->
OpenForInsert
(
file_name
,
"module_scope"
));
context
->
OpenForInsert
(
file_name
,
"module_scope"
));
CodedOutputStream
coded_out
(
output
.
get
());
CodedOutputStream
coded_out
(
output
.
get
());
bool
success
=
false
;
bool
success
=
false
;
string
code
=
""
;
std
::
string
code
=
""
;
tie
(
success
,
code
)
=
grpc_python_generator
::
GetServices
(
file
);
tie
(
success
,
code
)
=
grpc_python_generator
::
GetServices
(
file
);
if
(
success
)
{
if
(
success
)
{
coded_out
.
WriteRaw
(
code
.
data
(),
code
.
size
());
coded_out
.
WriteRaw
(
code
.
data
(),
code
.
size
());
...
...
This diff is collapsed.
Click to expand it.
Preview
0%
Loading
Try again
or
attach a new file
.
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Save comment
Cancel
Please
register
or
sign in
to comment