You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
36 lines
646 B
36 lines
646 B
8 years ago
|
package main
|
||
|
|
|
||
|
|
import (
|
||
|
|
"database/sql"
|
||
|
|
"sync"
|
||
|
|
"time"
|
||
|
|
|
||
|
|
"github.com/labstack/echo"
|
||
|
|
)
|
||
|
|
|
||
|
|
const statsSavingInterval = 1 * time.Minute
|
||
|
|
|
||
|
|
func persistStats(logger echo.Logger, db *sql.DB, stats *sync.Map) {
|
||
|
|
for {
|
||
|
|
time.Sleep(statsSavingInterval)
|
||
|
|
tx, err := db.Begin()
|
||
|
|
if err != nil {
|
||
|
|
logger.Error(err)
|
||
|
|
return
|
||
|
|
}
|
||
|
|
c := 0
|
||
|
|
stats.Range(func(id, views interface{}) bool {
|
||
|
|
stmt, _ := tx.Prepare("update notes set views = ? where id = ?")
|
||
|
|
_, err := stmt.Exec(views, id)
|
||
|
|
if err == nil {
|
||
|
|
c++
|
||
|
|
}
|
||
|
|
stmt.Close()
|
||
|
|
defer stats.Delete(id)
|
||
|
|
return true
|
||
|
|
})
|
||
|
|
tx.Commit()
|
||
|
|
logger.Infof("successfully persisted %d values", c)
|
||
|
|
}
|
||
|
|
}
|