mirror of https://github.com/status-im/fathom.git
100 lines
2.9 KiB
Go
100 lines
2.9 KiB
Go
package api
|
|
|
|
import (
|
|
"net/http"
|
|
"github.com/dannyvankooten/ana/models"
|
|
"github.com/dannyvankooten/ana/core"
|
|
"encoding/json"
|
|
)
|
|
|
|
// URL: /api/visits
|
|
var GetVisitsHandler = http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
|
stmt, err := core.DB.Prepare(`SELECT
|
|
id,
|
|
COALESCE(browser_name, '') AS browser_name,
|
|
COALESCE(browser_language, '') AS browser_language,
|
|
COALESCE(device_os, '') AS device_os,
|
|
ip_address,
|
|
path,
|
|
COALESCE(screen_resolution, '') AS screen_resolution,
|
|
timestamp
|
|
FROM visits
|
|
ORDER BY timestamp DESC
|
|
LIMIT ?`)
|
|
|
|
checkError(err)
|
|
defer stmt.Close()
|
|
|
|
limit := getRequestedLimit(r)
|
|
rows, err := stmt.Query(&limit)
|
|
checkError(err)
|
|
|
|
results := make([]models.Visit, 0)
|
|
defer rows.Close()
|
|
for rows.Next() {
|
|
var v models.Visit
|
|
err = rows.Scan(&v.ID, &v.BrowserName, &v.BrowserLanguage, &v.DeviceOS, &v.IpAddress, &v.Path, &v.ScreenResolution, &v.Timestamp);
|
|
checkError(err)
|
|
results = append(results, v)
|
|
}
|
|
|
|
err = rows.Err();
|
|
checkError(err)
|
|
|
|
w.Header().Set("Content-Type", "application/json")
|
|
json.NewEncoder(w).Encode(results)
|
|
})
|
|
|
|
// URL: /api/visits/count
|
|
var GetVisitsCountHandler = http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
|
period := getRequestedPeriod(r)
|
|
stmt, err := core.DB.Prepare(`SELECT COUNT(DISTINCT(ip_address)) FROM visits WHERE timestamp >= DATE_SUB(CURRENT_TIMESTAMP, INTERVAL ? day) AND timestamp <= CURRENT_TIMESTAMP`)
|
|
|
|
checkError(err)
|
|
defer stmt.Close()
|
|
|
|
var result int
|
|
stmt.QueryRow(period).Scan(&result)
|
|
|
|
w.Header().Set("Content-Type", "application/json")
|
|
json.NewEncoder(w).Encode(result)
|
|
})
|
|
|
|
// URL: /api/visits/count/realtime
|
|
var GetVisitsRealtimeCountHandler = http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
|
var result int
|
|
core.DB.QueryRow(`SELECT COUNT(DISTINCT(ip_address)) FROM visits WHERE timestamp >= DATE_SUB(CURRENT_TIMESTAMP, INTERVAL 3 HOUR_MINUTE) AND timestamp <= CURRENT_TIMESTAMP`).Scan(&result)
|
|
|
|
w.Header().Set("Content-Type", "application/json")
|
|
json.NewEncoder(w).Encode(result)
|
|
})
|
|
|
|
// URL: /api/visits/count/day
|
|
var GetVisitsDayCountHandler = http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
|
stmt, err := core.DB.Prepare(`SELECT
|
|
COUNT(DISTINCT(ip_address)) AS count, DATE_FORMAT(timestamp, '%Y-%m-%d') AS date_group
|
|
FROM visits
|
|
WHERE timestamp >= DATE_SUB(CURRENT_TIMESTAMP, INTERVAL ? DAY)
|
|
GROUP BY date_group`)
|
|
checkError(err)
|
|
defer stmt.Close()
|
|
|
|
period := getRequestedPeriod(r)
|
|
rows, err := stmt.Query(period)
|
|
checkError(err)
|
|
|
|
results := make([]Datapoint, 0)
|
|
defer rows.Close()
|
|
for rows.Next() {
|
|
v := Datapoint{}
|
|
err = rows.Scan(&v.Count, &v.Label);
|
|
checkError(err)
|
|
results = append(results, v)
|
|
}
|
|
|
|
results = fillDatapoints(period, results)
|
|
|
|
w.Header().Set("Content-Type", "application/json")
|
|
json.NewEncoder(w).Encode(results)
|
|
})
|