Java CredentialOptions类代码示例

import com.google.cloud.bigtable.config.CredentialOptions; //导入依赖的package包/类
/**
 * Helper function that either returns the mock Bigtable service supplied by
 * {@link #withBigtableService} or creates and returns an implementation that talks to
 * {@code Cloud Bigtable}.
 *
 * <p>Also populate the credentials option from {@link GcpOptions#getGcpCredential()} if the
 * default credentials are being used on {@link BigtableOptions}.
 */
@VisibleForTesting
BigtableService getBigtableService(PipelineOptions pipelineOptions) {
  if (getBigtableService() != null) {
    return getBigtableService();
  }

  BigtableOptions.Builder bigtableOptions = effectiveUserProvidedBigtableOptions();

  bigtableOptions.setUserAgent(pipelineOptions.getUserAgent());

  if (bigtableOptions.build().getCredentialOptions().getCredentialType()
    == CredentialOptions.CredentialType.DefaultCredentials) {
    bigtableOptions.setCredentialOptions(
      CredentialOptions.credential(
        pipelineOptions.as(GcpOptions.class).getGcpCredential()));
  }

  // Default option that should be forced
  bigtableOptions.setUseCachedDataPool(true);

  return new BigtableServiceImpl(bigtableOptions.build());
}
 

import com.google.cloud.bigtable.config.CredentialOptions; //导入依赖的package包/类
@Before
public void setup() throws Exception {
  PipelineOptionsFactory.register(BigtableTestOptions.class);
  options = TestPipeline.testingPipelineOptions().as(BigtableTestOptions.class);
  project = options.as(GcpOptions.class).getProject();

  bigtableOptions =
      new Builder()
          .setProjectId(project)
          .setInstanceId(options.getInstanceId())
          .setUserAgent("apache-beam-test")
          .build();

  session =
      new BigtableSession(
          bigtableOptions
              .toBuilder()
              .setCredentialOptions(
                  CredentialOptions.credential(options.as(GcpOptions.class).getGcpCredential()))
              .build());
  tableAdminClient = session.getTableAdminClient();
}
 

import com.google.cloud.bigtable.config.CredentialOptions; //导入依赖的package包/类
/** Tests that credentials are used from PipelineOptions if not supplied by BigtableOptions. */
@Test
public void testUsePipelineOptionsCredentialsIfNotSpecifiedInBigtableOptions() throws Exception {
  BigtableOptions options = BIGTABLE_OPTIONS.toBuilder()
      .setCredentialOptions(CredentialOptions.defaultCredentials())
      .build();
  GcpOptions pipelineOptions = PipelineOptionsFactory.as(GcpOptions.class);
  pipelineOptions.setGcpCredential(new TestCredential());
  BigtableService readService = BigtableIO.read()
      .withBigtableOptions(options)
      .withTableId("TEST-TABLE")
      .getBigtableConfig()
      .getBigtableService(pipelineOptions);
  BigtableService writeService = BigtableIO.write()
      .withBigtableOptions(options)
      .withTableId("TEST-TABLE")
      .getBigtableConfig()
      .getBigtableService(pipelineOptions);
  assertEquals(CredentialType.SuppliedCredentials,
      readService.getBigtableOptions().getCredentialOptions().getCredentialType());
  assertEquals(CredentialType.SuppliedCredentials,
      writeService.getBigtableOptions().getCredentialOptions().getCredentialType());
}
 

