refactor: remove computeIfAbsent

goto · Dec 11, 2024 · a10ec22 · a10ec22
1 parent 811d1ab
commit a10ec22
Showing 1 changed file with 7 additions and 4 deletions.
diff --git a/src/main/java/com/gotocompany/depot/maxcompute/model/MaxComputeProtobufConverterCache.java b/src/main/java/com/gotocompany/depot/maxcompute/model/MaxComputeProtobufConverterCache.java
@@ -32,10 +32,13 @@ public MaxComputeProtobufConverterCache(MaxComputeSinkConfig maxComputeSinkConfi
     }
 
     public TypeInfo getOrCreateTypeInfo(Descriptors.FieldDescriptor fieldDescriptor) {
-        return typeInfoCache.computeIfAbsent(fieldDescriptor.getFullName(), key -> {
-            ProtobufMaxComputeConverter protobufMaxComputeConverter = getConverter(fieldDescriptor);
-            return protobufMaxComputeConverter.convertTypeInfo(fieldDescriptor);
-        });
+        if (typeInfoCache.containsKey(fieldDescriptor.getFullName())) {
+            return typeInfoCache.get(fieldDescriptor.getFullName());
+        }
+        ProtobufMaxComputeConverter protobufMaxComputeConverter = getConverter(fieldDescriptor);
+        TypeInfo typeInfo = protobufMaxComputeConverter.convertTypeInfo(fieldDescriptor);
+        typeInfoCache.put(fieldDescriptor.getFullName(), typeInfo);
+        return typeInfo;
     }
 
     public TypeInfo getOrCreateTypeInfo(Descriptors.FieldDescriptor fieldDescriptor, Supplier<TypeInfo> supplier) {