一键多值的并发映射

Question

我需要一个线程安全且可以有多个值的 Map。我确实使用MultiMap 写了一个

    private MultiMap<String, String> uploadIdETagMap = new MultiValueMap<String, String>();

//Function to get tag value and add to map to that uploadId key
    public void uploadPartForMultiPartUpload() {
        try {
            UploadPartRequest partRequest = new UploadPartRequest();
            partRequest = partRequest.withBucketName(bucketName).withInputStream(inputStream).withKey(fileName)
                    .withPartNumber(partNumber).withUploadId(uploadId).withPartSize(sizeOfData)
                    .withLastPart(isLastPart);
            UploadPartResult result = s3Client.uploadPart(partRequest);
            uploadIdETagMap.put(uploadId, result.getPartETag());
        } catch (Exception e) {
            log.error("Error uploading part: {} in multi part upload.", partNumber, e);
        }
    }

    List<String> tagList = (List<String>) uploadIdETagMap.get(uploadId);

所以每次我都会为特定的 uplaodId 键获得新的标签值，最后我需要所有这些标签，并且同时会有其他上传处理。

在上面的代码中，每次都会添加tag 作为特定uploadId 键的MultiMap 的值。 上述代码有效，但MultiMap 和MultiValueMap 均已弃用且不是线程安全的。

所以任何建议我如何才能通过最新版本和线程安全来实现这一点。

Answer 1

tl;博士

personFavoriteColors.computeIfAbsent( 
    "Alice" , 
    ( x -> new CopyOnWriteArrayList<>() )
)
.add( "Purple" )
;

personFavoriteColors.computeIfAbsent( 
    "Alice" , 
    ( x -> new CopyOnWriteArrayList<>() )
)
.add( "Gold" )
;

personFavoriteColors.toString(): [紫、金]

Map::computeIfAbsent

Java 现在内置了 multimap 行为，并在 Map 接口上定义了新的 computeIfAbsent 方法。

定义你的地图。

ConcurrentMap< String , Set< String> > personFavoriteColors = new ConcurrentSkipListMap<>() ;

我们正在使用Map 实现，它根据您的要求实现了ConcurrentMap 接口以实现线程安全。 Java 提供了两种这样的实现。

使用单行创建一个新列表或设置以保存多个值。

personFavoriteColors.computeIfAbsent( 
    "Alice" , 
    ( x -> new CopyOnWriteArraySet<>() )   // Lambda expression. Run only if needed to run (if absent). 
)                                          // Returns the value of a map (a `Set` or `List` when desiring a multimap). 
.add( "Purple" )
;

您可以在拥有它的Map 之外访问Set（或List）。所以Set（或List）也必须是并发的。

这里我们使用并发集 CopyOnWriteArraySet 作为地图的值。

例如：

Set< String > favoriteColorsForAlice = personFavoriteColors.get( "Alice" ) ;
System.out.println( favoriteColorsForAlice ) ; 

看到这个code run live on IdeOne.com。

[紫、金]

Answer 2

对于您使用的任何Map<String,List<String>> 实现，您都可以执行以下操作：

String key;
String value;
map.compute(key, ((k,v)-> v == null ? new ArrayList<>() : v).add(value));

上面检查List<String> for key是否存在，如果不存在，则创建一个，否则使用现有的。在这两种情况下，都会将值添加到给定键的新列表或现有列表中。