apiserver: Close rows before reusing tx

Reusing tx before closing rows (might) result in the following error: [mysql] 2020/02/17 11:55:38 packets.go:427: busy buffer This closes #3098
kubeflow · Feb 18, 2020 · a45acb5 · a45acb5
1 parent d0ef0ae
commit a45acb5
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 4 deletions.
diff --git a/backend/src/apiserver/storage/db_status_store.go b/backend/src/apiserver/storage/db_status_store.go
@@ -46,10 +46,11 @@ func (s *DBStatusStore) InitializeDBStatusTable() error {
 		tx.Rollback()
 		return util.NewInternalServerError(err, "Failed to load database status.")
 	}
-	defer rows.Close()
+	next := rows.Next()
+	rows.Close()
 
 	// The table is not initialized
-	if !rows.Next() {
+	if !next {
 		sql, args, queryErr := sq.
 			Insert("db_statuses").
 			SetMap(defaultDBStatus).

diff --git a/backend/src/apiserver/storage/default_experiment_store.go b/backend/src/apiserver/storage/default_experiment_store.go
@@ -48,10 +48,11 @@ func (s *DefaultExperimentStore) initializeDefaultExperimentTable() error {
 		tx.Rollback()
 		return util.NewInternalServerError(err, "Failed to get default experiment.")
 	}
-	defer rows.Close()
+	next := rows.Next()
+	rows.Close()
 
 	// If the table is not initialized, then set the default value.
-	if !rows.Next() {
+	if !next {
 		sql, args, queryErr := sq.
 			Insert("default_experiments").
 			SetMap(defaultExperimentDBValue).