fbb: Adopt the recent lzhuf API change

Issue #41
la5nta · Jul 20, 2016 · af158b6 · af158b6
1 parent fe91210
commit af158b6
Showing 1 changed file with 7 additions and 5 deletions.
diff --git a/fbb/proposal.go b/fbb/proposal.go
@@ -120,15 +120,17 @@ func (p *Proposal) Message() (*Message, error) {
 // Data returns the decompressed raw message
 func (p *Proposal) Data() []byte {
 	var r io.ReadCloser
+	var err error
+
 	switch p.code {
 	case GzipProposal:
-		var err error
 		r, err = gzip.NewReader(bytes.NewBuffer(p.compressedData))
-		if err != nil {
-			panic(err) //TODO: Should return error
-		}
 	default:
-		r = lzhuf.NewB2Reader(bytes.NewBuffer(p.compressedData))
+		r, err = lzhuf.NewB2Reader(bytes.NewBuffer(p.compressedData))
+	}
+
+	if err != nil {
+		panic(err) //TODO: Should return error
 	}
 
 	var buf bytes.Buffer