LookupLevels 

LookupLevels 在 Paimon 中扮演着**“带缓存的、基于 Key 的数据查找引擎”**的角色。它的核心使命是:当需要根据主键(Key)查找某条数据时,能够高效地在 LSM-Tree 的多层(Levels)数据文件中定位到这条数据,并尽可能地利用本地缓存来避免昂贵的远程 I/O。

LookupLevels 封装了对一个 Levels 对象(代表了 LSM-Tree 的一个分区的所有数据文件分层信息)的查找操作。它本身不存储数据文件的元数据,而是依赖传入的 Levels 对象。它的主要职责可以概括为:

  • 提供统一的查找入口:通过 lookup(InternalRow key, int startLevel) 方法,屏蔽底层多层文件和缓存的复杂性。
  • 实现惰性本地缓存:当第一次需要在一个远程数据文件(SST 文件)中查找时,它会下载该文件,并在本地构建一个优化的、基于 Key 的查找索引文件(Lookup File)。
  • 管理缓存生命周期:通过 Caffeine 缓存管理这些本地查找文件,并能在远程文件被删除时(如 Compaction 后),自动清理对应的本地缓存。
  • 提供灵活的查找策略:通过 ValueProcessor 接口,支持不同场景的查找需求,例如:只需判断 Key 是否存在、需要获取完整的 Value、或需要获取 Value 在文件中的物理位置等。

关键成员变量(构造函数解析)

LookupLevels 的所有核心依赖都通过构造函数注入,这体现了良好的设计模式。

// ... existing code ...private final Levels levels;private final Comparator<InternalRow> keyComparator;private final RowCompactedSerializer keySerializer;private final ValueProcessor<T> valueProcessor;private final IOFunction<DataFileMeta, RecordReader<KeyValue>> fileReaderFactory;private final Function<String, File> localFileFactory;private final LookupStoreFactory lookupStoreFactory;private final Function<Long, BloomFilter.Builder> bfGenerator;private final Cache<String, LookupFile> lookupFileCache;private final Set<String> ownCachedFiles;public LookupLevels(Levels levels,Comparator<InternalRow> keyComparator,RowType keyType,ValueProcessor<T> valueProcessor,IOFunction<DataFileMeta, RecordReader<KeyValue>> fileReaderFactory,Function<String, File> localFileFactory,LookupStoreFactory lookupStoreFactory,Function<Long, BloomFilter.Builder> bfGenerator,Cache<String, LookupFile> lookupFileCache) {
// ... existing code ...
  • levels: 持有 LSM-Tree 的文件分层元数据,是查找的基础。
  • keyComparator & keySerializer: 用于主键的比较和序列化。
  • valueProcessor策略模式的核心。它决定了从远程文件读取数据后,在本地查找文件中存储什么,以及最终返回给调用者什么。
  • fileReaderFactory: 一个工厂,用于根据 DataFileMeta 创建能读取远程数据文件(如 Parquet)的 RecordReader
  • localFileFactory: 一个工厂,用于在本地磁盘上创建临时文件,作为查找文件的载体。
  • lookupStoreFactory: 一个工厂,用于创建真正的本地 KV 存储。Paimon 默认使用基于哈希的实现 (HashLookupStoreFactory)。
  • bfGenerator: 布隆过滤器(Bloom Filter)生成器,用于在创建本地查找文件时一并生成布隆过滤器,可以快速过滤掉不存在的 Key,避免磁盘 I/O。
  • lookupFileCache核心缓存,一个 Caffeine 缓存实例,Key 是远程数据文件的文件名,Value 是封装了本地查找文件的 LookupFile 对象。这个缓存可以在多个 LookupLevels 实例间共享。
  • ownCachedFiles: 一个 HashSet,用于追踪由当前 LookupLevels 实例创建并放入缓存的文件。这主要用于在当前实例关闭时,能准确地清理自己创建的缓存。

lookup(InternalRow key, int startLevel)

// ... existing code ...@Nullablepublic T lookup(InternalRow key, int startLevel) throws IOException {return LookupUtils.lookup(levels, key, startLevel, this::lookup, this::lookupLevel0);}
// ... existing code ...

这个方法将实际的查找逻辑委托给了 LookupUtils 工具类。LookupUtils.lookup 的逻辑遵循 LSM-Tree 的基本原则:

  1. 从 startLevel 开始,逐层向上查找(Level 0, Level 1, Level 2...)。
  2. 因为低层(Level 号码小)的数据比高层的数据更新,所以一旦在某一层找到了数据,就立刻返回,不再继续查找更高层。
  3. 对于 Level 0,由于文件之间 key range 可能重叠,需要遍历所有文件。
  4. 对于 Level 1 及以上,文件间的 key range 不重叠,因此可以通过二分查找快速定位到可能包含目标 key 的那个文件(这里的二分是根据DataFileMeta记录的最大和最小key)。

最终,无论是哪一层,定位到具体的 DataFileMeta 后,都会调用 LookupLevels 自己的 lookup(InternalRow key, DataFileMeta file) 方法。


lookup(InternalRow key, DataFileMeta file) 

// ... existing code ...@Nullableprivate T lookup(InternalRow key, DataFileMeta file) throws IOException {// 1. 尝试从缓存获取 LookupFileLookupFile lookupFile = lookupFileCache.getIfPresent(file.fileName());boolean newCreatedLookupFile = false;if (lookupFile == null) {// 2. 缓存未命中,调用 createLookupFile 创建一个新的lookupFile = createLookupFile(file);newCreatedLookupFile = true;}byte[] valueBytes;try {byte[] keyBytes = keySerializer.serializeToBytes(key);// 3. 使用 LookupFile 在本地进行查找valueBytes = lookupFile.get(keyBytes);} finally {if (newCreatedLookupFile) {// 4. 如果是新创建的,放入缓存供后续使用lookupFileCache.put(file.fileName(), lookupFile);}}if (valueBytes == null) {return null;}// 5. 使用 ValueProcessor 将从本地文件读出的字节数组,转换成最终结果return valueProcessor.readFromDisk(key, lookupFile.remoteFile().level(), valueBytes, file.fileName());}
// ... existing code ...

