-
-
Notifications
You must be signed in to change notification settings - Fork 160
/
cluster.go
121 lines (100 loc) · 3.71 KB
/
cluster.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
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
//go:build linux && cgo && !agent
package db
import (
"context"
"fmt"
"github.com/lxc/incus/v6/internal/server/db/cluster"
"github.com/lxc/incus/v6/internal/server/db/query"
"github.com/lxc/incus/v6/internal/version"
"github.com/lxc/incus/v6/shared/api"
)
// ClusterGroupToAPI is a convenience to convert a ClusterGroup db struct into
// an API cluster group struct.
func ClusterGroupToAPI(clusterGroup *cluster.ClusterGroup, nodes []string) *api.ClusterGroup {
c := &api.ClusterGroup{
ClusterGroupPut: api.ClusterGroupPut{
Description: clusterGroup.Description,
Members: nodes,
},
ClusterGroupPost: api.ClusterGroupPost{
Name: clusterGroup.Name,
},
}
return c
}
// GetClusterGroupNodes returns a list of nodes of the given cluster group.
func (c *ClusterTx) GetClusterGroupNodes(ctx context.Context, groupName string) ([]string, error) {
q := `SELECT nodes.name FROM nodes_cluster_groups
JOIN nodes ON nodes.id = nodes_cluster_groups.node_id
JOIN cluster_groups ON cluster_groups.id = nodes_cluster_groups.group_id
WHERE cluster_groups.name = ?`
return query.SelectStrings(ctx, c.tx, q, groupName)
}
// GetClusterGroupURIs returns all available ClusterGroup URIs.
// generator: ClusterGroup URIs
func (c *ClusterTx) GetClusterGroupURIs(ctx context.Context, filter cluster.ClusterGroupFilter) ([]string, error) {
var args []any
var sql string
if filter.Name != nil && filter.ID == nil {
sql = `SELECT cluster_groups.name FROM cluster_groups
WHERE cluster_groups.name = ? ORDER BY cluster_groups.name
`
args = []any{
filter.Name,
}
} else if filter.ID == nil && filter.Name == nil {
sql = `SELECT cluster_groups.name FROM cluster_groups ORDER BY cluster_groups.name`
args = []any{}
} else {
return nil, fmt.Errorf("No statement exists for the given Filter")
}
names, err := query.SelectStrings(ctx, c.tx, sql, args...)
if err != nil {
return nil, err
}
uris := make([]string, len(names))
for i, name := range names {
uris[i] = api.NewURL().Path(version.APIVersion, "cluster", "groups", name).String()
}
return uris, nil
}
// AddNodeToClusterGroup adds a given node to the given cluster group.
func (c *ClusterTx) AddNodeToClusterGroup(ctx context.Context, groupName string, nodeName string) error {
groupID, err := cluster.GetClusterGroupID(ctx, c.tx, groupName)
if err != nil {
return fmt.Errorf("Failed to get cluster group ID: %w", err)
}
nodeInfo, err := c.GetNodeByName(ctx, nodeName)
if err != nil {
return fmt.Errorf("Failed to get node info: %w", err)
}
_, err = c.tx.Exec(`INSERT INTO nodes_cluster_groups (node_id, group_id) VALUES(?, ?)`, nodeInfo.ID, groupID)
if err != nil {
return err
}
return nil
}
// RemoveNodeFromClusterGroup removes a given node from the given group name.
func (c *ClusterTx) RemoveNodeFromClusterGroup(ctx context.Context, groupName string, nodeName string) error {
groupID, err := cluster.GetClusterGroupID(ctx, c.tx, groupName)
if err != nil {
return fmt.Errorf("Failed to get cluster group ID: %w", err)
}
nodeInfo, err := c.GetNodeByName(ctx, nodeName)
if err != nil {
return fmt.Errorf("Failed to get node info: %w", err)
}
_, err = c.tx.Exec(`DELETE FROM nodes_cluster_groups WHERE node_id = ? AND group_id = ?`, nodeInfo.ID, groupID)
if err != nil {
return err
}
return nil
}
// GetClusterGroupsWithNode returns a list of cluster group names the given node belongs to.
func (c *ClusterTx) GetClusterGroupsWithNode(ctx context.Context, nodeName string) ([]string, error) {
q := `SELECT cluster_groups.name FROM nodes_cluster_groups
JOIN cluster_groups ON cluster_groups.id = nodes_cluster_groups.group_id
JOIN nodes ON nodes.id = nodes_cluster_groups.node_id
WHERE nodes.name = ?`
return query.SelectStrings(ctx, c.tx, q, nodeName)
}