swarm: for loop on reader

swarm/api/http/server.go

@@ -574,7 +574,23 @@ func (s *Server) HandleGet(w http.ResponseWriter, r *Request) {
 			contentType = typ
 		}
 		w.Header().Set("Content-Type", contentType)
-		http.ServeContent(w, &r.Request, "", time.Now(), reader)
+
+		var res []byte
+		var err error
+		for {
+			res, err = ioutil.ReadAll(reader)
+			if err != nil {
+				log.Error("handle.get", "ruid", r.ruid, "error", err)
+				time.Sleep(200 * time.Millisecond)
+				continue
+			}
+			log.Debug("handle.get.readall success", "ruid", r.ruid)
+			break
+		}
+
+		rdr := bytes.NewReader(res)
+
+		http.ServeContent(w, &r.Request, "", time.Now(), rdr)
 	case r.uri.Hash():
 		w.Header().Set("Content-Type", "text/plain")
 		w.WriteHeader(http.StatusOK)

swarm/storage/chunker.go

@@ -131,7 +131,7 @@ type TreeChunker struct {
 	The chunks are not meant to be validated by the chunker when joining. This
 	is because it is left to the DPA to decide which sources are trusted.
 */
-func TreeJoin(key Key, getter Getter, depth int) LazySectionReader {
+func TreeJoin(key Key, getter Getter, depth int) *LazyChunkReader {
 	return NewTreeJoiner(NewJoinerParams(key, getter, depth, DefaultChunkSize)).Join()
 }
 

swarm/storage/dpa.go

@@ -84,7 +84,7 @@ func NewDPA(store ChunkStore, params *DPAParams) *DPA {
 // Chunk retrieval blocks on netStore requests with a timeout so reader will
 // report error if retrieval of chunks within requested range time out.
 // It returns a reader with the chunk data and whether the content was encrypted
-func (self *DPA) Retrieve(key Key) (reader LazySectionReader, isEncrypted bool) {
+func (self *DPA) Retrieve(key Key) (reader *LazyChunkReader, isEncrypted bool) {
 	isEncrypted = len(key) > self.hashFunc().Size()
 	getter := NewHasherStore(self.ChunkStore, self.hashFunc, isEncrypted)
 	reader = TreeJoin(key, getter, 0)