Print reports and end at 8:10

Author: nbryan

main.go

@@ -19,33 +19,37 @@ func main() {
     tubeStations = LoadTubeStations()
 
     ids := []int{5937, 6043}
-    done := make(chan int)
+    report, done := make(chan string), make(chan int)
 
     for _, id := range ids {
-        go Dispatch(Robot{id, LatLong{0, 0}}, done)
+        go Dispatch(Robot{id, LatLong{0, 0}}, report, done)
     }
 
+    go PrintReports(report)
+
     for i := len(ids); i > 0; {
         <-done
         i--
     }
 }
 
-func Dispatch(r Robot, done chan int) {
-    fmt.Println("Dispatch")
+func Dispatch(r Robot, report chan string, done chan int) {
     ch := make(chan Instruction, 10) // Only allow 10 instructions in queue at a time
     go ReadInstructions(r.ID, ch)
 
     for instruction := range ch {
-        r.ReceiveInstruction(instruction)
-        fmt.Println(r.ID, instruction)
+        hour, min, _ := instruction.Time.Clock()
+        if hour < 8 || (hour == 8 && min < 10) { // End at 8:10
+            r.Instruct(instruction, report)
+        } else {
+            break
+        }
     }
 
     done <- 1
 }
 
 func ReadInstructions(id int, ch chan Instruction) {
-    fmt.Println("ReadInstructions")
     file, _ := os.Open(fmt.Sprintf("%v.csv", id))
     defer file.Close()
 
@@ -66,3 +70,9 @@ func ReadInstructions(id int, ch chan Instruction) {
 
     close(ch)
 }
+
+func PrintReports(report chan string) {
+    for {
+        fmt.Println(<-report)
+    }
+}

robot.go

@@ -1,6 +1,7 @@
 package main
 
 import (
+    "math/rand"
     "time"
     "fmt"
 )
@@ -10,20 +11,41 @@ type Robot struct{
     Location LatLong
 }
 
-func (r *Robot) ReceiveInstruction(instruction Instruction) {
+func (r *Robot) Instruct(instruction Instruction, report chan string) {
     r.Move(instruction.Location)
+    if r.IsNearTheTube() {
+        report <- fmt.Sprintf("%v, %v, %v, %v", r.ID, instruction.Time.Format("15:04:05"), r.Location, traffic())
+    }
 }
 
-const speed int64 = 100000 // Meters per hour (100kph)
+const speed int64 = 500000 // Meters per hour (500kph), pretty quick
 
 func (r *Robot) Move(newLocation LatLong) {
+    // Robots move at constant speed in a staright line
     startingLocation := LatLong{0, 0}
     if r.Location != startingLocation { // Assume it takes no time to move to starting location
         time.Sleep(time.Duration(int64(r.Location.DistanceFrom(newLocation)) * int64(time.Hour) / speed))
     }
     r.Location = newLocation
 }
 
-func (r *Robot) IsNearTheTube(tubeStations []TubeStation) bool {
-    return true
+func (r *Robot) IsNearTheTube() bool {
+    for _, station := range tubeStations { // Seems like a dreadful way to search for a nearby station
+        if r.Location.DistanceFrom(station.Location) <= 350.0 {
+            return true
+        }
+    }
+    return false
+}
+
+func traffic() string {
+    switch rand.Intn(3) {
+    case 0:
+        return "HEAVY"
+    case 1:
+        return "MODERATE"
+    case 2:
+        return "LIGHT"
+    }
+    return ""
 }