From 5806ef413f4fecec8d2e38f049ed0bbdd1d27720 Mon Sep 17 00:00:00 2001
From: Jacob Brewer <Jacob.brewer@bthree.uk>
Date: Mon, 27 Jan 2025 11:54:13 +0000
Subject: [PATCH] using new methods

---
 pkg/migrations/status.go | 26 ++++++++------------------
 1 file changed, 8 insertions(+), 18 deletions(-)

diff --git a/pkg/migrations/status.go b/pkg/migrations/status.go
index cd87564..8f9edca 100644
--- a/pkg/migrations/status.go
+++ b/pkg/migrations/status.go
@@ -2,31 +2,21 @@ package migrations
 
 import (
 	"fmt"
+	"sort"
 
 	"github.com/jacobbrewer1/goschema/pkg/models"
 )
 
 func (v *versioning) GetStatus() ([]*models.GoschemaMigrationVersion, error) {
-	sqlStmt := `
-		SELECT version
-		FROM goschema_migration_version
-		ORDER BY created_at DESC;
-	`
-
-	ids := make([]string, 0)
-	if err := v.db.Select(&ids, sqlStmt); err != nil {
-		return nil, fmt.Errorf("get status ids: %w", err)
+	versions, err := models.GetAllGoschemaMigrationVersion(v.db)
+	if err != nil {
+		return nil, fmt.Errorf("error getting goschema migration versions: %w", err)
 	}
 
-	versions := make([]*models.GoschemaMigrationVersion, len(ids))
-	for i, id := range ids {
-		ver, err := models.GoschemaMigrationVersionByVersion(v.db, id)
-		if err != nil {
-			return nil, fmt.Errorf("get status version by version: %w", err)
-		}
-
-		versions[i] = ver
-	}
+	// Sort the versions by created_at.
+	sort.Slice(versions, func(i, j int) bool {
+		return versions[i].CreatedAt.Before(versions[j].CreatedAt)
+	})
 
 	return versions, nil
 }