Java Bucket类代码示例

import com.google.api.services.storage.model.Bucket; //导入依赖的package包/类
@Test
public void testCreateBucket() throws IOException {
  GcsOptions pipelineOptions = gcsOptionsWithTestCredential();
  GcsUtil gcsUtil = pipelineOptions.getGcsUtil();

  Storage.Buckets mockStorageObjects = Mockito.mock(Storage.Buckets.class);
  Storage mockStorage = Mockito.mock(Storage.class);
  gcsUtil.setStorageClient(mockStorage);

  Storage.Buckets.Insert mockStorageInsert = Mockito.mock(Storage.Buckets.Insert.class);

  BackOff mockBackOff = BackOffAdapter.toGcpBackOff(FluentBackoff.DEFAULT.backoff());

  when(mockStorage.buckets()).thenReturn(mockStorageObjects);
  when(mockStorageObjects.insert(
         any(String.class), any(Bucket.class))).thenReturn(mockStorageInsert);
  when(mockStorageInsert.execute())
      .thenThrow(new SocketTimeoutException("SocketException"))
      .thenReturn(new Bucket());

  gcsUtil.createBucket("a", new Bucket(), mockBackOff, new FastNanoClockAndSleeper());
}
 

import com.google.api.services.storage.model.Bucket; //导入依赖的package包/类
@Test
public void testCreateBucketAccessErrors() throws IOException {
  GcsOptions pipelineOptions = gcsOptionsWithTestCredential();
  GcsUtil gcsUtil = pipelineOptions.getGcsUtil();

  Storage mockStorage = Mockito.mock(Storage.class);
  gcsUtil.setStorageClient(mockStorage);

  Storage.Buckets mockStorageObjects = Mockito.mock(Storage.Buckets.class);
  Storage.Buckets.Insert mockStorageInsert = Mockito.mock(Storage.Buckets.Insert.class);

  BackOff mockBackOff = BackOffAdapter.toGcpBackOff(FluentBackoff.DEFAULT.backoff());
  GoogleJsonResponseException expectedException =
      googleJsonResponseException(HttpStatusCodes.STATUS_CODE_FORBIDDEN,
          "Waves hand mysteriously", "These aren't the buckets you're looking for");

  when(mockStorage.buckets()).thenReturn(mockStorageObjects);
  when(mockStorageObjects.insert(
         any(String.class), any(Bucket.class))).thenReturn(mockStorageInsert);
  when(mockStorageInsert.execute())
      .thenThrow(expectedException);

  thrown.expect(AccessDeniedException.class);

  gcsUtil.createBucket("a", new Bucket(), mockBackOff, new FastNanoClockAndSleeper());
}
 

import com.google.api.services.storage.model.Bucket; //导入依赖的package包/类
@Test
public void testBucketAccessible() throws IOException {
  GcsOptions pipelineOptions = gcsOptionsWithTestCredential();
  GcsUtil gcsUtil = pipelineOptions.getGcsUtil();

  Storage mockStorage = Mockito.mock(Storage.class);
  gcsUtil.setStorageClient(mockStorage);

  Storage.Buckets mockStorageObjects = Mockito.mock(Storage.Buckets.class);
  Storage.Buckets.Get mockStorageGet = Mockito.mock(Storage.Buckets.Get.class);

  BackOff mockBackOff = BackOffAdapter.toGcpBackOff(
      FluentBackoff.DEFAULT.backoff());

  when(mockStorage.buckets()).thenReturn(mockStorageObjects);
  when(mockStorageObjects.get("testbucket")).thenReturn(mockStorageGet);
  when(mockStorageGet.execute())
      .thenThrow(new SocketTimeoutException("SocketException"))
      .thenReturn(new Bucket());

  assertTrue(gcsUtil.bucketAccessible(GcsPath.fromComponents("testbucket", "testobject"),
      mockBackOff, new FastNanoClockAndSleeper()));
}
 

import com.google.api.services.storage.model.Bucket; //导入依赖的package包/类
@Test
public void testGetBucket() throws IOException {
  GcsOptions pipelineOptions = gcsOptionsWithTestCredential();
  GcsUtil gcsUtil = pipelineOptions.getGcsUtil();

  Storage mockStorage = Mockito.mock(Storage.class);
  gcsUtil.setStorageClient(mockStorage);

  Storage.Buckets mockStorageObjects = Mockito.mock(Storage.Buckets.class);
  Storage.Buckets.Get mockStorageGet = Mockito.mock(Storage.Buckets.Get.class);

  BackOff mockBackOff = BackOffAdapter.toGcpBackOff(
      FluentBackoff.DEFAULT.backoff());

  when(mockStorage.buckets()).thenReturn(mockStorageObjects);
  when(mockStorageObjects.get("testbucket")).thenReturn(mockStorageGet);
  when(mockStorageGet.execute())
      .thenThrow(new SocketTimeoutException("SocketException"))
      .thenReturn(new Bucket());

  assertNotNull(gcsUtil.getBucket(GcsPath.fromComponents("testbucket", "testobject"),
      mockBackOff, new FastNanoClockAndSleeper()));
}
 

