Refresh the imdb ratings in a transaction

https://w000t.me/c140729392

backend/backend/imdb_ratings.go

@@ -25,13 +25,8 @@ type ImdbRating struct {
 	Updated time.Time `db:"updated_at"`
 }
 
-// UpsertImdbRating upsert a ImdbRating in the database
+// TxUpsertImdbRating upsert a ImdbRating in a transaction
-func UpsertImdbRating(db *sqlx.DB, rating *ImdbRating) error {
+func TxUpsertImdbRating(t *sqlx.Tx, rating *ImdbRating) error {
-	r, err := db.NamedQuery(upsertRatingQuery, rating)
+	_, err := t.NamedExec(upsertRatingQuery, rating)
-	if err != nil {
+	return err
-		return err
-	}
-	defer r.Close()
-
-	return nil
 }

backend/ratings/ratings.go

@@ -49,6 +49,11 @@ func Refresh(env *web.Env) error {
 		return err
 	}
 
+	tx, err := env.Database.Beginx()
+	if err != nil {
+		return err
+	}
+
 	// Read it
 	scanner := bufio.NewScanner(r)
 	for scanner.Scan() {
@@ -69,16 +74,24 @@ func Refresh(env *web.Env) error {
 			continue
 		}
 
-		movie := &backend.ImdbRating{
+		videoRating := &backend.ImdbRating{
 			ImdbID: elmts[0],
 			Rating: float32(rating),
 			Votes:  int(numVote),
 		}
-		err = backend.UpsertImdbRating(env.Database, movie)
+		err = backend.TxUpsertImdbRating(tx, videoRating)
 		if err != nil {
+			log.WithFields(logrus.Fields{
+				"error": err,
+			}).Error("got error while upserting rating, rollback!")
+			if rollbackErr := tx.Rollback(); err != nil {
+				log.WithFields(logrus.Fields{
+					"error": rollbackErr,
+				}).Error("unable to rollack")
+			}
 			return err
 		}
 	}
 
-	return nil
+	return tx.Commit()
 }