/
version_db.go
59 lines (50 loc) · 1.46 KB
/
version_db.go
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
package sqldb
import (
"context"
"encoding/json"
"code.cloudfoundry.org/bbs/db/sqldb/helpers"
"code.cloudfoundry.org/bbs/models"
"code.cloudfoundry.org/lager/v3"
)
const VersionID = "version"
func (db *SQLDB) SetVersion(tx helpers.Tx, ctx context.Context, logger lager.Logger, version *models.Version) error {
logger = logger.Session("db-set-version", lager.Data{"version": version})
logger.Debug("starting")
defer logger.Debug("complete")
versionJSON, err := json.Marshal(version)
if err != nil {
logger.Error("failed-marshalling-version", err)
return err
}
err = db.helper.RetryOnDeadlock(logger, func() error {
return db.setConfigurationValue(tx, ctx, logger, VersionID, string(versionJSON))
})
if err != nil {
return db.convertSQLError(err)
}
return nil
}
func (db *SQLDB) Version(tx helpers.Tx, ctx context.Context, logger lager.Logger) (*models.Version, error) {
logger = logger.Session("db-version")
logger.Debug("starting")
defer logger.Debug("complete")
var versionJSON string
err := db.helper.RetryOnDeadlock(logger, func() error {
var err error
versionJSON, err = db.getConfigurationValue(tx, ctx, logger, VersionID)
if err != nil {
return err
}
return nil
})
if err != nil {
return nil, db.convertSQLError(err)
}
var version models.Version
err = json.Unmarshal([]byte(versionJSON), &version)
if err != nil {
logger.Error("failed-to-deserialize-version", err)
return nil, models.ErrDeserialize
}
return &version, nil
}