From 1926982bd635751fd712a5b82b22dcd98a01e893 Mon Sep 17 00:00:00 2001
From: amorozov <amorozov@tswf.io>
Date: Sun, 23 Mar 2025 16:48:14 +0300
Subject: [PATCH] initial commit
---
go.mod | 26 +++
go.sum | 104 ++++++++++
pkg/gorm/migrate_pure_sqlite_driver.go | 269 +++++++++++++++++++++++++
3 files changed, 399 insertions(+)
create mode 100644 go.mod
create mode 100644 go.sum
create mode 100644 pkg/gorm/migrate_pure_sqlite_driver.go
diff --git a/go.mod b/go.mod
new file mode 100644
index 0000000..43cfa56
--- /dev/null
+++ b/go.mod
@@ -0,0 +1,26 @@
+module git.tswf.io/go-libraries/go-pure-sqlite
+
+go 1.23.4
+
+require (
+ github.com/clearcodecn/sqlite v1.4.0 // indirect
+ github.com/dustin/go-humanize v1.0.1 // indirect
+ github.com/golang-migrate/migrate/v4 v4.18.2 // indirect
+ github.com/google/uuid v1.6.0 // indirect
+ github.com/hashicorp/errwrap v1.1.0 // indirect
+ github.com/hashicorp/go-multierror v1.1.1 // indirect
+ github.com/jinzhu/inflection v1.0.0 // indirect
+ github.com/jinzhu/now v1.1.5 // indirect
+ github.com/mattn/go-isatty v0.0.20 // indirect
+ github.com/ncruces/go-strftime v0.1.9 // indirect
+ github.com/remyoudompheng/bigfft v0.0.0-20230129092748-24d4a6f8daec // indirect
+ go.uber.org/atomic v1.7.0 // indirect
+ golang.org/x/exp v0.0.0-20230315142452-642cacee5cc0 // indirect
+ golang.org/x/sys v0.30.0 // indirect
+ golang.org/x/text v0.21.0 // indirect
+ gorm.io/gorm v1.25.12 // indirect
+ modernc.org/libc v1.61.13 // indirect
+ modernc.org/mathutil v1.7.1 // indirect
+ modernc.org/memory v1.8.2 // indirect
+ modernc.org/sqlite v1.36.1 // indirect
+)
diff --git a/go.sum b/go.sum
new file mode 100644
index 0000000..4f232bb
--- /dev/null
+++ b/go.sum
@@ -0,0 +1,104 @@
+github.com/clearcodecn/sqlite v1.4.0 h1:ThG/FPhaPM8qjZhUBauaYdnt3+UCnXZHdjXDI8vd7RQ=
+github.com/clearcodecn/sqlite v1.4.0/go.mod h1:Iz6Xqbe1flEEk2Q63rBBFFDI1YkalIG2BgjcJO+APf0=
+github.com/davecgh/go-spew v1.1.0/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38=
+github.com/davecgh/go-spew v1.1.1/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38=
+github.com/dustin/go-humanize v1.0.0/go.mod h1:HtrtbFcZ19U5GC7JDqmcUSB87Iq5E25KnS6fMYU6eOk=
+github.com/dustin/go-humanize v1.0.1 h1:GzkhY7T5VNhEkwH0PVJgjz+fX1rhBrR7pRT3mDkpeCY=
+github.com/dustin/go-humanize v1.0.1/go.mod h1:Mu1zIs6XwVuF/gI1OepvI0qD18qycQx+mFykh5fBlto=
+github.com/golang-migrate/migrate/v4 v4.18.2 h1:2VSCMz7x7mjyTXx3m2zPokOY82LTRgxK1yQYKo6wWQ8=
+github.com/golang-migrate/migrate/v4 v4.18.2/go.mod h1:2CM6tJvn2kqPXwnXO/d3rAQYiyoIm180VsO8PRX6Rpk=
+github.com/google/go-cmp v0.5.3/go.mod h1:v8dTdLbMG2kIc/vJvl+f65V22dbkXbowE6jgT/gNBxE=
+github.com/google/uuid v1.3.0/go.mod h1:TIyPZe4MgqvfeYDBFedMoGGpEw/LqOeaOT+nhxU+yHo=
+github.com/google/uuid v1.6.0 h1:NIvaJDMOsjHA8n1jAhLSgzrAzy1Hgr+hNrb57e+94F0=
+github.com/google/uuid v1.6.0/go.mod h1:TIyPZe4MgqvfeYDBFedMoGGpEw/LqOeaOT+nhxU+yHo=
+github.com/hashicorp/errwrap v1.0.0/go.mod h1:YH+1FKiLXxHSkmPseP+kNlulaMuP3n2brvKWEqk/Jc4=
+github.com/hashicorp/errwrap v1.1.0 h1:OxrOeh75EUXMY8TBjag2fzXGZ40LB6IKw45YeGUDY2I=
+github.com/hashicorp/errwrap v1.1.0/go.mod h1:YH+1FKiLXxHSkmPseP+kNlulaMuP3n2brvKWEqk/Jc4=
+github.com/hashicorp/go-multierror v1.1.1 h1:H5DkEtf6CXdFp0N0Em5UCwQpXMWke8IA0+lD48awMYo=
+github.com/hashicorp/go-multierror v1.1.1/go.mod h1:iw975J/qwKPdAO1clOe2L8331t/9/fmwbPZ6JB6eMoM=
+github.com/jinzhu/inflection v1.0.0 h1:K317FqzuhWc8YvSVlFMCCUb36O/S9MCKRDI7QkRKD/E=
+github.com/jinzhu/inflection v1.0.0/go.mod h1:h+uFLlag+Qp1Va5pdKtLDYj+kHp5pxUVkryuEj+Srlc=
+github.com/jinzhu/now v1.1.4/go.mod h1:d3SSVoowX0Lcu0IBviAWJpolVfI5UJVZZ7cO71lE/z8=
+github.com/jinzhu/now v1.1.5 h1:/o9tlHleP7gOFmsnYNz3RGnqzefHA47wQpKrrdTIwXQ=
+github.com/jinzhu/now v1.1.5/go.mod h1:d3SSVoowX0Lcu0IBviAWJpolVfI5UJVZZ7cO71lE/z8=
+github.com/kballard/go-shellquote v0.0.0-20180428030007-95032a82bc51/go.mod h1:CzGEWj7cYgsdH8dAjBGEr58BoE7ScuLd+fwFZ44+/x8=
+github.com/mattn/go-isatty v0.0.12/go.mod h1:cbi8OIDigv2wuxKPP5vlRcQ1OAZbq2CE4Kysco4FUpU=
+github.com/mattn/go-isatty v0.0.20 h1:xfD0iDuEKnDkl03q4limB+vH+GxLEtL/jb4xVJSWWEY=
+github.com/mattn/go-isatty v0.0.20/go.mod h1:W+V8PltTTMOvKvAeJH7IuucS94S2C6jfK/D7dTCTo3Y=
+github.com/mattn/go-sqlite3 v1.14.14/go.mod h1:NyWgC/yNuGj7Q9rpYnZvas74GogHl5/Z4A/KQRfk6bU=
+github.com/ncruces/go-strftime v0.1.9 h1:bY0MQC28UADQmHmaF5dgpLmImcShSi2kHU9XLdhx/f4=
+github.com/ncruces/go-strftime v0.1.9/go.mod h1:Fwc5htZGVVkseilnfgOVb9mKy6w1naJmn9CehxcKcls=
+github.com/pmezard/go-difflib v1.0.0/go.mod h1:iKH77koFhYxTK1pcRnkKkqfTogsbg7gZNVY4sRDYZ/4=
+github.com/remyoudompheng/bigfft v0.0.0-20200410134404-eec4a21b6bb0/go.mod h1:qqbHyh8v60DhA7CoWK5oRCqLrMHRGoxYCSS9EjAz6Eo=
+github.com/remyoudompheng/bigfft v0.0.0-20230129092748-24d4a6f8daec h1:W09IVJc94icq4NjY3clb7Lk8O1qJ8BdBEF8z0ibU0rE=
+github.com/remyoudompheng/bigfft v0.0.0-20230129092748-24d4a6f8daec/go.mod h1:qqbHyh8v60DhA7CoWK5oRCqLrMHRGoxYCSS9EjAz6Eo=
+github.com/stretchr/objx v0.1.0/go.mod h1:HFkY916IF+rwdDfMAkV7OtwuqBVzrE8GR6GFx+wExME=
+github.com/stretchr/testify v1.3.0/go.mod h1:M5WIy9Dh21IEIfnGCwXGc5bZfKNJtfHm1UVUgZn+9EI=
+github.com/yuin/goldmark v1.2.1/go.mod h1:3hX8gzYuyVAZsxl0MRgGTJEmQBFcNTphYh9decYSb74=
+go.uber.org/atomic v1.7.0 h1:ADUqmZGgLDDfbSL9ZmPxKTybcoEYHgpYfELNoN+7hsw=
+go.uber.org/atomic v1.7.0/go.mod h1:fEN4uk6kAWBTFdckzkM89CLk9XfWZrxpCo0nPH17wJc=
+golang.org/x/crypto v0.0.0-20190308221718-c2843e01d9a2/go.mod h1:djNgcEr1/C05ACkg1iLfiJU5Ep61QUkGW8qpdssI0+w=
+golang.org/x/crypto v0.0.0-20191011191535-87dc89f01550/go.mod h1:yigFU9vqHzYiE8UmvKecakEJjdnWj3jj499lnFckfCI=
+golang.org/x/crypto v0.0.0-20200622213623-75b288015ac9/go.mod h1:LzIPMQfyMNhhGPhUkYOs5KpL4U8rLKemX1yGLhDgUto=
+golang.org/x/exp v0.0.0-20230315142452-642cacee5cc0 h1:pVgRXcIictcr+lBQIFeiwuwtDIs4eL21OuM9nyAADmo=
+golang.org/x/exp v0.0.0-20230315142452-642cacee5cc0/go.mod h1:CxIveKay+FTh1D0yPZemJVgC/95VzuuOLq5Qi4xnoYc=
+golang.org/x/mod v0.3.0/go.mod h1:s0Qsj1ACt9ePp/hMypM3fl4fZqREWJwdYDEqhRiZZUA=
+golang.org/x/net v0.0.0-20190404232315-eb5bcb51f2a3/go.mod h1:t9HGtf8HONx5eT2rtn7q6eTqICYqUVnKs3thJo3Qplg=
+golang.org/x/net v0.0.0-20190620200207-3b0461eec859/go.mod h1:z5CRVTTTmAJ677TzLLGU+0bjPO0LkuOLi4/5GtJWs/s=
+golang.org/x/net v0.0.0-20201021035429-f5854403a974/go.mod h1:sp8m0HH+o8qH0wwXwYZr8TS3Oi6o0r6Gce1SSxlDquU=
+golang.org/x/sync v0.0.0-20190423024810-112230192c58/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM=
+golang.org/x/sync v0.0.0-20201020160332-67f06af15bc9/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM=
+golang.org/x/sys v0.0.0-20190215142949-d0b11bdaac8a/go.mod h1:STP8DvDyc/dI5b8T5hshtkjS+E42TnysNCUPdjciGhY=
+golang.org/x/sys v0.0.0-20190412213103-97732733099d/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=
+golang.org/x/sys v0.0.0-20200116001909-b77594299b42/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=
+golang.org/x/sys v0.0.0-20200930185726-fdedc70b468f/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=
+golang.org/x/sys v0.0.0-20211007075335-d3039528d8ac/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
+golang.org/x/sys v0.6.0/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
+golang.org/x/sys v0.30.0 h1:QjkSwP/36a20jFYWkSue1YwXzLmsV5Gfq7Eiy72C1uc=
+golang.org/x/sys v0.30.0/go.mod h1:/VUhepiaJMQUp4+oa/7Zr1D23ma6VTLIYjOOTFZPUcA=
+golang.org/x/text v0.3.0/go.mod h1:NqM8EUOU14njkJ3fqMW+pc6Ldnwhi/IjpwHt7yyuwOQ=
+golang.org/x/text v0.3.3/go.mod h1:5Zoc/QRtKVWzQhOtBMvqHzDpF6irO9z98xDceosuGiQ=
+golang.org/x/text v0.14.0 h1:ScX5w1eTa3QqT8oi6+ziP7dTV1S2+ALU0bI+0zXKWiQ=
+golang.org/x/text v0.14.0/go.mod h1:18ZOQIKpY8NJVqYksKHtTdi31H5itFRjB5/qKTNYzSU=
+golang.org/x/text v0.21.0/go.mod h1:4IBbMaMmOPCJ8SecivzSH54+73PCFmPWxNTLm+vZkEQ=
+golang.org/x/tools v0.0.0-20180917221912-90fa682c2a6e/go.mod h1:n7NCudcB/nEzxVGmLbDWY5pfWTLqBcC2KZ6jyYvM4mQ=
+golang.org/x/tools v0.0.0-20191119224855-298f0cb1881e/go.mod h1:b+2E5dAYhXwXZwtnZ6UAqBI28+e2cm9otk0dWdXHAEo=
+golang.org/x/tools v0.0.0-20201124115921-2c860bdd6e78/go.mod h1:emZCQorbCU4vsT4fOWvOPXz4eW1wZW4PmDk9uLelYpA=
+golang.org/x/xerrors v0.0.0-20190717185122-a985d3407aa7/go.mod h1:I/5z698sn9Ka8TeJc9MKroUUfqBBauWjQqLJ2OPfmY0=
+golang.org/x/xerrors v0.0.0-20191011141410-1b5146add898/go.mod h1:I/5z698sn9Ka8TeJc9MKroUUfqBBauWjQqLJ2OPfmY0=
+golang.org/x/xerrors v0.0.0-20191204190536-9bdfabe68543/go.mod h1:I/5z698sn9Ka8TeJc9MKroUUfqBBauWjQqLJ2OPfmY0=
+golang.org/x/xerrors v0.0.0-20200804184101-5ec99f83aff1/go.mod h1:I/5z698sn9Ka8TeJc9MKroUUfqBBauWjQqLJ2OPfmY0=
+gorm.io/gorm v1.24.2/go.mod h1:DVrVomtaYTbqs7gB/x2uVvqnXzv0nqjB396B8cG4dBA=
+gorm.io/gorm v1.25.12 h1:I0u8i2hWQItBq1WfE0o2+WuL9+8L21K9e2HHSTE/0f8=
+gorm.io/gorm v1.25.12/go.mod h1:xh7N7RHfYlNc5EmcI/El95gXusucDrQnHXe0+CgWcLQ=
+lukechampine.com/uint128 v1.1.1/go.mod h1:c4eWIwlEGaxC/+H1VguhU4PHXNWDCDMUlWdIWl2j1gk=
+modernc.org/cc/v3 v3.36.0/go.mod h1:NFUHyPn4ekoC/JHeZFfZurN6ixxawE1BnVonP/oahEI=
+modernc.org/ccgo/v3 v3.0.0-20220428102840-41399a37e894/go.mod h1:eI31LL8EwEBKPpNpA4bU1/i+sKOwOrQy8D87zWUcRZc=
+modernc.org/ccgo/v3 v3.0.0-20220430103911-bc99d88307be/go.mod h1:bwdAnOoaIt8Ax9YdWGjxWsdkPcZyRPHqrOvJxaKAKGw=
+modernc.org/ccgo/v3 v3.16.4/go.mod h1:tGtX0gE9Jn7hdZFeU88slbTh1UtCYKusWOoCJuvkWsQ=
+modernc.org/ccgo/v3 v3.16.6/go.mod h1:tGtX0gE9Jn7hdZFeU88slbTh1UtCYKusWOoCJuvkWsQ=
+modernc.org/ccgo/v3 v3.16.8/go.mod h1:zNjwkizS+fIFDrDjIAgBSCLkWbJuHF+ar3QRn+Z9aws=
+modernc.org/ccorpus v1.11.6/go.mod h1:2gEUTrWqdpH2pXsmTM1ZkjeSrUWDpjMu2T6m29L/ErQ=
+modernc.org/httpfs v1.0.6/go.mod h1:7dosgurJGp0sPaRanU53W4xZYKh14wfzX420oZADeHM=
+modernc.org/libc v0.0.0-20220428101251-2d5f3daf273b/go.mod h1:p7Mg4+koNjc8jkqwcoFBJx7tXkpj00G77X7A72jXPXA=
+modernc.org/libc v1.16.0/go.mod h1:N4LD6DBE9cf+Dzf9buBlzVJndKr/iJHG97vGLHYnb5A=
+modernc.org/libc v1.16.1/go.mod h1:JjJE0eu4yeK7tab2n4S1w8tlWd9MxXLRzheaRnAKymU=
+modernc.org/libc v1.16.17/go.mod h1:hYIV5VZczAmGZAnG15Vdngn5HSF5cSkbvfz2B7GRuVU=
+modernc.org/libc v1.16.19/go.mod h1:p7Mg4+koNjc8jkqwcoFBJx7tXkpj00G77X7A72jXPXA=
+modernc.org/libc v1.61.13 h1:3LRd6ZO1ezsFiX1y+bHd1ipyEHIJKvuprv0sLTBwLW8=
+modernc.org/libc v1.61.13/go.mod h1:8F/uJWL/3nNil0Lgt1Dpz+GgkApWh04N3el3hxJcA6E=
+modernc.org/mathutil v1.2.2/go.mod h1:mZW8CKdRPY1v87qxC/wUdX5O1qDzXMP5TH3wjfpga6E=
+modernc.org/mathutil v1.4.1/go.mod h1:mZW8CKdRPY1v87qxC/wUdX5O1qDzXMP5TH3wjfpga6E=
+modernc.org/mathutil v1.7.1 h1:GCZVGXdaN8gTqB1Mf/usp1Y/hSqgI2vAGGP4jZMCxOU=
+modernc.org/mathutil v1.7.1/go.mod h1:4p5IwJITfppl0G4sUEDtCr4DthTaT47/N3aT6MhfgJg=
+modernc.org/memory v1.1.1/go.mod h1:/0wo5ibyrQiaoUoH7f9D8dnglAmILJ5/cxZlRECf+Nw=
+modernc.org/memory v1.8.2 h1:cL9L4bcoAObu4NkxOlKWBWtNHIsnnACGF/TbqQ6sbcI=
+modernc.org/memory v1.8.2/go.mod h1:ZbjSvMO5NQ1A2i3bWeDiVMxIorXwdClKE/0SZ+BMotU=
+modernc.org/opt v0.1.1/go.mod h1:WdSiB5evDcignE70guQKxYUl14mgWtbClRi5wmkkTX0=
+modernc.org/sqlite v1.18.1/go.mod h1:6ho+Gow7oX5V+OiOQ6Tr4xeqbx13UZ6t+Fw9IRUG4d4=
+modernc.org/sqlite v1.36.1 h1:bDa8BJUH4lg6EGkLbahKe/8QqoF8p9gArSc6fTqYhyQ=
+modernc.org/sqlite v1.36.1/go.mod h1:7MPwH7Z6bREicF9ZVUR78P1IKuxfZ8mRIDHD0iD+8TU=
+modernc.org/strutil v1.1.1/go.mod h1:DE+MQQ/hjKBZS2zNInV5hhcipt5rLPWkmpbGeW5mmdw=
+modernc.org/tcl v1.13.1/go.mod h1:XOLfOwzhkljL4itZkK6T72ckMgvj0BDsnKNdZVUOecw=
+modernc.org/token v1.0.0/go.mod h1:UGzOrNV1mAFSEB63lOFHIpNRUVMvYTc6yu1SMY/XTDM=
+modernc.org/z v1.5.1/go.mod h1:eWFB510QWW5Th9YGZT81s+LwvaAs3Q2yr4sP0rmLkv8=
diff --git a/pkg/gorm/migrate_pure_sqlite_driver.go b/pkg/gorm/migrate_pure_sqlite_driver.go
new file mode 100644
index 0000000..b7a90fb
--- /dev/null
+++ b/pkg/gorm/migrate_pure_sqlite_driver.go
@@ -0,0 +1,269 @@
+package gorm
+
+import (
+ "database/sql"
+ "fmt"
+ "io"
+ nurl "net/url"
+ "strconv"
+ "strings"
+
+ "github.com/golang-migrate/migrate/v4"
+ "github.com/hashicorp/go-multierror"
+ "go.uber.org/atomic"
+
+ "github.com/golang-migrate/migrate/v4/database"
+ _ "modernc.org/sqlite"
+)
+
+func init() {
+ database.Register("sqlite3", &Sqlite{})
+}
+
+var DefaultMigrationsTable = "schema_migrations"
+var (
+ ErrDatabaseDirty = fmt.Errorf("database is dirty")
+ ErrNilConfig = fmt.Errorf("no config")
+ ErrNoDatabaseName = fmt.Errorf("no database name")
+)
+
+type Config struct {
+ MigrationsTable string
+ DatabaseName string
+ NoTxWrap bool
+}
+
+type Sqlite struct {
+ db *sql.DB
+ isLocked atomic.Bool
+
+ config *Config
+}
+
+func PureSQLiteWithInstance(instance *sql.DB, config *Config) (database.Driver, error) {
+ if config == nil {
+ return nil, ErrNilConfig
+ }
+
+ if err := instance.Ping(); err != nil {
+ return nil, err
+ }
+
+ if len(config.MigrationsTable) == 0 {
+ config.MigrationsTable = DefaultMigrationsTable
+ }
+
+ mx := &Sqlite{
+ db: instance,
+ config: config,
+ }
+ if err := mx.ensureVersionTable(); err != nil {
+ return nil, err
+ }
+ return mx, nil
+}
+
+// ensureVersionTable checks if versions table exists and, if not, creates it.
+// Note that this function locks the database, which deviates from the usual
+// convention of "caller locks" in the Sqlite type.
+func (m *Sqlite) ensureVersionTable() (err error) {
+ if err = m.Lock(); err != nil {
+ return err
+ }
+
+ defer func() {
+ if e := m.Unlock(); e != nil {
+ if err == nil {
+ err = e
+ } else {
+ err = multierror.Append(err, e)
+ }
+ }
+ }()
+
+ query := fmt.Sprintf(`
+ CREATE TABLE IF NOT EXISTS %s (version uint64,dirty bool);
+ CREATE UNIQUE INDEX IF NOT EXISTS version_unique ON %s (version);
+ `, m.config.MigrationsTable, m.config.MigrationsTable)
+
+ if _, err := m.db.Exec(query); err != nil {
+ return err
+ }
+ return nil
+}
+
+func (m *Sqlite) Open(url string) (database.Driver, error) {
+ purl, err := nurl.Parse(url)
+ if err != nil {
+ return nil, err
+ }
+ dbfile := strings.Replace(migrate.FilterCustomQuery(purl).String(), "sqlite3://", "", 1)
+ db, err := sql.Open("sqlite3", dbfile)
+ if err != nil {
+ return nil, err
+ }
+
+ qv := purl.Query()
+
+ migrationsTable := qv.Get("x-migrations-table")
+ if len(migrationsTable) == 0 {
+ migrationsTable = DefaultMigrationsTable
+ }
+
+ noTxWrap := false
+ if v := qv.Get("x-no-tx-wrap"); v != "" {
+ noTxWrap, err = strconv.ParseBool(v)
+ if err != nil {
+ return nil, fmt.Errorf("x-no-tx-wrap: %s", err)
+ }
+ }
+
+ mx, err := PureSQLiteWithInstance(db, &Config{
+ DatabaseName: purl.Path,
+ MigrationsTable: migrationsTable,
+ NoTxWrap: noTxWrap,
+ })
+ if err != nil {
+ return nil, err
+ }
+ return mx, nil
+}
+
+func (m *Sqlite) Close() error {
+ return m.db.Close()
+}
+
+func (m *Sqlite) Drop() (err error) {
+ query := `SELECT name FROM sqlite_master WHERE type = 'table';`
+ tables, err := m.db.Query(query)
+ if err != nil {
+ return &database.Error{OrigErr: err, Query: []byte(query)}
+ }
+ defer func() {
+ if errClose := tables.Close(); errClose != nil {
+ err = multierror.Append(err, errClose)
+ }
+ }()
+
+ tableNames := make([]string, 0)
+ for tables.Next() {
+ var tableName string
+ if err := tables.Scan(&tableName); err != nil {
+ return err
+ }
+ if len(tableName) > 0 {
+ tableNames = append(tableNames, tableName)
+ }
+ }
+ if err := tables.Err(); err != nil {
+ return &database.Error{OrigErr: err, Query: []byte(query)}
+ }
+
+ if len(tableNames) > 0 {
+ for _, t := range tableNames {
+ query := "DROP TABLE " + t
+ err = m.executeQuery(query)
+ if err != nil {
+ return &database.Error{OrigErr: err, Query: []byte(query)}
+ }
+ }
+ query := "VACUUM"
+ _, err = m.db.Query(query)
+ if err != nil {
+ return &database.Error{OrigErr: err, Query: []byte(query)}
+ }
+ }
+
+ return nil
+}
+
+func (m *Sqlite) Lock() error {
+ if !m.isLocked.CAS(false, true) {
+ return database.ErrLocked
+ }
+ return nil
+}
+
+func (m *Sqlite) Unlock() error {
+ if !m.isLocked.CAS(true, false) {
+ return database.ErrNotLocked
+ }
+ return nil
+}
+
+func (m *Sqlite) Run(migration io.Reader) error {
+ migr, err := io.ReadAll(migration)
+ if err != nil {
+ return err
+ }
+ query := string(migr[:])
+
+ if m.config.NoTxWrap {
+ return m.executeQueryNoTx(query)
+ }
+ return m.executeQuery(query)
+}
+
+func (m *Sqlite) executeQuery(query string) error {
+ tx, err := m.db.Begin()
+ if err != nil {
+ return &database.Error{OrigErr: err, Err: "transaction start failed"}
+ }
+ if _, err := tx.Exec(query); err != nil {
+ if errRollback := tx.Rollback(); errRollback != nil {
+ err = multierror.Append(err, errRollback)
+ }
+ return &database.Error{OrigErr: err, Query: []byte(query)}
+ }
+ if err := tx.Commit(); err != nil {
+ return &database.Error{OrigErr: err, Err: "transaction commit failed"}
+ }
+ return nil
+}
+
+func (m *Sqlite) executeQueryNoTx(query string) error {
+ if _, err := m.db.Exec(query); err != nil {
+ return &database.Error{OrigErr: err, Query: []byte(query)}
+ }
+ return nil
+}
+
+func (m *Sqlite) SetVersion(version int, dirty bool) error {
+ tx, err := m.db.Begin()
+ if err != nil {
+ return &database.Error{OrigErr: err, Err: "transaction start failed"}
+ }
+
+ query := "DELETE FROM " + m.config.MigrationsTable
+ if _, err := tx.Exec(query); err != nil {
+ return &database.Error{OrigErr: err, Query: []byte(query)}
+ }
+
+ // Also re-write the schema version for nil dirty versions to prevent
+ // empty schema version for failed down migration on the first migration
+ // See: https://github.com/golang-migrate/migrate/issues/330
+ if version >= 0 || (version == database.NilVersion && dirty) {
+ query := fmt.Sprintf(`INSERT INTO %s (version, dirty) VALUES (?, ?)`, m.config.MigrationsTable)
+ if _, err := tx.Exec(query, version, dirty); err != nil {
+ if errRollback := tx.Rollback(); errRollback != nil {
+ err = multierror.Append(err, errRollback)
+ }
+ return &database.Error{OrigErr: err, Query: []byte(query)}
+ }
+ }
+
+ if err := tx.Commit(); err != nil {
+ return &database.Error{OrigErr: err, Err: "transaction commit failed"}
+ }
+
+ return nil
+}
+
+func (m *Sqlite) Version() (version int, dirty bool, err error) {
+ query := "SELECT version, dirty FROM " + m.config.MigrationsTable + " LIMIT 1"
+ err = m.db.QueryRow(query).Scan(&version, &dirty)
+ if err != nil {
+ return database.NilVersion, false, nil
+ }
+ return version, dirty, nil
+}