mirror of https://github.com/status-im/fathom.git
35 lines
828 B
Go
35 lines
828 B
Go
package datastore
|
|
|
|
import (
|
|
"database/sql"
|
|
"github.com/usefathom/fathom/pkg/models"
|
|
)
|
|
|
|
// GetPageByHostnameAndPath retrieves a page from the connected database
|
|
func GetPageByHostnameAndPath(hostname, path string) (*models.Page, error) {
|
|
p := &models.Page{}
|
|
query := dbx.Rebind(`SELECT * FROM pages WHERE hostname = ? AND path = ? LIMIT 1`)
|
|
err := dbx.Get(p, query, hostname, path)
|
|
if err != nil {
|
|
if err == sql.ErrNoRows {
|
|
return nil, ErrNoResults
|
|
}
|
|
|
|
return nil, err
|
|
}
|
|
|
|
return p, nil
|
|
}
|
|
|
|
// SavePage inserts the page model in the connected database
|
|
func SavePage(p *models.Page) error {
|
|
query := dbx.Rebind(`INSERT INTO pages(scheme, hostname, path) VALUES(?, ?, ?)`)
|
|
result, err := dbx.Exec(query, p.Scheme, p.Hostname, p.Path)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
p.ID, _ = result.LastInsertId()
|
|
return nil
|
|
}
|