mirror of https://github.com/status-im/fathom.git
53 lines
1.2 KiB
Go
53 lines
1.2 KiB
Go
package api
|
|
|
|
import (
|
|
"net/http"
|
|
"github.com/dannyvankooten/ana/core"
|
|
"encoding/json"
|
|
)
|
|
|
|
// URL: /api/countries
|
|
var GetCountriesHandler = http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
|
period := getRequestedPeriod(r)
|
|
|
|
// get total
|
|
stmt, err := core.DB.Prepare(`
|
|
SELECT
|
|
COUNT(DISTINCT(ip_address))
|
|
FROM visits
|
|
WHERE timestamp >= DATE_SUB(CURRENT_TIMESTAMP, INTERVAL ? DAY)`)
|
|
checkError(err)
|
|
defer stmt.Close()
|
|
var total float32
|
|
stmt.QueryRow(period).Scan(&total)
|
|
|
|
// get rows
|
|
stmt, err = core.DB.Prepare(`
|
|
SELECT
|
|
country,
|
|
COUNT(DISTINCT(ip_address)) AS count
|
|
FROM visits
|
|
WHERE timestamp >= DATE_SUB(CURRENT_TIMESTAMP, INTERVAL ? DAY) AND country IS NOT NULL
|
|
GROUP BY country
|
|
ORDER BY count DESC
|
|
LIMIT 10`)
|
|
checkError(err)
|
|
defer stmt.Close()
|
|
rows, err := stmt.Query(period)
|
|
checkError(err)
|
|
defer rows.Close()
|
|
|
|
results := make([]Datapoint, 0)
|
|
for rows.Next() {
|
|
var d Datapoint
|
|
err = rows.Scan(&d.Label, &d.Count);
|
|
checkError(err)
|
|
|
|
d.Percentage = float32(d.Count) / total * 100
|
|
results = append(results, d)
|
|
}
|
|
|
|
w.Header().Set("Content-Type", "application/json")
|
|
json.NewEncoder(w).Encode(results)
|
|
})
|