Serge Bazanski | 8d7843c | 2018-10-04 10:37:36 +0100 | [diff] [blame] | 1 | package main |
| 2 | |
Serge Bazanski | 4676508 | 2018-10-06 12:32:01 +0100 | [diff] [blame] | 3 | import ( |
Serge Bazanski | c7be4a1 | 2018-10-06 13:18:05 +0100 | [diff] [blame] | 4 | "context" |
| 5 | "fmt" |
Serge Bazanski | 4676508 | 2018-10-06 12:32:01 +0100 | [diff] [blame] | 6 | "net/http" |
Serge Bazanski | 69518ab | 2018-10-06 17:48:23 +0100 | [diff] [blame] | 7 | "sort" |
Serge Bazanski | de869df | 2018-10-06 13:55:49 +0100 | [diff] [blame] | 8 | "strings" |
Serge Bazanski | 8d7843c | 2018-10-04 10:37:36 +0100 | [diff] [blame] | 9 | |
Serge Bazanski | c7be4a1 | 2018-10-06 13:18:05 +0100 | [diff] [blame] | 10 | "github.com/gobuffalo/packr" |
Serge Bazanski | 4676508 | 2018-10-06 12:32:01 +0100 | [diff] [blame] | 11 | "github.com/golang/glog" |
Serge Bazanski | c7be4a1 | 2018-10-06 13:18:05 +0100 | [diff] [blame] | 12 | "github.com/q3k/statusz" |
Serge Bazanski | 69518ab | 2018-10-06 17:48:23 +0100 | [diff] [blame] | 13 | "vbom.ml/util/sortorder" |
Serge Bazanski | a758ef5 | 2018-10-06 17:54:25 +0100 | [diff] [blame^] | 14 | |
| 15 | "code.hackerspace.pl/q3k/topo/graph" |
| 16 | "code.hackerspace.pl/q3k/topo/state" |
Serge Bazanski | 4676508 | 2018-10-06 12:32:01 +0100 | [diff] [blame] | 17 | ) |
| 18 | |
| 19 | type ServiceConfig struct { |
| 20 | DebugListen string |
| 21 | } |
| 22 | |
| 23 | type Service struct { |
| 24 | gr *graph.Graph |
Serge Bazanski | a758ef5 | 2018-10-06 17:54:25 +0100 | [diff] [blame^] | 25 | stm *state.StateManager |
Serge Bazanski | 4676508 | 2018-10-06 12:32:01 +0100 | [diff] [blame] | 26 | config ServiceConfig |
| 27 | } |
| 28 | |
Serge Bazanski | a758ef5 | 2018-10-06 17:54:25 +0100 | [diff] [blame^] | 29 | func NewService(gr *graph.Graph, stm *state.StateManager, c ServiceConfig) *Service { |
Serge Bazanski | 4676508 | 2018-10-06 12:32:01 +0100 | [diff] [blame] | 30 | return &Service{ |
| 31 | gr: gr, |
Serge Bazanski | a758ef5 | 2018-10-06 17:54:25 +0100 | [diff] [blame^] | 32 | stm: stm, |
Serge Bazanski | 4676508 | 2018-10-06 12:32:01 +0100 | [diff] [blame] | 33 | config: c, |
| 34 | } |
| 35 | } |
| 36 | |
Serge Bazanski | c7be4a1 | 2018-10-06 13:18:05 +0100 | [diff] [blame] | 37 | type TopologyStatus struct { |
| 38 | } |
| 39 | |
| 40 | const topologyFragment = ` |
| 41 | <script src="/assets/viz.js"></script> |
| 42 | <script> |
| 43 | |
| 44 | var viz = new Viz({ workerURL: '/assets/full.render.js' }); |
| 45 | var xmlhttp = new XMLHttpRequest(); |
| 46 | xmlhttp.onreadystatechange = function() { |
| 47 | if (this.readyState == 4 && this.status == 200) { |
| 48 | var dot = this.responseText; |
| 49 | viz.renderSVGElement(dot) |
| 50 | .then(function(element) { |
Serge Bazanski | 69518ab | 2018-10-06 17:48:23 +0100 | [diff] [blame] | 51 | document.getElementById("graph").appendChild(element); |
Serge Bazanski | c7be4a1 | 2018-10-06 13:18:05 +0100 | [diff] [blame] | 52 | }); |
| 53 | } |
| 54 | }; |
| 55 | xmlhttp.open('GET', '/debug/graphviz'); |
| 56 | xmlhttp.send(); |
| 57 | |
| 58 | </script> |
Serge Bazanski | 69518ab | 2018-10-06 17:48:23 +0100 | [diff] [blame] | 59 | <div id="graph" style="text-align: center;"></div> |
Serge Bazanski | c7be4a1 | 2018-10-06 13:18:05 +0100 | [diff] [blame] | 60 | ` |
| 61 | |
Serge Bazanski | 4676508 | 2018-10-06 12:32:01 +0100 | [diff] [blame] | 62 | func (s *Service) Run() { |
Serge Bazanski | c7be4a1 | 2018-10-06 13:18:05 +0100 | [diff] [blame] | 63 | assets := packr.NewBox("./assets") |
| 64 | http.Handle("/assets/", http.StripPrefix("/assets/", http.FileServer(assets))) |
| 65 | |
Serge Bazanski | 4676508 | 2018-10-06 12:32:01 +0100 | [diff] [blame] | 66 | http.HandleFunc("/", func(w http.ResponseWriter, r *http.Request) { |
| 67 | http.Redirect(w, r, "/debug/status", http.StatusSeeOther) |
| 68 | }) |
Serge Bazanski | c7be4a1 | 2018-10-06 13:18:05 +0100 | [diff] [blame] | 69 | |
| 70 | http.HandleFunc("/debug/graphviz", func(w http.ResponseWriter, r *http.Request) { |
| 71 | fmt.Fprintf(w, "graph G {\n") |
Serge Bazanski | 69518ab | 2018-10-06 17:48:23 +0100 | [diff] [blame] | 72 | fmt.Fprintf(w, " ranksep = 2\n") |
| 73 | fmt.Fprintf(w, " splines = polyline\n") |
Serge Bazanski | de869df | 2018-10-06 13:55:49 +0100 | [diff] [blame] | 74 | fmt.Fprintf(w, " rankdir = LR\n") |
Serge Bazanski | c7be4a1 | 2018-10-06 13:18:05 +0100 | [diff] [blame] | 75 | for _, machine := range s.gr.Machines { |
Serge Bazanski | de869df | 2018-10-06 13:55:49 +0100 | [diff] [blame] | 76 | portNames := []string{} |
Serge Bazanski | c7be4a1 | 2018-10-06 13:18:05 +0100 | [diff] [blame] | 77 | for _, port := range machine.Ports { |
Serge Bazanski | de869df | 2018-10-06 13:55:49 +0100 | [diff] [blame] | 78 | name := fmt.Sprintf("<%s> %s", port.Name, port.Name) |
| 79 | portNames = append(portNames, name) |
Serge Bazanski | c7be4a1 | 2018-10-06 13:18:05 +0100 | [diff] [blame] | 80 | } |
Serge Bazanski | de869df | 2018-10-06 13:55:49 +0100 | [diff] [blame] | 81 | ports := strings.Join(portNames, "|") |
| 82 | fmt.Fprintf(w, " %s [shape=record label=\"{ %s | { %s }}\"]\n", machine.Name, machine.Name, ports) |
Serge Bazanski | c7be4a1 | 2018-10-06 13:18:05 +0100 | [diff] [blame] | 83 | } |
| 84 | for _, sw := range s.gr.Switches { |
Serge Bazanski | de869df | 2018-10-06 13:55:49 +0100 | [diff] [blame] | 85 | portNames := []string{} |
Serge Bazanski | 69518ab | 2018-10-06 17:48:23 +0100 | [diff] [blame] | 86 | portsOrdered := []*graph.SwitchPort{} |
Serge Bazanski | c7be4a1 | 2018-10-06 13:18:05 +0100 | [diff] [blame] | 87 | for _, port := range sw.Ports { |
Serge Bazanski | 69518ab | 2018-10-06 17:48:23 +0100 | [diff] [blame] | 88 | portsOrdered = append(portsOrdered, port) |
| 89 | } |
| 90 | sort.Slice(portsOrdered, func(i, j int) bool { |
| 91 | return sortorder.NaturalLess(portsOrdered[i].Name, portsOrdered[j].Name) |
| 92 | }) |
| 93 | for _, port := range portsOrdered { |
Serge Bazanski | de869df | 2018-10-06 13:55:49 +0100 | [diff] [blame] | 94 | name := fmt.Sprintf("<%s> %s", port.Name, port.Name) |
| 95 | portNames = append(portNames, name) |
Serge Bazanski | c7be4a1 | 2018-10-06 13:18:05 +0100 | [diff] [blame] | 96 | } |
Serge Bazanski | de869df | 2018-10-06 13:55:49 +0100 | [diff] [blame] | 97 | ports := strings.Join(portNames, "|") |
| 98 | fmt.Fprintf(w, " %s [shape=record label=\"{{ %s } | %s}\"]\n", sw.Name, ports, sw.Name) |
Serge Bazanski | c7be4a1 | 2018-10-06 13:18:05 +0100 | [diff] [blame] | 99 | } |
| 100 | for _, machine := range s.gr.Machines { |
| 101 | for _, port := range machine.Ports { |
| 102 | if port.OtherEnd == nil { |
| 103 | continue |
| 104 | } |
Serge Bazanski | 69518ab | 2018-10-06 17:48:23 +0100 | [diff] [blame] | 105 | fmt.Fprintf(w, " %s:%q:e -- %s:%q:w\n", machine.Name, port.Name, port.OtherEnd.Switch.Name, port.OtherEnd.Name) |
Serge Bazanski | c7be4a1 | 2018-10-06 13:18:05 +0100 | [diff] [blame] | 106 | } |
| 107 | } |
| 108 | fmt.Fprintf(w, "}\n") |
| 109 | }) |
| 110 | |
| 111 | statusz.AddStatusPart("Topology", topologyFragment, func(ctx context.Context) interface{} { |
| 112 | return &TopologyStatus{} |
| 113 | }) |
Serge Bazanski | 4676508 | 2018-10-06 12:32:01 +0100 | [diff] [blame] | 114 | glog.Infof("Debug listening on %s....", s.config.DebugListen) |
| 115 | http.ListenAndServe(s.config.DebugListen, nil) |
Serge Bazanski | 8d7843c | 2018-10-04 10:37:36 +0100 | [diff] [blame] | 116 | } |