From d63e854c6c83858bd33befde40124f7ecbc2af35 Mon Sep 17 00:00:00 2001 From: ddixit14 Date: Mon, 8 Jan 2024 11:55:40 -0500 Subject: [PATCH] chore: adding Value in fieldmethodname for enums --- ...ractTransportServiceStubClassComposer.java | 60 +++++++------------ 1 file changed, 23 insertions(+), 37 deletions(-) diff --git a/gapic-generator-java/src/main/java/com/google/api/generator/gapic/composer/common/AbstractTransportServiceStubClassComposer.java b/gapic-generator-java/src/main/java/com/google/api/generator/gapic/composer/common/AbstractTransportServiceStubClassComposer.java index 2d4bf377ca..eaefb4282d 100644 --- a/gapic-generator-java/src/main/java/com/google/api/generator/gapic/composer/common/AbstractTransportServiceStubClassComposer.java +++ b/gapic-generator-java/src/main/java/com/google/api/generator/gapic/composer/common/AbstractTransportServiceStubClassComposer.java @@ -1300,42 +1300,25 @@ private void createRequestParamsExtractorBodyForHttpBindings( for (HttpBindings.HttpBinding httpBindingFieldBinding : method.httpBindings().pathParameters()) { MethodInvocationExpr requestBuilderExpr = - createRequestFieldGetterExpr(requestVarExpr, httpBindingFieldBinding.name()); + createRequestFieldGetterExpr(requestVarExpr, httpBindingFieldBinding.name(),httpBindingFieldBinding.field().isEnum()); + Expr valueOfExpr = + MethodInvocationExpr.builder() + .setStaticReferenceType(TypeNode.STRING) + .setMethodName("valueOf") + .setArguments(requestBuilderExpr) + .build(); - if (httpBindingFieldBinding.isEnum()) { - Expr valueOfExpr = - MethodInvocationExpr.builder() - .setStaticReferenceType(TypeNode.INT_OBJECT) - .setMethodName("valueOf") - .setArguments(requestBuilderExpr) - .build(); - Expr paramsAddExpr = - MethodInvocationExpr.builder() - .setExprReferenceExpr(routingHeadersBuilderVarNonDeclExpr) - .setMethodName("add") - .setArguments( - ValueExpr.withValue(StringObjectValue.withValue(httpBindingFieldBinding.name())), - valueOfExpr) - .build(); - bodyStatements.add(ExprStatement.withExpr(paramsAddExpr)); - } else { - Expr valueOfExpr = - MethodInvocationExpr.builder() - .setStaticReferenceType(TypeNode.STRING) - .setMethodName("valueOf") - .setArguments(requestBuilderExpr) - .build(); - Expr paramsAddExpr = - MethodInvocationExpr.builder() - .setExprReferenceExpr(routingHeadersBuilderVarNonDeclExpr) - .setMethodName("add") - .setArguments( - ValueExpr.withValue(StringObjectValue.withValue(httpBindingFieldBinding.name())), - valueOfExpr) - .build(); - bodyStatements.add(ExprStatement.withExpr(paramsAddExpr)); - } + Expr paramsAddExpr = + MethodInvocationExpr.builder() + .setExprReferenceExpr(routingHeadersBuilderVarNonDeclExpr) + .setMethodName("add") + .setArguments( + ValueExpr.withValue(StringObjectValue.withValue(httpBindingFieldBinding.name())), + valueOfExpr) + .build(); + bodyStatements.add(ExprStatement.withExpr(paramsAddExpr)); } + returnExprBuilder .setExprReferenceExpr(routingHeadersBuilderVarNonDeclExpr) .setMethodName("build"); @@ -1378,7 +1361,7 @@ private void createRequestParamsExtractorBodyForRoutingHeaders( for (int i = 0; i < routingHeaderParams.size(); i++) { RoutingHeaderRule.RoutingHeaderParam routingHeaderParam = routingHeaderParams.get(i); MethodInvocationExpr requestFieldGetterExpr = - createRequestFieldGetterExpr(requestVarExpr, routingHeaderParam.fieldName()); + createRequestFieldGetterExpr(requestVarExpr, routingHeaderParam.fieldName(),false); Expr routingHeaderKeyExpr = ValueExpr.withValue(StringObjectValue.withValue(routingHeaderParam.key())); String pathTemplateName = @@ -1479,7 +1462,7 @@ private Expr fieldValuesNotNullConditionExpr( } private MethodInvocationExpr createRequestFieldGetterExpr( - VariableExpr requestVarExpr, String fieldName) { + VariableExpr requestVarExpr, String fieldName, Boolean isFieldEnum) { MethodInvocationExpr.Builder requestFieldGetterExprBuilder = MethodInvocationExpr.builder().setExprReferenceExpr(requestVarExpr); List descendantFields = Splitter.on(".").splitToList(fieldName); @@ -1488,7 +1471,10 @@ private MethodInvocationExpr createRequestFieldGetterExpr( for (int i = 0; i < descendantFields.size(); i++) { String currFieldName = descendantFields.get(i); String bindingFieldMethodName = - String.format("get%s", JavaStyle.toUpperCamelCase(currFieldName)); + String.format("get%s", JavaStyle.toUpperCamelCase(currFieldName)); + if (i==descendantFields.size()-1 && isFieldEnum) { + bindingFieldMethodName = bindingFieldMethodName+"Value"; + } requestFieldGetterExprBuilder = requestFieldGetterExprBuilder.setMethodName(bindingFieldMethodName); if (i < descendantFields.size() - 1) {