-
-
Save avinassh/5d476018db6d351302cb27ca2a12814d to your computer and use it in GitHub Desktop.
gossip
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
package main | |
import ( | |
"encoding/json" | |
"log" | |
"os" | |
"sync" | |
maelstrom "github.com/jepsen-io/maelstrom/demo/go" | |
) | |
type State struct { | |
messages []int | |
nodeId string | |
nodeIds []string | |
*sync.Mutex | |
} | |
func NewState() *State { | |
return &State{ | |
messages: make([]int, 0), | |
Mutex: &sync.Mutex{}, | |
} | |
} | |
func (s *State) addMsg(msg int) { | |
s.Lock() | |
defer s.Unlock() | |
s.messages = append(s.messages, msg) | |
} | |
func (s *State) addTopo(nodes []string) { | |
s.Lock() | |
defer s.Unlock() | |
s.nodeIds = nodes | |
} | |
func main() { | |
l := log.New(os.Stderr, "", 0) | |
l.Prefix() | |
n := maelstrom.NewNode() | |
state := NewState() | |
n.Handle("broadcast", func(msg maelstrom.Message) error { | |
type broadcastMsg struct { | |
Type string `json:"type"` | |
Message int `json:"message,omitempty"` | |
} | |
var body broadcastMsg | |
if err := json.Unmarshal(msg.Body, &body); err != nil { | |
return err | |
} | |
state.addMsg(body.Message) | |
for _, id := range state.nodeIds { | |
if id == state.nodeId { | |
continue | |
} | |
if id == msg.Dest { | |
continue | |
} | |
if err := n.Send(id, msg.Body); err != nil { | |
panic(err) | |
} | |
} | |
body.Message = 0 | |
body.Type = "broadcast_ok" | |
return n.Reply(msg, body) | |
}) | |
n.Handle("read", func(msg maelstrom.Message) error { | |
var body map[string]any | |
if err := json.Unmarshal(msg.Body, &body); err != nil { | |
return err | |
} | |
body["type"] = "read_ok" | |
body["messages"] = state.messages | |
return n.Reply(msg, body) | |
}) | |
n.Handle("topology", func(msg maelstrom.Message) error { | |
type topologyMSG struct { | |
Type string `json:"type"` | |
Topology map[string][]string `json:"topology,omitempty"` | |
} | |
var body topologyMSG | |
if err := json.Unmarshal(msg.Body, &body); err != nil { | |
return err | |
} | |
state.addTopo(body.Topology[state.nodeId]) | |
body.Type = "topology_ok" | |
body.Topology = nil | |
return n.Reply(msg, body) | |
}) | |
n.Handle("init", func(msg maelstrom.Message) error { | |
type initBody struct { | |
Type string `json:"type"` | |
MsgId int `json:"msg_id"` | |
NodeId string `json:"node_id"` | |
NodeIds []string `json:"node_ids"` | |
} | |
var body initBody | |
if err := json.Unmarshal(msg.Body, &body); err != nil { | |
return err | |
} | |
body.Type = "init_ok" | |
state.nodeId = body.NodeId | |
state.nodeIds = body.NodeIds | |
return n.Reply(msg, body) | |
}) | |
if err := n.Run(); err != nil { | |
log.Printf("ERROR: %s", err) | |
os.Exit(1) | |
} | |
} |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment