mirror of
https://github.com/stashapp/stash.git
synced 2026-04-20 05:52:40 +02:00
Add write mutex and max connection lifetime (#1211)
This commit is contained in:
parent
215737d6c5
commit
d93011a828
2 changed files with 7 additions and 5 deletions
|
|
@ -5,6 +5,7 @@ import (
|
|||
"errors"
|
||||
"fmt"
|
||||
"os"
|
||||
"sync"
|
||||
"time"
|
||||
|
||||
"github.com/fvbommel/sortorder"
|
||||
|
|
@ -20,6 +21,7 @@ import (
|
|||
)
|
||||
|
||||
var DB *sqlx.DB
|
||||
var WriteMu *sync.Mutex
|
||||
var dbPath string
|
||||
var appSchemaVersion uint = 19
|
||||
var databaseSchemaVersion uint
|
||||
|
|
@ -63,6 +65,7 @@ func Initialize(databasePath string) bool {
|
|||
|
||||
const disableForeignKeys = false
|
||||
DB = open(databasePath, disableForeignKeys)
|
||||
WriteMu = &sync.Mutex{}
|
||||
|
||||
return false
|
||||
}
|
||||
|
|
@ -77,6 +80,7 @@ func open(databasePath string, disableForeignKeys bool) *sqlx.DB {
|
|||
conn, err := sqlx.Open(sqlite3Driver, url)
|
||||
conn.SetMaxOpenConns(25)
|
||||
conn.SetMaxIdleConns(4)
|
||||
conn.SetConnMaxLifetime(30 * time.Second)
|
||||
if err != nil {
|
||||
logger.Fatalf("db.Open(): %q\n", err)
|
||||
}
|
||||
|
|
|
|||
|
|
@ -176,17 +176,15 @@ func (t *ReadTransaction) Tag() models.TagReader {
|
|||
}
|
||||
|
||||
type TransactionManager struct {
|
||||
// only allow one write transaction at a time
|
||||
c chan struct{}
|
||||
}
|
||||
|
||||
func NewTransactionManager() *TransactionManager {
|
||||
return &TransactionManager{
|
||||
c: make(chan struct{}, 1),
|
||||
}
|
||||
return &TransactionManager{}
|
||||
}
|
||||
|
||||
func (t *TransactionManager) WithTxn(ctx context.Context, fn func(r models.Repository) error) error {
|
||||
database.WriteMu.Lock()
|
||||
defer database.WriteMu.Unlock()
|
||||
return models.WithTxn(&transaction{Ctx: ctx}, fn)
|
||||
}
|
||||
|
||||
|
|
|
|||
Loading…
Reference in a new issue