41 lines
945 B
Go
41 lines
945 B
Go
package persistence
|
|
|
|
import (
|
|
"fmt"
|
|
|
|
"offline_twitter/scraper"
|
|
)
|
|
|
|
/**
|
|
* Save a Retweet. Do nothing if it already exists, because none of its parameters are modifiable.
|
|
*/
|
|
func (p Profile) SaveRetweet(r scraper.Retweet) error {
|
|
_, err := p.DB.Exec(`
|
|
insert into retweets (retweet_id, tweet_id, retweeted_by, retweeted_at)
|
|
values (?, ?, ?, ?)
|
|
on conflict do nothing
|
|
`,
|
|
r.RetweetID, r.TweetID, r.RetweetedByID, r.RetweetedAt.Unix(),
|
|
)
|
|
if err != nil {
|
|
return fmt.Errorf("Error executing SaveRetweet(%#v):\n %w", r, err)
|
|
}
|
|
return nil
|
|
}
|
|
|
|
/**
|
|
* Retrieve a Retweet by ID
|
|
*/
|
|
func (p Profile) GetRetweetById(id scraper.TweetID) (scraper.Retweet, error) {
|
|
var r scraper.Retweet
|
|
err := p.DB.Get(&r, `
|
|
select retweet_id, tweet_id, retweeted_by, retweeted_at
|
|
from retweets
|
|
where retweet_id = ?
|
|
`, id)
|
|
if err != nil {
|
|
return r, fmt.Errorf("Error executing GetRetweetById(%d):\n %w", id, err)
|
|
}
|
|
return r, nil
|
|
}
|