Examples with Expr com.google.api.generator.engine.ast.Expr used on opensource projects

Example 16 with Expr

use of com.google.api.generator.engine.ast.Expr in project gapic-generator-java by googleapis.

the class AbstractServiceClientClassComposer method createBackgroundResourceMethods.

private static List<MethodDefinition> createBackgroundResourceMethods(Service service, TypeStore typeStore) {
    List<MethodDefinition> methods = new ArrayList<>();
    VariableExpr stubVarExpr = VariableExpr.withVariable(Variable.builder().setType(typeStore.get(ClassNames.getServiceStubClassName(service))).setName("stub").build());
    MethodDefinition closeMethod = MethodDefinition.builder().setIsOverride(true).setScope(ScopeNode.PUBLIC).setIsFinal(true).setReturnType(TypeNode.VOID).setName("close").setBody(Arrays.asList(ExprStatement.withExpr(MethodInvocationExpr.builder().setExprReferenceExpr(stubVarExpr).setMethodName("close").build()))).build();
    methods.add(closeMethod);
    MethodDefinition shutdownMethod = MethodDefinition.builder().setIsOverride(true).setScope(ScopeNode.PUBLIC).setReturnType(TypeNode.VOID).setName("shutdown").setBody(Arrays.asList(ExprStatement.withExpr(MethodInvocationExpr.builder().setExprReferenceExpr(stubVarExpr).setMethodName("shutdown").build()))).build();
    methods.add(shutdownMethod);
    MethodDefinition isShutdownMethod = MethodDefinition.builder().setIsOverride(true).setScope(ScopeNode.PUBLIC).setReturnType(TypeNode.BOOLEAN).setName("isShutdown").setReturnExpr(MethodInvocationExpr.builder().setExprReferenceExpr(stubVarExpr).setMethodName("isShutdown").setReturnType(TypeNode.BOOLEAN).build()).build();
    methods.add(isShutdownMethod);
    MethodDefinition isTerminatedMethod = MethodDefinition.builder().setIsOverride(true).setScope(ScopeNode.PUBLIC).setReturnType(TypeNode.BOOLEAN).setName("isTerminated").setReturnExpr(MethodInvocationExpr.builder().setExprReferenceExpr(stubVarExpr).setMethodName("isTerminated").setReturnType(TypeNode.BOOLEAN).build()).build();
    methods.add(isTerminatedMethod);
    MethodDefinition shutdownNowMethod = MethodDefinition.builder().setIsOverride(true).setScope(ScopeNode.PUBLIC).setReturnType(TypeNode.VOID).setName("shutdownNow").setBody(Arrays.asList(ExprStatement.withExpr(MethodInvocationExpr.builder().setExprReferenceExpr(stubVarExpr).setMethodName("shutdownNow").build()))).build();
    methods.add(shutdownNowMethod);
    List<VariableExpr> arguments = Arrays.asList(VariableExpr.builder().setVariable(Variable.builder().setName("duration").setType(TypeNode.LONG).build()).build(), VariableExpr.builder().setVariable(Variable.builder().setName("unit").setType(typeStore.get("TimeUnit")).build()).build());
    MethodDefinition awaitTerminationMethod = MethodDefinition.builder().setIsOverride(true).setScope(ScopeNode.PUBLIC).setReturnType(TypeNode.BOOLEAN).setName("awaitTermination").setArguments(arguments.stream().map(v -> v.toBuilder().setIsDecl(true).build()).collect(Collectors.toList())).setThrowsExceptions(Arrays.asList(typeStore.get("InterruptedException"))).setReturnExpr(MethodInvocationExpr.builder().setExprReferenceExpr(stubVarExpr).setMethodName("awaitTermination").setArguments(arguments.stream().map(v -> (Expr) v).collect(Collectors.toList())).setReturnType(TypeNode.BOOLEAN).build()).build();
    methods.add(awaitTerminationMethod);
    return methods;
}

Example 17 with Expr

use of com.google.api.generator.engine.ast.Expr in project gapic-generator-java by googleapis.

