com.google.i18n.phonenumbers.metadata.source.BlockingMetadataBootstrappingGuard Maven / Gradle / Ivy
/*
* 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.google.i18n.phonenumbers.metadata.source;
import com.google.i18n.phonenumbers.MetadataLoader;
import com.google.i18n.phonenumbers.Phonemetadata.PhoneMetadata;
import com.google.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 needs to extend {@link MetadataContainer}
*/
final class BlockingMetadataBootstrappingGuard
implements MetadataBootstrappingGuard {
private final MetadataLoader metadataLoader;
private final MetadataParser metadataParser;
private final T metadataContainer;
private final Map 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 = read(phoneMetadataFile);
for (PhoneMetadata metadata : phoneMetadata) {
metadataContainer.accept(metadata);
}
loadedFiles.put(phoneMetadataFile, phoneMetadataFile);
}
private Collection 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);
}
}
}