package api import ( "net/http" "log" "github.com/dannyvankooten/ana/models" "github.com/dannyvankooten/ana/core" "encoding/json" ) // URL: /api/visits func GetVisitsHandler(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`) checkError(err) defer stmt.Close() rows, err := stmt.Query() 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/realtime func GetVisitsRealtimeCount(w http.ResponseWriter, r *http.Request) { row := core.DB.QueryRow(`SELECT COUNT(DISTINCT(id)) FROM visits WHERE timestamp >= DATE_SUB(CURRENT_TIMESTAMP, INTERVAL 3 HOUR_MINUTE)`) var result int row.Scan(&result) w.Header().Set("Content-Type", "application/json") json.NewEncoder(w).Encode(result) } // log fatal errors func checkError(err error) { if err != nil { log.Fatal(err) } }