mirror of
https://github.com/writefreely/writefreely
synced 2024-11-28 03:20:17 +00:00
eb6349f93a
This ensures the writefreely pkg can be used in other applications that need to load mysql themselves -- this can be done by building with the tag: wflib Ref T613
50 lines
1.2 KiB
Go
50 lines
1.2 KiB
Go
// +build sqlite,!wflib
|
|
|
|
/*
|
|
* Copyright © 2019 A Bunch Tell LLC.
|
|
*
|
|
* This file is part of WriteFreely.
|
|
*
|
|
* WriteFreely is free software: you can redistribute it and/or modify
|
|
* it under the terms of the GNU Affero General Public License, included
|
|
* in the LICENSE file in this source code package.
|
|
*/
|
|
|
|
package writefreely
|
|
|
|
import (
|
|
"database/sql"
|
|
"github.com/go-sql-driver/mysql"
|
|
"github.com/mattn/go-sqlite3"
|
|
"github.com/writeas/web-core/log"
|
|
"regexp"
|
|
)
|
|
|
|
func init() {
|
|
SQLiteEnabled = true
|
|
|
|
regex := func(re, s string) (bool, error) {
|
|
return regexp.MatchString(re, s)
|
|
}
|
|
sql.Register("sqlite3_with_regex", &sqlite3.SQLiteDriver{
|
|
ConnectHook: func(conn *sqlite3.SQLiteConn) error {
|
|
return conn.RegisterFunc("regexp", regex, true)
|
|
},
|
|
})
|
|
}
|
|
|
|
func (db *datastore) isDuplicateKeyErr(err error) bool {
|
|
if db.driverName == driverSQLite {
|
|
if err, ok := err.(sqlite3.Error); ok {
|
|
return err.Code == sqlite3.ErrConstraint
|
|
}
|
|
} else if db.driverName == driverMySQL {
|
|
if mysqlErr, ok := err.(*mysql.MySQLError); ok {
|
|
return mysqlErr.Number == mySQLErrDuplicateKey
|
|
}
|
|
} else {
|
|
log.Error("isDuplicateKeyErr: failed check for unrecognized driver '%s'", db.driverName)
|
|
}
|
|
|
|
return false
|
|
}
|