limit number of rows deleted by dashboard version cleanup

pkg/services/sqlstore/dashboard_version.go

@@ -2,6 +2,7 @@ package sqlstore
 
 
 import (
 import (
 	"fmt"
 	"fmt"
+	"strings"
 
 
 	"github.com/grafana/grafana/pkg/bus"
 	"github.com/grafana/grafana/pkg/bus"
 	m "github.com/grafana/grafana/pkg/models"
 	m "github.com/grafana/grafana/pkg/models"
@@ -69,6 +70,8 @@ func GetDashboardVersions(query *m.GetDashboardVersionsQuery) error {
 
 
 func DeleteExpiredVersions(cmd *m.DeleteExpiredVersionsCommand) error {
 func DeleteExpiredVersions(cmd *m.DeleteExpiredVersionsCommand) error {
 	return inTransaction(func(sess *DBSession) error {
 	return inTransaction(func(sess *DBSession) error {
+		const MAX_VERSIONS_TO_DELETE = 100
+
 		versionsToKeep := setting.DashboardVersionsToKeep
 		versionsToKeep := setting.DashboardVersionsToKeep
 		if versionsToKeep < 1 {
 		if versionsToKeep < 1 {
 			versionsToKeep = 1
 			versionsToKeep = 1
@@ -88,44 +91,26 @@ func DeleteExpiredVersions(cmd *m.DeleteExpiredVersionsCommand) error {
 				AND version < vtd.min + vtd.count - %v`
 				AND version < vtd.min + vtd.count - %v`
 
 
 		versionIdsToDeleteSubquery := fmt.Sprintf(versionIdsToDeleteSybqueryTemplate, versionsToKeep)
 		versionIdsToDeleteSubquery := fmt.Sprintf(versionIdsToDeleteSybqueryTemplate, versionsToKeep)
-		deleteExpiredSql := fmt.Sprintf(`DELETE FROM dashboard_version WHERE id IN (%s)`, versionIdsToDeleteSubquery)
-		expiredResponse, err := sess.Exec(deleteExpiredSql)
+		versions := []string{}
+		err := sess.SQL(versionIdsToDeleteSubquery).Find(&versions)
 		if err != nil {
 		if err != nil {
 			return err
 			return err
 		}
 		}
-		cmd.DeletedRows, _ = expiredResponse.RowsAffected()
-
-		return nil
-	})
-}
-
-// Short version of DashboardVersion for getting expired versions
-type DashboardVersionExp struct {
-	Id          int64 `json:"id"`
-	DashboardId int64 `json:"dashboardId"`
-	Version     int   `json:"version"`
-}
 
 
-func getVersionIDsToDelete(versions []DashboardVersionExp, versionsToKeep int) []interface{} {
-	versionIds := make([]interface{}, 0)
-
-	if len(versions) == 0 {
-		return versionIds
+		// Don't delete more than MAX_VERSIONS_TO_DELETE version per time
+		limit := MAX_VERSIONS_TO_DELETE
+		if len(versions) < MAX_VERSIONS_TO_DELETE {
+			limit = len(versions)
 		}
 		}
+		versions = versions[:limit]
 
 
-	currentDashboard := versions[0].DashboardId
-	count := 0
-	for _, v := range versions {
-		if v.DashboardId == currentDashboard {
-			count++
-		} else {
-			count = 1
-			currentDashboard = v.DashboardId
-		}
-		if count > versionsToKeep {
-			versionIds = append(versionIds, v.Id)
-		}
+		deleteExpiredSql := fmt.Sprintf(`DELETE FROM dashboard_version WHERE id IN (%s)`, strings.Join(versions, `,`))
+		expiredResponse, err := sess.Exec(deleteExpiredSql)
+		if err != nil {
+			return err
 		}
 		}
+		cmd.DeletedRows, _ = expiredResponse.RowsAffected()
 
 
-	return versionIds
+		return nil
+	})
 }
 }