src/Pure/General/sql.scala
author wenzelm
Sat, 19 Aug 2023 22:57:06 +0200
changeset 78540 a6d079e0575d
parent 78538 56e8458ba262
child 78544 3be0437759cf
permissions -rw-r--r--
clarified signature: filter batch;
Ignore whitespace changes - Everywhere: Within whitespace: At end of lines:
63788
3160826b92f8 clarified modules;
wenzelm
parents: 63784
diff changeset
     1
/*  Title:      Pure/General/sql.scala
63778
e06e899b78d0 clarified modules;
wenzelm
parents:
diff changeset
     2
    Author:     Makarius
e06e899b78d0 clarified modules;
wenzelm
parents:
diff changeset
     3
65006
632bdf7b8bab clarified modules;
wenzelm
parents: 65004
diff changeset
     4
Support for SQL databases: SQLite and PostgreSQL.
78351
9f2cfb9873bb tuned comments;
wenzelm
parents: 78348
diff changeset
     5
9f2cfb9873bb tuned comments;
wenzelm
parents: 78348
diff changeset
     6
See https://docs.oracle.com/en/java/javase/17/docs/api/java.sql/java/sql/Connection.html
63778
e06e899b78d0 clarified modules;
wenzelm
parents:
diff changeset
     7
*/
e06e899b78d0 clarified modules;
wenzelm
parents:
diff changeset
     8
e06e899b78d0 clarified modules;
wenzelm
parents:
diff changeset
     9
package isabelle
e06e899b78d0 clarified modules;
wenzelm
parents:
diff changeset
    10
69393
ed0824ef337e static type for Library.using: avoid Java 11 warnings on "Illegal reflective access";
wenzelm
parents: 69327
diff changeset
    11
65021
da8ae577d171 clarified Date storage;
wenzelm
parents: 65020
diff changeset
    12
import java.time.OffsetDateTime
78538
56e8458ba262 support for execute_batch: multiple statements in one round-trip;
wenzelm
parents: 78422
diff changeset
    13
import java.sql.{DriverManager, Connection, PreparedStatement, ResultSet, SQLException}
63779
9da65bc75610 more operations;
wenzelm
parents: 63778
diff changeset
    14
77037
164a21e5d568 support specific connection types, for additional operations;
wenzelm
parents: 76870
diff changeset
    15
import org.sqlite.jdbc4.JDBC4Connection
77039
2f09dc0e6dda support IPC via database server;
wenzelm
parents: 77037
diff changeset
    16
import org.postgresql.{PGConnection, PGNotification}
77037
164a21e5d568 support specific connection types, for additional operations;
wenzelm
parents: 76870
diff changeset
    17
65740
83388f09e9ab clarified signature;
wenzelm
parents: 65739
diff changeset
    18
import scala.collection.mutable
83388f09e9ab clarified signature;
wenzelm
parents: 65739
diff changeset
    19
63779
9da65bc75610 more operations;
wenzelm
parents: 63778
diff changeset
    20
75393
87ebf5a50283 clarified formatting, for the sake of scala3;
wenzelm
parents: 73367
diff changeset
    21