import com.google.api.services.storage.model.Bucket; //导入依赖的package包/类
private static Bucket createBucket(Storage storage, String projectId, String locationId, String bucketId) {
  try {
    Bucket bucket = new Bucket()
        .setLocation(locationId)
        .setName(bucketId)
        .setVersioning(new Bucket.Versioning().setEnabled(true));

    if (!StringUtils.isEmpty(locationId)) {
      bucket.setLocation(locationId);
    }

    return storage.buckets().insert(projectId, bucket).execute();
  } catch (IOException e) {
    throw new RuntimeException("Unable to create bucket", e);
  }
}
 

import com.google.api.services.storage.model.Bucket; //导入依赖的package包/类
/** Get all bucket names and create them. */
@SuppressWarnings("PMD.AvoidInstantiatingObjectsInLoops")
public static void initBuckets(final boolean create) {
    for (final String region : REGIONS) {
        final String bucketName = BUCKET_PREFIX + region.toLowerCase(Locale.ENGLISH);
        BUCKETS.put(region, bucketName);

        if (create) {
            try {
                LOGGER.debug("Creating bucket  '{}'", bucketName);
                client.buckets().insert(projectId,
                    new Bucket().setName(bucketName).setLocation(region)
                        .setStorageClass("DURABLE_REDUCED_AVAILABILITY")).execute();
            } catch (IOException e) {
                LOGGER.error("Create bucket exception", e);
            }
        }
    }
}
 

import com.google.api.services.storage.model.Bucket; //导入依赖的package包/类
@Test
public void testBadTTLWithUpdate() throws Exception {
  final Bucket bucket = new Bucket()
      .setName(BUCKET_NAME)
      .setLifecycle(
          new Bucket.Lifecycle().setRule(ImmutableList.of(new Rule()
                  .setCondition(new Rule.Condition().setAge(BAD_TTL))
                  .setAction(new Rule.Action().setType("Delete")))));

  // A get that returns a bucket should trigger a check/decorate/update
  executor.when(Storage.Buckets.Get.class, bucket);
  executor.passThruWhen(Storage.Buckets.Update.class,
      checkHasOneRuleLifecycle());

  underTest.perform(CREDENTIALS_ID, build, TaskListener.NULL);
}
 

import com.google.api.services.storage.model.Bucket; //导入依赖的package包/类
@Test
public void testReplaceComplexLifecycle() throws Exception {
  final Rule expireGoodTTL = new Rule()
      .setCondition(new Rule.Condition().setAge(TTL))
      .setAction(new Rule.Action().setType("Delete"));
  final Bucket bucket = new Bucket()
      .setName(BUCKET_NAME)
      .setLifecycle(new Bucket.Lifecycle().setRule(
          // Create a list with two good rules, to validate that
          // multi-clause rules get thrown out.
          ImmutableList.of(expireGoodTTL, expireGoodTTL)));

  // A get that returns a bucket should trigger a check/decorate/update
  executor.when(Storage.Buckets.Get.class, bucket);
  executor.passThruWhen(Storage.Buckets.Update.class,
      checkHasOneRuleLifecycle());

  underTest.perform(CREDENTIALS_ID, build, TaskListener.NULL);
}
 

import com.google.api.services.storage.model.Bucket; //导入依赖的package包/类
@Test
public void testBadAction() throws Exception {
  final Bucket bucket = new Bucket()
      .setName(BUCKET_NAME)
      .setLifecycle(
          new Bucket.Lifecycle().setRule(ImmutableList.of(new Rule()
                  .setCondition(new Rule.Condition().setAge(TTL))
                  .setAction(new Rule.Action().setType("Unknown")))));

  // A get that returns a bucket should trigger a check/decorate/update
  executor.when(Storage.Buckets.Get.class, bucket);
  executor.passThruWhen(Storage.Buckets.Update.class,
      checkHasOneRuleLifecycle());

  underTest.perform(CREDENTIALS_ID, build, TaskListener.NULL);
}
 

