mirror of https://github.com/status-im/fathom.git
52 lines
1.2 KiB
Go
52 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`)
|
||
|
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)
|
||
|
})
|