Parse URIs correctly

This commit is contained in:
Kegan Dougal 2020-05-28 12:28:21 +01:00
parent a6f995eb45
commit 2411007c4b

View File

@ -17,6 +17,8 @@ package sqlite3
import ( import (
"context" "context"
"errors"
"net/url"
"time" "time"
"golang.org/x/crypto/ed25519" "golang.org/x/crypto/ed25519"
@ -44,7 +46,19 @@ func NewDatabase(
serverKey ed25519.PublicKey, serverKey ed25519.PublicKey,
serverKeyID gomatrixserverlib.KeyID, serverKeyID gomatrixserverlib.KeyID,
) (*Database, error) { ) (*Database, error) {
db, err := sqlutil.Open(internal.SQLiteDriverName(), dataSourceName, nil) uri, err := url.Parse(dataSourceName)
if err != nil {
return nil, err
}
var cs string
if uri.Opaque != "" { // file:filename.db
cs = uri.Opaque
} else if uri.Path != "" { // file:///path/to/filename.db
cs = uri.Path
} else {
return nil, errors.New("no filename or path in connect string")
}
db, err := sqlutil.Open(internal.SQLiteDriverName(), cs, nil)
if err != nil { if err != nil {
return nil, err return nil, err
} }