root / src / Ganeti / Query / Network.hs @ beb9c009
History | View | Annotate | Download (7.2 kB)
1 |
{-| Implementation of the Ganeti Query2 node group queries. |
---|---|
2 |
|
3 |
-} |
4 |
|
5 |
{- |
6 |
|
7 |
Copyright (C) 2012, 2013 Google Inc. |
8 |
|
9 |
This program is free software; you can redistribute it and/or modify |
10 |
it under the terms of the GNU General Public License as published by |
11 |
the Free Software Foundation; either version 2 of the License, or |
12 |
(at your option) any later version. |
13 |
|
14 |
This program is distributed in the hope that it will be useful, but |
15 |
WITHOUT ANY WARRANTY; without even the implied warranty of |
16 |
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU |
17 |
General Public License for more details. |
18 |
|
19 |
You should have received a copy of the GNU General Public License |
20 |
along with this program; if not, write to the Free Software |
21 |
Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA |
22 |
02110-1301, USA. |
23 |
|
24 |
-} |
25 |
|
26 |
module Ganeti.Query.Network |
27 |
( getGroupConnection |
28 |
, getNetworkUuid |
29 |
, instIsConnected |
30 |
, Runtime |
31 |
, fieldsMap |
32 |
, collectLiveData |
33 |
) where |
34 |
|
35 |
-- FIXME: everything except Runtime(..) and fieldsMap |
36 |
-- is only exported for testing. |
37 |
|
38 |
import qualified Data.Map as Map |
39 |
import Data.Maybe (fromMaybe, mapMaybe) |
40 |
import Data.List (find, foldl', intercalate) |
41 |
|
42 |
import Ganeti.JSON |
43 |
import Ganeti.Network |
44 |
import Ganeti.Objects |
45 |
import Ganeti.Query.Language |
46 |
import Ganeti.Query.Common |
47 |
import Ganeti.Query.Types |
48 |
import Ganeti.Types |
49 |
|
50 |
-- | There is no actual runtime. |
51 |
data Runtime = Runtime |
52 |
|
53 |
networkFields :: FieldList Network Runtime |
54 |
networkFields = |
55 |
[ (FieldDefinition "name" "Network" QFTText "Name", |
56 |
FieldSimple (rsNormal . networkName), QffNormal) |
57 |
, (FieldDefinition "network" "Subnet" QFTText "IPv4 subnet", |
58 |
FieldSimple (rsNormal . networkNetwork), QffNormal) |
59 |
, (FieldDefinition "gateway" "Gateway" QFTOther "IPv4 gateway", |
60 |
FieldSimple (rsMaybeUnavail . networkGateway), QffNormal) |
61 |
, (FieldDefinition "network6" "IPv6Subnet" QFTOther "IPv6 subnet", |
62 |
FieldSimple (rsMaybeUnavail . networkNetwork6), QffNormal) |
63 |
, (FieldDefinition "gateway6" "IPv6Gateway" QFTOther "IPv6 gateway", |
64 |
FieldSimple (rsMaybeUnavail . networkGateway6), QffNormal) |
65 |
, (FieldDefinition "mac_prefix" "MacPrefix" QFTOther "MAC address prefix", |
66 |
FieldSimple (rsMaybeUnavail . networkMacPrefix), QffNormal) |
67 |
, (FieldDefinition "free_count" "FreeCount" QFTNumber "Number of available\ |
68 |
\ addresses", |
69 |
FieldSimple (rsMaybeNoData . fmap getFreeCount . createAddressPool), |
70 |
QffNormal) |
71 |
, (FieldDefinition "map" "Map" QFTText "Actual mapping", |
72 |
FieldSimple (rsMaybeNoData . fmap getMap . createAddressPool), |
73 |
QffNormal) |
74 |
, (FieldDefinition "reserved_count" "ReservedCount" QFTNumber |
75 |
"Number of reserved addresses", |
76 |
FieldSimple (rsMaybeNoData . fmap getReservedCount . createAddressPool), |
77 |
QffNormal) |
78 |
, (FieldDefinition "group_list" "GroupList" QFTOther |
79 |
"List of nodegroups (group name, NIC mode, NIC link)", |
80 |
FieldConfig (\cfg -> rsNormal . getGroupConnections cfg . networkUuid), |
81 |
QffNormal) |
82 |
, (FieldDefinition "group_cnt" "NodeGroups" QFTNumber "Number of nodegroups", |
83 |
FieldConfig (\cfg -> rsNormal . length . getGroupConnections cfg |
84 |
. networkUuid), QffNormal) |
85 |
, (FieldDefinition "inst_list" "InstanceList" QFTOther "List of instances", |
86 |
FieldConfig (\cfg -> rsNormal . getInstances cfg . networkUuid), |
87 |
QffNormal) |
88 |
, (FieldDefinition "inst_cnt" "Instances" QFTNumber "Number of instances", |
89 |
FieldConfig (\cfg -> rsNormal . length . getInstances cfg |
90 |
. networkUuid), QffNormal) |
91 |
, (FieldDefinition "external_reservations" "ExternalReservations" QFTText |
92 |
"External reservations", |
93 |
FieldSimple getExtReservationsString, QffNormal) |
94 |
] ++ |
95 |
timeStampFields ++ |
96 |
uuidFields "Network" ++ |
97 |
serialFields "Network" ++ |
98 |
tagsFields |
99 |
|
100 |
-- | The group fields map. |
101 |
fieldsMap :: FieldMap Network Runtime |
102 |
fieldsMap = |
103 |
Map.fromList $ map (\v@(f, _, _) -> (fdefName f, v)) networkFields |
104 |
|
105 |
-- TODO: the following fields are not implemented yet: external_reservations |
106 |
|
107 |
-- | Given a network's UUID, this function lists all connections from |
108 |
-- the network to nodegroups including the respective mode and links. |
109 |
getGroupConnections :: ConfigData -> String -> [(String, String, String)] |
110 |
getGroupConnections cfg network_uuid = |
111 |
mapMaybe (getGroupConnection network_uuid) |
112 |
((Map.elems . fromContainer . configNodegroups) cfg) |
113 |
|
114 |
-- | Given a network's UUID and a node group, this function assembles |
115 |
-- a tuple of the group's name, the mode and the link by which the |
116 |
-- network is connected to the group. Returns 'Nothing' if the network |
117 |
-- is not connected to the group. |
118 |
getGroupConnection :: String -> NodeGroup -> Maybe (String, String, String) |
119 |
getGroupConnection network_uuid group = |
120 |
let networks = fromContainer . groupNetworks $ group |
121 |
in case Map.lookup network_uuid networks of |
122 |
Nothing -> Nothing |
123 |
Just net -> |
124 |
Just (groupName group, getNicMode net, getNicLink net) |
125 |
|
126 |
-- | Retrieves the network's mode and formats it human-readable, |
127 |
-- also in case it is not available. |
128 |
getNicMode :: PartialNicParams -> String |
129 |
getNicMode nic_params = |
130 |
maybe "-" nICModeToRaw $ nicpModeP nic_params |
131 |
|
132 |
-- | Retrieves the network's link and formats it human-readable, also in |
133 |
-- case it it not available. |
134 |
getNicLink :: PartialNicParams -> String |
135 |
getNicLink nic_params = fromMaybe "-" (nicpLinkP nic_params) |
136 |
|
137 |
-- | Retrieves the network's instances' names. |
138 |
getInstances :: ConfigData -> String -> [String] |
139 |
getInstances cfg network_uuid = |
140 |
map instName (filter (instIsConnected cfg network_uuid) |
141 |
((Map.elems . fromContainer . configInstances) cfg)) |
142 |
|
143 |
-- | Helper function that checks if an instance is linked to the given network. |
144 |
instIsConnected :: ConfigData -> String -> Instance -> Bool |
145 |
instIsConnected cfg network_uuid inst = |
146 |
network_uuid `elem` mapMaybe (getNetworkUuid cfg) |
147 |
(mapMaybe nicNetwork (instNics inst)) |
148 |
|
149 |
-- | Helper function to look up a network's UUID by its name |
150 |
getNetworkUuid :: ConfigData -> String -> Maybe String |
151 |
getNetworkUuid cfg name = |
152 |
let net = find (\n -> name == fromNonEmpty (networkName n)) |
153 |
((Map.elems . fromContainer . configNetworks) cfg) |
154 |
in fmap networkUuid net |
155 |
|
156 |
-- | Computes the reservations list for a network. |
157 |
-- |
158 |
-- This doesn't use the netmask for validation of the length, instead |
159 |
-- simply iterating over the reservations string. |
160 |
getReservations :: Ip4Network -> String -> [Ip4Address] |
161 |
getReservations (Ip4Network net _) = |
162 |
reverse . |
163 |
fst . |
164 |
foldl' (\(accu, addr) c -> |
165 |
let addr' = nextIp4Address addr |
166 |
accu' = case c of |
167 |
'1' -> addr:accu |
168 |
'0' -> accu |
169 |
_ -> -- FIXME: the reservations string |
170 |
-- should be a proper type |
171 |
accu |
172 |
in (accu', addr')) ([], net) |
173 |
|
174 |
-- | Computes the external reservations as string for a network. |
175 |
getExtReservationsString :: Network -> ResultEntry |
176 |
getExtReservationsString net = |
177 |
let addrs = getReservations (networkNetwork net) |
178 |
(fromMaybe "" $ networkExtReservations net) |
179 |
in rsNormal . intercalate ", " $ map show addrs |
180 |
|
181 |
-- | Dummy function for collecting live data (which networks don't have). |
182 |
collectLiveData :: Bool -> ConfigData -> [Network] -> IO [(Network, Runtime)] |
183 |
collectLiveData _ _ = return . map (\n -> (n, Runtime)) |