mirror of
https://github.com/binwiederhier/ntfy.git
synced 2025-05-30 18:29:16 +02:00
Cache
This commit is contained in:
parent
b775e6dfce
commit
67922b0ae5
8 changed files with 175 additions and 6 deletions
server
61
server/cache.go
Normal file
61
server/cache.go
Normal file
|
@ -0,0 +1,61 @@
|
|||
package server
|
||||
|
||||
import (
|
||||
"database/sql"
|
||||
"time"
|
||||
_ "github.com/mattn/go-sqlite3" // SQLite driver
|
||||
)
|
||||
|
||||
const (
|
||||
createTableQuery = `CREATE TABLE IF NOT EXISTS messages (
|
||||
id VARCHAR(20) PRIMARY KEY,
|
||||
time INT NOT NULL,
|
||||
topic VARCHAR(64) NOT NULL,
|
||||
message VARCHAR(1024) NOT NULL
|
||||
)`
|
||||
insertQuery = `INSERT INTO messages (id, time, topic, message) VALUES (?, ?, ?, ?)`
|
||||
pruneOlderThanQuery = `DELETE FROM messages WHERE time < ?`
|
||||
)
|
||||
|
||||
type cache struct {
|
||||
db *sql.DB
|
||||
insert *sql.Stmt
|
||||
prune *sql.Stmt
|
||||
}
|
||||
|
||||
func newCache(filename string) (*cache, error) {
|
||||
db, err := sql.Open("sqlite3", filename)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
if _, err := db.Exec(createTableQuery); err != nil {
|
||||
return nil, err
|
||||
}
|
||||
insert, err := db.Prepare(insertQuery)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
prune, err := db.Prepare(pruneOlderThanQuery)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
return &cache{
|
||||
db: db,
|
||||
insert: insert,
|
||||
prune: prune,
|
||||
}, nil
|
||||
}
|
||||
|
||||
func (c *cache) Load() (map[string]*topic, error) {
|
||||
|
||||
}
|
||||
|
||||
func (c *cache) Add(m *message) error {
|
||||
_, err := c.insert.Exec(m.ID, m.Time, m.Topic, m.Message)
|
||||
return err
|
||||
}
|
||||
|
||||
func (c *cache) Prune(olderThan time.Duration) error {
|
||||
_, err := c.prune.Exec(time.Now().Add(-1 * olderThan).Unix())
|
||||
return err
|
||||
}
|
Loading…
Add table
Add a link
Reference in a new issue