adyxax/tfstated
1
0
Fork 0
Code Issues 47 Pull requests Releases Packages Wiki Activity Actions

feat(tfstated): allow configuration of version history limit via environment variable

Browse source
This commit is contained in:
Julien Dessaux 2024-10-18 00:01:50 +02:00
parent 4c2032e1f0
commit 1ebf0341dd
Signed by: adyxax
GPG key ID: F92E51B86E07177E
5 changed files with 43 additions and 9 deletions
Download patch file Download diff file Expand all files Collapse all files

20
pkg/database/db.go
View file

@ -26,10 +26,11 @@ func initDB(ctx context.Context, url string) (*sql.DB, error) {
}
type DB struct {
ctx context.Context
dataEncryptionKey scrypto.AES256Key
readDB *sql.DB
writeDB *sql.DB
ctx context.Context
dataEncryptionKey scrypto.AES256Key
readDB *sql.DB
versionsHistoryLimit int
writeDB *sql.DB
}
func NewDB(ctx context.Context, url string) (*DB, error) {
@ -56,9 +57,10 @@ func NewDB(ctx context.Context, url string) (*DB, error) {
writeDB.SetMaxOpenConns(1)
db := DB{
ctx: ctx,
readDB: readDB,
writeDB: writeDB,
ctx: ctx,
readDB: readDB,
versionsHistoryLimit: 64,
writeDB: writeDB,
}
if _, err = db.Exec("PRAGMA foreign_keys = ON"); err != nil {
return nil, err
@ -101,3 +103,7 @@ func (db *DB) QueryRow(query string, args ...any) *sql.Row {
func (db *DB) SetDataEncryptionKey(s string) error {
return db.dataEncryptionKey.FromBase64(s)
}
func (db *DB) SetVersionsHistoryLimit(n int) {
db.versionsHistoryLimit = n
}