Skip to content

Commit

Permalink
samples: add long timeout sample (#27)
Browse files Browse the repository at this point in the history
* docs: add long timeout sample

* fix lint

* rm unused variable

* fix package

* 🦉 Updates from OwlBot post-processor

See https://github.com/googleapis/repo-automation-bots/blob/main/packages/owl-bot/README.md

Co-authored-by: Owl Bot <gcf-owl-bot[bot]@users.noreply.github.com>
  • Loading branch information
changsongd and gcf-owl-bot[bot] authored Apr 11, 2022
1 parent c75d5f0 commit 82829d4
Show file tree
Hide file tree
Showing 2 changed files with 123 additions and 0 deletions.
Original file line number Diff line number Diff line change
@@ -0,0 +1,66 @@
/*
* Copyright 2022 Google LLC
*
* 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.
*/

package com.example.optimizationai;

// [START cloudoptimization_long_timeout]

import com.google.cloud.optimization.v1.FleetRoutingClient;
import com.google.cloud.optimization.v1.FleetRoutingSettings;
import com.google.cloud.optimization.v1.OptimizeToursRequest;
import com.google.cloud.optimization.v1.OptimizeToursResponse;
import com.google.protobuf.Duration;
import com.google.protobuf.TextFormat;
import java.io.FileInputStream;
import java.io.InputStream;
import java.io.InputStreamReader;
import java.io.Reader;

/**
* This is an example to send a request to Cloud Fleet Routing synchronous API via Java API Client.
*/
public class SyncApiWithLongTimeout {
public static void longTimeout() throws Exception {
// TODO(developer): Replace these variables before running the sample.
String projectParent = "projects/{YOUR_GCP_PROJECT_ID}";
String modelPath = "YOUR_MODEL_PATH";
longTimeout(projectParent, modelPath);
}

public static void longTimeout(String projectParent, String modelPath) throws Exception {
int timeoutSeconds = 100;
InputStream modelInputstream = new FileInputStream(modelPath);
Reader modelInputStreamReader = new InputStreamReader(modelInputstream);
OptimizeToursRequest.Builder requestBuilder =
OptimizeToursRequest.newBuilder()
.setTimeout(Duration.newBuilder().setSeconds(timeoutSeconds).build())
.setParent(projectParent);
TextFormat.getParser().merge(modelInputStreamReader, requestBuilder);

// Checks the gRPC connection every 5 mins and keeps it alive.
FleetRoutingClient fleetRoutingClientClient =
FleetRoutingClient.create(
FleetRoutingSettings.newBuilder()
.setTransportChannelProvider(
FleetRoutingSettings.defaultGrpcTransportProviderBuilder()
.setKeepAliveTime(org.threeten.bp.Duration.ofSeconds(300))
.build())
.build());
OptimizeToursResponse response = fleetRoutingClientClient.optimizeTours(requestBuilder.build());
System.out.println(response.toString());
}
}
// [END cloudoptimization_long_timeout]
Original file line number Diff line number Diff line change
@@ -0,0 +1,57 @@
/*
* Copyright 2022 Google LLC
*
* 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.
*/

package com.example.optimizationai;

import static com.google.common.truth.Truth.assertThat;

import java.io.ByteArrayOutputStream;
import java.io.PrintStream;
import org.junit.After;
import org.junit.Before;
import org.junit.Test;

/** Tests for SyncApiWithLongTimeout sample. */
public class SyncApiWithLongTimeoutTest {
private static final String PROJECT_ID = System.getenv("GOOGLE_CLOUD_PROJECT");
private static final String PROJECT_PARENT = String.format("projects/%s", PROJECT_ID);
private static final String MODEL_PATH = "resources/sync_request.textproto";

private ByteArrayOutputStream bout;
private PrintStream out;
private PrintStream originalPrintStream;

@Before
public void setUp() {
bout = new ByteArrayOutputStream();
out = new PrintStream(bout);
originalPrintStream = System.out;
System.setOut(out);
}

@After
public void tearDown() {
System.out.flush();
System.setOut(originalPrintStream);
}

@Test
public void testSyncApi() throws Exception {
SyncApiWithLongTimeout.longTimeout(PROJECT_PARENT, MODEL_PATH);
String got = bout.toString();
assertThat(got).contains("routes");
}
}

0 comments on commit 82829d4

Please sign in to comment.