core/commands: Fixed net diag command not outputting in chosen format

2c019b54 · Matt Bell · Juan Batiz-Benet · ae5e2443 · 2c019b54
Commit 2c019b54 authored Jan 24, 2015 by Matt Bell Committed by Juan Batiz-Benet Jan 26, 2015
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 0 deletions

core/commands/diag.go core/commands/diag.go +2 -0

No files found.
--- a/core/commands/diag.go
+++ b/core/commands/diag.go
@@ -106,6 +106,7 @@ connected peers and latencies between them.
 		switch vis {
 		case visD3:
 			res.SetOutput(bytes.NewReader(diag.GetGraphJson(info)))
+			return
 		case visDot:
 			var buf bytes.Buffer
 			w := diag.DotWriter{W: &buf}
@@ -115,6 +116,7 @@ connected peers and latencies between them.
 				return
 			}
 			res.SetOutput(io.Reader(&buf))
+			return
 		}

 		output, err := stdDiagOutputMarshal(standardDiagOutput(info))