这个过程清晰地展示了 “按需加载、惰性创建” 的缓存模式。而 createLookupFile 方法则是将远程数据转化为本地索引的核心:

  1. 创建本地空文件localFileFactory.apply(...)
  2. 创建本地写入器lookupStoreFactory.createWriter(...)
  3. 创建远程读取器fileReaderFactory.apply(file)
  4. 数据转换与写入:循环读取远程文件中的每条 KeyValue,通过 valueProcessor.persistToDisk(...) 处理后,写入本地文件。
  5. 封装返回:将创建好的本地文件及其读取器封装成 LookupFile 对象返回。

createLookupFile(DataFileMeta file) 

这个函数是 LookupLevels 实现惰性本地缓存机制的关键。它的主要作用是:当需要在一个远程数据文件(DataFileMeta)中进行查找,但本地缓存又不存在时,由该函数负责将远程数据文件转换成本地优化的、可快速查找的索引文件(LookupFile

下面我们分步骤、层层递进地解析它的实现逻辑。

// ... existing code ...private LookupFile createLookupFile(DataFileMeta file) throws IOException {// 1. 创建本地临时文件File localFile = localFileFactory.apply(file.fileName());if (!localFile.createNewFile()) {throw new IOException("Can not create new file: " + localFile);}// 2. 创建本地 KV 存储的写入器LookupStoreWriter kvWriter =lookupStoreFactory.createWriter(localFile, bfGenerator.apply(file.rowCount()));LookupStoreFactory.Context context;try (RecordReader<KeyValue> reader = fileReaderFactory.apply(file)) {// 3. 数据从远程读取并写入本地KeyValue kv;if (valueProcessor.withPosition()) {// ... (处理带位置信息的场景)} else {// ... (处理不带位置信息的场景)}} catch (IOException e) {// 4. 异常处理:删除不完整的本地文件FileIOUtils.deleteFileOrDirectory(localFile);throw e;} finally {// 5. 关闭写入器并获取上下文context = kvWriter.close();}// 6. 注册到 ownCachedFilesownCachedFiles.add(file.fileName());// 7. 创建并返回 LookupFile 实例return new LookupFile(localFile,file,lookupStoreFactory.createReader(localFile, context),() -> ownCachedFiles.remove(file.fileName()));}
// ... existing code ...

创建本地临时文件 (localFileFactory.apply)

File localFile = localFileFactory.apply(file.fileName());
  • 作用: 为即将创建的本地查找文件在磁盘上预留一个位置。
  • 递归分析:
    • localFileFactory 是一个 Function<String, File> 类型的函数,在 LookupLevels 实例化时由外部传入。
    • 在典型的 KeyValueFileStoreWrite 中,它的实现是调用 ioManager.createChannel(prefix).getPathFile()
    • ioManager 是 Paimon 的 I/O 管理器,它负责在配置的临时目录(java.io.tmpdir 或用户指定的目录)下创建文件,并保证文件名唯一,通常会添加随机后缀。
    • 文件名的前缀由 LookupFile.localFilePrefix(...) 生成,格式为 分区信息_bucket号_远程文件名,这保证了不同来源的文件生成的本地文件不会冲突。

创建本地 KV 存储的写入器 (lookupStoreFactory.createWriter)

LookupStoreWriter kvWriter =lookupStoreFactory.createWriter(localFile, bfGenerator.apply(file.rowCount()));
  • 作用: 获取一个能向第一步创建的 localFile 中写入键值对的 LookupStoreWriter 对象。
  • 递归分析:
    • lookupStoreFactory: 这是本地查找文件格式的工厂。Paimon 支持多种格式,由 CoreOptions.LOOKUP_LOCAL_FILE_TYPE 配置决定。
      • HASH : 返回 HashLookupStoreFactory。它创建的 HashLookupStoreWriter 会构建一个基于哈希表的本地文件,提供 O(1) 的平均查找复杂度。
      • SORT: 返回 SortLookupStoreFactory。它创建的 SortLookupStoreWriter 会构建一个基于排序键的本地文件,通过二分查找进行定位。
    • bfGenerator.apply(file.rowCount()): 这是一个布隆过滤器(Bloom Filter)生成器。createWriter 会接收这个 BloomFilter.Builder,并在写入数据的同时,将所有的 Key 添加到布隆过滤器中。这个布隆过滤器最终也会被序列化到本地文件中,用于快速过滤掉不存在的 Key。

数据从远程读取并写入本地

try (RecordReader<KeyValue> reader = fileReaderFactory.apply(file)) {// ... 循环读取和写入 ...
}

这是数据转换的核心步骤。

  • 作用: 从远程的 SST 文件(如 Parquet)中逐条读取 KeyValue,然后通过 kvWriter 写入本地查找文件。
  • 递归分析:
    • fileReaderFactory.apply(file): 创建一个 RecordReader 来读取远程的 DataFileMeta。这个读取器知道如何解析 Parquet/ORC/Avro 文件格式。
    • reader.readBatch(): 为了效率,数据是按批次读取的。
    • valueProcessor.withPosition(): 这是一个判断,询问当前的 ValueProcessor 策略是否需要原始数据在文件中的物理位置(Position)。
      • 如果为 true (如 PositionedKeyValueProcessor),则会调用 valueProcessor.persistToDisk(kv, batch.returnedPosition())
      • 如果为 false (如 KeyValueProcessor 或 ContainsValueProcessor),则调用 valueProcessor.persistToDisk(kv)
    • valueProcessor.persistToDisk(...): 这是策略模式的应用。ValueProcessor 决定了最终写入本地查找文件的 value 是什么。
      • KeyValueProcessor: 写入完整的 valuesequenceNumber 和 RowKind
      • ContainsValueProcessor: 写入一个空字节数组,极大地节省空间。
      • PositionedKeyValueProcessor: 写入 rowPosition 以及可选的 value 等信息。
    • kvWriter.put(keyBytes, valueBytes): 将序列化后的 key 和 value 写入本地查找文件。kvWriter 内部会处理哈希冲突(HASH 模式)或排序(SORT 模式),并同步更新布隆过滤器。

异常处理

catch (IOException e) {FileIOUtils.deleteFileOrDirectory(localFile);throw e;
}
  • 作用: 这是一个健壮性保证。如果在数据转换过程中发生任何 I/O 异常(例如网络中断、磁盘写满),这个 catch 块会确保被创建出来但不完整的本地临时文件被删除,避免留下垃圾文件。

关闭写入器并获取上下文

  • 作用kvWriter.close() 是一个至关重要的步骤。它会完成所有收尾工作,例如:
    • 将内存中的缓冲区(buffer)刷写到磁盘。
    • 写入文件元数据(metadata),比如布隆过滤器的序列化数据、哈希表的元信息、索引块等。
    • 返回一个 LookupStoreFactory.Context 对象,这个对象包含了读取该文件所必需的元信息(比如文件总大小、布隆过滤器在文件中的偏移量等)。这个 context 会在后续创建 LookupStoreReader 时被传入。

注册到 ownCachedFiles

ownCachedFiles.add(file.fileName());
  • 作用: 将这个新创建的本地文件的远程文件名记录在 ownCachedFiles 这个 Set 中。这用于追踪当前 LookupLevels 实例创建了哪些缓存。当这个实例被 close() 时,它只会清理自己创建的缓存,而不会影响其他实例创建的缓存。

创建并返回 LookupFile 实例

return new LookupFile(localFile,file,lookupStoreFactory.createReader(localFile, context),() -> ownCachedFiles.remove(file.fileName()));
  • 作用: 将所有资源封装成一个 LookupFile 对象返回。
  • 递归分析:
    • new LookupFile(...)LookupFile 是对一个本地查找文件的完整封装。
    • lookupStoreFactory.createReader(localFile, context): 使用与写入时相同的工厂,并传入之前获取的 context,来创建一个 LookupStoreReader。这个 reader 知道如何解析这个本地文件并执行快速查找。
    • () -> ownCachedFiles.remove(file.fileName()): 传入一个 Runnable 回调。这个回调会在 LookupFile 从 Caffeine 缓存中被移除时调用,从而将文件名从 ownCachedFiles 中也移除,保持状态同步。


ValueProcessor 策略接口

LookupLevels 提供了三种 ValueProcessor 实现,以应对不同场景:

  1. KeyValueProcessor: 用于需要获取完整 KeyValue 的场景。它会将 valuesequenceNumber 和 valueKind 都序列化后存入本地文件。
  2. ContainsValueProcessor: 用于仅需判断 Key 是否存在的场景(例如 Lookup Join)。它的 persistToDisk 直接返回一个空字节数组,极大地减小了本地索引文件的大小。readFromDisk 则直接返回 true
  3. PositionedKeyValueProcessor: 用于需要知道数据在原文件中物理位置的场景,这对于实现 Deletion Vector(删除向量)至关重要。它可以选择是否将 value 也一并存入本地文件。

生命周期与缓存清理

LookupLevels 实现了两个接口来管理生命周期:

  • Closeable: 当 LookupLevels 对象关闭时(例如一个 Flink Task 结束),它的 close() 方法会被调用,从而将它自己创建的那些缓存项(记录在 ownCachedFiles 中)从共享的 lookupFileCache 中移除。
  • Levels.DropFileCallback: 它把自己注册为 Levels 的一个回调。当 Compaction 等操作导致某个远程数据文件被删除时,Levels 会通过 notifyDropFile 方法通知 LookupLevels,后者会立即将该文件对应的本地缓存项从 lookupFileCache 中移除,保证了缓存与实际数据的一致性。

总结

LookupLevels 是 Paimon 高性能查找能力的关键。它通过惰性本地缓存可插拔的 KV 存储灵活的 ValueProcessor 策略,巧妙地解决了在存算分离架构下,如何高效地对远程大规模数据进行点查的问题。其设计兼顾了性能、灵活性和健壮的缓存管理,是理解 Paimon 读取链路的一个绝佳范例。

lookup 方法的调用链

LookupLevels.lookup 方法本身非常简洁,它将所有复杂的逻辑都委托给了 LookupUtils.lookup

// ... existing code ...@Nullablepublic T lookup(InternalRow key, int startLevel) throws IOException {return LookupUtils.lookup(levels, key, startLevel, this::lookup, this::lookupLevel0);}
// ... existing code ...

这里的关键是理解两个方法引用:this::lookup 和 this::lookupLevel0。它们分别指向 LookupLevels 类中其他同名但参数不同的 lookup 和 lookupLevel0 私有方法

  • this::lookup 对应的是 private T lookup(InternalRow key, SortedRun level)
  • this::lookupLevel0 对应的是 private T lookupLevel0(InternalRow key, TreeSet<DataFileMeta> level0)

现在,我们来看 LookupUtils.lookup 的源码,看看它是如何使用这两个方法引用的。

第一层:LookupUtils.lookup(levels, ...) - 顶层循环

LookupUtils.java

// ... existing code ...public static <T> T lookup(Levels levels,InternalRow key,int startLevel,BiFunctionWithIOE<InternalRow, SortedRun, T> lookup, // 对应 this::lookupBiFunctionWithIOE<InternalRow, TreeSet<DataFileMeta>, T> level0Lookup) // 对应 this::lookupLevel0throws IOException {T result = null;// 1. 从 startLevel 开始,逐层向上查找for (int i = startLevel; i < levels.numberOfLevels(); i++) {if (i == 0) {// 2. 如果是 Level 0,调用传入的 level0Lookup 函数result = level0Lookup.apply(key, levels.level0());} else {// 3. 如果是 Level 1+,调用传入的 lookup 函数SortedRun level = levels.runOfLevel(i);result = lookup.apply(key, level);}// 4. 一旦找到结果,立即跳出循环if (result != null) {break;}}return result;}
// ... existing code ...

调用过程分析:

  1. LookupUtils.lookup 接收到 LookupLevels 传来的两个方法引用,分别命名为 lookup 和 level0Lookup
  2. 它开始一个 for 循环,从 startLevel 开始遍历 LSM-Tree 的每一层。
  3. 当 i == 0 时:它调用 level0Lookup.apply(...)。这实际上就是调用了 LookupLevels.lookupLevel0(key, levels.level0())
  4. 当 i > 0 时:它调用 lookup.apply(...)。这实际上就是调用了 LookupLevels.lookup(key, levels.runOfLevel(i))
  5. 只要任何一次调用返回了非 null 的结果,循环就会终止,并返回该结果。这符合 LSM-Tree 从新数据(低 Level)向旧数据(高 Level)查找的原则。

第二层:深入 LookupLevels 的私有方法

现在我们来看被调用的那两个私有方法内部又做了什么。

LookupLevels.lookupLevel0

// ... existing code ...@Nullableprivate T lookupLevel0(InternalRow key, TreeSet<DataFileMeta> level0) throws IOException {return LookupUtils.lookupLevel0(keyComparator, key, level0, this::lookup);}
// ... existing code ...

这里又出现了一次委托!它调用了 LookupUtils.lookupLevel0,并且又传递了一个方法引用 this::lookup。这次的 this::lookup 对应的是 private T lookup(InternalRow key, DataFileMeta file)

LookupLevels.lookup(key, level)

// ... existing code ...@Nullableprivate T lookup(InternalRow key, SortedRun level) throws IOException {return LookupUtils.lookup(keyComparator, key, level, this::lookup);}
// ... existing code ...

同样,这里也委托给了 LookupUtils.lookup 的另一个重载版本,并再次传递了方法引用 this::lookup,同样对应 private T lookup(InternalRow key, DataFileMeta file)


第三层:LookupUtils 的具体查找逻辑

现在我们进入 LookupUtils 的具体实现,看看它们如何使用第三层传递进来的 this::lookup 方法引用。

LookupUtils.lookupLevel0

// ... existing code ...public static <T> T lookupLevel0(Comparator<InternalRow> keyComparator,InternalRow target,TreeSet<DataFileMeta> level0,BiFunctionWithIOE<InternalRow, DataFileMeta, T> lookup) // 对应 this::lookup(key, file)throws IOException {T result = null;// 遍历 Level 0 的所有文件for (DataFileMeta file : level0) {// 检查 key 是否在文件的 [minKey, maxKey] 范围内if (keyComparator.compare(file.maxKey(), target) >= 0&& keyComparator.compare(file.minKey(), target) <= 0) {// 如果在范围内,就调用传入的 lookup 函数result = lookup.apply(target, file);if (result != null) {// 找到就返回(Level 0 内部文件按新旧排序,所以第一个找到的就是最新的)return result;}}}return null;}
// ... existing code ...

LookupUtils.lookup(keyComparator, ...)

LookupUtils.java

// ... existing code ...public static <T> T lookup(Comparator<InternalRow> keyComparator,InternalRow target,SortedRun level,BiFunctionWithIOE<InternalRow, DataFileMeta, T> lookup) // 对应 this::lookup(key, file)throws IOException {// ...// 对 Level 1+ 的文件列表进行二分查找,找到可能包含 key 的那个文件// ... (binary search logic) ...List<DataFileMeta> files = level.files();int left = 0;int right = files.size() - 1;// binary search restart positions to find the restart position immediately before the// targetKeywhile (left < right) {int mid = (left + right) / 2;if (keyComparator.compare(files.get(mid).maxKey(), target) < 0) {// Key at "mid.max" is < "target".  Therefore all// files at or before "mid" are uninteresting.left = mid + 1;} else {// Key at "mid.max" is >= "target".  Therefore all files// after "mid" are uninteresting.right = mid;}}int index = right;// ...// 如果找到了对应的文件if (index < files.size()&& keyComparator.compare(files.get(index).minKey(), target) <= 0) {// 调用传入的 lookup 函数return lookup.apply(target, files.get(index));}return null;}
// ... existing code ...

第四层:最终的执行体

经过层层传递,最终所有逻辑都汇聚到了 LookupLevels.lookup(InternalRow key, DataFileMeta file) 这个方法。

// ... existing code ...@Nullableprivate T lookup(InternalRow key, DataFileMeta file) throws IOException {// 1. 检查本地缓存LookupFile lookupFile = lookupFileCache.getIfPresent(file.fileName());// 2. 如果缓存没有,创建本地查找文件if (lookupFile == null) {lookupFile = createLookupFile(file);// ...}// 3. 在本地查找文件中执行 get 操作valueBytes = lookupFile.get(keyBytes);// ...// 4. 处理结果并返回return valueProcessor.readFromDisk(...);}
// ... existing code ...

这个方法是真正干活的地方:检查和使用缓存、创建本地查找文件、从本地文件中读取数据。

总结调用链

整个过程是一个精巧的委托链条:

  1. LookupLevels.lookup(key, startLevel) (入口)

    • 委托给 LookupUtils.lookup(levels, ...),并告诉它:“当你需要处理 Level 0 时,请调用我的 lookupLevel0 方法;当你需要处理 Level 1+ 时,请调用我的 lookup(key, level) 方法。”
  2. LookupUtils.lookup(levels, ...) (顶层循环)

    • 根据当前层级,调用 LookupLevels 对应的私有方法。
  3. LookupLevels.lookupLevel0 或 LookupLevels.lookup(key, level) (中间层)

    • 这两个方法再次委托给 LookupUtils 中更具体的实现 (lookupLevel0 或 lookup(keyComparator, ...) ),并告诉它们:“当你定位到具体要查哪个 DataFileMeta 文件时,请调用我的 lookup(key, file) 方法。”
  4. LookupUtils.lookupLevel0 或 LookupUtils.lookup(keyComparator, ...) (定位层)

    • 它们负责遍历(Level 0)或二分查找(Level 1+)来定位到具体的文件。
    • 一旦定位到 DataFileMeta,就调用最终的方法引用,即 LookupLevels.lookup(key, file)
  5. LookupLevels.lookup(key, file) (执行层)

    • 这是所有逻辑的终点,负责与缓存和本地文件系统交互,完成最终的查找。

这种设计将**“遍历/查找策略”(在 LookupUtils 中)和“具体执行逻辑”**(在 LookupLevels 中)完美地解耦开来,使得代码结构清晰,复用性强。

LookupFile

LookupFile 是 Paimon 本地查找缓存机制的物理载体和逻辑封装。当 LookupLevels 决定将一个远程数据文件(SST 文件)缓存到本地时,最终产物就是一个 LookupFile 对象。这个对象不仅代表了本地磁盘上的一个物理文件,还封装了对该文件的所有操作和生命周期管理。

LookupFile 的职责非常清晰和集中:

  • 封装本地查找文件:它持有一个本地 File 对象和一个 LookupStoreReader,提供了对这个本地优化文件的统一访问入口。
  • 关联远程文件:它内部保存了原始远程文件 DataFileMeta 的引用,明确了这个本地缓存的来源。
  • 提供查找功能:通过 get(byte[] key) 方法,利用内部的 LookupStoreReader 在本地文件上执行快速的 Key-Value 查找。
  • 管理生命周期:通过 close() 方法,负责关闭文件读取器、删除本地物理文件,并执行必要的回调。
  • 作为缓存的 ValueLookupFile 对象本身就是 Caffeine 缓存中的 Value,与远程文件名(Key)一一对应。

关键成员变量(构造函数解析)

LookupFile 的所有核心组件都在构造时传入,清晰地定义了它的构成。

// ... existing code ...private final File localFile;private final DataFileMeta remoteFile;private final LookupStoreReader reader;private final Runnable callback;private long requestCount;private long hitCount;private boolean isClosed = false;public LookupFile(File localFile, DataFileMeta remoteFile, LookupStoreReader reader, Runnable callback) {this.localFile = localFile;this.remoteFile = remoteFile;this.reader = reader;this.callback = callback;}
// ... existing code ...
  • localFile: 指向本地磁盘上物理文件的 java.io.File 对象。这是查找操作的物理基础。
  • remoteFile: 对应的远程数据文件的元数据(DataFileMeta)。用于追溯来源和调试。
  • reader核心查找器。这是一个 LookupStoreReader 接口的实例,它知道如何解析 localFile 的二进制格式并高效地查找 Key。具体的实现可能是 HashLookupStoreReader 或 SortLookupStoreReader,取决于创建时使用的 LookupStoreFactory
  • callback: 一个 Runnable 对象。这是一个非常重要的回调函数。当 LookupFile 被关闭或从缓存中移除时,这个回调会被执行。在 LookupLevels 中,这个回调的作用是 ownCachedFiles.remove(file.fileName()),用于维护 LookupLevels 内部的状态一致性。
  • requestCounthitCount: 用于统计这个本地文件的访问情况,便于监控缓存效率。
  • isClosed: 状态标记,防止对一个已经关闭的 LookupFile 进行操作。

get(byte[] key)

// ... existing code ...@Nullablepublic byte[] get(byte[] key) throws IOException {checkArgument(!isClosed);requestCount++;byte[] res = reader.lookup(key);if (res != null) {hitCount++;}return res;}
// ... existing code ...

这是 LookupFile 最主要的功能方法。它将查找请求直接委托给内部的 reader.lookup(key)LookupStoreReader 会利用布隆过滤器、哈希索引或二分查找等技术,在 localFile 中快速定位并返回与 key 对应的 value 字节数组。

close(RemovalCause cause)

// ... existing code ...public void close(RemovalCause cause) throws IOException {reader.close();isClosed = true;callback.run();LOG.info("Delete Lookup file {} due to {}. Access stats: requestCount={}, hitCount={}, size={}KB",
// ... existing code ...FileIOUtils.deleteFileOrDirectory(localFile);}
// ... existing code ...

这是 LookupFile 的生命周期终点。它执行一系列清理操作:

  1. reader.close(): 关闭底层的 LookupStoreReader,释放文件句柄等资源。
  2. isClosed = true: 更新状态。
  3. callback.run(): 执行回调,通知其创建者(LookupLevels)自己已被销毁。
  4. LOG.info(...): 打印一条详细的日志,说明文件被删除的原因(RemovalCause)、访问统计和文件大小,这对于问题排查和性能调优非常有价值。
  5. FileIOUtils.deleteFileOrDirectory(localFile)从本地磁盘上删除物理文件,释放磁盘空间。

静态工厂与缓存集成 (createCache)

LookupFile 类还包含一组非常重要的静态方法,用于创建和配置 Caffeine 缓存。这使得缓存的创建逻辑与 LookupFile 本身紧密耦合,体现了其作为“可缓存对象”的设计意图。

// ... existing code ...public static Cache<String, LookupFile> createCache(Duration fileRetention, MemorySize maxDiskSize) {return Caffeine.newBuilder().expireAfterAccess(fileRetention).maximumWeight(maxDiskSize.getKibiBytes()).weigher(LookupFile::fileWeigh).removalListener(LookupFile::removalCallback).executor(Runnable::run).build();}
// ... existing code ...

这里配置了 Caffeine 缓存的几个关键策略:

  • .expireAfterAccess(fileRetention)基于访问时间的淘汰策略。如果一个 LookupFile 在 fileRetention 时间内没有被访问过,它就会被标记为过期,并可能被缓存淘汰。这是避免冷数据永久占用磁盘空间的关键。
  • .maximumWeight(maxDiskSize.getKibiBytes())基于容量的淘汰策略。限制了所有本地查找文件占用的总磁盘空间。
  • .weigher(LookupFile::fileWeigh): 指定了如何计算每个缓存项的“权重”。fileWeigh 方法返回本地文件的大小(以 KB 为单位)。当总权重超过 maximumWeight 时,Caffeine 会开始淘汰缓存项。
  • .removalListener(LookupFile::removalCallback)核心清理机制。当一个 LookupFile 因为任何原因(过期、容量超限、手动删除)被缓存移除时,removalCallback 方法会被调用。这个回调方法会调用 lookupFile.close(cause),从而触发前面分析的清理流程,确保本地物理文件被删除。

 Caffeine 内部有自己的维护线程,用于处理过期检查、淘汰计算等任务。默认情况下,为了不阻塞这些核心维护线程,removalListener 会被提交到一个默认的 Executor (通常是 ForkJoinPool.commonPool()) 中异步执行

然而,Paimon 在这里通过 .executor(Runnable::run) 改变了默认行为。

Runnable::run 是一个方法引用,它等价于一个 Executor 的实现:

// 伪代码
class SameThreadExecutor implements Executor {@Overridepublic void execute(Runnable command) {command.run(); // 直接在当前线程执行,而不是提交到线程池}
}

所以,.executor(Runnable::run) 的确切含义是:

“请不要将 removalListener 的任务异步地提交到线程池,而是在触发这个移除事件的那个线程中,立即、同步地执行它。”

为什么 Paimon 要这么做?

选择同步执行而不是异步执行,通常是出于以下考虑:

  1. 及时性和确定性LookupFile 的关闭操作涉及到文件 I/O(删除本地文件)。通过同步执行,可以确保一旦 Caffeine 决定移除一个文件,这个文件会立即被删除。这避免了在异步执行模型下可能出现的延迟——即缓存已经认为条目被移除了,但对应的磁盘文件由于线程池繁忙等原因,可能过了一小段时间才被真正删除。对于需要精确控制磁盘空间的场景,这种确定性很重要。

  2. 简化资源管理: 同步执行可以简化并发控制。由于回调是在 Caffeine 的内部维护线程中同步执行的,可以减少多线程并发访问和修改 LookupFile 状态的复杂性。

  3. 任务轻量lookupFile.close() 操作虽然是 I/O,但删除单个文件通常是一个非常快速的操作。Paimon 的设计者可能认为这个操作足够轻量,不会对 Caffeine 的内部线程造成明显的阻塞,因此同步执行的开销是可以接受的。

本地文件名生成 (localFilePrefix)

// ... existing code ...public static String localFilePrefix(RowType partitionType, BinaryRow partition, int bucket, String remoteFileName) {if (partition.getFieldCount() == 0) {return String.format("%s-%s", bucket, remoteFileName);} else {String partitionString = partToSimpleString(partitionType, partition, "-", 20);return String.format("%s-%s-%s", partitionString, bucket, remoteFileName);}}
// ... existing code ...

这个静态方法定义了本地查找文件的命名规则,格式为 分区信息-bucket号-远程文件名。这样做的好处是:

  • 唯一性:保证了不同分区、不同 Bucket、不同远程文件生成的本地缓存文件不会重名。
  • 可读性:通过文件名就可以大致了解这个缓存文件对应的数据来源,方便人工排查问题。

DataFileMeta vs. LookupFile

这两者代表了不同层面的东西,它们的存储位置也不同:

  • DataFileMeta对象本身在内存中,它描述的数据在远程磁盘上。

    • DataFileMeta 是一个轻量级的 Java 对象,它本身存在于 JVM 内存中。
    • 它并不包含实际的数据,而是数据的“元信息”,比如数据文件的路径(通常在 HDFS、S3 等远程存储上)、文件大小、行数、主键的最小/最大值等。
    • 你可以把它理解为一个指向远程数据文件的“指针”或“描述符”。
  • LookupFile对象本身在内存中,它管理的数据在本地磁盘上。

    • LookupFile 也是一个 Java 对象,存在于 JVM 内存中(具体来说,是作为 lookupFileCache 的 value)。
    • 它的核心职责是管理一个为了加速查询而创建在本地磁盘上的临时文件。这个本地文件是远程数据文件(由 DataFileMeta 指向)的一个经过优化的副本。
    • 当需要进行 lookup 时,Paimon 会将远程文件拉取到本地,并可能转换成更适合快速随机查找的格式(如哈希索引文件),这个本地文件由 LookupFile 对象来管理。

Caffeine 自带写磁盘功能?

这是一个非常关键的问题。答案是:不,Caffeine 本身是一个纯粹的、高性能的 内存 缓存库。 它不提供原生的将缓存项写入磁盘的功能。

那么 Paimon 是如何实现用它来管理磁盘文件的呢?

Paimon 在这里使用了一种非常巧妙的设计模式,利用 Caffeine 的能力来管理本地磁盘文件的 生命周期,而不是用它来存储文件内容本身。

具体实现机制如下:

  1. 缓存的不是文件内容,而是文件句柄:Paimon 存入 Caffeine 缓存的 value 不是文件的字节内容,而是 LookupFile 这个 Java 对象。这个对象内部持有了本地磁盘文件的路径。

  2. 权重计算基于磁盘大小:Caffeine 的容量限制是通过 weigher 来计算每个缓存项的“权重”的。Paimon 在配置 lookupFileCache 时,提供的 weigher (LookupFile::fileWeigh) 计算的不是 LookupFile 对象的内存大小,而是它所指向的本地磁盘文件的实际大小

    // ... existing code ...
    public static Cache<String, LookupFile> createCache(Duration fileRetention, MemorySize maxDiskSize) {return Caffeine.newBuilder().expireAfterAccess(fileRetention)// 这里的 maximumWeight 是基于磁盘大小的.maximumWeight(maxDiskSize.getKibiBytes())// weigher 计算的是本地文件在磁盘上的大小.weigher(LookupFile::fileWeigh).removalListener(LookupFile::removalCallback).executor(Runnable::run).build();
    }private static int fileWeigh(String file, LookupFile lookupFile) {// 返回本地文件的大小(以 KB 为单位)return fileKibiBytes(lookupFile.localFile);
    }
    // ... existing code ...
    

    这样一来,Caffeine 的容量管理就从“内存大小”变成了“磁盘空间大小”。

  3. 利用淘汰监听器删除文件:当缓存占用的总“权重”(即总磁盘空间)超过阈值时,Caffeine 会根据其淘汰策略(如 LRU)移除一个 LookupFile 对象。此时,会触发注册的 removalListener (LookupFile::removalCallback)。在这个监听器中,Paimon 的代码会主动地去删除该 LookupFile 对象所对应的本地磁盘文件,从而释放磁盘空间。

    // ... existing code ...
    private static void removalCallback(String file, LookupFile lookupFile, RemovalCause cause) {if (lookupFile != null) {try {// 当缓存项被淘汰时,关闭并删除对应的本地磁盘文件lookupFile.close(cause);} catch (IOException e) {throw new UncheckedIOException(e);}}
    }public void close(RemovalCause cause) throws IOException {reader.close();isClosed = true;callback.run();LOG.info("Delete Lookup file {} due to {}. Access stats: requestCount={}, hitCount={}, size={}KB",localFile.getName(),cause,requestCount,hitCount,localFile.length() >> 10);FileIOUtils.deleteFileOrDirectory(localFile);}
    // ... existing code ...
    

综上所述,Paimon 并不是在使用 Caffeine 的某个隐藏的磁盘功能,而是将 Caffeine 作为了一个非常高效的本地磁盘文件生命周期管理器。Caffeine 负责决定“何时”淘汰哪个文件,而 Paimon 的代码则负责执行实际的“文件删除”操作。这是一个非常优雅的组合设计。

本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。
如若转载,请注明出处:http://www.pswp.cn/pingmian/87793.shtml
繁体地址,请注明出处:http://hk.pswp.cn/pingmian/87793.shtml
英文地址,请注明出处:http://en.pswp.cn/pingmian/87793.shtml

如若内容造成侵权/违法违规/事实不符,请联系英文站点网进行投诉反馈email:809451989@qq.com,一经查实,立即删除!

相关文章

Ruby大会演讲实录:Baklib 如何用 AI 重构内容管理赛道

“2015 年成都 Ruby 大会时&#xff0c;我们还在做大数据项目&#xff1b;2025 年的今天&#xff0c;Baklib 已服务 800 多家企业。” 在 RubyConf China 2025 的演讲台上&#xff0c;Baklib 创始人Song以十年对比开篇&#xff0c;讲述了从技术爱好者到企业服务创业者的蜕变&am…

408第三季part2 - 计算机网络 - 传输层II

理解第一次和第二次握手是不能携带数据&#xff0c;只能消耗一个序号后面挥手也有第一次和第三次题目建立连接是1000&#xff0c;FIN挥手是5001&#xff0c;这两个是不会带数据的所以字节数范围是1001-50005000-10011 4000c再次理解还可以叫快速重传题目服务器想要100确认号客…

揭秘图像LLM:从像素到语言的智能转换

图像LLM是怎么工作 图像LLM(多模态大语言模型)的核心是将图像转化为语言模型能理解的“语言”,并与文本深度融合。以下结合CLIP、DALL-E、GPT-4V等主流模型,通过具体例子说明其工作机制: 一、图像→特征向量:从像素到“密码” 例子:识别“戴墨镜的猫” 视觉编码器提取…

十、K8s集群资源合理化分配

十、K8s集群资源合理化分配 文章目录 十、K8s集群资源合理化分配1、K8s 资源限制 ResourceQuota1.1 什么是ResourceQuota&#xff1f;1.2 ResourceQuota通常用于如下场景&#xff1a;1.3 基于租户和团队的资源限制1.4 基于命名空间的资源限制 2、K8s 资源限制 LimitRange2.1 设…

Android 13 设置界面会判断当前屏幕的大小,如果是大屏,则为左右屏显示

1.前言 在13.0的系统rom定制化开发中,在某些时候,在大屏设备中,设置新增了左右分屏的功能,就是 左边显示主菜单,右边显示一级菜单的功能,某些情况下不需要,接下来关闭这个功能 2.设置界面会判断当前屏幕的大小,如果是大屏,则为左右屏显示的核心类 packages/apps/Sett…

算法学习day15----蓝桥杯--进制转换

python代码&#xff1a;def jinzhizhuanhuan(x, n):ans 0while x > 0:ans x % nx // nreturn ans sum 0 for i in range(1, 2025):if jinzhizhuanhuan(i, 2) jinzhizhuanhuan(i, 4):sum 1 print(sum)虽然结束了期末周&#xff0c;但是又变懒了不少&#xff0c;水一个先…

OpenStack扩展

一、OpenStack &#xff08;开源云计算管理平台&#xff09;的核心服务组件及功能计算服务&#xff08;Compute - Nova&#xff09;功能&#xff1a;是 OpenStack 计算资源管理的核心组件&#xff0c;负责虚拟机实例全生命周期管理&#xff0c;像实例的创建、启动、暂停、迁移、…

【LeetCode 热题 100】41. 缺失的第一个正数——(解法二)原地哈希

Problem: 41. 缺失的第一个正数 题目&#xff1a;给你一个未排序的整数数组 nums &#xff0c;请你找出其中没有出现的最小的正整数。 请你实现时间复杂度为 O(n) 并且只使用常数级别额外空间的解决方案。 【LeetCode 热题 100】41. 缺失的第一个正数——&#xff08;解法一&am…

C#上位机之Modbus通信协议!

文章目录前言一、Modbus概念二、使用步骤1.使用Modbus准备2.使用步骤三、Modbus RTU 与 Modbus ASCII对比前言 Modbus通信协议&#xff01; 一、Modbus概念 从站设备编码&#xff08;从站地址、单元ID&#xff09;&#xff0c;一主多从。 存储区&#xff1a;0-线圈状态、1-输…

前后端分离架构下的跨域问题与解决方案

在现代Web开发中&#xff0c;特别是随着前后端分离架构的普及&#xff0c;跨域问题成为了开发者必须面对的一个重要议题。本文将详细介绍什么是跨域问题、其产生的原因以及如何从前端和后端两个角度来解决这个问题&#xff0c;并提供一些实用的代码示例。一、跨域问题概述1. 定…

搜索数据建设系列之数据架构重构

导读 主要概述百度搜索业务数据建设的创新实践&#xff0c;重点围绕宽表模型设计、计算引擎优化和新一代业务服务交付模式&#xff08;图灵3.0开发模式&#xff09;三大方向&#xff0c;解决了传统数仓在搜索场景下面临的诸多挑战&#xff0c;实现了搜索数据建设的高效、稳定、…

2025年渗透测试面试题总结-2025年HW(护网面试) 29(题目+回答)

安全领域各种资源&#xff0c;学习文档&#xff0c;以及工具分享、前沿信息分享、POC、EXP分享。不定期分享各种好玩的项目及好用的工具&#xff0c;欢迎关注。、 目录 2025年HW(护网面试) 29 1. 样本分析思路 2. Linux GDB分析样本示例 3. 应急案例&#xff1a;WebShell后…

动态编程入门第二节:委托与事件 - Unity 开发者的高级回调与通信艺术

动态编程入门第一节&#xff1a;C# 反射 - Unity 开发者的超级工具箱 动态编程入门第二节&#xff1a;委托与事件 - Unity 开发者的高级回调与通信艺术 上次我们聊了 C# 反射&#xff0c;它让程序拥有了在运行时“看清自己”的能力。但光能看清还不够&#xff0c;我们还需要让…

降低网络安全中的人为风险:以人为本的路径

有效降低网络安全中的人为风险&#xff0c;关键在于采取以人为本的方法。这种方法的核心在于通过高效的培训和实践&#xff0c;使员工掌握安全知识、践行安全行为&#xff0c;并最终培育出安全且相互支持的文化氛围。 诚然&#xff0c;技术和政策必须为良好的安全行为提供支持、…

opencv裁剪和编译

opencv裁剪和编译 0. 准备工作 0.1 下载和安装Eigen 地址 https://eigen.tuxfamily.org/index.php?titleMain_Page对于opencv编译&#xff0c;需要增加EIGEN_INCLUDE_PATH和开启WITH_EIGEN -DWITH_EIGENON -DEIGEN_INCLUDE_PATH./3rd/eigen-3.4.01. 实际脚本 编译脚本如下: ch…

小白成长之路-mysql数据基础(三)

文章目录一、主从复制二、案例总结一、主从复制 1、master开启二进制日志记录2、slave开启IO进程&#xff0c;从master中读取二进制日志并写入slave的中继日志3、slave开启SQL进程&#xff0c;从中继日志中读取二进制日志并进行重放4、最终&#xff0c;达到slave与master中数据…

通过 Windows 共享文件夹 + 手机访问(SMB协议)如何实现

通过 Windows 共享文件夹 手机访问&#xff08;SMB协议&#xff09; 实现 PC 和安卓手机局域网文件共享&#xff0c;具体步骤如下&#xff1a; &#x1f4cc; 前置条件 电脑和手机连接同一局域网&#xff08;同一个Wi-Fi或路由器&#xff09;。关闭防火墙或放行SMB端口&#…

【Python3教程】Python3高级篇之正则表达式

博主介绍:✌全网粉丝23W+,CSDN博客专家、Java领域优质创作者,掘金/华为云/阿里云/InfoQ等平台优质作者、专注于Java技术领域✌ 技术范围:SpringBoot、SpringCloud、Vue、SSM、HTML、Nodejs、Python、MySQL、PostgreSQL、大数据、物联网、机器学习等设计与开发。 感兴趣的可…

Redis--黑马点评--达人探店功能实现详解

达人探店发布探店笔记探店笔记类似于点评网站的评价&#xff0c;往往是图文结合&#xff0c;对应的表有两个&#xff1a;tb_blog&#xff1a;探店笔记表&#xff0c;包含笔记中的标题、文字、图片等tb_blog_comments&#xff1a;其他用户对探店笔记的评价tb_blog表结构如下&…

一探 3D 互动展厅的神奇构造​

3D 互动展厅的神奇之处&#xff0c;离不开一系列先进技术的强力支撑 。其中&#xff0c;VR(虚拟现实)技术无疑是核心亮点之一。通过佩戴 VR 设备&#xff0c;观众仿佛被瞬间 “传送” 到一个全新的世界&#xff0c;能够全身心地沉浸其中&#xff0c;360 度无死角地观察周围的一…