-
-
Notifications
You must be signed in to change notification settings - Fork 165
/
operations.go
68 lines (56 loc) · 1.94 KB
/
operations.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
65
66
67
68
//go:build linux && cgo && !agent
package db
import (
"context"
"github.com/lxc/incus/v6/internal/server/db/cluster"
"github.com/lxc/incus/v6/internal/server/db/operationtype"
"github.com/lxc/incus/v6/internal/server/db/query"
)
// GetAllNodesWithOperations returns a list of nodes that have operations in any project.
func (c *ClusterTx) GetAllNodesWithOperations(ctx context.Context) ([]string, error) {
stmt := `
SELECT DISTINCT nodes.address
FROM operations
JOIN nodes ON nodes.id = operations.node_id
`
return query.SelectStrings(ctx, c.tx, stmt)
}
// GetNodesWithOperations returns a list of nodes that have operations.
func (c *ClusterTx) GetNodesWithOperations(ctx context.Context, project string) ([]string, error) {
stmt := `
SELECT DISTINCT nodes.address
FROM operations
LEFT OUTER JOIN projects ON projects.id = operations.project_id
JOIN nodes ON nodes.id = operations.node_id
WHERE projects.name = ? OR operations.project_id IS NULL
`
return query.SelectStrings(ctx, c.tx, stmt, project)
}
// GetOperationsOfType returns a list operations that belong to the specified project and have the desired type.
func (c *ClusterTx) GetOperationsOfType(ctx context.Context, projectName string, opType operationtype.Type) ([]cluster.Operation, error) {
var ops []cluster.Operation
stmt := `
SELECT operations.id, operations.uuid, operations.type, nodes.address
FROM operations
LEFT JOIN projects on projects.id = operations.project_id
JOIN nodes on nodes.id = operations.node_id
WHERE (projects.name = ? OR operations.project_id IS NULL) and operations.type = ?
`
rows, err := c.tx.QueryContext(ctx, stmt, projectName, opType)
if err != nil {
return nil, err
}
defer func() { _ = rows.Close() }()
for rows.Next() {
var op cluster.Operation
err := rows.Scan(&op.ID, &op.UUID, &op.Type, &op.NodeAddress)
if err != nil {
return nil, err
}
ops = append(ops, op)
}
if rows.Err() != nil {
return nil, err
}
return ops, nil
}