object SQL {
65006
632bdf7b8bab clarified modules;
wenzelm
parents: 65004
diff changeset
    22
  /** SQL language **/
632bdf7b8bab clarified modules;
wenzelm
parents: 65004
diff changeset
    23
65730
7ae61e72a678 clarified signature;
wenzelm
parents: 65729
diff changeset
    24
  type Source = String
7ae61e72a678 clarified signature;
wenzelm
parents: 65729
diff changeset
    25
7ae61e72a678 clarified signature;
wenzelm
parents: 65729
diff changeset
    26
63778
e06e899b78d0 clarified modules;
wenzelm
parents:
diff changeset
    27
  /* concrete syntax */
e06e899b78d0 clarified modules;
wenzelm
parents:
diff changeset
    28
65321
2b1cd063e0b2 proper SQL syntax (according of PostgreSQL);
wenzelm
parents: 65319
diff changeset
    29
  def escape_char(c: Char): String =
63778
e06e899b78d0 clarified modules;
wenzelm
parents:
diff changeset
    30
    c match {
e06e899b78d0 clarified modules;
wenzelm
parents:
diff changeset
    31
      case '\u0000' => "\\0"
e06e899b78d0 clarified modules;
wenzelm
parents:
diff changeset
    32
      case '\'' => "\\'"
e06e899b78d0 clarified modules;
wenzelm
parents:
diff changeset
    33
      case '\"' => "\\\""
e06e899b78d0 clarified modules;
wenzelm
parents:
diff changeset
    34
      case '\b' => "\\b"
e06e899b78d0 clarified modules;
wenzelm
parents:
diff changeset
    35
      case '\n' => "\\n"
e06e899b78d0 clarified modules;
wenzelm
parents:
diff changeset
    36
      case '\r' => "\\r"
e06e899b78d0 clarified modules;
wenzelm
parents:
diff changeset
    37
      case '\t' => "\\t"
e06e899b78d0 clarified modules;
wenzelm
parents:
diff changeset
    38
      case '\u001a' => "\\Z"
e06e899b78d0 clarified modules;
wenzelm
parents:
diff changeset
    39
      case '\\' => "\\\\"
e06e899b78d0 clarified modules;
wenzelm
parents:
diff changeset
    40
      case _ => c.toString
e06e899b78d0 clarified modules;
wenzelm
parents:
diff changeset
    41
    }
e06e899b78d0 clarified modules;
wenzelm
parents:
diff changeset
    42
65730
7ae61e72a678 clarified signature;
wenzelm
parents: 65729
diff changeset
    43
  def string(s: String): Source =
71601
97ccf48c2f0c misc tuning based on hints by IntelliJ IDEA;
wenzelm
parents: 69980
diff changeset
    44
    s.iterator.map(escape_char).mkString("'", "", "'")
63778
e06e899b78d0 clarified modules;
wenzelm
parents:
diff changeset
    45
65730
7ae61e72a678 clarified signature;
wenzelm
parents: 65729
diff changeset
    46
  def ident(s: String): Source =
65677
7d25b8dbdbfa proper quote (amending 546020a98a91): e.g. relevant for "ISABELLE_BUILD_OPTIONS";
wenzelm
parents: 65673
diff changeset
    47
    Long_Name.implode(Long_Name.explode(s).map(a => quote(a.replace("\"", "\"\""))))
63779
9da65bc75610 more operations;
wenzelm
parents: 63778
diff changeset
    48
65730
7ae61e72a678 clarified signature;
wenzelm
parents: 65729
diff changeset
    49
  def enclose(s: Source): Source = "(" + s + ")"
7ae61e72a678 clarified signature;
wenzelm
parents: 65729
diff changeset
    50
  def enclosure(ss: Iterable[Source]): Source = ss.mkString("(", ", ", ")")
63791
c6cbdfaae19e more operations;
wenzelm
parents: 63790
diff changeset
    51
77381
a86e346b20d8 misc tuning and clarification: more uniform use of optional "sql" in SQL.Table.delete/select;
wenzelm
parents: 77377
diff changeset
    52
  def separate(sql: Source): Source =
a86e346b20d8 misc tuning and clarification: more uniform use of optional "sql" in SQL.Table.delete/select;
wenzelm
parents: 77377
diff changeset
    53
    (if (sql.isEmpty || sql.startsWith(" ")) "" else " ") + sql
a86e346b20d8 misc tuning and clarification: more uniform use of optional "sql" in SQL.Table.delete/select;
wenzelm
parents: 77377
diff changeset
    54
77403
be8e14c7da80 clarified signature, although "sql" argument is de-facto mandatory;
wenzelm
parents: 77381
diff changeset
    55
  def select(columns: List[Column] = Nil, distinct: Boolean = false, sql: Source = ""): Source =
65774
1001fb86d7f7 clarified signature;
wenzelm
parents: 65748
diff changeset
    56
    "SELECT " + (if (distinct) "DISTINCT " else "") +
77403
be8e14c7da80 clarified signature, although "sql" argument is de-facto mandatory;
wenzelm
parents: 77381
diff changeset
    57
    (if (columns.isEmpty) "*" else commas(columns.map(_.ident))) + " FROM " + sql
65644
7ef438495a02 support for qualified names, which are not quoted (e.g. for SQLite);
wenzelm
parents: 65641
diff changeset
    58
65775
123f2c0995b8 tuned signature;
wenzelm
parents: 65774
diff changeset
    59
  val join_outer: Source = " LEFT OUTER JOIN "
123f2c0995b8 tuned signature;
wenzelm
parents: 65774
diff changeset
    60
  val join_inner: Source = " INNER JOIN "
65776
373d708898d4 tuned signature;
wenzelm
parents: 65775
diff changeset
    61
  def join(outer: Boolean): Source = if (outer) join_outer else join_inner
65775
123f2c0995b8 tuned signature;
wenzelm
parents: 65774
diff changeset
    62
77370
47c2ac81ddd4 clarified signature: more robust operations;
wenzelm
parents: 77368
diff changeset
    63
  def infix(op: Source, args: Iterable[Source]): Source = {
47c2ac81ddd4 clarified signature: more robust operations;
wenzelm
parents: 77368
diff changeset
    64
    val body = args.iterator.filter(_.nonEmpty).mkString(" " + op + " ")
47c2ac81ddd4 clarified signature: more robust operations;
wenzelm
parents: 77368
diff changeset
    65
    if_proper(body, enclose(body))
47c2ac81ddd4 clarified signature: more robust operations;
wenzelm
parents: 77368
diff changeset
    66
  }
47c2ac81ddd4 clarified signature: more robust operations;
wenzelm
parents: 77368
diff changeset
    67
47c2ac81ddd4 clarified signature: more robust operations;
wenzelm
parents: 77368
diff changeset
    68
  def AND(args: Iterable[Source]): Source = infix("AND", args)
47c2ac81ddd4 clarified signature: more robust operations;
wenzelm
parents: 77368
diff changeset
    69
  def OR(args: Iterable[Source]): Source = infix("OR", args)
47c2ac81ddd4 clarified signature: more robust operations;
wenzelm
parents: 77368
diff changeset
    70
47c2ac81ddd4 clarified signature: more robust operations;
wenzelm
parents: 77368
diff changeset
    71
  def and(args: Source*): Source = AND(args)
47c2ac81ddd4 clarified signature: more robust operations;
wenzelm
parents: 77368
diff changeset
    72
  def or(args: Source*): Source = OR(args)
47c2ac81ddd4 clarified signature: more robust operations;
wenzelm
parents: 77368
diff changeset
    73
47c2ac81ddd4 clarified signature: more robust operations;
wenzelm
parents: 77368
diff changeset
    74
  val TRUE: Source = "TRUE"
47c2ac81ddd4 clarified signature: more robust operations;
wenzelm
parents: 77368
diff changeset
    75
  val FALSE: Source = "FALSE"
47c2ac81ddd4 clarified signature: more robust operations;
wenzelm
parents: 77368
diff changeset
    76
78152
d4f387339494 tuned signature: more operations;
wenzelm
parents: 78151
diff changeset
    77
  def equal(sql: Source, x: Int): Source = sql + " = " + x
d4f387339494 tuned signature: more operations;
wenzelm
parents: 78151
diff changeset
    78
  def equal(sql: Source, x: Long): Source = sql + " = " + x
d4f387339494 tuned signature: more operations;
wenzelm
parents: 78151
diff changeset
    79
  def equal(sql: Source, x: String): Source = sql + " = " + string(x)
77375
324f5821a4a4 clarified signature: more concise operations;
wenzelm
parents: 77370
diff changeset
    80
324f5821a4a4 clarified signature: more concise operations;
wenzelm
parents: 77370
diff changeset
    81
  def member(sql: Source, set: Iterable[String]): Source =
77370
47c2ac81ddd4 clarified signature: more robust operations;
wenzelm
parents: 77368
diff changeset
    82
    if (set.isEmpty) FALSE
77375
324f5821a4a4 clarified signature: more concise operations;
wenzelm
parents: 77370
diff changeset
    83
    else OR(set.iterator.map(equal(sql, _)).toList)
65804
wenzelm
parents: 65780
diff changeset
    84
77370
47c2ac81ddd4 clarified signature: more robust operations;
wenzelm
parents: 77368
diff changeset
    85
  def where(sql: Source): Source = if_proper(sql, " WHERE " + sql)
78153
55a6aa77f3d8 tuned signature: more operations;
wenzelm
parents: 78152
diff changeset
    86
  def where_and(args: Source*): Source = where(and(args:_*))
55a6aa77f3d8 tuned signature: more operations;
wenzelm
parents: 78152
diff changeset
    87
  def where_or(args: Source*): Source = where(or(args:_*))
77370
47c2ac81ddd4 clarified signature: more robust operations;
wenzelm
parents: 77368
diff changeset
    88
47c2ac81ddd4 clarified signature: more robust operations;
wenzelm
parents: 77368
diff changeset
    89
65008
ed2eedf786f3 clarified handling of SQL.Type;
wenzelm
parents: 65007
diff changeset
    90
  /* types */
ed2eedf786f3 clarified handling of SQL.Type;
wenzelm
parents: 65007
diff changeset
    91
75393
87ebf5a50283 clarified formatting, for the sake of scala3;
wenzelm
parents: 73367
diff changeset
    92
  object Type extends Enumeration {
65011
42bffd1637f0 support for type Boolean;
wenzelm
parents: 65010
diff changeset
    93
    val Boolean = Value("BOOLEAN")
65008
ed2eedf786f3 clarified handling of SQL.Type;
wenzelm
parents: 65007
diff changeset
    94
    val Int = Value("INTEGER")
ed2eedf786f3 clarified handling of SQL.Type;
wenzelm
parents: 65007
diff changeset
    95
    val Long = Value("BIGINT")
ed2eedf786f3 clarified handling of SQL.Type;
wenzelm
parents: 65007
diff changeset
    96
    val Double = Value("DOUBLE PRECISION")
ed2eedf786f3 clarified handling of SQL.Type;
wenzelm
parents: 65007
diff changeset
    97
    val String = Value("TEXT")
ed2eedf786f3 clarified handling of SQL.Type;
wenzelm
parents: 65007
diff changeset
    98
    val Bytes = Value("BLOB")
65014
97a622d01609 support for type Date;
wenzelm
parents: 65013
diff changeset
    99
    val Date = Value("TIMESTAMP WITH TIME ZONE")
65008
ed2eedf786f3 clarified handling of SQL.Type;
wenzelm
parents: 65007
diff changeset
   100
  }
ed2eedf786f3 clarified handling of SQL.Type;
wenzelm
parents: 65007
diff changeset
   101
65730
7ae61e72a678 clarified signature;
wenzelm
parents: 65729
diff changeset
   102
  def sql_type_default(T: Type.Value): Source = T.toString
65013
wenzelm
parents: 65012
diff changeset
   103
65730
7ae61e72a678 clarified signature;
wenzelm
parents: 65729
diff changeset
   104
  def sql_type_sqlite(T: Type.Value): Source =
65019
0ca91e5d52df tuned signature;
wenzelm
parents: 65018
diff changeset
   105
    if (T == Type.Boolean) "INTEGER"
0ca91e5d52df tuned signature;
wenzelm
parents: 65018
diff changeset
   106
    else if (T == Type.Date) "TEXT"
0ca91e5d52df tuned signature;
wenzelm
parents: 65018
diff changeset
   107
    else sql_type_default(T)
65013
wenzelm
parents: 65012
diff changeset
   108
65730
7ae61e72a678 clarified signature;
wenzelm
parents: 65729
diff changeset
   109
  def sql_type_postgresql(T: Type.Value): Source =
65019
0ca91e5d52df tuned signature;
wenzelm
parents: 65018
diff changeset
   110
    if (T == Type.Bytes) "BYTEA"
0ca91e5d52df tuned signature;
wenzelm
parents: 65018
diff changeset
   111
    else sql_type_default(T)
65008
ed2eedf786f3 clarified handling of SQL.Type;
wenzelm
parents: 65007
diff changeset
   112
ed2eedf786f3 clarified handling of SQL.Type;
wenzelm
parents: 65007
diff changeset
   113
63779
9da65bc75610 more operations;
wenzelm
parents: 63778
diff changeset
   114
  /* columns */
9da65bc75610 more operations;
wenzelm
parents: 63778
diff changeset
   115
75393
87ebf5a50283 clarified formatting, for the sake of scala3;
wenzelm
parents: 73367
diff changeset
   116
  object Column {
65280
ef37f5236794 prefer non-strict default;
wenzelm
parents: 65022
diff changeset
   117
    def bool(name: String, strict: Boolean = false, primary_key: Boolean = false): Column =
65018
8b36f225bbee clarified signature;
wenzelm
parents: 65014
diff changeset
   118
      Column(name, Type.Boolean, strict, primary_key)
65280
ef37f5236794 prefer non-strict default;
wenzelm
parents: 65022
diff changeset
   119
    def int(name: String, strict: Boolean = false, primary_key: Boolean = false): Column =
65018
8b36f225bbee clarified signature;
wenzelm
parents: 65014
diff changeset
   120
      Column(name, Type.Int, strict, primary_key)
65280
ef37f5236794 prefer non-strict default;
wenzelm
parents: 65022
diff changeset
   121
    def long(name: String, strict: Boolean = false, primary_key: Boolean = false): Column =
65018
8b36f225bbee clarified signature;
wenzelm
parents: 65014
diff changeset
   122
      Column(name, Type.Long, strict, primary_key)
65280
ef37f5236794 prefer non-strict default;
wenzelm
parents: 65022
diff changeset
   123
    def double(name: String, strict: Boolean = false, primary_key: Boolean = false): Column =
65018
8b36f225bbee clarified signature;
wenzelm
parents: 65014
diff changeset
   124
      Column(name, Type.Double, strict, primary_key)
65280
ef37f5236794 prefer non-strict default;
wenzelm
parents: 65022
diff changeset
   125
    def string(name: String, strict: Boolean = false, primary_key: Boolean = false): Column =
65018
8b36f225bbee clarified signature;
wenzelm
parents: 65014
diff changeset
   126
      Column(name, Type.String, strict, primary_key)
65280
ef37f5236794 prefer non-strict default;
wenzelm
parents: 65022
diff changeset
   127
    def bytes(name: String, strict: Boolean = false, primary_key: Boolean = false): Column =
65018
8b36f225bbee clarified signature;
wenzelm
parents: 65014
diff changeset
   128
      Column(name, Type.Bytes, strict, primary_key)
65280
ef37f5236794 prefer non-strict default;
wenzelm
parents: 65022
diff changeset
   129
    def date(name: String, strict: Boolean = false, primary_key: Boolean = false): Column =
65018
8b36f225bbee clarified signature;
wenzelm
parents: 65014
diff changeset
   130
      Column(name, Type.Date, strict, primary_key)
63779
9da65bc75610 more operations;
wenzelm
parents: 63778
diff changeset
   131
  }
9da65bc75610 more operations;
wenzelm
parents: 63778
diff changeset
   132
65018
8b36f225bbee clarified signature;
wenzelm
parents: 65014
diff changeset
   133
  sealed case class Column(
75393
87ebf5a50283 clarified formatting, for the sake of scala3;
wenzelm
parents: 73367
diff changeset
   134
    name: String,
87ebf5a50283 clarified formatting, for the sake of scala3;
wenzelm
parents: 73367
diff changeset
   135
    T: Type.Value,
87ebf5a50283 clarified formatting, for the sake of scala3;
wenzelm
parents: 73367
diff changeset
   136
    strict: Boolean = false,
87ebf5a50283 clarified formatting, for the sake of scala3;
wenzelm
parents: 73367
diff changeset
   137
    primary_key: Boolean = false,
87ebf5a50283 clarified formatting, for the sake of scala3;
wenzelm
parents: 73367
diff changeset
   138
    expr: SQL.Source = ""
87ebf5a50283 clarified formatting, for the sake of scala3;
wenzelm
parents: 73367
diff changeset
   139
  ) {
66857
f8f42289c4df tuned signature;
wenzelm
parents: 66856
diff changeset
   140
    def make_primary_key: Column = copy(primary_key = true)
f8f42289c4df tuned signature;
wenzelm
parents: 66856
diff changeset
   141
65691
2229276a1f99 eliminated redundant type SQL.View;
wenzelm
parents: 65690
diff changeset
   142
    def apply(table: Table): Column =
2229276a1f99 eliminated redundant type SQL.View;
wenzelm
parents: 65690
diff changeset
   143
      Column(Long_Name.qualify(table.name, name), T, strict = strict, primary_key = primary_key)
65644
7ef438495a02 support for qualified names, which are not quoted (e.g. for SQLite);
wenzelm
parents: 65641
diff changeset
   144
65780
8baf789b1537 allow column with defining expression;
wenzelm
parents: 65778
diff changeset
   145
    def ident: Source =
8baf789b1537 allow column with defining expression;
wenzelm
parents: 65778
diff changeset
   146
      if (expr == "") SQL.ident(name)
8baf789b1537 allow column with defining expression;
wenzelm
parents: 65778
diff changeset
   147
      else enclose(expr) + " AS " + SQL.ident(name)
65695
4edac706bc5e tuned signature;
wenzelm
parents: 65691
diff changeset
   148
65730
7ae61e72a678 clarified signature;
wenzelm
parents: 65729
diff changeset
   149
    def decl(sql_type: Type.Value => Source): Source =
65695
4edac706bc5e tuned signature;
wenzelm
parents: 65691
diff changeset
   150
      ident + " " + sql_type(T) + (if (strict || primary_key) " NOT NULL" else "")
63781
af9fe0b6b78e support for (single) primary key;
wenzelm
parents: 63780
diff changeset
   151
66856
6b90c688a6dc tuned signature;
wenzelm
parents: 65918
diff changeset
   152
    def defined: String = ident + " IS NOT NULL"
6b90c688a6dc tuned signature;
wenzelm
parents: 65918
diff changeset
   153
    def undefined: String = ident + " IS NULL"
6b90c688a6dc tuned signature;
wenzelm
parents: 65918
diff changeset
   154
78152
d4f387339494 tuned signature: more operations;
wenzelm
parents: 78151
diff changeset
   155
    def equal(x: Int): Source = SQL.equal(ident, x)
d4f387339494 tuned signature: more operations;
wenzelm
parents: 78151
diff changeset
   156
    def equal(x: Long): Source = SQL.equal(ident, x)
d4f387339494 tuned signature: more operations;
wenzelm
parents: 78151
diff changeset
   157
    def equal(x: String): Source = SQL.equal(ident, x)
d4f387339494 tuned signature: more operations;
wenzelm
parents: 78151
diff changeset
   158
d4f387339494 tuned signature: more operations;
wenzelm
parents: 78151
diff changeset
   159
    def where_equal(x: Int): Source = SQL.where(equal(x))
d4f387339494 tuned signature: more operations;
wenzelm
parents: 78151
diff changeset
   160
    def where_equal(x: Long): Source = SQL.where(equal(x))
d4f387339494 tuned signature: more operations;
wenzelm
parents: 78151
diff changeset
   161
    def where_equal(x: String): Source = SQL.where(equal(x))
d4f387339494 tuned signature: more operations;
wenzelm
parents: 78151
diff changeset
   162
77375
324f5821a4a4 clarified signature: more concise operations;
wenzelm
parents: 77370
diff changeset
   163
    def member(set: Iterable[String]): Source = SQL.member(ident, set)
324f5821a4a4 clarified signature: more concise operations;
wenzelm
parents: 77370
diff changeset
   164
    def where_member(set: Iterable[String]): Source = SQL.where(member(set))
65593
607f7ad07a60 tuned signature;
wenzelm
parents: 65327
diff changeset
   165
78151
2fdf3d8a94e6 clarified signature;
wenzelm
parents: 77681
diff changeset
   166
    def max: Column = copy(expr = "MAX(" + ident + ")")
2fdf3d8a94e6 clarified signature;
wenzelm
parents: 77681
diff changeset
   167
65730
7ae61e72a678 clarified signature;
wenzelm
parents: 65729
diff changeset
   168
    override def toString: Source = ident
63779
9da65bc75610 more operations;
wenzelm
parents: 63778
diff changeset
   169
  }
9da65bc75610 more operations;
wenzelm
parents: 63778
diff changeset
   170
76870
c6cdf2a641f4 clarified signature: more explicit types;
wenzelm
parents: 76529
diff changeset
   171
  def order_by(columns: List[Column], descending: Boolean = false): Source =
c6cdf2a641f4 clarified signature: more explicit types;
wenzelm
parents: 76529
diff changeset
   172
    " ORDER BY " + columns.mkString(", ") + (if (descending) " DESC" else "")
c6cdf2a641f4 clarified signature: more explicit types;
wenzelm
parents: 76529
diff changeset
   173
63780
163244cefb4e more operations;
wenzelm
parents: 63779
diff changeset
   174
163244cefb4e more operations;
wenzelm
parents: 63779
diff changeset
   175
  /* tables */
163244cefb4e more operations;
wenzelm
parents: 63779
diff changeset
   176
75393
87ebf5a50283 clarified formatting, for the sake of scala3;
wenzelm
parents: 73367
diff changeset
   177
  sealed case class Table(name: String, columns: List[Column], body: Source = "") {
63781
af9fe0b6b78e support for (single) primary key;
wenzelm
parents: 63780
diff changeset
   178
    Library.duplicates(columns.map(_.name)) match {
af9fe0b6b78e support for (single) primary key;
wenzelm
parents: 63780
diff changeset
   179
      case Nil =>
af9fe0b6b78e support for (single) primary key;
wenzelm
parents: 63780
diff changeset
   180
      case bad => error("Duplicate column names " + commas_quote(bad) + " for table " + quote(name))
af9fe0b6b78e support for (single) primary key;
wenzelm
parents: 63780
diff changeset
   181
    }
af9fe0b6b78e support for (single) primary key;
wenzelm
parents: 63780
diff changeset
   182
65730
7ae61e72a678 clarified signature;
wenzelm
parents: 65729
diff changeset
   183
    def ident: Source = SQL.ident(name)
65691
2229276a1f99 eliminated redundant type SQL.View;
wenzelm
parents: 65690
diff changeset
   184
65730
7ae61e72a678 clarified signature;
wenzelm
parents: 65729
diff changeset
   185
    def query: Source =
65696
3f53a05c1266 tuned signature;
wenzelm
parents: 65695
diff changeset
   186
      if (body == "") error("Missing SQL body for table " + quote(name))
3f53a05c1266 tuned signature;
wenzelm
parents: 65695
diff changeset
   187
      else SQL.enclose(body)
65649
0818da4f67bb tuned signature;
wenzelm
parents: 65648
diff changeset
   188
65778
666a1bac126b tuned signature;
wenzelm
parents: 65776
diff changeset
   189
    def query_named: Source = query + " AS " + SQL.ident(name)
65702
7c6a91deb212 tuned signature;
wenzelm
parents: 65701
diff changeset
   190
78392
27c2fa1db6ed more uniform guard (!exists_table(table)): avoid "ALTER TABLE" on already existing table, which could lead to deadlocks if this is presently locked;
wenzelm
parents: 78391
diff changeset
   191
    def create(sql_type: Type.Value => Source): Source = {
65325
981df08de0ab more general primary_key;
wenzelm
parents: 65323
diff changeset
   192
      val primary_key =
981df08de0ab more general primary_key;
wenzelm
parents: 65323
diff changeset
   193
        columns.filter(_.primary_key).map(_.name) match {
981df08de0ab more general primary_key;
wenzelm
parents: 65323
diff changeset
   194
          case Nil => Nil
981df08de0ab more general primary_key;
wenzelm
parents: 65323
diff changeset
   195
          case keys => List("PRIMARY KEY " + enclosure(keys))
981df08de0ab more general primary_key;
wenzelm
parents: 65323
diff changeset
   196
        }
78392
27c2fa1db6ed more uniform guard (!exists_table(table)): avoid "ALTER TABLE" on already existing table, which could lead to deadlocks if this is presently locked;
wenzelm
parents: 78391
diff changeset
   197
      "CREATE TABLE " + ident + " " + enclosure(columns.map(_.decl(sql_type)) ::: primary_key)
63781
af9fe0b6b78e support for (single) primary key;
wenzelm
parents: 63780
diff changeset
   198
    }
af9fe0b6b78e support for (single) primary key;
wenzelm
parents: 63780
diff changeset
   199
77365
a10fa2112854 clarified signature;
wenzelm
parents: 77364
diff changeset
   200
    def insert_cmd(cmd: Source = "INSERT", sql: Source = ""): Source =
77377
82fdc7cf9d44 tuned whitespace in generated SQL;
wenzelm
parents: 77375
diff changeset
   201
      cmd + " INTO " + ident + " VALUES " + enclosure(columns.map(_ => "?")) + SQL.separate(sql)
63791
c6cbdfaae19e more operations;
wenzelm
parents: 63790
diff changeset
   202
77365
a10fa2112854 clarified signature;
wenzelm
parents: 77364
diff changeset
   203
    def insert(sql: Source = ""): Source = insert_cmd(sql = sql)
65703
cead65c19f2e more direct insert_permissive statement, which avoids somewhat fragile nested transactions;
wenzelm
parents: 65702
diff changeset
   204
65730
7ae61e72a678 clarified signature;
wenzelm
parents: 65729
diff changeset
   205
    def delete(sql: Source = ""): Source =
77377
82fdc7cf9d44 tuned whitespace in generated SQL;
wenzelm
parents: 77375
diff changeset
   206
      "DELETE FROM " + ident + SQL.separate(sql)
65319
64da14387b2c more operations;
wenzelm
parents: 65292
diff changeset
   207
77537
1bbf29ec9ce3 more operations;
wenzelm
parents: 77527
diff changeset
   208
    def update(update_columns: List[Column] = Nil, sql: Source = ""): Source =
1bbf29ec9ce3 more operations;
wenzelm
parents: 77527
diff changeset
   209
      "UPDATE " + ident + " SET " + commas(update_columns.map(c => c.ident + " = ?")) +
1bbf29ec9ce3 more operations;
wenzelm
parents: 77527
diff changeset
   210
        SQL.separate(sql)
1bbf29ec9ce3 more operations;
wenzelm
parents: 77527
diff changeset
   211
65774
1001fb86d7f7 clarified signature;
wenzelm
parents: 65748
diff changeset
   212
    def select(
77381
a86e346b20d8 misc tuning and clarification: more uniform use of optional "sql" in SQL.Table.delete/select;
wenzelm
parents: 77377
diff changeset
   213
      select_columns: List[Column] = Nil,
a86e346b20d8 misc tuning and clarification: more uniform use of optional "sql" in SQL.Table.delete/select;
wenzelm
parents: 77377
diff changeset
   214
      distinct: Boolean = false,
a86e346b20d8 misc tuning and clarification: more uniform use of optional "sql" in SQL.Table.delete/select;
wenzelm
parents: 77377
diff changeset
   215
      sql: Source = ""
77403
be8e14c7da80 clarified signature, although "sql" argument is de-facto mandatory;
wenzelm
parents: 77381
diff changeset
   216
    ): Source = SQL.select(select_columns, distinct = distinct, sql = ident + SQL.separate(sql))
63790
3d723062dc70 more operations;
wenzelm
parents: 63788
diff changeset
   217
65730
7ae61e72a678 clarified signature;
wenzelm
parents: 65729
diff changeset
   218
    override def toString: Source = ident
63780
163244cefb4e more operations;
wenzelm
parents: 63779
diff changeset
   219
  }
63790
3d723062dc70 more operations;
wenzelm
parents: 63788
diff changeset
   220
3d723062dc70 more operations;
wenzelm
parents: 63788
diff changeset
   221
77596
dd8b08729458 clarified signature;
wenzelm
parents: 77591
diff changeset
   222
  /* table groups */
dd8b08729458 clarified signature;
wenzelm
parents: 77591
diff changeset
   223
dd8b08729458 clarified signature;
wenzelm
parents: 77591
diff changeset
   224
  object Tables {
dd8b08729458 clarified signature;
wenzelm
parents: 77591
diff changeset
   225
    def list(list: List[Table]): Tables = new Tables(list)
dd8b08729458 clarified signature;
wenzelm
parents: 77591
diff changeset
   226
    def apply(args: Table*): Tables = list(args.toList)
dd8b08729458 clarified signature;
wenzelm
parents: 77591
diff changeset
   227
  }
dd8b08729458 clarified signature;
wenzelm
parents: 77591
diff changeset
   228
dd8b08729458 clarified signature;
wenzelm
parents: 77591
diff changeset
   229
  final class Tables private(val list: List[Table]) extends Iterable[Table] {
dd8b08729458 clarified signature;
wenzelm
parents: 77591
diff changeset
   230
    override def toString: String = list.mkString("SQL.Tables(", ", ", ")")
dd8b08729458 clarified signature;
wenzelm
parents: 77591
diff changeset
   231
dd8b08729458 clarified signature;
wenzelm
parents: 77591
diff changeset
   232
    def iterator: Iterator[Table] = list.iterator
dd8b08729458 clarified signature;
wenzelm
parents: 77591
diff changeset
   233
dd8b08729458 clarified signature;
wenzelm
parents: 77591
diff changeset
   234
    // requires transaction
78355
9fbc6a043268 support trace of transaction_lock via property "isabelle.transaction_log";
wenzelm
parents: 78354
diff changeset
   235
    def lock(db: Database, create: Boolean = false): Boolean = {
78154
8a7df40375ae tuned signature: more operations;
wenzelm
parents: 78153
diff changeset
   236
      if (create) foreach(db.create_table(_))
77596
dd8b08729458 clarified signature;
wenzelm
parents: 77591
diff changeset
   237
      val sql = db.lock_tables(list)
78355
9fbc6a043268 support trace of transaction_lock via property "isabelle.transaction_log";
wenzelm
parents: 78354
diff changeset
   238
      if (sql.nonEmpty) { db.execute_statement(sql); true }
9fbc6a043268 support trace of transaction_lock via property "isabelle.transaction_log";
wenzelm
parents: 78354
diff changeset
   239
      else false
77596
dd8b08729458 clarified signature;
wenzelm
parents: 77591
diff changeset
   240
    }
dd8b08729458 clarified signature;
wenzelm
parents: 77591
diff changeset
   241
  }
dd8b08729458 clarified signature;
wenzelm
parents: 77591
diff changeset
   242
78187
2df0f3604a67 clarified signature: more explicit class SQL.Data;
wenzelm
parents: 78167
diff changeset
   243
  abstract class Data(table_prefix: String = "") {
78389
41e8ae87184d clarified signature: eliminate SQL.Tables.empty to avoid confusion (see also 0bd366fad888);
wenzelm
parents: 78386
diff changeset
   244
    def tables: Tables
78187
2df0f3604a67 clarified signature: more explicit class SQL.Data;
wenzelm
parents: 78167
diff changeset
   245
2df0f3604a67 clarified signature: more explicit class SQL.Data;
wenzelm
parents: 78167
diff changeset
   246
    def transaction_lock[A](
2df0f3604a67 clarified signature: more explicit class SQL.Data;
wenzelm
parents: 78167
diff changeset
   247
      db: Database,
78207
8e1941d3f703 clarified signature: more options;
wenzelm
parents: 78187
diff changeset
   248
      create: Boolean = false,
78356
974dbe256a37 more informative trace;
wenzelm
parents: 78355
diff changeset
   249
      label: String = "",
78369
ba71ea02d965 more robust Java/Scala multithreading: transaction is always connection.synchronized;
wenzelm
parents: 78366
diff changeset
   250
      log: Logger = new System_Logger
78207
8e1941d3f703 clarified signature: more options;
wenzelm
parents: 78187
diff changeset
   251
    )(body: => A): A = {
78369
ba71ea02d965 more robust Java/Scala multithreading: transaction is always connection.synchronized;
wenzelm
parents: 78366
diff changeset
   252
      db.transaction_lock(tables, create = create, label = label, log = log)(body)
78207
8e1941d3f703 clarified signature: more options;
wenzelm
parents: 78187
diff changeset
   253
    }
78187
2df0f3604a67 clarified signature: more explicit class SQL.Data;
wenzelm
parents: 78167
diff changeset
   254
78266
d8c99a497502 clarified signature;
wenzelm
parents: 78264
diff changeset
   255
    def make_table(columns: List[Column], body: String = "", name: String = ""): Table = {
78187
2df0f3604a67 clarified signature: more explicit class SQL.Data;
wenzelm
parents: 78167
diff changeset
   256
      val table_name =
2df0f3604a67 clarified signature: more explicit class SQL.Data;
wenzelm
parents: 78167
diff changeset
   257
        List(proper_string(table_prefix), proper_string(name)).flatten.mkString("_")
2df0f3604a67 clarified signature: more explicit class SQL.Data;
wenzelm
parents: 78167
diff changeset
   258
      require(table_name.nonEmpty, "Undefined database table name")
2df0f3604a67 clarified signature: more explicit class SQL.Data;
wenzelm
parents: 78167
diff changeset
   259
      Table(table_name, columns, body = body)
2df0f3604a67 clarified signature: more explicit class SQL.Data;
wenzelm
parents: 78167
diff changeset
   260
    }
2df0f3604a67 clarified signature: more explicit class SQL.Data;
wenzelm
parents: 78167
diff changeset
   261
  }
2df0f3604a67 clarified signature: more explicit class SQL.Data;
wenzelm
parents: 78167
diff changeset
   262
77596
dd8b08729458 clarified signature;
wenzelm
parents: 77591
diff changeset
   263
65012
wenzelm
parents: 65011
diff changeset
   264
wenzelm
parents: 65011
diff changeset
   265
  /** SQL database operations **/
wenzelm
parents: 65011
diff changeset
   266
65740
83388f09e9ab clarified signature;
wenzelm
parents: 65739
diff changeset
   267
  /* statements */
83388f09e9ab clarified signature;
wenzelm
parents: 65739
diff changeset
   268
78538
56e8458ba262 support for execute_batch: multiple statements in one round-trip;
wenzelm
parents: 78422
diff changeset
   269
  class Batch_Error(val results: List[Int]) extends SQLException
56e8458ba262 support for execute_batch: multiple statements in one round-trip;
wenzelm
parents: 78422
diff changeset
   270
75393
87ebf5a50283 clarified formatting, for the sake of scala3;
wenzelm
parents: 73367
diff changeset
   271
  class Statement private[SQL](val db: Database, val rep: PreparedStatement) extends AutoCloseable {
65740
83388f09e9ab clarified signature;
wenzelm
parents: 65739
diff changeset
   272
    stmt =>
83388f09e9ab clarified signature;
wenzelm
parents: 65739
diff changeset
   273
75393
87ebf5a50283 clarified formatting, for the sake of scala3;
wenzelm
parents: 73367
diff changeset
   274
    object bool {
73340
0ffcad1f6130 tuned --- fewer warnings;
wenzelm
parents: 73337
diff changeset
   275
      def update(i: Int, x: Boolean): Unit = rep.setBoolean(i, x)
75393
87ebf5a50283 clarified formatting, for the sake of scala3;
wenzelm
parents: 73367
diff changeset
   276
      def update(i: Int, x: Option[Boolean]): Unit = {
65748
1f4a80e80c88 tuned signature;
wenzelm
parents: 65741
diff changeset
   277
        if (x.isDefined) update(i, x.get)
1f4a80e80c88 tuned signature;
wenzelm
parents: 65741
diff changeset
   278
        else rep.setNull(i, java.sql.Types.BOOLEAN)
1f4a80e80c88 tuned signature;
wenzelm
parents: 65741
diff changeset
   279
      }
65740
83388f09e9ab clarified signature;
wenzelm
parents: 65739
diff changeset
   280
    }
75393
87ebf5a50283 clarified formatting, for the sake of scala3;
wenzelm
parents: 73367
diff changeset
   281
    object int {
73340
0ffcad1f6130 tuned --- fewer warnings;
wenzelm
parents: 73337
diff changeset
   282
      def update(i: Int, x: Int): Unit = rep.setInt(i, x)
75393
87ebf5a50283 clarified formatting, for the sake of scala3;
wenzelm
parents: 73367
diff changeset
   283
      def update(i: Int, x: Option[Int]): Unit = {
65748
1f4a80e80c88 tuned signature;
wenzelm
parents: 65741
diff changeset
   284
        if (x.isDefined) update(i, x.get)
1f4a80e80c88 tuned signature;
wenzelm
parents: 65741
diff changeset
   285
        else rep.setNull(i, java.sql.Types.INTEGER)
1f4a80e80c88 tuned signature;
wenzelm
parents: 65741
diff changeset
   286
      }
65740
83388f09e9ab clarified signature;
wenzelm
parents: 65739
diff changeset
   287
    }
75393
87ebf5a50283 clarified formatting, for the sake of scala3;
wenzelm
parents: 73367
diff changeset
   288
    object long {
73340
0ffcad1f6130 tuned --- fewer warnings;
wenzelm
parents: 73337
diff changeset
   289
      def update(i: Int, x: Long): Unit = rep.setLong(i, x)
75393
87ebf5a50283 clarified formatting, for the sake of scala3;
wenzelm
parents: 73367
diff changeset
   290
      def update(i: Int, x: Option[Long]): Unit = {
65748
1f4a80e80c88 tuned signature;
wenzelm
parents: 65741
diff changeset
   291
        if (x.isDefined) update(i, x.get)
1f4a80e80c88 tuned signature;
wenzelm
parents: 65741
diff changeset
   292
        else rep.setNull(i, java.sql.Types.BIGINT)
1f4a80e80c88 tuned signature;
wenzelm
parents: 65741
diff changeset
   293
      }
65740
83388f09e9ab clarified signature;
wenzelm
parents: 65739
diff changeset
   294
    }
75393
87ebf5a50283 clarified formatting, for the sake of scala3;
wenzelm
parents: 73367
diff changeset
   295
    object double {
73340
0ffcad1f6130 tuned --- fewer warnings;
wenzelm
parents: 73337
diff changeset
   296
      def update(i: Int, x: Double): Unit = rep.setDouble(i, x)
75393
87ebf5a50283 clarified formatting, for the sake of scala3;
wenzelm
parents: 73367
diff changeset
   297
      def update(i: Int, x: Option[Double]): Unit = {
65748
1f4a80e80c88 tuned signature;
wenzelm
parents: 65741
diff changeset
   298
        if (x.isDefined) update(i, x.get)
1f4a80e80c88 tuned signature;
wenzelm
parents: 65741
diff changeset
   299
        else rep.setNull(i, java.sql.Types.DOUBLE)
1f4a80e80c88 tuned signature;
wenzelm
parents: 65741
diff changeset
   300
      }
1f4a80e80c88 tuned signature;
wenzelm
parents: 65741
diff changeset
   301
    }
75393
87ebf5a50283 clarified formatting, for the sake of scala3;
wenzelm
parents: 73367
diff changeset
   302
    object string {
73340
0ffcad1f6130 tuned --- fewer warnings;
wenzelm
parents: 73337
diff changeset
   303
      def update(i: Int, x: String): Unit = rep.setString(i, x)
65748
1f4a80e80c88 tuned signature;
wenzelm
parents: 65741
diff changeset
   304
      def update(i: Int, x: Option[String]): Unit = update(i, x.orNull)
65740
83388f09e9ab clarified signature;
wenzelm
parents: 65739
diff changeset
   305
    }
75393
87ebf5a50283 clarified formatting, for the sake of scala3;
wenzelm
parents: 73367
diff changeset
   306
    object bytes {
87ebf5a50283 clarified formatting, for the sake of scala3;
wenzelm
parents: 73367
diff changeset
   307
      def update(i: Int, bytes: Bytes): Unit = {
65748
1f4a80e80c88 tuned signature;
wenzelm
parents: 65741
diff changeset
   308
        if (bytes == null) rep.setBytes(i, null)
1f4a80e80c88 tuned signature;
wenzelm
parents: 65741
diff changeset
   309
        else rep.setBinaryStream(i, bytes.stream(), bytes.length)
1f4a80e80c88 tuned signature;
wenzelm
parents: 65741
diff changeset
   310
      }
1f4a80e80c88 tuned signature;
wenzelm
parents: 65741
diff changeset
   311
      def update(i: Int, bytes: Option[Bytes]): Unit = update(i, bytes.orNull)
65740
83388f09e9ab clarified signature;
wenzelm
parents: 65739
diff changeset
   312
    }
75393
87ebf5a50283 clarified formatting, for the sake of scala3;
wenzelm
parents: 73367
diff changeset
   313
    object date {
65748
1f4a80e80c88 tuned signature;
wenzelm
parents: 65741
diff changeset
   314
      def update(i: Int, date: Date): Unit = db.update_date(stmt, i, date)
1f4a80e80c88 tuned signature;
wenzelm
parents: 65741
diff changeset
   315
      def update(i: Int, date: Option[Date]): Unit = update(i, date.orNull)
1f4a80e80c88 tuned signature;
wenzelm
parents: 65741
diff changeset
   316
    }
65740
83388f09e9ab clarified signature;
wenzelm
parents: 65739
diff changeset
   317
83388f09e9ab clarified signature;
wenzelm
parents: 65739
diff changeset
   318
    def execute(): Boolean = rep.execute()
78538
56e8458ba262 support for execute_batch: multiple statements in one round-trip;
wenzelm
parents: 78422
diff changeset
   319
78540
a6d079e0575d clarified signature: filter batch;
wenzelm
parents: 78538
diff changeset
   320
    def execute_batch(batch: Statement => IterableOnce[Boolean]): Unit = {
a6d079e0575d clarified signature: filter batch;
wenzelm
parents: 78538
diff changeset
   321
      var proper = false
a6d079e0575d clarified signature: filter batch;
wenzelm
parents: 78538
diff changeset
   322
      for (ok <- batch(this).iterator if ok) { rep.addBatch(); proper = true }
a6d079e0575d clarified signature: filter batch;
wenzelm
parents: 78538
diff changeset
   323
      if (proper) {
a6d079e0575d clarified signature: filter batch;
wenzelm
parents: 78538
diff changeset
   324
        val res = rep.executeBatch()
a6d079e0575d clarified signature: filter batch;
wenzelm
parents: 78538
diff changeset
   325
        if (!res.forall(i => i >= 0 || i == java.sql.Statement.SUCCESS_NO_INFO)) {
a6d079e0575d clarified signature: filter batch;
wenzelm
parents: 78538
diff changeset
   326
          throw new Batch_Error(res.toList)
a6d079e0575d clarified signature: filter batch;
wenzelm
parents: 78538
diff changeset
   327
        }
78538
56e8458ba262 support for execute_batch: multiple statements in one round-trip;
wenzelm
parents: 78422
diff changeset
   328
      }
56e8458ba262 support for execute_batch: multiple statements in one round-trip;
wenzelm
parents: 78422
diff changeset
   329
    }
56e8458ba262 support for execute_batch: multiple statements in one round-trip;
wenzelm
parents: 78422
diff changeset
   330
65740
83388f09e9ab clarified signature;
wenzelm
parents: 65739
diff changeset
   331
    def execute_query(): Result = new Result(this, rep.executeQuery())
83388f09e9ab clarified signature;
wenzelm
parents: 65739
diff changeset
   332
78353
wenzelm
parents: 78352
diff changeset
   333
    override def close(): Unit = rep.close()
65740
83388f09e9ab clarified signature;
wenzelm
parents: 65739
diff changeset
   334
  }
83388f09e9ab clarified signature;
wenzelm
parents: 65739
diff changeset
   335
83388f09e9ab clarified signature;
wenzelm
parents: 65739
diff changeset
   336
63790
3d723062dc70 more operations;
wenzelm
parents: 63788
diff changeset
   337
  /* results */
3d723062dc70 more operations;
wenzelm
parents: 63788
diff changeset
   338
78352
10f8f12c61b0 proper close() operation;
wenzelm
parents: 78351
diff changeset
   339
  class Result private[SQL](val stmt: Statement, val rep: ResultSet) extends AutoCloseable {
65740
83388f09e9ab clarified signature;
wenzelm
parents: 65739
diff changeset
   340
    res =>
83388f09e9ab clarified signature;
wenzelm
parents: 65739
diff changeset
   341
83388f09e9ab clarified signature;
wenzelm
parents: 65739
diff changeset
   342
    def next(): Boolean = rep.next()
83388f09e9ab clarified signature;
wenzelm
parents: 65739
diff changeset
   343
75393
87ebf5a50283 clarified formatting, for the sake of scala3;
wenzelm
parents: 73367
diff changeset
   344
    def iterator[A](get: Result => A): Iterator[A] = new Iterator[A] {
65740
83388f09e9ab clarified signature;
wenzelm
parents: 65739
diff changeset
   345
      private var _next: Boolean = res.next()
83388f09e9ab clarified signature;
wenzelm
parents: 65739
diff changeset
   346
      def hasNext: Boolean = _next
73337
0af9e7e4476f tuned --- fewer warnings;
wenzelm
parents: 71601
diff changeset
   347
      def next(): A = { val x = get(res); _next = res.next(); x }
65740
83388f09e9ab clarified signature;
wenzelm
parents: 65739
diff changeset
   348
    }
83388f09e9ab clarified signature;
wenzelm
parents: 65739
diff changeset
   349
83388f09e9ab clarified signature;
wenzelm
parents: 65739
diff changeset
   350
    def bool(column: Column): Boolean = rep.getBoolean(column.name)
83388f09e9ab clarified signature;
wenzelm
parents: 65739
diff changeset
   351
    def int(column: Column): Int = rep.getInt(column.name)
83388f09e9ab clarified signature;
wenzelm
parents: 65739
diff changeset
   352
    def long(column: Column): Long = rep.getLong(column.name)
83388f09e9ab clarified signature;
wenzelm
parents: 65739
diff changeset
   353
    def double(column: Column): Double = rep.getDouble(column.name)
75393
87ebf5a50283 clarified formatting, for the sake of scala3;
wenzelm
parents: 73367
diff changeset
   354
    def string(column: Column): String = {
65740
83388f09e9ab clarified signature;
wenzelm
parents: 65739
diff changeset
   355
      val s = rep.getString(column.name)
83388f09e9ab clarified signature;
wenzelm
parents: 65739
diff changeset
   356
      if (s == null) "" else s
83388f09e9ab clarified signature;
wenzelm
parents: 65739
diff changeset
   357
    }
75393
87ebf5a50283 clarified formatting, for the sake of scala3;
wenzelm
parents: 73367
diff changeset
   358
    def bytes(column: Column): Bytes = {
65740
83388f09e9ab clarified signature;
wenzelm
parents: 65739
diff changeset
   359
      val bs = rep.getBytes(column.name)
83388f09e9ab clarified signature;
wenzelm
parents: 65739
diff changeset
   360
      if (bs == null) Bytes.empty else Bytes(bs)
83388f09e9ab clarified signature;
wenzelm
parents: 65739
diff changeset
   361
    }
83388f09e9ab clarified signature;
wenzelm
parents: 65739
diff changeset
   362
    def date(column: Column): Date = stmt.db.date(res, column)
83388f09e9ab clarified signature;
wenzelm
parents: 65739
diff changeset
   363
71601
97ccf48c2f0c misc tuning based on hints by IntelliJ IDEA;
wenzelm
parents: 69980
diff changeset
   364
    def timing(c1: Column, c2: Column, c3: Column): Timing =
65741
wenzelm
parents: 65740
diff changeset
   365
      Timing(Time.ms(long(c1)), Time.ms(long(c2)), Time.ms(long(c3)))
wenzelm
parents: 65740
diff changeset
   366
75393
87ebf5a50283 clarified formatting, for the sake of scala3;
wenzelm
parents: 73367
diff changeset
   367
    def get[A](column: Column, f: Column => A): Option[A] = {
65740
83388f09e9ab clarified signature;
wenzelm
parents: 65739
diff changeset
   368
      val x = f(column)
77598
6370d9e5ab50 proper support for Option[Date] columns;
wenzelm
parents: 77596
diff changeset
   369
      if (rep.wasNull || x == null) None else Some(x)
65740
83388f09e9ab clarified signature;
wenzelm
parents: 65739
diff changeset
   370
    }
71601
97ccf48c2f0c misc tuning based on hints by IntelliJ IDEA;
wenzelm
parents: 69980
diff changeset
   371
    def get_bool(column: Column): Option[Boolean] = get(column, bool)
97ccf48c2f0c misc tuning based on hints by IntelliJ IDEA;
wenzelm
parents: 69980
diff changeset
   372
    def get_int(column: Column): Option[Int] = get(column, int)
97ccf48c2f0c misc tuning based on hints by IntelliJ IDEA;
wenzelm
parents: 69980
diff changeset
   373
    def get_long(column: Column): Option[Long] = get(column, long)
97ccf48c2f0c misc tuning based on hints by IntelliJ IDEA;
wenzelm
parents: 69980
diff changeset
   374
    def get_double(column: Column): Option[Double] = get(column, double)
97ccf48c2f0c misc tuning based on hints by IntelliJ IDEA;
wenzelm
parents: 69980
diff changeset
   375
    def get_string(column: Column): Option[String] = get(column, string)
97ccf48c2f0c misc tuning based on hints by IntelliJ IDEA;
wenzelm
parents: 69980
diff changeset
   376
    def get_bytes(column: Column): Option[Bytes] = get(column, bytes)
97ccf48c2f0c misc tuning based on hints by IntelliJ IDEA;
wenzelm
parents: 69980
diff changeset
   377
    def get_date(column: Column): Option[Date] = get(column, date)
78352
10f8f12c61b0 proper close() operation;
wenzelm
parents: 78351
diff changeset
   378
10f8f12c61b0 proper close() operation;
wenzelm
parents: 78351
diff changeset
   379
    override def close(): Unit = rep.close()
63790
3d723062dc70 more operations;
wenzelm
parents: 63788
diff changeset
   380
  }
65006
632bdf7b8bab clarified modules;
wenzelm
parents: 65004
diff changeset
   381
65740
83388f09e9ab clarified signature;
wenzelm
parents: 65739
diff changeset
   382
83388f09e9ab clarified signature;
wenzelm
parents: 65739
diff changeset
   383
  /* database */
83388f09e9ab clarified signature;
wenzelm
parents: 65739
diff changeset
   384
75393
87ebf5a50283 clarified formatting, for the sake of scala3;
wenzelm
parents: 73367
diff changeset
   385
  trait Database extends AutoCloseable {
65740
83388f09e9ab clarified signature;
wenzelm
parents: 65739
diff changeset
   386
    db =>
83388f09e9ab clarified signature;
wenzelm
parents: 65739
diff changeset
   387
77542
2da5562114c5 clarified signature: more uniform operations;
wenzelm
parents: 77541
diff changeset
   388
    def is_sqlite: Boolean = isInstanceOf[SQLite.Database]
2da5562114c5 clarified signature: more uniform operations;
wenzelm
parents: 77541
diff changeset
   389
    def is_postgresql: Boolean = isInstanceOf[PostgreSQL.Database]
76363
f7174238b5e3 no compression for database server: let PostgreSQL/TOAST do the job;
wenzelm
parents: 76122
diff changeset
   390
78389
41e8ae87184d clarified signature: eliminate SQL.Tables.empty to avoid confusion (see also 0bd366fad888);
wenzelm
parents: 78386
diff changeset
   391
    def vacuum(tables: List[SQL.Table] = Nil): Unit =
78390
a84f8fca0833 clarified "vacuum" (again, reverting 0bd366fad888);
wenzelm
parents: 78389
diff changeset
   392
      if (is_sqlite) execute_statement("VACUUM")  // always FULL
a84f8fca0833 clarified "vacuum" (again, reverting 0bd366fad888);
wenzelm
parents: 78389
diff changeset
   393
      else if (tables.isEmpty) execute_statement("VACUUM FULL")
a84f8fca0833 clarified "vacuum" (again, reverting 0bd366fad888);
wenzelm
parents: 78389
diff changeset
   394
      else if (postgresql_major_version.get <= 10) {
a84f8fca0833 clarified "vacuum" (again, reverting 0bd366fad888);
wenzelm
parents: 78389
diff changeset
   395
        for (t <- tables) execute_statement("VACUUM " + t.ident)
78273
95a3bb4d7e38 clarified "vacuum" operation for various database versions (PostgreSQL <= 10 is strictly speaking obsolete, but still used on some test machines);
wenzelm
parents: 78271
diff changeset
   396
      }
78390
a84f8fca0833 clarified "vacuum" (again, reverting 0bd366fad888);
wenzelm
parents: 78389
diff changeset
   397
      else execute_statement("VACUUM" + commas(tables.map(_.ident)))
77348
885842575e2a more uniform operations;
wenzelm
parents: 77346
diff changeset
   398
77527
790085b1002f more detailed table "isabelle_build_serial": allow to monitor activity of build_process instances;
wenzelm
parents: 77403
diff changeset
   399
    def now(): Date
790085b1002f more detailed table "isabelle_build_serial": allow to monitor activity of build_process instances;
wenzelm
parents: 77403
diff changeset
   400
65740
83388f09e9ab clarified signature;
wenzelm
parents: 65739
diff changeset
   401
65008
ed2eedf786f3 clarified handling of SQL.Type;
wenzelm
parents: 65007
diff changeset
   402
    /* types */
ed2eedf786f3 clarified handling of SQL.Type;
wenzelm
parents: 65007
diff changeset
   403
65730
7ae61e72a678 clarified signature;
wenzelm
parents: 65729
diff changeset
   404
    def sql_type(T: Type.Value): Source
65008
ed2eedf786f3 clarified handling of SQL.Type;
wenzelm
parents: 65007
diff changeset
   405
ed2eedf786f3 clarified handling of SQL.Type;
wenzelm
parents: 65007
diff changeset
   406
65006
632bdf7b8bab clarified modules;
wenzelm
parents: 65004
diff changeset
   407
    /* connection */
632bdf7b8bab clarified modules;
wenzelm
parents: 65004
diff changeset
   408
632bdf7b8bab clarified modules;
wenzelm
parents: 65004
diff changeset
   409
    def connection: Connection
632bdf7b8bab clarified modules;
wenzelm
parents: 65004
diff changeset
   410
77037
164a21e5d568 support specific connection types, for additional operations;
wenzelm
parents: 76870
diff changeset
   411
    def sqlite_connection: Option[JDBC4Connection] =
164a21e5d568 support specific connection types, for additional operations;
wenzelm
parents: 76870
diff changeset
   412
      connection match { case conn: JDBC4Connection => Some(conn) case _ => None }
164a21e5d568 support specific connection types, for additional operations;
wenzelm
parents: 76870
diff changeset
   413
164a21e5d568 support specific connection types, for additional operations;
wenzelm
parents: 76870
diff changeset
   414
    def postgresql_connection: Option[PGConnection] =
164a21e5d568 support specific connection types, for additional operations;
wenzelm
parents: 76870
diff changeset
   415
      connection match { case conn: PGConnection => Some(conn) case _ => None }
164a21e5d568 support specific connection types, for additional operations;
wenzelm
parents: 76870
diff changeset
   416
164a21e5d568 support specific connection types, for additional operations;
wenzelm
parents: 76870
diff changeset
   417
    def the_sqlite_connection: JDBC4Connection =
164a21e5d568 support specific connection types, for additional operations;
wenzelm
parents: 76870
diff changeset
   418
      sqlite_connection getOrElse
164a21e5d568 support specific connection types, for additional operations;
wenzelm
parents: 76870
diff changeset
   419
        error("SQLite connection expected, but found " + connection.getClass.getName)
164a21e5d568 support specific connection types, for additional operations;
wenzelm
parents: 76870
diff changeset
   420
164a21e5d568 support specific connection types, for additional operations;
wenzelm
parents: 76870
diff changeset
   421
    def the_postgresql_connection: PGConnection =
164a21e5d568 support specific connection types, for additional operations;
wenzelm
parents: 76870
diff changeset
   422
      postgresql_connection getOrElse
164a21e5d568 support specific connection types, for additional operations;
wenzelm
parents: 76870
diff changeset
   423
        error("PostgreSQL connection expected, but found " + connection.getClass.getName)
164a21e5d568 support specific connection types, for additional operations;
wenzelm
parents: 76870
diff changeset
   424
78273
95a3bb4d7e38 clarified "vacuum" operation for various database versions (PostgreSQL <= 10 is strictly speaking obsolete, but still used on some test machines);
wenzelm
parents: 78271
diff changeset
   425
    def postgresql_major_version: Option[Int] =
95a3bb4d7e38 clarified "vacuum" operation for various database versions (PostgreSQL <= 10 is strictly speaking obsolete, but still used on some test machines);
wenzelm
parents: 78271
diff changeset
   426
      if (is_postgresql) {
95a3bb4d7e38 clarified "vacuum" operation for various database versions (PostgreSQL <= 10 is strictly speaking obsolete, but still used on some test machines);
wenzelm
parents: 78271
diff changeset
   427
        def err(s: String): Nothing = error("Bad PostgreSQL version " + s)
95a3bb4d7e38 clarified "vacuum" operation for various database versions (PostgreSQL <= 10 is strictly speaking obsolete, but still used on some test machines);
wenzelm
parents: 78271
diff changeset
   428
95a3bb4d7e38 clarified "vacuum" operation for various database versions (PostgreSQL <= 10 is strictly speaking obsolete, but still used on some test machines);
wenzelm
parents: 78271
diff changeset
   429
        the_postgresql_connection.getParameterStatus("server_version") match {
95a3bb4d7e38 clarified "vacuum" operation for various database versions (PostgreSQL <= 10 is strictly speaking obsolete, but still used on some test machines);
wenzelm
parents: 78271
diff changeset
   430
          case null => err("null")
95a3bb4d7e38 clarified "vacuum" operation for various database versions (PostgreSQL <= 10 is strictly speaking obsolete, but still used on some test machines);
wenzelm
parents: 78271
diff changeset
   431
          case str =>
95a3bb4d7e38 clarified "vacuum" operation for various database versions (PostgreSQL <= 10 is strictly speaking obsolete, but still used on some test machines);
wenzelm
parents: 78271
diff changeset
   432
            str.iterator.takeWhile(Symbol.is_ascii_digit).mkString match {
95a3bb4d7e38 clarified "vacuum" operation for various database versions (PostgreSQL <= 10 is strictly speaking obsolete, but still used on some test machines);
wenzelm
parents: 78271
diff changeset
   433
              case Value.Int(m) => Some(m)
95a3bb4d7e38 clarified "vacuum" operation for various database versions (PostgreSQL <= 10 is strictly speaking obsolete, but still used on some test machines);
wenzelm
parents: 78271
diff changeset
   434
              case _ => err(quote(str))
95a3bb4d7e38 clarified "vacuum" operation for various database versions (PostgreSQL <= 10 is strictly speaking obsolete, but still used on some test machines);
wenzelm
parents: 78271
diff changeset
   435
            }
95a3bb4d7e38 clarified "vacuum" operation for various database versions (PostgreSQL <= 10 is strictly speaking obsolete, but still used on some test machines);
wenzelm
parents: 78271
diff changeset
   436
        }
95a3bb4d7e38 clarified "vacuum" operation for various database versions (PostgreSQL <= 10 is strictly speaking obsolete, but still used on some test machines);
wenzelm
parents: 78271
diff changeset
   437
      }
95a3bb4d7e38 clarified "vacuum" operation for various database versions (PostgreSQL <= 10 is strictly speaking obsolete, but still used on some test machines);
wenzelm
parents: 78271
diff changeset
   438
      else None
95a3bb4d7e38 clarified "vacuum" operation for various database versions (PostgreSQL <= 10 is strictly speaking obsolete, but still used on some test machines);
wenzelm
parents: 78271
diff changeset
   439
78353
wenzelm
parents: 78352
diff changeset
   440
    override def close(): Unit = connection.close()
65006
632bdf7b8bab clarified modules;
wenzelm
parents: 65004
diff changeset
   441
78369
ba71ea02d965 more robust Java/Scala multithreading: transaction is always connection.synchronized;
wenzelm
parents: 78366
diff changeset
   442
    def transaction[A](body: => A): A = connection.synchronized {
78383
d032bf604e93 more robust: exclude accidental nesting (synchronized block is re-entrant);
wenzelm
parents: 78381
diff changeset
   443
      require(connection.getAutoCommit(), "transaction already active")
65006
632bdf7b8bab clarified modules;
wenzelm
parents: 65004
diff changeset
   444
      try {
632bdf7b8bab clarified modules;
wenzelm
parents: 65004
diff changeset
   445
        connection.setAutoCommit(false)
65022
cda3d36aceb2 proper transaction for PostgreSQL;
wenzelm
parents: 65021
diff changeset
   446
        try {
cda3d36aceb2 proper transaction for PostgreSQL;
wenzelm
parents: 65021
diff changeset
   447
          val result = body
75776
72e77c8307ec tuned signature, following hints by IntelliJ IDEA;
wenzelm
parents: 75393
diff changeset
   448
          connection.commit()
65022
cda3d36aceb2 proper transaction for PostgreSQL;
wenzelm
parents: 65021
diff changeset
   449
          result
cda3d36aceb2 proper transaction for PostgreSQL;
wenzelm
parents: 65021
diff changeset
   450
        }
78386
ee588c4b5557 more elementary transaction implementation (despite fda3f7a158b9 and 9da65bc75610);
wenzelm
parents: 78384
diff changeset
   451
        catch { case exn: Throwable => connection.rollback(); throw exn }
65006
632bdf7b8bab clarified modules;
wenzelm
parents: 65004
diff changeset
   452
      }
78383
d032bf604e93 more robust: exclude accidental nesting (synchronized block is re-entrant);
wenzelm
parents: 78381
diff changeset
   453
      finally { connection.setAutoCommit(true) }
65006
632bdf7b8bab clarified modules;
wenzelm
parents: 65004
diff changeset
   454
    }
632bdf7b8bab clarified modules;
wenzelm
parents: 65004
diff changeset
   455
78355
9fbc6a043268 support trace of transaction_lock via property "isabelle.transaction_log";
wenzelm
parents: 78354
diff changeset
   456
    def transaction_lock[A](
9fbc6a043268 support trace of transaction_lock via property "isabelle.transaction_log";
wenzelm
parents: 78354
diff changeset
   457
      tables: Tables,
9fbc6a043268 support trace of transaction_lock via property "isabelle.transaction_log";
wenzelm
parents: 78354
diff changeset
   458
      create: Boolean = false,
78356
974dbe256a37 more informative trace;
wenzelm
parents: 78355
diff changeset
   459
      label: String = "",
78355
9fbc6a043268 support trace of transaction_lock via property "isabelle.transaction_log";
wenzelm
parents: 78354
diff changeset
   460
      log: Logger = new System_Logger
9fbc6a043268 support trace of transaction_lock via property "isabelle.transaction_log";
wenzelm
parents: 78354
diff changeset
   461
    )(body: => A): A = {
78358
f5cf8e500dee clarified isabelle.transaction_log: support time_min (in ms);
wenzelm
parents: 78356
diff changeset
   462
      val prop = "isabelle.transaction_trace"
f5cf8e500dee clarified isabelle.transaction_log: support time_min (in ms);
wenzelm
parents: 78356
diff changeset
   463
      val trace_min =
f5cf8e500dee clarified isabelle.transaction_log: support time_min (in ms);
wenzelm
parents: 78356
diff changeset
   464
        System.getProperty(prop, "") match {
78365
bb5e2a7198e3 more standard time unit;
wenzelm
parents: 78363
diff changeset
   465
          case Value.Seconds(t) => t
78358
f5cf8e500dee clarified isabelle.transaction_log: support time_min (in ms);
wenzelm
parents: 78356
diff changeset
   466
          case "true" => Time.min
f5cf8e500dee clarified isabelle.transaction_log: support time_min (in ms);
wenzelm
parents: 78356
diff changeset
   467
          case "false" | "" => Time.max
f5cf8e500dee clarified isabelle.transaction_log: support time_min (in ms);
wenzelm
parents: 78356
diff changeset
   468
          case s => error("Bad system property " + prop + ": " + quote(s))
f5cf8e500dee clarified isabelle.transaction_log: support time_min (in ms);
wenzelm
parents: 78356
diff changeset
   469
        }
78355
9fbc6a043268 support trace of transaction_lock via property "isabelle.transaction_log";
wenzelm
parents: 78354
diff changeset
   470
78362
8da30ae02dda global transaction_count;
wenzelm
parents: 78361
diff changeset
   471
      val trace_count = - SQL.transaction_count()
78355
9fbc6a043268 support trace of transaction_lock via property "isabelle.transaction_log";
wenzelm
parents: 78354
diff changeset
   472
      val trace_start = Time.now()
78361
b625cdabf963 tuned output;
wenzelm
parents: 78358
diff changeset
   473
      var trace_nl = false
78355
9fbc6a043268 support trace of transaction_lock via property "isabelle.transaction_log";
wenzelm
parents: 78354
diff changeset
   474
78380
f8e3b228670c tuned output;
wenzelm
parents: 78378
diff changeset
   475
      def trace(msg: String): Unit = {
78358
f5cf8e500dee clarified isabelle.transaction_log: support time_min (in ms);
wenzelm
parents: 78356
diff changeset
   476
        val trace_time = Time.now() - trace_start
78381
9c86b609eeb6 removed junk (amending f8e3b228670c);
wenzelm
parents: 78380
diff changeset
   477
        if (trace_time >= trace_min) {
78363
fca9ec5a615c tuned output;
wenzelm
parents: 78362
diff changeset
   478
          val nl = if (trace_nl) "" else { trace_nl = true; "\n" }
78361
b625cdabf963 tuned output;
wenzelm
parents: 78358
diff changeset
   479
          log(nl + trace_time + " transaction " + trace_count +
b625cdabf963 tuned output;
wenzelm
parents: 78358
diff changeset
   480
            if_proper(label, " " + label) + ": " + msg)
78355
9fbc6a043268 support trace of transaction_lock via property "isabelle.transaction_log";
wenzelm
parents: 78354
diff changeset
   481
        }
9fbc6a043268 support trace of transaction_lock via property "isabelle.transaction_log";
wenzelm
parents: 78354
diff changeset
   482
      }
9fbc6a043268 support trace of transaction_lock via property "isabelle.transaction_log";
wenzelm
parents: 78354
diff changeset
   483
78373
2deecde7f1f6 more informative trace;
wenzelm
parents: 78371
diff changeset
   484
      try {
2deecde7f1f6 more informative trace;
wenzelm
parents: 78371
diff changeset
   485
        val res =
2deecde7f1f6 more informative trace;
wenzelm
parents: 78371
diff changeset
   486
          transaction {
2deecde7f1f6 more informative trace;
wenzelm
parents: 78371
diff changeset
   487
            trace("begin")
2deecde7f1f6 more informative trace;
wenzelm
parents: 78371
diff changeset
   488
            if (tables.lock(db, create = create)) {
78380
f8e3b228670c tuned output;
wenzelm
parents: 78378
diff changeset
   489
              trace("locked " + commas_quote(tables.list.map(_.name)))
78373
2deecde7f1f6 more informative trace;
wenzelm
parents: 78371
diff changeset
   490
            }
2deecde7f1f6 more informative trace;
wenzelm
parents: 78371
diff changeset
   491
            val res = Exn.capture { body }
2deecde7f1f6 more informative trace;
wenzelm
parents: 78371
diff changeset
   492
            trace("end")
2deecde7f1f6 more informative trace;
wenzelm
parents: 78371
diff changeset
   493
            res
78371
928e031b7c52 tuned output;
wenzelm
parents: 78370
diff changeset
   494
          }
78373
2deecde7f1f6 more informative trace;
wenzelm
parents: 78371
diff changeset
   495
        trace("commit")
2deecde7f1f6 more informative trace;
wenzelm
parents: 78371
diff changeset
   496
        Exn.release(res)
2deecde7f1f6 more informative trace;
wenzelm
parents: 78371
diff changeset
   497
      }
2deecde7f1f6 more informative trace;
wenzelm
parents: 78371
diff changeset
   498
      catch { case exn: Throwable => trace("crash"); throw exn }
78355
9fbc6a043268 support trace of transaction_lock via property "isabelle.transaction_log";
wenzelm
parents: 78354
diff changeset
   499
    }
77596
dd8b08729458 clarified signature;
wenzelm
parents: 77591
diff changeset
   500
dd8b08729458 clarified signature;
wenzelm
parents: 77591
diff changeset
   501
    def lock_tables(tables: List[Table]): Source = ""  // PostgreSQL only
77590
edc96be6b939 explicit locking for PostgreSQL --- neither available nor required for SQLite;
wenzelm
parents: 77552
diff changeset
   502
65006
632bdf7b8bab clarified modules;
wenzelm
parents: 65004
diff changeset
   503
65740
83388f09e9ab clarified signature;
wenzelm
parents: 65739
diff changeset
   504
    /* statements and results */
83388f09e9ab clarified signature;
wenzelm
parents: 65739
diff changeset
   505
83388f09e9ab clarified signature;
wenzelm
parents: 65739
diff changeset
   506
    def statement(sql: Source): Statement =
83388f09e9ab clarified signature;
wenzelm
parents: 65739
diff changeset
   507
      new Statement(db, connection.prepareStatement(sql))
65006
632bdf7b8bab clarified modules;
wenzelm
parents: 65004
diff changeset
   508
65740
83388f09e9ab clarified signature;
wenzelm
parents: 65739
diff changeset
   509
    def using_statement[A](sql: Source)(f: Statement => A): A =
83388f09e9ab clarified signature;
wenzelm
parents: 65739
diff changeset
   510
      using(statement(sql))(f)
65006
632bdf7b8bab clarified modules;
wenzelm
parents: 65004
diff changeset
   511
77541
9d9b30741fc4 tuned signature: reduce boilerplate;
wenzelm
parents: 77540
diff changeset
   512
    def execute_statement(sql: Source, body: Statement => Unit = _ => ()): Unit =
9d9b30741fc4 tuned signature: reduce boilerplate;
wenzelm
parents: 77540
diff changeset
   513
      using_statement(sql) { stmt => body(stmt); stmt.execute() }
77540
c537905c2125 tuned signature;
wenzelm
parents: 77537
diff changeset
   514
78538
56e8458ba262 support for execute_batch: multiple statements in one round-trip;
wenzelm
parents: 78422
diff changeset
   515
    def execute_batch_statement(
56e8458ba262 support for execute_batch: multiple statements in one round-trip;
wenzelm
parents: 78422
diff changeset
   516
      sql: Source,
78540
a6d079e0575d clarified signature: filter batch;
wenzelm
parents: 78538
diff changeset
   517
      batch: Statement => IterableOnce[Boolean] = _ => Nil
78538
56e8458ba262 support for execute_batch: multiple statements in one round-trip;
wenzelm
parents: 78422
diff changeset
   518
    ): Unit = using_statement(sql) { stmt => stmt.execute_batch(batch) }
56e8458ba262 support for execute_batch: multiple statements in one round-trip;
wenzelm
parents: 78422
diff changeset
   519
77552
080422b3d914 clarified signature: reduce boilerplate;
wenzelm
parents: 77543
diff changeset
   520
    def execute_query_statement[A, B](
080422b3d914 clarified signature: reduce boilerplate;
wenzelm
parents: 77543
diff changeset
   521
      sql: Source,
080422b3d914 clarified signature: reduce boilerplate;
wenzelm
parents: 77543
diff changeset
   522
      make_result: Iterator[A] => B,
080422b3d914 clarified signature: reduce boilerplate;
wenzelm
parents: 77543
diff changeset
   523
      get: Result => A
78352
10f8f12c61b0 proper close() operation;
wenzelm
parents: 78351
diff changeset
   524
    ): B = {
10f8f12c61b0 proper close() operation;
wenzelm
parents: 78351
diff changeset
   525
      using_statement(sql) { stmt =>
10f8f12c61b0 proper close() operation;
wenzelm
parents: 78351
diff changeset
   526
        using(stmt.execute_query()) { res => make_result(res.iterator(get)) }
10f8f12c61b0 proper close() operation;
wenzelm
parents: 78351
diff changeset
   527
      }
10f8f12c61b0 proper close() operation;
wenzelm
parents: 78351
diff changeset
   528
    }
77552
080422b3d914 clarified signature: reduce boilerplate;
wenzelm
parents: 77543
diff changeset
   529
080422b3d914 clarified signature: reduce boilerplate;
wenzelm
parents: 77543
diff changeset
   530
    def execute_query_statementO[A](sql: Source, get: Result => A): Option[A] =
080422b3d914 clarified signature: reduce boilerplate;
wenzelm
parents: 77543
diff changeset
   531
      execute_query_statement[A, Option[A]](sql, _.nextOption, get)
080422b3d914 clarified signature: reduce boilerplate;
wenzelm
parents: 77543
diff changeset
   532
080422b3d914 clarified signature: reduce boilerplate;
wenzelm
parents: 77543
diff changeset
   533
    def execute_query_statementB(sql: Source): Boolean =
78352
10f8f12c61b0 proper close() operation;
wenzelm
parents: 78351
diff changeset
   534
      using_statement(sql)(stmt => using(stmt.execute_query())(_.next()))
77552
080422b3d914 clarified signature: reduce boilerplate;
wenzelm
parents: 77543
diff changeset
   535
65748
1f4a80e80c88 tuned signature;
wenzelm
parents: 65741
diff changeset
   536
    def update_date(stmt: Statement, i: Int, date: Date): Unit
65740
83388f09e9ab clarified signature;
wenzelm
parents: 65739
diff changeset
   537
    def date(res: Result, column: Column): Date
65006
632bdf7b8bab clarified modules;
wenzelm
parents: 65004
diff changeset
   538
65730
7ae61e72a678 clarified signature;
wenzelm
parents: 65729
diff changeset
   539
    def insert_permissive(table: Table, sql: Source = ""): Source
65703
cead65c19f2e more direct insert_permissive statement, which avoids somewhat fragile nested transactions;
wenzelm
parents: 65702
diff changeset
   540
65006
632bdf7b8bab clarified modules;
wenzelm
parents: 65004
diff changeset
   541
65669
d2f19b4a16ae support for views;
wenzelm
parents: 65668
diff changeset
   542
    /* tables and views */
65006
632bdf7b8bab clarified modules;
wenzelm
parents: 65004
diff changeset
   543
78375
234f2ff9afe6 clarified signature: more specific exists_table --- avoid retrieving full list beforehand;
wenzelm
parents: 78373
diff changeset
   544
    def get_tables(pattern: String = "%"): List[String] = {
65740
83388f09e9ab clarified signature;
wenzelm
parents: 65739
diff changeset
   545
      val result = new mutable.ListBuffer[String]
78375
234f2ff9afe6 clarified signature: more specific exists_table --- avoid retrieving full list beforehand;
wenzelm
parents: 78373
diff changeset
   546
      val rs = connection.getMetaData.getTables(null, null, pattern, null)
65740
83388f09e9ab clarified signature;
wenzelm
parents: 65739
diff changeset
   547
      while (rs.next) { result += rs.getString(3) }
83388f09e9ab clarified signature;
wenzelm
parents: 65739
diff changeset
   548
      result.toList
83388f09e9ab clarified signature;
wenzelm
parents: 65739
diff changeset
   549
    }
65006
632bdf7b8bab clarified modules;
wenzelm
parents: 65004
diff changeset
   550
78375
234f2ff9afe6 clarified signature: more specific exists_table --- avoid retrieving full list beforehand;
wenzelm
parents: 78373
diff changeset
   551
    def exists_table(name: String): Boolean = {
234f2ff9afe6 clarified signature: more specific exists_table --- avoid retrieving full list beforehand;
wenzelm
parents: 78373
diff changeset
   552
      val escape = connection.getMetaData.getSearchStringEscape
234f2ff9afe6 clarified signature: more specific exists_table --- avoid retrieving full list beforehand;
wenzelm
parents: 78373
diff changeset
   553
      val pattern =
234f2ff9afe6 clarified signature: more specific exists_table --- avoid retrieving full list beforehand;
wenzelm
parents: 78373
diff changeset
   554
        name.iterator.map(c =>
234f2ff9afe6 clarified signature: more specific exists_table --- avoid retrieving full list beforehand;
wenzelm
parents: 78373
diff changeset
   555
          (if (c == '_' || c == '%' || c == escape(0)) escape else "") + c).mkString
234f2ff9afe6 clarified signature: more specific exists_table --- avoid retrieving full list beforehand;
wenzelm
parents: 78373
diff changeset
   556
      get_tables(pattern = pattern).nonEmpty
234f2ff9afe6 clarified signature: more specific exists_table --- avoid retrieving full list beforehand;
wenzelm
parents: 78373
diff changeset
   557
    }
234f2ff9afe6 clarified signature: more specific exists_table --- avoid retrieving full list beforehand;
wenzelm
parents: 78373
diff changeset
   558
234f2ff9afe6 clarified signature: more specific exists_table --- avoid retrieving full list beforehand;
wenzelm
parents: 78373
diff changeset
   559
    def exists_table(table: Table): Boolean = exists_table(table.name)
234f2ff9afe6 clarified signature: more specific exists_table --- avoid retrieving full list beforehand;
wenzelm
parents: 78373
diff changeset
   560
78392
27c2fa1db6ed more uniform guard (!exists_table(table)): avoid "ALTER TABLE" on already existing table, which could lead to deadlocks if this is presently locked;
wenzelm
parents: 78391
diff changeset
   561
    def create_table(table: Table, sql: Source = ""): Unit = {
27c2fa1db6ed more uniform guard (!exists_table(table)): avoid "ALTER TABLE" on already existing table, which could lead to deadlocks if this is presently locked;
wenzelm
parents: 78391
diff changeset
   562
      if (!exists_table(table)) {
27c2fa1db6ed more uniform guard (!exists_table(table)): avoid "ALTER TABLE" on already existing table, which could lead to deadlocks if this is presently locked;
wenzelm
parents: 78391
diff changeset
   563
        execute_statement(table.create(sql_type) + SQL.separate(sql))
27c2fa1db6ed more uniform guard (!exists_table(table)): avoid "ALTER TABLE" on already existing table, which could lead to deadlocks if this is presently locked;
wenzelm
parents: 78391
diff changeset
   564
        if (is_postgresql) {
27c2fa1db6ed more uniform guard (!exists_table(table)): avoid "ALTER TABLE" on already existing table, which could lead to deadlocks if this is presently locked;
wenzelm
parents: 78391
diff changeset
   565
          for (column <- table.columns if column.T == SQL.Type.Bytes) {
27c2fa1db6ed more uniform guard (!exists_table(table)): avoid "ALTER TABLE" on already existing table, which could lead to deadlocks if this is presently locked;
wenzelm
parents: 78391
diff changeset
   566
            execute_statement(
27c2fa1db6ed more uniform guard (!exists_table(table)): avoid "ALTER TABLE" on already existing table, which could lead to deadlocks if this is presently locked;
wenzelm
parents: 78391
diff changeset
   567
              "ALTER TABLE " + table + " ALTER COLUMN " + column + " SET STORAGE EXTERNAL")
27c2fa1db6ed more uniform guard (!exists_table(table)): avoid "ALTER TABLE" on already existing table, which could lead to deadlocks if this is presently locked;
wenzelm
parents: 78391
diff changeset
   568
          }
77681
1db732e6c3d2 back to compression in Isabelle/Scala (in contrast to f7174238b5e3), e.g. relevant for old_command_timings_blob, but also for prospective heaps;
wenzelm
parents: 77665
diff changeset
   569
        }
1db732e6c3d2 back to compression in Isabelle/Scala (in contrast to f7174238b5e3), e.g. relevant for old_command_timings_blob, but also for prospective heaps;
wenzelm
parents: 77665
diff changeset
   570
      }
1db732e6c3d2 back to compression in Isabelle/Scala (in contrast to f7174238b5e3), e.g. relevant for old_command_timings_blob, but also for prospective heaps;
wenzelm
parents: 77665
diff changeset
   571
    }
65006
632bdf7b8bab clarified modules;
wenzelm
parents: 65004
diff changeset
   572
78392
27c2fa1db6ed more uniform guard (!exists_table(table)): avoid "ALTER TABLE" on already existing table, which could lead to deadlocks if this is presently locked;
wenzelm
parents: 78391
diff changeset
   573
    def create_view(table: Table): Unit = {
27c2fa1db6ed more uniform guard (!exists_table(table)): avoid "ALTER TABLE" on already existing table, which could lead to deadlocks if this is presently locked;
wenzelm
parents: 78391
diff changeset
   574
      if (!exists_table(table)) {
77540
c537905c2125 tuned signature;
wenzelm
parents: 77537
diff changeset
   575
        execute_statement("CREATE VIEW " + table + " AS " + { table.query; table.body })
65690
wenzelm
parents: 65677
diff changeset
   576
      }
65691
2229276a1f99 eliminated redundant type SQL.View;
wenzelm
parents: 65690
diff changeset
   577
    }
65006
632bdf7b8bab clarified modules;
wenzelm
parents: 65004
diff changeset
   578
  }
78362
8da30ae02dda global transaction_count;
wenzelm
parents: 78361
diff changeset
   579
8da30ae02dda global transaction_count;
wenzelm
parents: 78361
diff changeset
   580
8da30ae02dda global transaction_count;
wenzelm
parents: 78361
diff changeset
   581
  private val transaction_count = Counter.make()
63778
e06e899b78d0 clarified modules;
wenzelm
parents:
diff changeset
   582
}
65006
632bdf7b8bab clarified modules;
wenzelm
parents: 65004
diff changeset
   583
