d3fe6b0f32
- Add nolint:unparam for readOnly param in sqlite_open.go - Handle cmd.Help() error in wisp.go - Handle rows.Close() error in migration 028 - Handle targetStore.Close() error in create.go - Update Nix vendorHash for current dependencies Cherry-picked from PR #769
56 lines
1.4 KiB
Go
56 lines
1.4 KiB
Go
package doctor
|
|
|
|
import (
|
|
"fmt"
|
|
"os"
|
|
"strings"
|
|
"time"
|
|
)
|
|
|
|
//nolint:unparam // readOnly is used to add mode=ro to connection string
|
|
func sqliteConnString(path string, readOnly bool) string {
|
|
path = strings.TrimSpace(path)
|
|
if path == "" {
|
|
return ""
|
|
}
|
|
|
|
// Best-effort: honor the same env var viper uses (BD_LOCK_TIMEOUT).
|
|
busy := 30 * time.Second
|
|
if v := strings.TrimSpace(os.Getenv("BD_LOCK_TIMEOUT")); v != "" {
|
|
if d, err := time.ParseDuration(v); err == nil {
|
|
busy = d
|
|
}
|
|
}
|
|
busyMs := int64(busy / time.Millisecond)
|
|
|
|
// If it's already a URI, append pragmas if absent.
|
|
if strings.HasPrefix(path, "file:") {
|
|
conn := path
|
|
sep := "?"
|
|
if strings.Contains(conn, "?") {
|
|
sep = "&"
|
|
}
|
|
if readOnly && !strings.Contains(conn, "mode=") {
|
|
conn += sep + "mode=ro"
|
|
sep = "&"
|
|
}
|
|
if !strings.Contains(conn, "_pragma=busy_timeout") {
|
|
conn += fmt.Sprintf("%s_pragma=busy_timeout(%d)", sep, busyMs)
|
|
sep = "&"
|
|
}
|
|
if !strings.Contains(conn, "_pragma=foreign_keys") {
|
|
conn += sep + "_pragma=foreign_keys(ON)"
|
|
sep = "&"
|
|
}
|
|
if !strings.Contains(conn, "_time_format=") {
|
|
conn += sep + "_time_format=sqlite"
|
|
}
|
|
return conn
|
|
}
|
|
|
|
if readOnly {
|
|
return fmt.Sprintf("file:%s?mode=ro&_pragma=foreign_keys(ON)&_pragma=busy_timeout(%d)&_time_format=sqlite", path, busyMs)
|
|
}
|
|
return fmt.Sprintf("file:%s?_pragma=foreign_keys(ON)&_pragma=busy_timeout(%d)&_time_format=sqlite", path, busyMs)
|
|
}
|