Merge pull request #53 from scroot/master

pubsub/pub TCP未关闭
ipfs · Jul 3, 2017 · 2da86eb · 2da86eb
2 parents e577e10 + 0666f00
commit 2da86eb
Showing 1 changed file with 9 additions and 3 deletions.
diff --git a/shell.go b/shell.go
@@ -737,11 +737,17 @@ func (s *Shell) PubSubSubscribe(topic string) (*PubSubSubscription, error) {
 	return newPubSubSubscription(resp), nil
 }
 
-func (s *Shell) PubSubPublish(topic, data string) error {
-	_, err := s.newRequest(context.Background(), "pubsub/pub", topic, data).Send(s.httpcli)
+func (s *Shell) PubSubPublish(topic, data string) (err error) {
+	resp, err := s.newRequest(context.Background(), "pubsub/pub", topic, data).Send(s.httpcli)
 	if err != nil {
-		return err
+		return
 	}
+	defer func(){
+		err1 := resp.Close()
+		if err == nil {
+			err = err1
+		}
+	}()
 
 	return nil
 }