632bdf7b8bab clarified modules;
wenzelm
parents: 65004
diff changeset
   584
632bdf7b8bab clarified modules;
wenzelm
parents: 65004
diff changeset
   585
632bdf7b8bab clarified modules;
wenzelm
parents: 65004
diff changeset
   586
/** SQLite **/
632bdf7b8bab clarified modules;
wenzelm
parents: 65004
diff changeset
   587
75393
87ebf5a50283 clarified formatting, for the sake of scala3;
wenzelm
parents: 73367
diff changeset
   588
object SQLite {
65021
da8ae577d171 clarified Date storage;
wenzelm
parents: 65020
diff changeset
   589
  // see https://www.sqlite.org/lang_datefunc.html
da8ae577d171 clarified Date storage;
wenzelm
parents: 65020
diff changeset
   590
  val date_format: Date.Format = Date.Format("uuuu-MM-dd HH:mm:ss.SSS x")
da8ae577d171 clarified Date storage;
wenzelm
parents: 65020
diff changeset
   591
75393
87ebf5a50283 clarified formatting, for the sake of scala3;
wenzelm
parents: 73367
diff changeset
   592
  lazy val init_jdbc: Unit = {
65890
1b004f5974af refer to already extracted library files, to avoid tmp files produced by SQLiteJDBCLoader, which tend to remain after JVM crash;
wenzelm
parents: 65804
diff changeset
   593
    val lib_path = Path.explode("$ISABELLE_SQLITE_HOME/" + Platform.jvm_platform)
76529
ded37aade88e clarified signature;
wenzelm
parents: 76363
diff changeset
   594
    val lib_name = File.get_file(lib_path).file_name
ded37aade88e clarified signature;
wenzelm
parents: 76363
diff changeset
   595
65890
1b004f5974af refer to already extracted library files, to avoid tmp files produced by SQLiteJDBCLoader, which tend to remain after JVM crash;
wenzelm
parents: 65804
diff changeset
   596
    System.setProperty("org.sqlite.lib.path", File.platform_path(lib_path))
1b004f5974af refer to already extracted library files, to avoid tmp files produced by SQLiteJDBCLoader, which tend to remain after JVM crash;
wenzelm
parents: 65804
diff changeset
   597
    System.setProperty("org.sqlite.lib.name", lib_name)
1b004f5974af refer to already extracted library files, to avoid tmp files produced by SQLiteJDBCLoader, which tend to remain after JVM crash;
wenzelm
parents: 65804
diff changeset
   598
1b004f5974af refer to already extracted library files, to avoid tmp files produced by SQLiteJDBCLoader, which tend to remain after JVM crash;
wenzelm
parents: 65804
diff changeset
   599
    Class.forName("org.sqlite.JDBC")
1b004f5974af refer to already extracted library files, to avoid tmp files produced by SQLiteJDBCLoader, which tend to remain after JVM crash;
wenzelm
parents: 65804
diff changeset
   600
  }
65292
e3bd1e7ddd23 more robust JDBC initialization, e.g. required for Isabelle/jEdit startup;
wenzelm
parents: 65280
diff changeset
   601
78163
c6d4b1a00ad7 clarified signature;
wenzelm
parents: 78154
diff changeset
   602
  def open_database(path: Path, restrict: Boolean = false): Database = {
65292
e3bd1e7ddd23 more robust JDBC initialization, e.g. required for Isabelle/jEdit startup;
wenzelm
parents: 65280
diff changeset
   603
    init_jdbc
65006
632bdf7b8bab clarified modules;
wenzelm
parents: 65004
diff changeset
   604
    val path0 = path.expand
632bdf7b8bab clarified modules;
wenzelm
parents: 65004
diff changeset
   605
    val s0 = File.platform_path(path0)
632bdf7b8bab clarified modules;
wenzelm
parents: 65004
diff changeset
   606
    val s1 = if (Platform.is_windows) s0.replace('\\', '/') else s0
632bdf7b8bab clarified modules;
wenzelm
parents: 65004
diff changeset
   607
    val connection = DriverManager.getConnection("jdbc:sqlite:" + s1)
78163
c6d4b1a00ad7 clarified signature;
wenzelm
parents: 78154
diff changeset
   608
    val db = new Database(path0.toString, connection)
c6d4b1a00ad7 clarified signature;
wenzelm
parents: 78154
diff changeset
   609
c6d4b1a00ad7 clarified signature;
wenzelm
parents: 78154
diff changeset
   610
    try { if (restrict) File.restrict(path0) }
c6d4b1a00ad7 clarified signature;
wenzelm
parents: 78154
diff changeset
   611
    catch { case exn: Throwable => db.close(); throw exn }
c6d4b1a00ad7 clarified signature;
wenzelm
parents: 78154
diff changeset
   612
c6d4b1a00ad7 clarified signature;
wenzelm
parents: 78154
diff changeset
   613
    db
65006
632bdf7b8bab clarified modules;
wenzelm
parents: 65004
diff changeset
   614
  }
632bdf7b8bab clarified modules;
wenzelm
parents: 65004
diff changeset
   615
75393
87ebf5a50283 clarified formatting, for the sake of scala3;
wenzelm
parents: 73367
diff changeset
   616
  class Database private[SQLite](name: String, val connection: Connection) extends SQL.Database {
65007
b6a1a1d42f5d clarified toString;
wenzelm
parents: 65006
diff changeset
   617
    override def toString: String = name
77664
f5d3ade80d15 more specific vacuum operation, which is also relevant to PostgreSQL;
wenzelm
parents: 77598
diff changeset
   618
77527
790085b1002f more detailed table "isabelle_build_serial": allow to monitor activity of build_process instances;
wenzelm
parents: 77403
diff changeset
   619
    override def now(): Date = Date.now()
790085b1002f more detailed table "isabelle_build_serial": allow to monitor activity of build_process instances;
wenzelm
parents: 77403
diff changeset
   620
65730
7ae61e72a678 clarified signature;
wenzelm
parents: 65729
diff changeset
   621
    def sql_type(T: SQL.Type.Value): SQL.Source = SQL.sql_type_sqlite(T)
65011
42bffd1637f0 support for type Boolean;
wenzelm
parents: 65010
diff changeset
   622
65748
1f4a80e80c88 tuned signature;
wenzelm
parents: 65741
diff changeset
   623
    def update_date(stmt: SQL.Statement, i: Int, date: Date): Unit =
1f4a80e80c88 tuned signature;
wenzelm
parents: 65741
diff changeset
   624
      if (date == null) stmt.string(i) = (null: String)
1f4a80e80c88 tuned signature;
wenzelm
parents: 65741
diff changeset
   625
      else stmt.string(i) = date_format(date)
65598
5deef985e38e allow null;
wenzelm
parents: 65594
diff changeset
   626
65740
83388f09e9ab clarified signature;
wenzelm
parents: 65739
diff changeset
   627
    def date(res: SQL.Result, column: SQL.Column): Date =
77598
6370d9e5ab50 proper support for Option[Date] columns;
wenzelm
parents: 77596
diff changeset
   628
      proper_string(res.string(column)) match {
6370d9e5ab50 proper support for Option[Date] columns;
wenzelm
parents: 77596
diff changeset
   629
        case None => null
6370d9e5ab50 proper support for Option[Date] columns;
wenzelm
parents: 77596
diff changeset
   630
        case Some(s) => date_format.parse(s)
6370d9e5ab50 proper support for Option[Date] columns;
wenzelm
parents: 77596
diff changeset
   631
      }
65021
da8ae577d171 clarified Date storage;
wenzelm
parents: 65020
diff changeset
   632
65730
7ae61e72a678 clarified signature;
wenzelm
parents: 65729
diff changeset
   633
    def insert_permissive(table: SQL.Table, sql: SQL.Source = ""): SQL.Source =
77365
a10fa2112854 clarified signature;
wenzelm
parents: 77364
diff changeset
   634
      table.insert_cmd(cmd = "INSERT OR IGNORE", sql = sql)
65006
632bdf7b8bab clarified modules;
wenzelm
parents: 65004
diff changeset
   635
  }
632bdf7b8bab clarified modules;
wenzelm
parents: 65004
diff changeset
   636
}
632bdf7b8bab clarified modules;
wenzelm
parents: 65004
diff changeset
   637
