package count import ( "github.com/dannyvankooten/ana/db" ) // RealtimeVisitors returns the total number of visitors in the last 3 minutes func RealtimeVisitors() int { var result int db.Conn.QueryRow(` SELECT COUNT(DISTINCT(pv.visitor_id)) FROM pageviews pv WHERE pv.timestamp >= DATE_SUB(CURRENT_TIMESTAMP, INTERVAL 3 HOUR_MINUTE) AND pv.timestamp <= CURRENT_TIMESTAMP`).Scan(&result) return result } // Visitors returns the number of total visitors between the given timestamps func Visitors(before int64, after int64) float64 { // get total stmt, err := db.Conn.Prepare(` SELECT SUM(t.count) FROM total_visitors t WHERE UNIX_TIMESTAMP(t.date) <= ? AND UNIX_TIMESTAMP(t.date) >= ?`) checkError(err) defer stmt.Close() var total float64 stmt.QueryRow(before, after).Scan(&total) return total } // VisitorsPerDay returns a point slice containing visitor data per day func VisitorsPerDay(before int64, after int64) []Point { stmt, err := db.Conn.Prepare(`SELECT SUM(t.count) AS count, DATE_FORMAT(t.date, '%Y-%m-%d') AS date_group FROM total_visitors t WHERE UNIX_TIMESTAMP(t.date) <= ? AND UNIX_TIMESTAMP(t.date) >= ? GROUP BY date_group`) checkError(err) defer stmt.Close() rows, err := stmt.Query(before, after) checkError(err) var results []Point defer rows.Close() for rows.Next() { p := Point{} err = rows.Scan(&p.Value, &p.Label) checkError(err) results = append(results, p) } results = fill(after, before, results) return results } // CreateVisitorTotals aggregates visitor data into daily totals func CreateVisitorTotals(since string) { stmt, err := db.Conn.Prepare(` SELECT COUNT(DISTINCT(pv.visitor_id)) AS count, DATE_FORMAT(pv.timestamp, "%Y-%m-%d") AS date_group FROM pageviews pv WHERE pv.timestamp > ? GROUP BY date_group`) checkError(err) defer stmt.Close() rows, err := stmt.Query(since) checkError(err) defer rows.Close() db.Conn.Exec("START TRANSACTION") for rows.Next() { var t Total err = rows.Scan(&t.Count, &t.Date) checkError(err) stmt, err := db.Conn.Prepare(`INSERT INTO total_visitors( count, date ) VALUES( ?, ? ) ON DUPLICATE KEY UPDATE count = ?`) checkError(err) defer stmt.Close() _, err = stmt.Exec( t.Count, t.Date, t.Count, ) checkError(err) } db.Conn.Exec("COMMIT") }