-
Notifications
You must be signed in to change notification settings - Fork 9
/
dao.go
64 lines (54 loc) · 1.33 KB
/
dao.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
60
61
62
63
64
package impl
import (
"context"
"database/sql"
"fmt"
"github.com/infraboard/cmdb/apps/rds"
)
func (s *service) save(ctx context.Context, h *rds.Rds) error {
var (
stmt *sql.Stmt
err error
)
tx, err := s.db.BeginTx(ctx, nil)
if err != nil {
return err
}
defer func() {
if err != nil {
if err := tx.Rollback(); err != nil {
s.log.Errorf("rollback error, %s", err)
}
} else {
if err := tx.Commit(); err != nil {
s.log.Errorf("commit error, %s", err)
}
}
}()
// 生成描写信息的Hash
if err := h.GenHash(); err != nil {
return err
}
// err = impl.SaveResource(ctx, tx, h.Resource)
// if err != nil {
// return err
// }
// 避免SQL注入, 请使用Prepare
stmt, err = tx.Prepare(insertRdsSQL)
if err != nil {
return err
}
defer stmt.Close()
desc := h.Describe
_, err = stmt.Exec(
h.Resource.Meta.Id, desc.EngineType, desc.EngineVersion, desc.InstanceClass, desc.ClassType, desc.ExportType,
desc.NetworkType, desc.Type, desc.DbMaxQuantity, desc.AccountMaxQuantity, desc.MaxConnections,
desc.MaxIops, desc.Collation, desc.TimeZone, desc.StorageType, desc.SecurityIpMode,
desc.SecurityIpListToString(), desc.ConnectionMode, desc.IpType, desc.DeployMode,
desc.Port, desc.ExtraToJson(),
)
if err != nil {
return fmt.Errorf("save rds resource describe error, %s", err)
}
return err
}