add read super lock api

Signed-off-by: lou <alex1988@outlook.com>
jibudata · Aug 16, 2024 · 0f492e3 · 0f492e3
1 parent a993a43
commit 0f492e3
Show file tree

Hide file tree

Showing 4 changed files with 169 additions and 36 deletions.
diff --git a/weed/storage/backend/udm/api/v1/storage.pb.go b/weed/storage/backend/udm/api/v1/storage.pb.go
diff --git a/weed/storage/backend/udm/api/v1/storage.proto b/weed/storage/backend/udm/api/v1/storage.proto
@@ -16,6 +16,9 @@ service UDMStorage {
   rpc CacheFile (FileKey) returns (CacheFileReply) {
   }
 
+  rpc ReadSuperBlock (FileKey) returns (ReadSuperBlockReply) {
+  }
+
   rpc DeleteFile (FileKey) returns (google.protobuf.Empty) {
   }
 }
@@ -37,3 +40,7 @@ message FileInfo {
 message CacheFileReply {
   string cache_file = 1;
 }
+
+message ReadSuperBlockReply {
+  bytes data = 1;
+}
diff --git a/weed/storage/backend/udm/api/v1/storage_grpc.pb.go b/weed/storage/backend/udm/api/v1/storage_grpc.pb.go
diff --git a/weed/storage/backend/udm/grpc_client.go b/weed/storage/backend/udm/grpc_client.go
@@ -127,6 +127,10 @@ func (cs *ClientSet) DeleteFile(ctx context.Context, key string) error {
 }
 
 func (cs *ClientSet) ReadAt(ctx context.Context, key string, offset, length int64) ([]byte, error) {
+	if isSuperBlock(offset, length) {
+		return cs.ReadSuperBlock(ctx, key)
+	}
+
 	res, err := cs.storageClient.CacheFile(ctx, &pb.FileKey{
 		Key: key,
 	})
@@ -149,3 +153,20 @@ func (cs *ClientSet) ReadAt(ctx context.Context, key string, offset, length int6
 
 	return buffer[:n], nil
 }
+
+func (cs *ClientSet) ReadSuperBlock(ctx context.Context, key string) ([]byte, error) {
+	res, err := cs.storageClient.ReadSuperBlock(ctx, &pb.FileKey{
+		Key: key,
+	})
+	if err != nil {
+		return nil, err
+	}
+
+	return res.Data, nil
+}
+
+const superBlockSize = 8
+
+func isSuperBlock(offset, length int64) bool {
+	return offset == 0 && length == superBlockSize
+}