the class RetrySettingsComposer method createRetryCodesDefinitionsBlock.

public static BlockStatement createRetryCodesDefinitionsBlock(Service service, GapicServiceConfig serviceConfig, VariableExpr retryCodesDefinitionsClassMemberVarExpr) {
    TypeNode definitionsType = TypeNode.withReference(ConcreteReference.builder().setClazz(ImmutableMap.Builder.class).setGenerics(retryCodesDefinitionsClassMemberVarExpr.type().reference().generics()).build());
    VariableExpr definitionsVarExpr = VariableExpr.withVariable(Variable.builder().setType(definitionsType).setName("definitions").build());
    List<Expr> bodyExprs = new ArrayList<>();
    // Create the first expr.
    bodyExprs.add(AssignmentExpr.builder().setVariableExpr(definitionsVarExpr.toBuilder().setIsDecl(true).build()).setValueExpr(MethodInvocationExpr.builder().setStaticReferenceType(FIXED_TYPESTORE.get("ImmutableMap")).setMethodName("builder").setReturnType(definitionsVarExpr.type()).build()).build());
    for (Map.Entry<String, List<Code>> codeEntry : serviceConfig.getAllRetryCodes(service).entrySet()) {
        bodyExprs.add(createRetryCodeDefinitionExpr(codeEntry.getKey(), codeEntry.getValue(), definitionsVarExpr));
    }
    // Reassign the new codes.
    bodyExprs.add(AssignmentExpr.builder().setVariableExpr(retryCodesDefinitionsClassMemberVarExpr).setValueExpr(MethodInvocationExpr.builder().setExprReferenceExpr(definitionsVarExpr).setMethodName("build").setReturnType(retryCodesDefinitionsClassMemberVarExpr.type()).build()).build());
    // Put everything together.
    return BlockStatement.builder().setIsStatic(true).setBody(bodyExprs.stream().map(e -> ExprStatement.withExpr(e)).collect(Collectors.toList())).build();
}

Example 18 with Expr

use of com.google.api.generator.engine.ast.Expr in project gapic-generator-java by googleapis.

the class RetrySettingsComposer method createLroRetrySettingsExpr.

