Fixes for multiple Pinecone peers

2 files changed, 6 insertions, 3 deletions

diff --git a/cmd/dendrite-demo-pinecone/conn/client.go b/cmd/dendrite-demo-pinecone/conn/client.go
index 14b648a3..e3cc0468 100644
--- a/cmd/dendrite-demo-pinecone/conn/client.go
+++ b/cmd/dendrite-demo-pinecone/conn/client.go
@@ -37,7 +37,7 @@ func ConnectToPeer(pRouter *pineconeRouter.Router, peer string) error {
 	_, err := pRouter.Connect(
 		parent,
 		pineconeRouter.ConnectionZone("static"),
-		pineconeRouter.PeerTypeRemote,
+		pineconeRouter.ConnectionPeerType(pineconeRouter.PeerTypeRemote),
 		pineconeRouter.ConnectionURI(peer),
 	)
 	return err
diff --git a/cmd/dendrite-demo-pinecone/main.go b/cmd/dendrite-demo-pinecone/main.go
index 8ed3f349..cc7238ea 100644
--- a/cmd/dendrite-demo-pinecone/main.go
+++ b/cmd/dendrite-demo-pinecone/main.go
@@ -110,7 +110,10 @@ func main() {
 				continue
 			}
 
-			port, err := pRouter.Connect(conn, pineconeRouter.PeerTypeRemote)
+			port, err := pRouter.Connect(
+				conn,
+				pineconeRouter.ConnectionPeerType(pineconeRouter.PeerTypeRemote),
+			)
 			if err != nil {
 				logrus.WithError(err).Error("pSwitch.Connect failed")
 				continue
@@ -242,7 +245,7 @@ func main() {
 		if _, err = pRouter.Connect(
 			conn,
 			pineconeRouter.ConnectionZone("websocket"),
-			pineconeRouter.PeerTypeRemote,
+			pineconeRouter.ConnectionPeerType(pineconeRouter.PeerTypeRemote),
 		); err != nil {
 			logrus.WithError(err).Error("Failed to connect WebSocket peer to Pinecone switch")
 		}