Fix stash problem

dsg-uwaterloo · Feb 16, 2024 · e52968c · e52968c
1 parent eef7505
commit e52968c
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/pkg/oramnode/server.go b/pkg/oramnode/server.go
@@ -153,6 +153,9 @@ func (o *oramNodeServer) readAllBuckets(buckets []int, storageID int) (blocksFro
 	if err != nil {
 		return nil, fmt.Errorf("unable to get bucket ids for early reshuffle path; %v", err)
 	}
+	for _, bucket := range buckets {
+		blocksFromReadBucket[bucket] = make(map[string]string)
+	}
 	readBucketResponseChan := make(chan readBucketResponse)
 	batches := distributeBucketIDs(buckets, o.parameters.RedisPipelineSize)
 	for _, bucketIDs := range batches {
@@ -166,9 +169,6 @@ func (o *oramNodeServer) readAllBuckets(buckets []int, storageID int) (blocksFro
 			return nil, fmt.Errorf("unable to read bucket; %s", err)
 		}
 		for bucket, blockValues := range response.bucketValues {
-			if blocksFromReadBucket[bucket] == nil {
-				blocksFromReadBucket[bucket] = make(map[string]string)
-			}
 			for block, value := range blockValues {
 				blocksFromReadBucket[bucket][block] = value
 			}