Beispiel #1
0
func main() {
	var historyResult HistoryResult
	var waitGroup sync.WaitGroup
	startTime := time.Now()
	// Load config
	cfg, err := config.GetConfigData()
	if err != nil {
		log.Fatal(err)
	}
	// Connect to DB
	db, err := sql.Open("mysql", fmt.Sprintf("%s:%s@tcp(%s)/%s", cfg.DB.User, cfg.DB.Password, cfg.DB.Host, cfg.DB.Database))
	if err != nil {
		log.Fatal(err)
	}
	defer db.Close()
	// Request server-list from lists.sa-mp.com
	log.Println("Download server-list from lists.sa-mp.com.")
	var serversAddr []string
	var currentAddr int
	request := serverlist.NewRequest(cfg.Updater.Version)
	if err := request.Exec(); err != nil {
		log.Fatal("Error (request.Exec()):", err)
	}
	serversAddr, err = request.ReadAll()
	if err != nil {
		log.Fatal("Error (request.ReadAll()):", err)
	}
	// Clear database table
	if err := clearServersInDB(db); err != nil {
		log.Fatal("Error (clearServersInDB()):", err)
	}
	// Create threads
	waitGroup.Add(maxQueryThreads)
	for i := 0; i < maxQueryThreads; i++ {
		go processLine(&serversAddr, &currentAddr, &historyResult, db, &waitGroup)
	}
	// Wait threads
	waitGroup.Wait()
	// Write history to database
	writeHistoryToDB(db, &historyResult)
	// Print results
	log.Println("Execution time:", time.Since(startTime))
	log.Println("Servers offline:", historyResult.ServersOnline, "of", historyResult.ServersTotal)
	log.Println("Slots used:", historyResult.SlotsUsed, "of", historyResult.SlotsTotal)
}
Beispiel #2
0
func main() {
	// Load config
	cfg, err := config.GetConfigData()
	if err != nil {
		log.Fatal(err)
	}
	// Initialize database
	db, err := sql.Open("mysql", fmt.Sprintf("%s:%s@tcp(%s)/%s", cfg.DB.User, cfg.DB.Password, cfg.DB.Host, cfg.DB.Database))
	if err != nil {
		log.Fatal(err)
	}
	defer db.Close()

	// Initialize HTTP-server
	router := mux.NewRouter()
	router.HandleFunc("/api/general", func(w http.ResponseWriter, r *http.Request) {
		data, err := GetGeneralData(db)
		if err != nil {
			response.JSONResponseError(&w, http.StatusInternalServerError, err.Error())
			return
		}
		response.JSONResponse(&w, http.StatusOK, data)
	})

	router.HandleFunc("/api/history/servers/{range}", func(w http.ResponseWriter, r *http.Request) {
		var days int
		vars := mux.Vars(r)
		requestRange := vars["range"]
		switch requestRange {
		case "day":
			days = 1
		case "week":
			days = 7
		case "month":
			days = 30
		case "year":
			days = 365
		default:
			response.JSONResponseError(&w, http.StatusInternalServerError, "Incorrect range")
			return
		}
		data, err := GetHistoryServers(db, days)
		if err != nil {
			response.JSONResponseError(&w, http.StatusInternalServerError, err.Error())
			return
		}
		response.JSONResponse(&w, http.StatusOK, data)
	})

	router.HandleFunc("/api/history/players/{range}", func(w http.ResponseWriter, r *http.Request) {
		var days int
		vars := mux.Vars(r)
		requestRange := vars["range"]
		switch requestRange {
		case "day":
			days = 1
		case "week":
			days = 7
		case "month":
			days = 30
		case "year":
			days = 365
		default:
			response.JSONResponseError(&w, http.StatusInternalServerError, "Incorrect range")
			return
		}
		data, err := GetHistoryPlayers(db, days)
		if err != nil {
			response.JSONResponseError(&w, http.StatusInternalServerError, err.Error())
			return
		}
		response.JSONResponse(&w, http.StatusOK, data)
	})

	router.PathPrefix("/").Handler(http.FileServer(http.Dir("./public")))
	http.Handle("/", router)
	log.Println(fmt.Sprintf("Listening %s...", cfg.Main.Listen))
	if err := http.ListenAndServe(cfg.Main.Listen, nil); err != nil {
		log.Fatal("ListenAndServe: ", err)
	}
}