Better way to match top songs from external sources (Last.fm)

db/migration/20211023184825_add_order_title_to_media_file.go

@@ -0,0 +1,62 @@
+package migrations
+
+import (
+	"database/sql"
+	"strings"
+
+	"github.com/kennygrant/sanitize"
+	"github.com/navidrome/navidrome/log"
+
+	"github.com/pressly/goose"
+)
+
+func init() {
+	goose.AddMigration(upAddOrderTitleToMediaFile, downAddOrderTitleToMediaFile)
+}
+
+func upAddOrderTitleToMediaFile(tx *sql.Tx) error {
+	_, err := tx.Exec(`
+alter table main.media_file
+	add order_title varchar null collate NOCASE;
+create index if not exists media_file_order_title
+    on media_file (order_title);
+`)
+	if err != nil {
+		return err
+	}
+
+	return upAddOrderTitleToMediaFile_populateOrderTitle(tx)
+}
+
+//goland:noinspection GoSnakeCaseUsage
+func upAddOrderTitleToMediaFile_populateOrderTitle(tx *sql.Tx) error {
+	rows, err := tx.Query(`select id, title from media_file`)
+	if err != nil {
+		return err
+	}
+	defer rows.Close()
+
+	stmt, err := tx.Prepare("update media_file set order_title = ? where id = ?")
+	if err != nil {
+		return err
+	}
+
+	var id, title string
+	for rows.Next() {
+		err = rows.Scan(&id, &title)
+		if err != nil {
+			return err
+		}
+
+		orderTitle := strings.TrimSpace(sanitize.Accents(title))
+		_, err = stmt.Exec(orderTitle, id)
+		if err != nil {
+			log.Error("Error setting media_file's order_title", "title", title, "id", id, err)
+		}
+	}
+	return rows.Err()
+}
+
+func downAddOrderTitleToMediaFile(tx *sql.Tx) error {
+	return nil
+}