diff --git a/main.go b/main.go index a0ed004..1cef70a 100644 --- a/main.go +++ b/main.go @@ -45,6 +45,7 @@ func handleConnection(conn *Connection) { log_error("player %s failed to get random planet: %v", conn.PlayerName(), err) return } + planet.Arrive(conn) if planet.planets == 1 { fmt.Fprintf(conn, "you are in the system %s. There is %d planet here.\n", planet.name, planet.planets) } else { @@ -68,8 +69,8 @@ func handleConnection(conn *Connection) { if otherPlanet.name == planet.name { continue } - go func(p System) { - dist := planetDistance(*planet, p) + go func(p *System) { + dist := planetDistance(*planet, *p) delay := time.Duration(int64(dist * 100000000)) time.Sleep(delay) mu.Lock() @@ -78,7 +79,23 @@ func handleConnection(conn *Connection) { }(otherPlanet) } case "broadcast": - + msg := strings.Join(parts[1:], " ") + log_info("player %s is broadcasting message %s", conn.PlayerName(), msg) + for _, otherSystem := range index { + if otherSystem.name == planet.name { + log_info("skpping duplicate system %s", planet.name) + continue + } + go func(s *System) { + log_info("message reached planet %s with %d inhabitants", s.name, s.NumInhabitants()) + dist := planetDistance(*planet, *s) * 0.5 + delay := time.Duration(int64(dist * 100000000)) + time.Sleep(delay) + s.EachConn(func(conn *Connection) { + fmt.Fprintln(conn, msg) + }) + }(otherSystem) + } case "quit": return default: diff --git a/system.go b/system.go index b6ec45d..1d0e6a2 100644 --- a/system.go +++ b/system.go @@ -8,13 +8,43 @@ import ( ) var ( - index map[int]System + index map[int]*System ) type System struct { x, y, z float64 planets int name string + + players map[*Connection]bool +} + +func (s *System) Arrive(p *Connection) { + log_info("player %s has arrived at system %s", p.PlayerName(), s.name) + if s.players == nil { + s.players = make(map[*Connection]bool, 8) + } + s.players[p] = true +} + +func (s *System) Leave(p *Connection) { + delete(s.players, p) +} + +func (s *System) EachConn(fn func(*Connection)) { + if s.players == nil { + return + } + for conn, _ := range s.players { + fn(conn) + } +} + +func (s *System) NumInhabitants() int { + if s.players == nil { + return 0 + } + return len(s.players) } func (e System) Store(db *sql.DB) { @@ -53,14 +83,14 @@ func planetDistance(p1, p2 System) float64 { return dist3d(p1.x, p1.y, p1.z, p2.x, p2.y, p2.z) } -func indexPlanets(db *sql.DB) map[int]System { +func indexPlanets(db *sql.DB) map[int]*System { rows, err := db.Query(`select * from planets`) if err != nil { log_error("unable to select all planets: %v", err) return nil } defer rows.Close() - index = make(map[int]System, 551) + index = make(map[int]*System, 551) for rows.Next() { var id int p := System{} @@ -68,7 +98,7 @@ func indexPlanets(db *sql.DB) map[int]System { log_info("unable to scan planet row: %v", err) continue } - index[id] = p + index[id] = &p } return index } @@ -81,5 +111,5 @@ func randomPlanet() (*System, error) { pick := rand.Intn(n) planet := index[pick] - return &planet, nil + return planet, nil }