import com.google.api.services.storage.model.Bucket; //导入依赖的package包/类
@Test
public void testBadCondition() throws Exception {
  final Bucket bucket = new Bucket()
      .setName(BUCKET_NAME)
      .setLifecycle(
          new Bucket.Lifecycle().setRule(ImmutableList.of(new Rule()
                  .setCondition(new Rule.Condition().setNumNewerVersions(3))
                  .setAction(new Rule.Action().setType("Delete")))));

  // A get that returns a bucket should trigger a check/decorate/update
  executor.when(Storage.Buckets.Get.class, bucket);
  executor.passThruWhen(Storage.Buckets.Update.class,
      checkHasOneRuleLifecycle());

  underTest.perform(CREDENTIALS_ID, build, TaskListener.NULL);
}
 

import com.google.api.services.storage.model.Bucket; //导入依赖的package包/类
@Test
public void testBadComplexCondition() throws Exception {
  final Bucket bucket = new Bucket()
      .setName(BUCKET_NAME)
      .setLifecycle(
          new Bucket.Lifecycle().setRule(ImmutableList.of(new Rule()
                  .setCondition(new Rule.Condition()
                      .setAge(TTL)
                      .setNumNewerVersions(3))
                  .setAction(new Rule.Action().setType("Delete")))));

  // A get that returns a bucket should trigger a check/decorate/update
  executor.when(Storage.Buckets.Get.class, bucket);
  executor.passThruWhen(Storage.Buckets.Update.class,
      checkHasOneRuleLifecycle());

  underTest.perform(CREDENTIALS_ID, build, TaskListener.NULL);
}
 

import com.google.api.services.storage.model.Bucket; //导入依赖的package包/类
@Test
public void testFailingBucketCheck() throws Exception {
  final Bucket bucket = new Bucket().setName(BUCKET_NAME);

  FakeUpload underTest = new FakeUpload(BUCKET_URI,
      new MockUploadModule(executor),
      FAKE_DETAILS,
      bucket);

  // A get that returns a bucket should trigger a check/decorate/update
  executor.when(Storage.Buckets.Get.class, new Bucket());
  executor.passThruWhen(Storage.Buckets.Update.class,
      checkSameBucket(bucket));

  underTest
      .perform(CREDENTIALS_ID, build, build.getWorkspace(),
          TaskListener.NULL);
}
 

import com.google.api.services.storage.model.Bucket; //导入依赖的package包/类
@Test
public void testPassingBucketCheck() throws Exception {
  final Bucket bucket = new Bucket().setName(BUCKET_NAME);

  FakeUpload underTest = new FakeUpload(BUCKET_URI,
      new MockUploadModule(executor),
      FAKE_DETAILS,
      null /* pass the bucket check */);

  // A get that passes our check should incur no further RPC
  executor.when(Storage.Buckets.Get.class, bucket);

  underTest
      .perform(CREDENTIALS_ID, build, build.getWorkspace(),
          TaskListener.NULL);
}
 

import com.google.api.services.storage.model.Bucket; //导入依赖的package包/类
@Test
public void testPassingBucketCheckAfterNotFoundThenConflict()
    throws Exception {
  final Bucket bucket = new Bucket().setName(BUCKET_NAME);

  FakeUpload underTest = new FakeUpload(BUCKET_URI,
      new MockUploadModule(executor),
      FAKE_DETAILS,
      bucket);

  executor.throwWhen(Storage.Buckets.Get.class, notFoundException);
  executor.throwWhen(Storage.Buckets.Insert.class, conflictException);
  // Verify that our final "get" handles updating the bucket as well
  executor.when(Storage.Buckets.Get.class, new Bucket());
  executor.passThruWhen(Storage.Buckets.Update.class,
      checkSameBucket(bucket));

  underTest
      .perform(CREDENTIALS_ID, build, build.getWorkspace(),
          TaskListener.NULL);
}
 

import com.google.api.services.storage.model.Bucket; //导入依赖的package包/类
/**
 * Helper for converting a StorageResourceId + Bucket into a GoogleCloudStorageItemInfo.
 */
public static GoogleCloudStorageItemInfo createItemInfoForBucket(
    StorageResourceId resourceId, Bucket bucket) {
  Preconditions.checkArgument(resourceId != null, "resourceId must not be null");
  Preconditions.checkArgument(bucket != null, "bucket must not be null");
  Preconditions.checkArgument(
      resourceId.isBucket(), "resourceId must be a Bucket. resourceId: %s", resourceId);
  Preconditions.checkArgument(
      resourceId.getBucketName().equals(bucket.getName()),
      "resourceId.getBucketName() must equal bucket.getName(): '%s' vs '%s'",
      resourceId.getBucketName(), bucket.getName());

  // For buckets, size is 0.
  return new GoogleCloudStorageItemInfo(resourceId, bucket.getTimeCreated().getValue(),
      0, bucket.getLocation(), bucket.getStorageClass());
}