Skip to content
Snippets Groups Projects
Select Git revision
  • c31c2e5dd238b995d09c62ac6e2a0bab17c97eb1
  • develop default protected
  • congestioncontrol
  • merge-v-data-collection-spammer-0.8.2
  • WIP-merge-v-data-collection-spammer-0.8.2
  • merge-v-data-collection-spammer-0.7.7
  • tmp
  • test-masterpow-fixing
  • test-masterpow
  • test-echo
  • v-data-collection
  • v-data-collection-spammer
  • tmp-dump-spam-info
  • dump-msg-info-0.3.1
  • test-dump-message-info
  • spammer-exprandom
  • extra/tutorial
  • without_tipselection
  • hacking-docker-network
  • hacking-docker-network-0.2.3
  • master
  • v0.2.3
22 results

go.mod

Blame
  • This project manages its dependencies using Go Modules. Learn more
    data_stream.go 2.19 KiB
    package httpserver
    
    import (
        "fmt"
        "github.com/iotaledger/goshimmer/plugins/analysis/server"
        "github.com/iotaledger/goshimmer/plugins/analysis/webinterface/recordedevents"
        "github.com/iotaledger/goshimmer/plugins/analysis/webinterface/types"
        "golang.org/x/net/websocket"
    )
    
    var nodes = make(map[string]bool)
    var links = make(map[string]map[string]bool)
    
    func dataStream(ws *websocket.Conn) {
        eventHandlers := &types.EventHandlers{
            AddNode:         func(nodeId string) {
                if _, exists := nodes[nodeId]; !exists {
                    nodes[nodeId] = true
    
                    fmt.Fprint(ws, "A"+nodeId)
                }
            },
            RemoveNode:      func(nodeId string) {
                if _, exists := nodes[nodeId]; exists {
                    delete(nodes, nodeId)
    
                    fmt.Fprint(ws, "a"+nodeId)
                }
            },
            ConnectNodes:    func(sourceId string, targetId string) { fmt.Fprint(ws, "C"+sourceId+targetId) },
            DisconnectNodes: func(sourceId string, targetId string) { fmt.Fprint(ws, "c"+sourceId+targetId) },
            NodeOnline:      func(nodeId string) { fmt.Fprint(ws, "O"+nodeId) },
            NodeOffline:     func(nodeId string) { fmt.Fprint(ws, "o"+nodeId) },
        }
    
        server.Events.AddNode.Attach(eventHandlers.AddNode)
        server.Events.RemoveNode.Attach(eventHandlers.RemoveNode)
        server.Events.ConnectNodes.Attach(eventHandlers.ConnectNodes)
        server.Events.DisconnectNodes.Attach(eventHandlers.DisconnectNodes)
        server.Events.NodeOnline.Attach(eventHandlers.NodeOnline)
        server.Events.NodeOffline.Attach(eventHandlers.NodeOffline)
    
        go recordedevents.Replay(eventHandlers, 0)
    
        buf := make([]byte, 1)
        readFromWebsocket:
        for {
            if _, err := ws.Read(buf); err != nil {
                break readFromWebsocket
            }
    
            fmt.Fprint(ws, "_")
        }
    
        server.Events.AddNode.Detach(eventHandlers.AddNode)
        server.Events.RemoveNode.Detach(eventHandlers.RemoveNode)
        server.Events.ConnectNodes.Detach(eventHandlers.ConnectNodes)
        server.Events.DisconnectNodes.Detach(eventHandlers.DisconnectNodes)
        server.Events.NodeOnline.Detach(eventHandlers.NodeOnline)
        server.Events.NodeOffline.Detach(eventHandlers.NodeOffline)
    }