1 files changed, 79 insertions, 0 deletions
diff --git a/repackaged/libphonenumber/src/com/android/i18n/phonenumbers/metadata/source/BlockingMetadataBootstrappingGuard.java b/repackaged/libphonenumber/src/com/android/i18n/phonenumbers/metadata/source/BlockingMetadataBootstrappingGuard.java
new file mode 100644
index 00000000..ddb2fb9f
--- /dev/null
+++ b/repackaged/libphonenumber/src/com/android/i18n/phonenumbers/metadata/source/BlockingMetadataBootstrappingGuard.java
@@ -0,0 +1,79 @@
+/* GENERATED SOURCE. DO NOT MODIFY. */
+/*
+ * Copyright (C) 2022 The Libphonenumber Authors
+ *
+ * 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.android.i18n.phonenumbers.metadata.source;
+
+import com.android.i18n.phonenumbers.MetadataLoader;
+import com.android.i18n.phonenumbers.Phonemetadata.PhoneMetadata;
+import com.android.i18n.phonenumbers.metadata.init.MetadataParser;
+import java.io.InputStream;
+import java.util.Collection;
+import java.util.Map;
+import java.util.concurrent.ConcurrentHashMap;
+
+/**
+ * A blocking implementation of {@link MetadataBootstrappingGuard}. Can be used for both single-file
+ * (bulk) and multi-file metadata
+ *
+ * @param <T> needs to extend {@link MetadataContainer}
+ */
+final class BlockingMetadataBootstrappingGuard<T extends MetadataContainer>
+    implements MetadataBootstrappingGuard<T> {
+
+  private final MetadataLoader metadataLoader;
+  private final MetadataParser metadataParser;
+  private final T metadataContainer;
+  private final Map<String, String> loadedFiles; // identity map
+
+  BlockingMetadataBootstrappingGuard(
+      MetadataLoader metadataLoader, MetadataParser metadataParser, T metadataContainer) {
+    this.metadataLoader = metadataLoader;
+    this.metadataParser = metadataParser;
+    this.metadataContainer = metadataContainer;
+    this.loadedFiles = new ConcurrentHashMap<>();
+  }
+
+  @Override
+  public T getOrBootstrap(String phoneMetadataFile) {
+    if (!loadedFiles.containsKey(phoneMetadataFile)) {
+      bootstrapMetadata(phoneMetadataFile);
+    }
+    return metadataContainer;
+  }
+
+  private synchronized void bootstrapMetadata(String phoneMetadataFile) {
+    // Additional check is needed because multiple threads could pass the first check when calling
+    // getOrBootstrap() at the same time for unloaded metadata file
+    if (loadedFiles.containsKey(phoneMetadataFile)) {
+      return;
+    }
+    Collection<PhoneMetadata> phoneMetadata = read(phoneMetadataFile);
+    for (PhoneMetadata metadata : phoneMetadata) {
+      metadataContainer.accept(metadata);
+    }
+    loadedFiles.put(phoneMetadataFile, phoneMetadataFile);
+  }
+
+  private Collection<PhoneMetadata> read(String phoneMetadataFile) {
+    try {
+      InputStream metadataStream = metadataLoader.loadMetadata(phoneMetadataFile);
+      return metadataParser.parse(metadataStream);
+    } catch (IllegalArgumentException | IllegalStateException e) {
+      throw new IllegalStateException("Failed to read file " + phoneMetadataFile, e);
+    }
+  }
+}