private static Expr createLroRetrySettingsExpr(Service service, Method method, GapicServiceConfig serviceConfig) {
    Expr lroRetrySettingsExpr = MethodInvocationExpr.builder().setStaticReferenceType(FIXED_TYPESTORE.get("RetrySettings")).setMethodName("newBuilder").build();
    long initialPollDelayMillis = LRO_DEFAULT_INITIAL_POLL_DELAY_MILLIS;
    double pollDelayMultiplier = LRO_DEFAULT_POLL_DELAY_MULTIPLIER;
    long maxPollDelayMillis = LRO_DEFAULT_MAX_POLL_DELAY_MILLIS;
    long totalPollTimeoutMillis = LRO_DEFAULT_TOTAL_POLL_TIMEOUT_MILLIS;
    if (serviceConfig.hasLroRetrySetting(service, method)) {
        Optional<GapicLroRetrySettings> lroRetrySettingsOpt = serviceConfig.getLroRetrySetting(service, method);
        if (lroRetrySettingsOpt.isPresent()) {
            GapicLroRetrySettings lroRetrySettings = lroRetrySettingsOpt.get();
            initialPollDelayMillis = lroRetrySettings.initialPollDelayMillis();
            pollDelayMultiplier = lroRetrySettings.pollDelayMultiplier();
            maxPollDelayMillis = lroRetrySettings.maxPollDelayMillis();
            totalPollTimeoutMillis = lroRetrySettings.totalPollTimeoutMillis();
        }
    }
    lroRetrySettingsExpr = MethodInvocationExpr.builder().setExprReferenceExpr(lroRetrySettingsExpr).setMethodName("setInitialRetryDelay").setArguments(createDurationOfMillisExpr(toValExpr(initialPollDelayMillis))).build();
    lroRetrySettingsExpr = MethodInvocationExpr.builder().setExprReferenceExpr(lroRetrySettingsExpr).setMethodName("setRetryDelayMultiplier").setArguments(toValExpr(pollDelayMultiplier)).build();
    lroRetrySettingsExpr = MethodInvocationExpr.builder().setExprReferenceExpr(lroRetrySettingsExpr).setMethodName("setMaxRetryDelay").setArguments(createDurationOfMillisExpr(toValExpr(maxPollDelayMillis))).build();
    Expr zeroDurationExpr = EnumRefExpr.builder().setType(FIXED_TYPESTORE.get("Duration")).setName("ZERO").build();
    // TODO(miraleung): Add an "// ignored" comment here.
    lroRetrySettingsExpr = MethodInvocationExpr.builder().setExprReferenceExpr(lroRetrySettingsExpr).setMethodName("setInitialRpcTimeout").setArguments(zeroDurationExpr).build();
    // TODO(miraleung): Add an "// ignored" comment here.
    lroRetrySettingsExpr = MethodInvocationExpr.builder().setExprReferenceExpr(lroRetrySettingsExpr).setMethodName("setRpcTimeoutMultiplier").setArguments(toValExpr(LRO_DEFAULT_MAX_RPC_TIMEOUT)).build();
    // TODO(miraleung): Add an "// ignored" comment here.
    lroRetrySettingsExpr = MethodInvocationExpr.builder().setExprReferenceExpr(lroRetrySettingsExpr).setMethodName("setMaxRpcTimeout").setArguments(zeroDurationExpr).build();
    lroRetrySettingsExpr = MethodInvocationExpr.builder().setExprReferenceExpr(lroRetrySettingsExpr).setMethodName("setTotalTimeout").setArguments(createDurationOfMillisExpr(toValExpr(totalPollTimeoutMillis))).build();
    lroRetrySettingsExpr = MethodInvocationExpr.builder().setExprReferenceExpr(lroRetrySettingsExpr).setMethodName("build").build();
    return lroRetrySettingsExpr;
}

Example 19 with Expr

use of com.google.api.generator.engine.ast.Expr in project gapic-generator-java by googleapis.

the class RetrySettingsComposer method createRetryParamDefinitionsBlock.

public static BlockStatement createRetryParamDefinitionsBlock(Service service, GapicServiceConfig serviceConfig, VariableExpr retryParamDefinitionsClassMemberVarExpr) {
    List<Expr> bodyExprs = new ArrayList<>();
    TypeNode definitionsType = TypeNode.withReference(ConcreteReference.builder().setClazz(ImmutableMap.Builder.class).setGenerics(retryParamDefinitionsClassMemberVarExpr.type().reference().generics()).build());
    VariableExpr definitionsVarExpr = VariableExpr.withVariable(Variable.builder().setType(definitionsType).setName("definitions").build());
    VariableExpr settingsVarExpr = VariableExpr.withVariable(Variable.builder().setType(FIXED_TYPESTORE.get("RetrySettings")).setName("settings").build());
    // Create the first two exprs.
    bodyExprs.add(AssignmentExpr.builder().setVariableExpr(definitionsVarExpr.toBuilder().setIsDecl(true).build()).setValueExpr(MethodInvocationExpr.builder().setStaticReferenceType(FIXED_TYPESTORE.get("ImmutableMap")).setMethodName("builder").setReturnType(definitionsVarExpr.type()).build()).build());
    bodyExprs.add(AssignmentExpr.builder().setVariableExpr(settingsVarExpr.toBuilder().setIsDecl(true).build()).setValueExpr(ValueExpr.createNullExpr()).build());
    // Build the settings object for each config.
    for (Map.Entry<String, GapicRetrySettings> settingsEntry : serviceConfig.getAllGapicRetrySettings(service).entrySet()) {
        bodyExprs.addAll(createRetrySettingsExprs(settingsEntry.getKey(), settingsEntry.getValue(), settingsVarExpr, definitionsVarExpr));
    }
    // Reassign the new settings.
    bodyExprs.add(AssignmentExpr.builder().setVariableExpr(retryParamDefinitionsClassMemberVarExpr).setValueExpr(MethodInvocationExpr.builder().setExprReferenceExpr(definitionsVarExpr).setMethodName("build").setReturnType(retryParamDefinitionsClassMemberVarExpr.type()).build()).build());
    // Put everything together.
    return BlockStatement.builder().setIsStatic(true).setBody(bodyExprs.stream().map(e -> ExprStatement.withExpr(e)).collect(Collectors.toList())).build();
}

