diff options
| -rw-r--r-- | main.go | 47 |
1 files changed, 41 insertions, 6 deletions
diff --git a/main.go b/main.go index 36b8c7e..f5f395d 100644 --- a/main.go +++ b/main.go @@ -17,15 +17,16 @@ import ( var ( metrics map[string]float64 - mutex = &sync.Mutex{} - starBufferChannel = make(chan stargalaxy) + mutex = &sync.Mutex{} + starBufferChannel = make(chan structs.Stargalaxy, 1000000) + currentStarBuffer int64 = 0 ) // struct bundling the star and the galaxy index it comes from -type stargalaxy struct { - star structs.Star2D - index int -} +// type stargalaxy struct { +// star structs.Star2D +// index int64 +// } // indexHandler handles incomming requests on the / endpoint func indexHandler(w http.ResponseWriter, r *http.Request) { @@ -230,6 +231,38 @@ func getstarHandler(w http.ResponseWriter, r *http.Request) { _, _ = fmt.Fprintf(w, "%v", star) } +func fillStarBufferChannel() { + fmt.Println("Filling the star buffer channel") + + // get the list of stars using the currentStarBuffer value + // the currentStarBuffer value is a counter keeping track of which galaxy is going to + // be inserted into the StarBufferChannel next + listofstars := listofstars(currentStarBuffer) + + for _, star := range *listofstars { + starBufferChannel <- structs.Stargalaxy{star, currentStarBuffer} + } + + // increase the currentStarBuffer counter + currentStarBuffer += 1 +} + +func providestarsHandler(w http.ResponseWriter, r *http.Request) { + fmt.Println("Providing stars from the StarBufferHandler") + + // if there are no stars in the star buffer channel, fill the star buffer channel width stars + if len(starBufferChannel) == 0 { + fillStarBufferChannel() + } + + stargalaxy := <-starBufferChannel + marshaledStargalaxy, _ := json.Marshal(stargalaxy) + + _, _ = fmt.Fprintf(w, "%v", string(marshaledStargalaxy)) + + fmt.Println("Done providing stars from the StarBufferHandler") +} + func main() { router := mux.NewRouter() @@ -240,6 +273,8 @@ func main() { router.HandleFunc("/metrics", metricHandler).Methods("GET", "POST") router.HandleFunc("/getstar", getstarHandler).Methods("GET") + router.HandleFunc("/providestars/{treeindex}", providestarsHandler).Methods("GET") + fmt.Println("Manager Container up") log.Fatal(http.ListenAndServe(":80", router)) } |
