You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

42 lines
1.0 KiB

  1. package main
  2. import (
  3. "fmt"
  4. "log"
  5. "net/http"
  6. "os"
  7. "github.com/fatih/color"
  8. "github.com/gorilla/handlers"
  9. mgo "gopkg.in/mgo.v2"
  10. )
  11. var cellCollection *mgo.Collection
  12. func main() {
  13. //connect with mongodb
  14. readMongodbConfig("./mongodbConfig.json")
  15. session, err := getSession()
  16. check(err)
  17. cellCollection = getCollection(session, "cells")
  18. if len(os.Args) > 1 {
  19. if os.Args[1] == "-dataset" {
  20. color.Blue("starting to read dataset")
  21. readDataset("cell_towers.csv")
  22. color.Blue("finished reading dataset")
  23. }
  24. }
  25. //http server start
  26. readServerConfig("./serverConfig.json")
  27. color.Green("server running")
  28. fmt.Print("port: ")
  29. color.Green(serverConfig.ServerPort)
  30. router := NewRouter()
  31. headersOk := handlers.AllowedHeaders([]string{"X-Requested-With", "Access-Control-Allow-Origin"})
  32. originsOk := handlers.AllowedOrigins([]string{"*"})
  33. methodsOk := handlers.AllowedMethods([]string{"GET", "HEAD", "POST", "PUT", "OPTIONS"})
  34. log.Fatal(http.ListenAndServe(":"+serverConfig.ServerPort, handlers.CORS(originsOk, headersOk, methodsOk)(router)))
  35. }