104 lines
1.9 KiB
Go
104 lines
1.9 KiB
Go
package database
|
|
|
|
import (
|
|
"crowsnest/internal/model"
|
|
"database/sql"
|
|
)
|
|
|
|
type ResponseModel struct {
|
|
DB *sql.DB
|
|
DbDriver string
|
|
}
|
|
|
|
// TODO docstring
|
|
func (m *ResponseModel) All() ([]model.Response, error) {
|
|
stmt := `
|
|
SELECT url, content, fetchDate, processed
|
|
FROM responses
|
|
`
|
|
rows, err := m.DB.Query(stmt)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
responses := []model.Response{}
|
|
for rows.Next() {
|
|
r := model.Response{}
|
|
err := rows.Scan(&r.Url, &r.Content, &r.FetchDate, &r.Processed)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
responses = append(responses, r)
|
|
}
|
|
|
|
if err = rows.Err(); err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return responses, nil
|
|
}
|
|
|
|
// TODO docstring
|
|
func (m *ResponseModel) UnprocessedUrls() ([]string, error) {
|
|
stmt := `
|
|
SELECT url
|
|
FROM responses
|
|
WHERE NOT processed
|
|
`
|
|
rows, err := m.DB.Query(stmt)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
urls := make([]string, 0)
|
|
for rows.Next() {
|
|
r := ""
|
|
err := rows.Scan(&r)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
urls = append(urls, r)
|
|
}
|
|
|
|
if err = rows.Err(); err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return urls, nil
|
|
}
|
|
|
|
// TODO docstring
|
|
func (m *ResponseModel) GetByUrl(url string) (model.Response, error) {
|
|
stmt := `
|
|
SELECT url, content, fetchDate, processed
|
|
FROM responses
|
|
WHERE url = ?
|
|
`
|
|
|
|
res := model.Response{}
|
|
row := m.DB.QueryRow(stmt, url)
|
|
err := row.Scan(&res.Url, &res.Content, &res.FetchDate, &res.Processed)
|
|
|
|
return res, err
|
|
}
|
|
|
|
// TODO docstring
|
|
func (m *ResponseModel) Insert(url string, content string) error {
|
|
// insert response
|
|
stmt := `INSERT INTO responses (url, content) VALUES (?, ?)`
|
|
_, err := m.DB.Exec(stmt, url, content)
|
|
|
|
return err
|
|
}
|
|
|
|
// TODO docstring
|
|
func (m *ResponseModel) Processed(url string) error {
|
|
// insert response
|
|
stmt := `UPDATE responses SET processed = true WHERE url = ?`
|
|
_, err := m.DB.Exec(stmt, url)
|
|
|
|
return err
|
|
}
|