632bdf7b8bab clarified modules;
wenzelm
parents: 65004
diff changeset
   638
632bdf7b8bab clarified modules;
wenzelm
parents: 65004
diff changeset
   639
632bdf7b8bab clarified modules;
wenzelm
parents: 65004
diff changeset
   640
/** PostgreSQL **/
632bdf7b8bab clarified modules;
wenzelm
parents: 65004
diff changeset
   641
78351
9f2cfb9873bb tuned comments;
wenzelm
parents: 78348
diff changeset
   642
// see https://www.postgresql.org/docs/current/index.html
9f2cfb9873bb tuned comments;
wenzelm
parents: 78348
diff changeset
   643
// see https://jdbc.postgresql.org/documentation
9f2cfb9873bb tuned comments;
wenzelm
parents: 78348
diff changeset
   644
75393
87ebf5a50283 clarified formatting, for the sake of scala3;
wenzelm
parents: 73367
diff changeset
   645
object PostgreSQL {
75967
ff164add75cd maintain "uuid" column in session build database, to identity the original build process uniquely;
wenzelm
parents: 75966
diff changeset
   646
  type Source = SQL.Source
ff164add75cd maintain "uuid" column in session build database, to identity the original build process uniquely;
wenzelm
parents: 75966
diff changeset
   647
78347
72fc2ff08e07 clarified signature: more operations;
wenzelm
parents: 78345
diff changeset
   648
  lazy val init_jdbc: Unit = Class.forName("org.postgresql.Driver")
72fc2ff08e07 clarified signature: more operations;
wenzelm
parents: 78345
diff changeset
   649
78345
545da61f5989 clarified signature;
wenzelm
parents: 78344
diff changeset
   650
  val default_server: SSH.Server = SSH.local_server(port = 5432)
65006
632bdf7b8bab clarified modules;
wenzelm
parents: 65004
diff changeset
   651
78378
2f16f23baefd tuned source structure;
wenzelm
parents: 78375
diff changeset
   652
  def open_database(
2f16f23baefd tuned source structure;
wenzelm
parents: 78375
diff changeset
   653
    user: String,
2f16f23baefd tuned source structure;
wenzelm
parents: 78375
diff changeset
   654
    password: String,
2f16f23baefd tuned source structure;
wenzelm
parents: 78375
diff changeset
   655
    database: String = "",
2f16f23baefd tuned source structure;
wenzelm
parents: 78375
diff changeset
   656
    server: SSH.Server = default_server,
2f16f23baefd tuned source structure;
wenzelm
parents: 78375
diff changeset
   657
    server_close: Boolean = false,
78400
63d55ba90a9f more options for performance tuning;
wenzelm
parents: 78392
diff changeset
   658
    synchronous_commit: String = "off"
78378
2f16f23baefd tuned source structure;
wenzelm
parents: 78375
diff changeset
   659
  ): Database = {
2f16f23baefd tuned source structure;
wenzelm
parents: 78375
diff changeset
   660
    init_jdbc
2f16f23baefd tuned source structure;
wenzelm
parents: 78375
diff changeset
   661
2f16f23baefd tuned source structure;
wenzelm
parents: 78375
diff changeset
   662
    if (user.isEmpty) error("Undefined database user")
2f16f23baefd tuned source structure;
wenzelm
parents: 78375
diff changeset
   663
    if (server.host.isEmpty) error("Undefined database server host")
2f16f23baefd tuned source structure;
wenzelm
parents: 78375
diff changeset
   664
    if (server.port <= 0) error("Undefined database server port")
2f16f23baefd tuned source structure;
wenzelm
parents: 78375
diff changeset
   665
2f16f23baefd tuned source structure;
wenzelm
parents: 78375
diff changeset
   666
    val name = proper_string(database) getOrElse user
2f16f23baefd tuned source structure;
wenzelm
parents: 78375
diff changeset
   667
    val url = "jdbc:postgresql://" + server.host + ":" + server.port + "/" + name
2f16f23baefd tuned source structure;
wenzelm
parents: 78375
diff changeset
   668
    val ssh = server.ssh_system.ssh_session
78422
dcaf6f33d94d tuned output;
wenzelm
parents: 78402
diff changeset
   669
    val print =
dcaf6f33d94d tuned output;
wenzelm
parents: 78402
diff changeset
   670
      "server " + user + "@" + server + "/" + name + if_proper(ssh, " via ssh " + ssh.get)
78378
2f16f23baefd tuned source structure;
wenzelm
parents: 78375
diff changeset
   671
2f16f23baefd tuned source structure;
wenzelm
parents: 78375
diff changeset
   672
    val connection = DriverManager.getConnection(url, user, password)
78400
63d55ba90a9f more options for performance tuning;
wenzelm
parents: 78392
diff changeset
   673
    val db = new Database(connection, print, server, server_close)
63d55ba90a9f more options for performance tuning;
wenzelm
parents: 78392
diff changeset
   674
63d55ba90a9f more options for performance tuning;
wenzelm
parents: 78392
diff changeset
   675
    try {
78402
e25f1d343fa7 clarified options;
wenzelm
parents: 78400
diff changeset
   676
      if (synchronous_commit.nonEmpty) {
e25f1d343fa7 clarified options;
wenzelm
parents: 78400
diff changeset
   677
        db.execute_statement("SET synchronous_commit = " + SQL.string(synchronous_commit))
e25f1d343fa7 clarified options;
wenzelm
parents: 78400
diff changeset
   678
      }
78400
63d55ba90a9f more options for performance tuning;
wenzelm
parents: 78392
diff changeset
   679
    }
63d55ba90a9f more options for performance tuning;
wenzelm
parents: 78392
diff changeset
   680
    catch { case exn: Throwable => db.close(); throw exn }
63d55ba90a9f more options for performance tuning;
wenzelm
parents: 78392
diff changeset
   681
63d55ba90a9f more options for performance tuning;
wenzelm
parents: 78392
diff changeset
   682
    db
78378
2f16f23baefd tuned source structure;
wenzelm
parents: 78375
diff changeset
   683
  }
2f16f23baefd tuned source structure;
wenzelm
parents: 78375
diff changeset
   684
78366
aa4ea5398ab8 clarified signature: more operations;
wenzelm
parents: 78365
diff changeset
   685
  def open_server(
aa4ea5398ab8 clarified signature: more operations;
wenzelm
parents: 78365
diff changeset
   686
    options: Options,
aa4ea5398ab8 clarified signature: more operations;
wenzelm
parents: 78365
diff changeset
   687
    host: String = "",
aa4ea5398ab8 clarified signature: more operations;
wenzelm
parents: 78365
diff changeset
   688
    port: Int = 0,
aa4ea5398ab8 clarified signature: more operations;
wenzelm
parents: 78365
diff changeset
   689
    ssh_host: String = "",
aa4ea5398ab8 clarified signature: more operations;
wenzelm
parents: 78365
diff changeset
   690
    ssh_port: Int = 0,
aa4ea5398ab8 clarified signature: more operations;
wenzelm
parents: 78365
diff changeset
   691
    ssh_user: String = ""
aa4ea5398ab8 clarified signature: more operations;
wenzelm
parents: 78365
diff changeset
   692
  ): SSH.Server = {
aa4ea5398ab8 clarified signature: more operations;
wenzelm
parents: 78365
diff changeset
   693
    val server_host = proper_string(host).getOrElse(default_server.host)
aa4ea5398ab8 clarified signature: more operations;
wenzelm
parents: 78365
diff changeset
   694
    val server_port = if (port > 0) port else default_server.port
aa4ea5398ab8 clarified signature: more operations;
wenzelm
parents: 78365
diff changeset
   695
aa4ea5398ab8 clarified signature: more operations;
wenzelm
parents: 78365
diff changeset
   696
    if (ssh_host.isEmpty) SSH.local_server(host = server_host, port = server_port)
aa4ea5398ab8 clarified signature: more operations;
wenzelm
parents: 78365
diff changeset
   697
    else {
aa4ea5398ab8 clarified signature: more operations;
wenzelm
parents: 78365
diff changeset
   698
      SSH.open_server(options, host = ssh_host, port = ssh_port, user = ssh_user,
aa4ea5398ab8 clarified signature: more operations;
wenzelm
parents: 78365
diff changeset
   699
        remote_host = server_host, remote_port = server_port)
aa4ea5398ab8 clarified signature: more operations;
wenzelm
parents: 78365
diff changeset
   700
    }
aa4ea5398ab8 clarified signature: more operations;
wenzelm
parents: 78365
diff changeset
   701
  }
aa4ea5398ab8 clarified signature: more operations;
wenzelm
parents: 78365
diff changeset
   702
78347
72fc2ff08e07 clarified signature: more operations;
wenzelm
parents: 78345
diff changeset
   703
  def open_database_server(
72fc2ff08e07 clarified signature: more operations;
wenzelm
parents: 78345
diff changeset
   704
    options: Options,
72fc2ff08e07 clarified signature: more operations;
wenzelm
parents: 78345
diff changeset
   705
    user: String = "",
72fc2ff08e07 clarified signature: more operations;
wenzelm
parents: 78345
diff changeset
   706
    password: String = "",
72fc2ff08e07 clarified signature: more operations;
wenzelm
parents: 78345
diff changeset
   707
    database: String = "",
78366
aa4ea5398ab8 clarified signature: more operations;
wenzelm
parents: 78365
diff changeset
   708
    server: SSH.Server = SSH.no_server,
78347
72fc2ff08e07 clarified signature: more operations;
wenzelm
parents: 78345
diff changeset
   709
    host: String = "",
72fc2ff08e07 clarified signature: more operations;
wenzelm
parents: 78345
diff changeset
   710
    port: Int = 0,
72fc2ff08e07 clarified signature: more operations;
wenzelm
parents: 78345
diff changeset
   711
    ssh_host: String = "",
72fc2ff08e07 clarified signature: more operations;
wenzelm
parents: 78345
diff changeset
   712
    ssh_port: Int = 0,
78400
63d55ba90a9f more options for performance tuning;
wenzelm
parents: 78392
diff changeset
   713
    ssh_user: String = "",
78402
e25f1d343fa7 clarified options;
wenzelm
parents: 78400
diff changeset
   714
    synchronous_commit: String = ""
78347
72fc2ff08e07 clarified signature: more operations;
wenzelm
parents: 78345
diff changeset
   715
  ): PostgreSQL.Database = {
78366
aa4ea5398ab8 clarified signature: more operations;
wenzelm
parents: 78365
diff changeset
   716
    val db_server =
aa4ea5398ab8 clarified signature: more operations;
wenzelm
parents: 78365
diff changeset
   717
      if (server.defined) server
78347
72fc2ff08e07 clarified signature: more operations;
wenzelm
parents: 78345
diff changeset
   718
      else {
78366
aa4ea5398ab8 clarified signature: more operations;
wenzelm
parents: 78365
diff changeset
   719
        open_server(options, host = host, port = port, ssh_host = ssh_host,
aa4ea5398ab8 clarified signature: more operations;
wenzelm
parents: 78365
diff changeset
   720
          ssh_port = ssh_port, ssh_user = ssh_user)
78347
72fc2ff08e07 clarified signature: more operations;
wenzelm
parents: 78345
diff changeset
   721
      }
78366
aa4ea5398ab8 clarified signature: more operations;
wenzelm
parents: 78365
diff changeset
   722
    val server_close = !server.defined
78347
72fc2ff08e07 clarified signature: more operations;
wenzelm
parents: 78345
diff changeset
   723
    try {
72fc2ff08e07 clarified signature: more operations;
wenzelm
parents: 78345
diff changeset
   724
      open_database(user = user, password = password, database = database,
78400
63d55ba90a9f more options for performance tuning;
wenzelm
parents: 78392
diff changeset
   725
        server = db_server, server_close = server_close, synchronous_commit = synchronous_commit)
78347
72fc2ff08e07 clarified signature: more operations;
wenzelm
parents: 78345
diff changeset
   726
    }
78366
aa4ea5398ab8 clarified signature: more operations;
wenzelm
parents: 78365
diff changeset
   727
    catch { case exn: Throwable if server_close => db_server.close(); throw exn }
78347
72fc2ff08e07 clarified signature: more operations;
wenzelm
parents: 78345
diff changeset
   728
  }
65292
e3bd1e7ddd23 more robust JDBC initialization, e.g. required for Isabelle/jEdit startup;
wenzelm
parents: 65280
diff changeset
   729
65009
eda9366bbfac remote database access via ssh port forwarding;
wenzelm
parents: 65008
diff changeset
   730
  class Database private[PostgreSQL](
75393
87ebf5a50283 clarified formatting, for the sake of scala3;
wenzelm
parents: 73367
diff changeset
   731
    val connection: Connection,
78347
72fc2ff08e07 clarified signature: more operations;
wenzelm
parents: 78345
diff changeset
   732
    print: String,
72fc2ff08e07 clarified signature: more operations;
wenzelm
parents: 78345
diff changeset
   733
    server: SSH.Server,
72fc2ff08e07 clarified signature: more operations;
wenzelm
parents: 78345
diff changeset
   734
    server_close: Boolean
75393
87ebf5a50283 clarified formatting, for the sake of scala3;
wenzelm
parents: 73367
diff changeset
   735
  ) extends SQL.Database {
78347
72fc2ff08e07 clarified signature: more operations;
wenzelm
parents: 78345
diff changeset
   736
    override def toString: String = print
65008
ed2eedf786f3 clarified handling of SQL.Type;
wenzelm
parents: 65007
diff changeset
   737
77527
790085b1002f more detailed table "isabelle_build_serial": allow to monitor activity of build_process instances;
wenzelm
parents: 77403
diff changeset
   738
    override def now(): Date = {
790085b1002f more detailed table "isabelle_build_serial": allow to monitor activity of build_process instances;
wenzelm
parents: 77403
diff changeset
   739
      val now = SQL.Column.date("now")
77552
080422b3d914 clarified signature: reduce boilerplate;
wenzelm
parents: 77543
diff changeset
   740
      execute_query_statementO[Date]("SELECT NOW() as " + now.ident, res => res.date(now))
080422b3d914 clarified signature: reduce boilerplate;
wenzelm
parents: 77543
diff changeset
   741
        .getOrElse(error("Failed to get current date/time from database server " + toString))
77527
790085b1002f more detailed table "isabelle_build_serial": allow to monitor activity of build_process instances;
wenzelm
parents: 77403
diff changeset
   742
    }
790085b1002f more detailed table "isabelle_build_serial": allow to monitor activity of build_process instances;
wenzelm
parents: 77403
diff changeset
   743
65730
7ae61e72a678 clarified signature;
wenzelm
parents: 65729
diff changeset
   744
    def sql_type(T: SQL.Type.Value): SQL.Source = SQL.sql_type_postgresql(T)
65009
eda9366bbfac remote database access via ssh port forwarding;
wenzelm
parents: 65008
diff changeset
   745
65021
da8ae577d171 clarified Date storage;
wenzelm
parents: 65020
diff changeset
   746
    // see https://jdbc.postgresql.org/documentation/head/8-date-time.html
65748
1f4a80e80c88 tuned signature;
wenzelm
parents: 65741
diff changeset
   747
    def update_date(stmt: SQL.Statement, i: Int, date: Date): Unit =
65740
83388f09e9ab clarified signature;
wenzelm
parents: 65739
diff changeset
   748
      if (date == null) stmt.rep.setObject(i, null)
69980
f2e3adfd916f tuned signature;
wenzelm
parents: 69393
diff changeset
   749
      else stmt.rep.setObject(i, OffsetDateTime.from(date.to(Date.timezone_utc).rep))
65598
5deef985e38e allow null;
wenzelm
parents: 65594
diff changeset
   750
75393
87ebf5a50283 clarified formatting, for the sake of scala3;
wenzelm
parents: 73367
diff changeset
   751
    def date(res: SQL.Result, column: SQL.Column): Date = {
65740
83388f09e9ab clarified signature;
wenzelm
parents: 65739
diff changeset
   752
      val obj = res.rep.getObject(column.name, classOf[OffsetDateTime])
65704
aa9a7a753296 more robust;
wenzelm
parents: 65703
diff changeset
   753
      if (obj == null) null else Date.instant(obj.toInstant)
aa9a7a753296 more robust;
wenzelm
parents: 65703
diff changeset
   754
    }
65021
da8ae577d171 clarified Date storage;
wenzelm
parents: 65020
diff changeset
   755
65730
7ae61e72a678 clarified signature;
wenzelm
parents: 65729
diff changeset
   756
    def insert_permissive(table: SQL.Table, sql: SQL.Source = ""): SQL.Source =
77377
82fdc7cf9d44 tuned whitespace in generated SQL;
wenzelm
parents: 77375
diff changeset
   757
      table.insert_cmd(sql = if_proper(sql, sql + " ") + "ON CONFLICT DO NOTHING")
65703
cead65c19f2e more direct insert_permissive statement, which avoids somewhat fragile nested transactions;
wenzelm
parents: 65702
diff changeset
   758
77039
2f09dc0e6dda support IPC via database server;
wenzelm
parents: 77037
diff changeset
   759
77590
edc96be6b939 explicit locking for PostgreSQL --- neither available nor required for SQLite;
wenzelm
parents: 77552
diff changeset
   760
    /* explicit locking: only applicable to PostgreSQL within transaction context */
edc96be6b939 explicit locking for PostgreSQL --- neither available nor required for SQLite;
wenzelm
parents: 77552
diff changeset
   761
    // see https://www.postgresql.org/docs/current/sql-lock.html
edc96be6b939 explicit locking for PostgreSQL --- neither available nor required for SQLite;
wenzelm
parents: 77552
diff changeset
   762
    // see https://www.postgresql.org/docs/current/explicit-locking.html
edc96be6b939 explicit locking for PostgreSQL --- neither available nor required for SQLite;
wenzelm
parents: 77552
diff changeset
   763
77596
dd8b08729458 clarified signature;
wenzelm
parents: 77591
diff changeset
   764
    override def lock_tables(tables: List[SQL.Table]): PostgreSQL.Source =
78253
12d54a78bc0e more robust;
wenzelm
parents: 78207
diff changeset
   765
      if_proper(tables, "LOCK TABLE " + tables.mkString(", ") + " IN ACCESS EXCLUSIVE MODE")
77590
edc96be6b939 explicit locking for PostgreSQL --- neither available nor required for SQLite;
wenzelm
parents: 77552
diff changeset
   766
edc96be6b939 explicit locking for PostgreSQL --- neither available nor required for SQLite;
wenzelm
parents: 77552
diff changeset
   767
77039
2f09dc0e6dda support IPC via database server;
wenzelm
parents: 77037
diff changeset
   768
    /* notifications: IPC via database server */
2f09dc0e6dda support IPC via database server;
wenzelm
parents: 77037
diff changeset
   769
    // see https://www.postgresql.org/docs/current/sql-notify.html
2f09dc0e6dda support IPC via database server;
wenzelm
parents: 77037
diff changeset
   770
2f09dc0e6dda support IPC via database server;
wenzelm
parents: 77037
diff changeset
   771
    def listen(name: String): Unit =
77541
9d9b30741fc4 tuned signature: reduce boilerplate;
wenzelm
parents: 77540
diff changeset
   772
      execute_statement("LISTEN " + SQL.ident(name))
77039
2f09dc0e6dda support IPC via database server;
wenzelm
parents: 77037
diff changeset
   773
2f09dc0e6dda support IPC via database server;
wenzelm
parents: 77037
diff changeset
   774
    def unlisten(name: String = "*"): Unit =
77541
9d9b30741fc4 tuned signature: reduce boilerplate;
wenzelm
parents: 77540
diff changeset
   775
      execute_statement("UNLISTEN " + (if (name == "*") name else SQL.ident(name)))
77039
2f09dc0e6dda support IPC via database server;
wenzelm
parents: 77037
diff changeset
   776
2f09dc0e6dda support IPC via database server;
wenzelm
parents: 77037
diff changeset
   777
    def notify(name: String, payload: String = ""): Unit =
77543
97b5547f2b17 tuned whitespace and braces;
wenzelm
parents: 77542
diff changeset
   778
      execute_statement("NOTIFY " + SQL.ident(name) + if_proper(payload, ", " + SQL.string(payload)))
77039
2f09dc0e6dda support IPC via database server;
wenzelm
parents: 77037
diff changeset
   779
2f09dc0e6dda support IPC via database server;
wenzelm
parents: 77037
diff changeset
   780
    def get_notifications(): List[PGNotification] =
2f09dc0e6dda support IPC via database server;
wenzelm
parents: 77037
diff changeset
   781
      the_postgresql_connection.getNotifications() match {
2f09dc0e6dda support IPC via database server;
wenzelm
parents: 77037
diff changeset
   782
        case null => Nil
2f09dc0e6dda support IPC via database server;
wenzelm
parents: 77037
diff changeset
   783
        case array => array.toList
2f09dc0e6dda support IPC via database server;
wenzelm
parents: 77037
diff changeset
   784
      }
2f09dc0e6dda support IPC via database server;
wenzelm
parents: 77037
diff changeset
   785
2f09dc0e6dda support IPC via database server;
wenzelm
parents: 77037
diff changeset
   786
78347
72fc2ff08e07 clarified signature: more operations;
wenzelm
parents: 78345
diff changeset
   787
    override def close(): Unit = { super.close(); if (server_close) server.close() }
65006
632bdf7b8bab clarified modules;
wenzelm
parents: 65004
diff changeset
   788
  }
632bdf7b8bab clarified modules;
wenzelm
parents: 65004
diff changeset
   789
}