49 lines
859 B
Go
49 lines
859 B
Go
package database
|
|
|
|
import (
|
|
"database/sql"
|
|
)
|
|
|
|
func (db *Type) VisitorGet() (uint64, error) {
|
|
var (
|
|
row *sql.Row
|
|
count uint64
|
|
err error
|
|
)
|
|
|
|
if row = db.sql.QueryRow("SELECT count FROM visitor_count WHERE id = 0"); row == nil {
|
|
return 0, nil
|
|
}
|
|
|
|
if err = row.Scan(&count); err != nil && err != sql.ErrNoRows {
|
|
return 0, err
|
|
}
|
|
|
|
if err == sql.ErrNoRows {
|
|
return 0, nil
|
|
}
|
|
|
|
return count, nil
|
|
}
|
|
|
|
func (db *Type) VisitorIncrement() (err error) {
|
|
if _, err = db.sql.Exec("UPDATE visitor_count SET count = count + 1 WHERE id = 0"); err != nil && err != sql.ErrNoRows {
|
|
return err
|
|
}
|
|
|
|
// TODO: err is always nil even if there is no rows for some reason, check sql.Result instead
|
|
|
|
if err == sql.ErrNoRows {
|
|
_, err = db.sql.Exec(
|
|
`INSERT INTO visitor_count(
|
|
id, count
|
|
) values(?, ?)`,
|
|
0, 0,
|
|
)
|
|
|
|
return err
|
|
}
|
|
|
|
return nil
|
|
}
|