fix block bug

xitongsys · Nov 11, 2019 · 522c7a9 · 522c7a9
1 parent 86471e0
commit 522c7a9
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 6 deletions.
diff --git a/README.md b/README.md
@@ -1,4 +1,4 @@
-# parquet-go v1.4.2
+# parquet-go v1.4.3
 [![Travis Status for xitongsys/parquet-go](https://travis-ci.org/xitongsys/parquet-go.svg?branch=master&label=linux+build)](https://travis-ci.org/xitongsys/parquet-go)
 [![godoc for xitongsys/parquet-go](https://godoc.org/github.com/nathany/looper?status.svg)](http://godoc.org/github.com/xitongsys/parquet-go)
 

diff --git a/layout/rowgroup.go b/layout/rowgroup.go
@@ -2,11 +2,11 @@ package layout
 
 import (
 	"errors"
-	
+
 	"github.com/xitongsys/parquet-go/common"
-	"github.com/xitongsys/parquet-go/source"
-	"github.com/xitongsys/parquet-go/schema"
 	"github.com/xitongsys/parquet-go/parquet"
+	"github.com/xitongsys/parquet-go/schema"
+	"github.com/xitongsys/parquet-go/source"
 )
 
 //RowGroup stores the RowGroup in parquet file
@@ -77,6 +77,7 @@ func ReadRowGroup(rowGroupHeader *parquet.RowGroup, PFile source.ParquetFile, sc
 					default:
 						err = errors.New("unknown error")
 					}
+					close(doneChan)
 				}
 			}()
 

diff --git a/writer/writer.go b/writer/writer.go
@@ -203,11 +203,12 @@ func (self *ParquetWriter) flushObjs() error {
 					default:
 						err = errors.New("unknown error")
 					}
+					close(doneChan)
 				}
+				doneChan <- 0
 			}()
 
 			if e <= b {
-				doneChan <- 0
 				return
 			}
 
@@ -234,7 +235,6 @@ func (self *ParquetWriter) flushObjs() error {
 				err = err2
 			}
 
-			doneChan <- 0
 		}(int(bgn), int(end), c)
 	}