func (db *Db) PersistNewsletter(newsletter *newsletters.Newsletter) {
stmt, err := db.connection.Prepare("UPDATE broccoli_newsletters SET title=?, pubdate=? WHERE id = ?")
if err != nil {
fmt.Printf("There was an error\n")
}
result, err := stmt.Exec(newsletter.GetTitle(), newsletter.GetPubDate(), newsletter.GetId())
rows_affected, err := result.RowsAffected()
if rows_affected == 0 {
// Use INSERT
stmt, err = db.connection.Prepare("INSERT INTO broccoli_newsletters(title, pubdate) VALUES (?, ?)")
_, err = stmt.Exec(newsletter.GetTitle(), newsletter.GetPubDate())
}
// Now, store the newsletter sources
sources := newsletter.GetSources()
sources_lastchecked := newsletter.GetSourcesLastChecked()
newsletter_id := newsletter.GetId()
for i, src := range sources {
src_id := (*src).GetId()
// Use update
stmt, err = db.connection.Prepare("UPDATE broccoli_newsletters_sources SET newsletter_id=?, source_id=?, source_lastchecked=?")
result, err = stmt.Exec(newsletter_id, src_id, sources_lastchecked[i])
rows_affected, err = result.RowsAffected()
if rows_affected == 0 {
// Use insert
stmt, err = db.connection.Prepare("INSERT broccoli_newsletters_sources(newsletter_id, source_id, source_lastchecked) VALUES (?, ?, ?)")
_, err = stmt.Exec(newsletter.GetId(), newsletter_id, src_id, sources_lastchecked[i])
}
}
}
// WARNING: Mutation of newsletter here
func (db *Db) FetchNewsletterSources(newsletter *newsletters.Newsletter) {
output, err := db.connection.Query("SELECT source_id, source_lastchecked FROM broccoli_newsletters_sources WHERE newsletter_id = ?", newsletter.GetId())
if err != nil {
fmt.Printf("There was an error fetching all the source IDs\n%s\n", err)
}
for output.Next() {
var source_id int
var source_lastchecked int64
output.Scan(&source_id, &source_lastchecked)
fmt.Printf("Source last checked = %d\n", source_lastchecked)
source := db.FetchSource(source_id)
newsletter.AddSource(source, source_lastchecked)
}
}