Example 20 with Expr

use of com.google.api.generator.engine.ast.Expr in project gapic-generator-java by googleapis.

the class RetrySettingsComposer method createBatchingBuilderSettingsExpr.

public static Expr createBatchingBuilderSettingsExpr(String settingsGetterMethodName, GapicBatchingSettings batchingSettings, VariableExpr builderVarExpr) {
    Expr batchingSettingsBuilderExpr = MethodInvocationExpr.builder().setStaticReferenceType(FIXED_TYPESTORE.get("BatchingSettings")).setMethodName("newBuilder").build();
    batchingSettingsBuilderExpr = MethodInvocationExpr.builder().setExprReferenceExpr(batchingSettingsBuilderExpr).setMethodName("setElementCountThreshold").setArguments(toValExpr(batchingSettings.elementCountThreshold())).build();
    batchingSettingsBuilderExpr = MethodInvocationExpr.builder().setExprReferenceExpr(batchingSettingsBuilderExpr).setMethodName("setRequestByteThreshold").setArguments(toValExpr(batchingSettings.requestByteThreshold())).build();
    batchingSettingsBuilderExpr = MethodInvocationExpr.builder().setExprReferenceExpr(batchingSettingsBuilderExpr).setMethodName("setDelayThreshold").setArguments(createDurationOfMillisExpr(toValExpr(batchingSettings.delayThresholdMillis()))).build();
    // FlowControlSettings.
    Expr flowControlSettingsExpr = MethodInvocationExpr.builder().setStaticReferenceType(FIXED_TYPESTORE.get("FlowControlSettings")).setMethodName("newBuilder").build();
    if (batchingSettings.flowControlElementLimit() != null) {
        flowControlSettingsExpr = MethodInvocationExpr.builder().setExprReferenceExpr(flowControlSettingsExpr).setMethodName("setMaxOutstandingElementCount").setArguments(toValExpr(batchingSettings.flowControlElementLimit())).build();
    }
    if (batchingSettings.flowControlByteLimit() != null) {
        flowControlSettingsExpr = MethodInvocationExpr.builder().setExprReferenceExpr(flowControlSettingsExpr).setMethodName("setMaxOutstandingRequestBytes").setArguments(toValExpr(batchingSettings.flowControlByteLimit())).build();
    }
    flowControlSettingsExpr = MethodInvocationExpr.builder().setExprReferenceExpr(flowControlSettingsExpr).setMethodName("setLimitExceededBehavior").setArguments(EnumRefExpr.builder().setType(FIXED_TYPESTORE.get("LimitExceededBehavior")).setName(JavaStyle.toUpperCamelCase(batchingSettings.flowControlLimitExceededBehavior().name().toLowerCase())).build()).build();
    flowControlSettingsExpr = MethodInvocationExpr.builder().setExprReferenceExpr(flowControlSettingsExpr).setMethodName("build").build();
    batchingSettingsBuilderExpr = MethodInvocationExpr.builder().setExprReferenceExpr(batchingSettingsBuilderExpr).setMethodName("setFlowControlSettings").setArguments(flowControlSettingsExpr).build();
    batchingSettingsBuilderExpr = MethodInvocationExpr.builder().setExprReferenceExpr(batchingSettingsBuilderExpr).setMethodName("build").build();
    // Put everything together.
    Expr builderExpr = MethodInvocationExpr.builder().setExprReferenceExpr(builderVarExpr).setMethodName(settingsGetterMethodName).build();
    return MethodInvocationExpr.builder().setExprReferenceExpr(builderExpr).setMethodName("setBatchingSettings").setArguments(batchingSettingsBuilderExpr).build();
}