From b39cd0afeca6a3dce248d9725c225c8cfd324327 Mon Sep 17 00:00:00 2001
From: Jeromy <why@ipfs.io>
Date: Wed, 17 Aug 2016 18:40:08 -0700
Subject: [PATCH] swarm: make stream.Protocol() return type protocol.ID

---
 swarm_stream.go | 7 ++++---
 1 file changed, 4 insertions(+), 3 deletions(-)

diff --git a/swarm_stream.go b/swarm_stream.go
index 8671981..dc365e8 100644
--- a/swarm_stream.go
+++ b/swarm_stream.go
@@ -2,6 +2,7 @@ package swarm
 
 import (
 	inet "github.com/libp2p/go-libp2p/p2p/net"
+	protocol "github.com/libp2p/go-libp2p/p2p/protocol"
 
 	ps "github.com/jbenet/go-peerstream"
 )
@@ -10,7 +11,7 @@ import (
 // our Conn and Swarm (instead of just the ps.Conn and ps.Swarm)
 type Stream struct {
 	stream   *ps.Stream
-	protocol string
+	protocol protocol.ID
 }
 
 // Stream returns the underlying peerstream.Stream
@@ -44,11 +45,11 @@ func (s *Stream) Close() error {
 	return s.stream.Close()
 }
 
-func (s *Stream) Protocol() string {
+func (s *Stream) Protocol() protocol.ID {
 	return s.protocol
 }
 
-func (s *Stream) SetProtocol(p string) {
+func (s *Stream) SetProtocol(p protocol.ID) {
 	s.protocol = p
 }
 
-- 
GitLab