Forráskód Böngészése

:gear: be consistent and consume context while opening stream

Ettore Di Giacinto 3 éve
szülő
commit
516904fd41
2 módosított fájl, 2 hozzáadás és 2 törlés
  1. 1 1
      pkg/services/files.go
  2. 1 1
      pkg/services/services.go

+ 1 - 1
pkg/services/files.go

@@ -156,7 +156,7 @@ func ReceiveFile(ctx context.Context, ledger *blockchain.Ledger, n *node.Node, l
 				}
 
 				// Open a stream
-				stream, err := n.Host().NewStream(context.Background(), d, protocol.FileProtocol.ID())
+				stream, err := n.Host().NewStream(ctx, d, protocol.FileProtocol.ID())
 				if err != nil {
 					l.Debugf("failed to dial %s, retrying in 5 seconds", d)
 					continue

+ 1 - 1
pkg/services/services.go

@@ -170,8 +170,8 @@ func ConnectToService(ctx context.Context, ledger *blockchain.Ledger, node *node
 				go copyStream(closer, conn, stream)
 				<-closer
 
-				conn.Close()
 				stream.Close()
+				conn.Close()
 				ll.Infof("(service %s) Done handling %s", serviceID, l.Addr().String())
 			}()
 		}