import com.google.cloud.bigtable.config.CredentialOptions; //导入依赖的package包/类
/** Tests that credentials are not used from PipelineOptions if supplied by BigtableOptions. */
@Test
public void testDontUsePipelineOptionsCredentialsIfSpecifiedInBigtableOptions() throws Exception {
  BigtableOptions options = BIGTABLE_OPTIONS.toBuilder()
      .setCredentialOptions(CredentialOptions.nullCredential())
      .build();
  GcpOptions pipelineOptions = PipelineOptionsFactory.as(GcpOptions.class);
  pipelineOptions.setGcpCredential(new TestCredential());
  BigtableService readService = BigtableIO.read()
      .withBigtableOptions(options)
      .withTableId("TEST-TABLE")
      .getBigtableConfig()
      .getBigtableService(pipelineOptions);
  BigtableService writeService = BigtableIO.write()
      .withBigtableOptions(options)
      .withTableId("TEST-TABLE")
      .getBigtableConfig()
      .getBigtableService(pipelineOptions);
  assertEquals(CredentialType.None,
      readService.getBigtableOptions().getCredentialOptions().getCredentialType());
  assertEquals(CredentialType.None,
      writeService.getBigtableOptions().getCredentialOptions().getCredentialType());
}
 

import com.google.cloud.bigtable.config.CredentialOptions; //导入依赖的package包/类
@Override
public BigtableConnection call() throws Exception {
    final CredentialOptions credentials = this.credentials.build();

    final RetryOptions retryOptions = new RetryOptions.Builder()
        .addStatusToRetryOn(Status.Code.UNKNOWN)
        .setAllowRetriesWithoutTimestamp(true)
        .build();

    final BulkOptions bulkOptions = batchSize
        .map(integer -> new BulkOptions.Builder().setBulkMaxRowKeyCount(integer).build())
        .orElseGet(() -> new BulkOptions.Builder().build());

    final BigtableOptions options = new BigtableOptions.Builder()
        .setProjectId(project)
        .setInstanceId(instance)
        .setUserAgent(USER_AGENT)
        .setDataChannelCount(64)
        .setCredentialOptions(credentials)
        .setRetryOptions(retryOptions)
        .setBulkOptions(bulkOptions)
        .build();

    final BigtableSession session = new BigtableSession(options);

    final BigtableTableAdminClient adminClient =
        new BigtableTableTableAdminClientImpl(async, session.getTableAdminClient(), project,
            instance);

    final BigtableMutator mutator =
        new BigtableMutatorImpl(async, session, disableBulkMutations, flushIntervalSeconds);

    final BigtableDataClient client =
        new BigtableDataClientImpl(async, session, mutator, project, instance);

    return new GrpcBigtableConnection(async, project, instance, session, mutator, adminClient,
        client);
}
 

import com.google.cloud.bigtable.config.CredentialOptions; //导入依赖的package包/类
@Override
public CredentialOptions build() throws Exception {
    // XXX: You have to leave the input stream open for BigtableSession to use it.
    // This does 'leak' an input stream, but it's only once, so we'll live with it for now.
    // Reported here: https://github.com/GoogleCloudPlatform/cloud-bigtable-client/issues/534
    return CredentialOptions.jsonCredentials(Files.newInputStream(path));
}
 

import com.google.cloud.bigtable.config.CredentialOptions; //导入依赖的package包/类
@Test
public void shouldBuildDefaultOptions() throws Exception {
    final DefaultCredentialsBuilder credentialsBuilder = new DefaultCredentialsBuilder();
    final CredentialOptions credentialOptions = credentialsBuilder.build();
    final CredentialOptions.CredentialType type = credentialOptions.getCredentialType();
    assertThat(type, is(CredentialOptions.CredentialType.DefaultCredentials));
}
 

import com.google.cloud.bigtable.config.CredentialOptions; //导入依赖的package包/类
@Override
public CredentialOptions build() {
    return CredentialOptions.credential(new ComputeEngineCredentials());
}
 

import com.google.cloud.bigtable.config.CredentialOptions; //导入依赖的package包/类
@Override
public CredentialOptions build() {
    return CredentialOptions.p12Credential(serviceAccount, keyFile);
}
 

import com.google.cloud.bigtable.config.CredentialOptions; //导入依赖的package包/类
@Override
public CredentialOptions build() {
  return CredentialOptions.defaultCredentials();
}
 

import com.google.cloud.bigtable.config.CredentialOptions; //导入依赖的package包/类
public CredentialOptions build() throws Exception;