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.
37 lines
664 B
37 lines
664 B
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() |
|
if c > 0 { |
|
logger.Infof("successfully persisted %d values", c) |
|
} |
|
} |
|
}
|
|
|