1 /*
2  *  Make.org Core API
3  *  Copyright (C) 2020 Make.org
4  *
5  * This program is free software: you can redistribute it and/or modify
6  *  it under the terms of the GNU Affero General Public License as
7  *  published by the Free Software Foundation, either version 3 of the
8  *  License, or (at your option) any later version.
9  *
10  *  This program is distributed in the hope that it will be useful,
11  *  but WITHOUT ANY WARRANTY; without even the implied warranty of
12  *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
13  *  GNU Affero General Public License for more details.
14  *
15  *  You should have received a copy of the GNU Affero General Public License
16  *  along with this program.  If not, see <https://www.gnu.org/licenses/>.
17  *
18  */
19 
20 package org.make.api.migrations.db
21 
22 import java.sql.{Connection, ResultSet}
23 import scala.collection.mutable.ArrayBuffer
24 import org.flywaydb.core.api.migration.{BaseJavaMigration, Context}
25 
26 abstract class Migration extends BaseJavaMigration {
27 
28   override def migrate(context: Context): Unit = {
29     val connection = context.getConnection
30     val autoCommit = connection.getAutoCommit
31     connection.setAutoCommit(false)
32     migrate(connection)
33     connection.commit()
34     connection.setAutoCommit(autoCommit)
35   }
36 
37   def migrate(connection: Connection): Unit
38 }
39 
40 object SqlTools {
41 
42   implicit class ImprovedResultSet(set: ResultSet) {
43 
44     @SuppressWarnings(Array("org.wartremover.warts.While", "org.wartremover.warts.MutableDataStructures"))
45     def toListOfRows(columns: List[String]): ArrayBuffer[List[String]] = {
46       val rows = ArrayBuffer[List[String]]()
47       while (set.next()) rows.append(columns.map(set.getString))
48       rows
49     }
50 
51     @SuppressWarnings(Array("org.wartremover.warts.While", "org.wartremover.warts.MutableDataStructures"))
52     def toMaps(columns: String*): ArrayBuffer[Map[String, String]] = {
53       val rows = ArrayBuffer[Map[String, String]]()
54       while (set.next()) rows.append(columns.map(k => (k -> set.getString(k))).toMap)
55       rows
56     }
57   }
58 }