Add getFileChecksum and getPathsByFileId (#116)

Commonjava · Sep 19, 2024 · 659135e · 659135e
1 parent 21a5fb5
commit 659135e
Show file tree

Hide file tree

Showing 4 changed files with 48 additions and 0 deletions.
diff --git a/common/src/main/java/org/commonjava/storage/pathmapped/spi/PathDB.java b/common/src/main/java/org/commonjava/storage/pathmapped/spi/PathDB.java
@@ -15,6 +15,7 @@
  */
 package org.commonjava.storage.pathmapped.spi;
 
+import org.commonjava.storage.pathmapped.model.FileChecksum;
 import org.commonjava.storage.pathmapped.model.PathMap;
 import org.commonjava.storage.pathmapped.model.Reclaim;
 
@@ -26,6 +27,10 @@
 
 public interface PathDB
 {
+    FileChecksum getFileChecksum( String checksum );
+
+    Set<String> getPathsByFileId( String fileId );
+
     enum FileType {
         all, file, dir;
     };

diff --git a/...stax/src/main/java/org/commonjava/storage/pathmapped/pathdb/datastax/CassandraPathDB.java b/...stax/src/main/java/org/commonjava/storage/pathmapped/pathdb/datastax/CassandraPathDB.java
@@ -934,4 +934,21 @@ public void purgeFilesystem( Filesystem filesystem )
             filesystemMapper.delete( filesystem.getFilesystem() );
         }
     }
+
+    @Override
+    public FileChecksum getFileChecksum( String checksum )
+    {
+        return fileChecksumMapper.get( checksum );
+    }
+
+    @Override
+    public Set<String> getPathsByFileId( String fileId )
+    {
+        ReverseMap reverseMap = reverseMapMapper.get( fileId );
+        if ( reverseMap != null )
+        {
+            return reverseMap.getPaths();
+        }
+        return emptySet();
+    }
 }
diff --git a/pathdb/jpa/src/main/java/org/commonjava/storage/pathmapped/pathdb/jpa/JPAPathDB.java b/pathdb/jpa/src/main/java/org/commonjava/storage/pathmapped/pathdb/jpa/JPAPathDB.java
@@ -15,6 +15,7 @@
  */
 package org.commonjava.storage.pathmapped.pathdb.jpa;
 
+import org.commonjava.storage.pathmapped.model.FileChecksum;
 import org.commonjava.storage.pathmapped.pathdb.jpa.model.JpaPathKey;
 import org.commonjava.storage.pathmapped.pathdb.jpa.model.JpaPathMap;
 import org.commonjava.storage.pathmapped.pathdb.jpa.model.JpaReclaim;
@@ -56,6 +57,18 @@ public JPAPathDB( String persistenceUnitName )
         entitymanager = factory.createEntityManager();
     }
 
+    @Override
+    public FileChecksum getFileChecksum( String checksum )
+    {
+        return null;
+    }
+
+    @Override
+    public Set<String> getPathsByFileId( String fileId )
+    {
+        return null;
+    }
+
     public List<PathMap> list( String fileSystem, String path, FileType fileType )
     {
         if ( path.endsWith( "/" ) )

diff --git a/storage/src/main/java/org/commonjava/storage/pathmapped/metrics/MeasuredPathDB.java b/storage/src/main/java/org/commonjava/storage/pathmapped/metrics/MeasuredPathDB.java
@@ -16,6 +16,7 @@
 package org.commonjava.storage.pathmapped.metrics;
 
 import org.commonjava.o11yphant.metrics.MetricsManager;
+import org.commonjava.storage.pathmapped.model.FileChecksum;
 import org.commonjava.storage.pathmapped.model.PathMap;
 import org.commonjava.storage.pathmapped.model.Reclaim;
 import org.commonjava.storage.pathmapped.spi.PathDB;
@@ -45,6 +46,18 @@ public MeasuredPathDB( PathDB decorated, MetricsManager metricsManager, String m
         this.metricPrefix = metricPrefix;
     }
 
+    @Override
+    public FileChecksum getFileChecksum( String checksum )
+    {
+        return measure( () -> decorated.getFileChecksum( checksum ), "getFileChecksum" );
+    }
+
+    @Override
+    public Set<String> getPathsByFileId( String fileId )
+    {
+        return measure( () -> decorated.getPathsByFileId( fileId ), "getPathsByFileId" );
+    }
+
     @Override
     public List<PathMap> list( String fileSystem, String path, FileType fileType )
     {