1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
|
package database
import (
"database/sql"
"errors"
"fmt"
"time"
"git.adyxax.org/adyxax/tfstated/pkg/model"
)
func (db *DB) LoadVersionById(id int) (*model.Version, error) {
version := model.Version{
Id: id,
}
var (
created int64
encryptedData []byte
)
err := db.QueryRow(
`SELECT account_id, state_id, data, lock, created FROM versions WHERE id = ?;`,
id).Scan(
&version.AccountId,
&version.StateId,
&encryptedData,
&version.Lock,
&created)
if err != nil {
if errors.Is(err, sql.ErrNoRows) {
return nil, nil
}
return nil, fmt.Errorf("failed to load version id %d from database: %w", id, err)
}
version.Created = time.Unix(created, 0)
version.Data, err = db.dataEncryptionKey.DecryptAES256(encryptedData)
if err != nil {
return nil, fmt.Errorf("failed to decrypt version %d data: %w", id, err)
}
return &version, nil
}
func (db *DB) LoadVersionsByState(state *model.State) ([]model.Version, error) {
rows, err := db.Query(
`SELECT account_id, created, data, id, lock
FROM versions
WHERE state_id = ?
ORDER BY id DESC;`, state.Id)
if err != nil {
return nil, fmt.Errorf("failed to load versions from database: %w", err)
}
defer rows.Close()
versions := make([]model.Version, 0)
for rows.Next() {
var version model.Version
var created int64
err = rows.Scan(&version.AccountId, &created, &version.Data, &version.Id, &version.Lock)
if err != nil {
return nil, fmt.Errorf("failed to load version from row: %w", err)
}
version.Created = time.Unix(created, 0)
versions = append(versions, version)
}
if err := rows.Err(); err != nil {
return nil, fmt.Errorf("failed to load versions from rows: %w", err)
}
return versions, nil
}
|
