mirrors/eclair
1
0
Fork 0
mirror of https://github.com/ACINQ/eclair.git synced 2025-02-23 06:35:11 +01:00
Code Issues Projects Releases Wiki Activity

dot graph are now exported as a string

Browse source
This commit is contained in:
pm47 2017-02-03 19:46:18 +01:00
parent 3a3b8c6eb1
commit ea3480b9f8
1 changed files with 6 additions and 8 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

14
eclair-node/src/main/scala/fr/acinq/eclair/router/Router.scala
View file

@ -1,12 +1,12 @@
package fr.acinq.eclair.router
import java.io.{ByteArrayOutputStream, OutputStreamWriter}
import java.io.{BufferedWriter, StringWriter}
import akka.actor.{Actor, ActorLogging, ActorRef, Props}
import akka.pattern.pipe
import fr.acinq.bitcoin.BinaryData
import fr.acinq.bitcoin.Crypto.PublicKey
import fr.acinq.bitcoin.Script.{pay2wsh, write}
import fr.acinq.bitcoin.{BinaryData, Transaction}
import fr.acinq.eclair._
import fr.acinq.eclair.blockchain.{GetTx, GetTxResponse, WatchEventSpent, WatchSpent}
import fr.acinq.eclair.channel._
@ -209,7 +209,7 @@ object Router {
.map(desc => Hop(desc.a, desc.b, updates(desc)))
}
def graph2dot(nodes: Map[BinaryData, NodeAnnouncement], channels: Map[Long, ChannelAnnouncement])(implicit ec: ExecutionContext): Future[BinaryData] = Future {
def graph2dot(nodes: Map[BinaryData, NodeAnnouncement], channels: Map[Long, ChannelAnnouncement])(implicit ec: ExecutionContext): Future[String] = Future {
case class DescEdge(channelId: Long) extends DefaultEdge
val g = new SimpleGraph[BinaryData, DescEdge](classOf[DescEdge])
channels.foreach(d => {
@ -218,7 +218,7 @@ object Router {
g.addEdge(d._2.nodeId1, d._2.nodeId2, new DescEdge(d._1))
})
val vertexIDProvider = new ComponentNameProvider[BinaryData]() {
override def getName(nodeId: BinaryData): String = "\"" + nodeId.toString() + "\""
override def getName(nodeId: BinaryData): String = "\"" + nodeId.toString() + "\""
}
val edgeLabelProvider = new ComponentNameProvider[DescEdge]() {
override def getName(e: DescEdge): String = e.channelId.toString
@ -233,14 +233,12 @@ object Router {
}
}
val exporter = new DOTExporter[BinaryData, DescEdge](vertexIDProvider, null, edgeLabelProvider, vertexAttributeProvider, null)
val bos = new ByteArrayOutputStream()
val writer = new OutputStreamWriter(bos)
val writer = new BufferedWriter(new StringWriter())
try {
exporter.exportGraph(g, writer)
bos.toByteArray
writer.toString
} finally {
writer.close()
bos.close()
}
}