51
51
import com .google .api .generator .engine .ast .VariableExpr ;
52
52
import com .google .api .generator .gapic .composer .comment .StubCommentComposer ;
53
53
import com .google .api .generator .gapic .composer .store .TypeStore ;
54
- import com .google .api .generator .gapic .composer .utils .ClassNames ;
55
54
import com .google .api .generator .gapic .composer .utils .PackageChecker ;
56
55
import com .google .api .generator .gapic .model .GapicClass ;
57
56
import com .google .api .generator .gapic .model .GapicClass .Kind ;
@@ -204,15 +203,15 @@ public GapicClass generate(GapicContext context, Service service) {
204
203
.setName (className )
205
204
.setExtendsType (
206
205
typeStore .get (getTransportContext ().classNames ().getServiceStubClassName (service )))
207
- .setStatements (classStatements )
208
206
.setMethods (
209
207
createClassMethods (
210
208
context ,
211
209
service ,
212
210
typeStore ,
213
211
classMemberVarExprs ,
214
212
callableClassMemberVarExprs ,
215
- protoMethodNameToDescriptorVarExprs ))
213
+ protoMethodNameToDescriptorVarExprs , classStatements ))
214
+ .setStatements (classStatements )
216
215
.build ();
217
216
return GapicClass .create (kind , classDef );
218
217
}
@@ -249,7 +248,8 @@ protected List<MethodDefinition> createOperationsStubGetterMethod(
249
248
}
250
249
251
250
protected abstract Expr createTransportSettingsInitExpr (
252
- Method method , VariableExpr transportSettingsVarExpr , VariableExpr methodDescriptorVarExpr );
251
+ Method method , VariableExpr transportSettingsVarExpr , VariableExpr methodDescriptorVarExpr ,
252
+ List <Statement > classStatements );
253
253
254
254
protected List <MethodDefinition > createGetMethodDescriptorsMethod (
255
255
Service service ,
@@ -430,7 +430,8 @@ protected List<MethodDefinition> createClassMethods(
430
430
TypeStore typeStore ,
431
431
Map <String , VariableExpr > classMemberVarExprs ,
432
432
Map <String , VariableExpr > callableClassMemberVarExprs ,
433
- Map <String , VariableExpr > protoMethodNameToDescriptorVarExprs ) {
433
+ Map <String , VariableExpr > protoMethodNameToDescriptorVarExprs ,
434
+ List <Statement > classStatements ) {
434
435
List <MethodDefinition > javaMethods = new ArrayList <>();
435
436
javaMethods .addAll (createStaticCreatorMethods (service , typeStore , "newBuilder" ));
436
437
javaMethods .addAll (
@@ -440,7 +441,8 @@ protected List<MethodDefinition> createClassMethods(
440
441
typeStore ,
441
442
classMemberVarExprs ,
442
443
callableClassMemberVarExprs ,
443
- protoMethodNameToDescriptorVarExprs ));
444
+ protoMethodNameToDescriptorVarExprs ,
445
+ classStatements ));
444
446
javaMethods .addAll (
445
447
createGetMethodDescriptorsMethod (service , typeStore , protoMethodNameToDescriptorVarExprs ));
446
448
javaMethods .addAll (
@@ -541,7 +543,8 @@ protected List<MethodDefinition> createConstructorMethods(
541
543
TypeStore typeStore ,
542
544
Map <String , VariableExpr > classMemberVarExprs ,
543
545
Map <String , VariableExpr > callableClassMemberVarExprs ,
544
- Map <String , VariableExpr > protoMethodNameToDescriptorVarExprs ) {
546
+ Map <String , VariableExpr > protoMethodNameToDescriptorVarExprs ,
547
+ List <Statement > classStatements ) {
545
548
TypeNode stubSettingsType =
546
549
typeStore .get (getTransportContext ().classNames ().getServiceStubSettingsClassName (service ));
547
550
VariableExpr settingsVarExpr =
@@ -666,7 +669,7 @@ protected List<MethodDefinition> createConstructorMethods(
666
669
m ,
667
670
javaStyleMethodNameToTransportSettingsVarExprs .get (
668
671
JavaStyle .toLowerCamelCase (m .name ())),
669
- protoMethodNameToDescriptorVarExprs .get (m .name ())))
672
+ protoMethodNameToDescriptorVarExprs .get (m .name ()), classStatements ))
670
673
.collect (Collectors .toList ()));
671
674
secondCtorStatements .addAll (
672
675
secondCtorExprs .stream ().map (ExprStatement ::withExpr ).collect (Collectors .toList ()));
0 commit comments