Conditions | 10 |
Total Lines | 105 |
Code Lines | 61 |
Lines | 0 |
Ratio | 0 % |
Tests | 45 |
CRAP Score | 10.0244 |
Changes | 0 |
Small methods make your code easier to understand, in particular if combined with a good name. Besides, if your method is small, finding a good name is usually much easier.
For example, if you find yourself adding comments to a method's body, this is usually a good sign to extract the commented part to a new method, and use the comment as a starting point when coming up with a good name for this new method.
Commonly applied refactorings include:
If many parameters/temporary variables are present:
Complex classes like build.managers.flow_builder.FlowBuilder._build_int_sink_flows() often do a lot of different things. To break such a class down, we need to identify a cohesive component within that class. A common approach to find such a component is to look for fields/methods that share the same prefixes, or suffixes.
Once you have determined the fields that belong together, you can apply the Extract Class refactoring. If the component makes sense as a sub-class, Extract Subclass is also a candidate, and is often faster.
1 | """flow_builder module responsible for building and mapping flows.""" |
||
190 | def _build_int_sink_flows( |
||
191 | 1 | self, |
|
192 | uni_dst_key: Literal["uni_a", "uni_z"], |
||
193 | 1 | evc: dict, |
|
194 | stored_flows: dict[int, list[dict]], |
||
195 | ) -> list[dict]: |
||
196 | """ |
||
197 | Build INT sink flows. |
||
198 | |||
199 | At the INT sink, one flow becomes many: |
||
200 | 1. Before the proxy, we do add_int_metadata as an INT hop. |
||
201 | We need to keep the set_queue |
||
202 | 2. After the proxy, we do send_report and pop_int and output |
||
203 | We only use table 0 for #1. |
||
204 | We use table X (2 or 3) for #2. for pop_int and output |
||
205 | """ |
||
206 | new_flows = [] |
||
207 | dst_uni = evc[uni_dst_key] |
||
208 | proxy_port = dst_uni["proxy_port"] |
||
209 | 1 | dpid = dst_uni["switch"] |
|
210 | 1 | ||
211 | 1 | for flow in stored_flows[ |
|
212 | 1 | utils.get_cookie(evc["id"], settings.MEF_COOKIE_PREFIX) |
|
213 | ]: |
||
214 | 1 | # Only consider this sink's dpid flows |
|
215 | if flow["switch"] != dpid: |
||
216 | continue |
||
217 | # Only consider flows coming from NNI interfaces |
||
218 | 1 | if flow["flow"]["match"]["in_port"] == dst_uni["port_number"]: |
|
219 | 1 | continue |
|
220 | |||
221 | 1 | new_int_flow_tbl_0_tcp = copy.deepcopy(flow) |
|
222 | 1 | ||
223 | if not new_int_flow_tbl_0_tcp: |
||
224 | 1 | continue |
|
225 | |||
226 | 1 | utils.set_new_cookie(new_int_flow_tbl_0_tcp) |
|
227 | utils.set_owner(new_int_flow_tbl_0_tcp) |
||
228 | utils.set_instructions_from_actions(new_int_flow_tbl_0_tcp) |
||
229 | |||
230 | # Save for pos-proxy flows |
||
231 | 1 | new_int_flow_tbl_0_pos = copy.deepcopy(new_int_flow_tbl_0_tcp) |
|
232 | 1 | new_int_flow_tbl_x_pos = copy.deepcopy(new_int_flow_tbl_0_tcp) |
|
233 | 1 | ||
234 | # Prepare TCP Flow for Table 0 PRE proxy |
||
235 | if not utils.is_intra_switch_evc(evc): |
||
236 | 1 | new_int_flow_tbl_0_tcp["flow"]["match"]["dl_type"] = settings.IPv4 |
|
237 | 1 | new_int_flow_tbl_0_tcp["flow"]["match"]["nw_proto"] = settings.TCP |
|
238 | utils.set_priority(new_int_flow_tbl_0_tcp) |
||
239 | |||
240 | 1 | # Add telemetry, keep set_queue, output to the proxy port. |
|
241 | 1 | output_port_no = proxy_port.source.port_number |
|
242 | 1 | for instruction in new_int_flow_tbl_0_tcp["flow"]["instructions"]: |
|
243 | 1 | if instruction["instruction_type"] == "apply_actions": |
|
244 | # Keep set_queue |
||
245 | actions = utils.modify_actions( |
||
246 | 1 | instruction["actions"], |
|
247 | 1 | ["pop_vlan", "push_vlan", "set_vlan", "output"], |
|
248 | 1 | remove=True, |
|
249 | ) |
||
250 | 1 | actions.insert(0, {"action_type": "add_int_metadata"}) |
|
251 | actions.append( |
||
252 | {"action_type": "output", "port": output_port_no} |
||
253 | ) |
||
254 | instruction["actions"] = actions |
||
255 | 1 | ||
256 | 1 | # Prepare UDP Flow for Table 0 |
|
257 | new_int_flow_tbl_0_udp = copy.deepcopy(new_int_flow_tbl_0_tcp) |
||
258 | new_int_flow_tbl_0_udp["flow"]["match"]["nw_proto"] = settings.UDP |
||
259 | 1 | ||
260 | new_flows.append(copy.deepcopy(new_int_flow_tbl_0_tcp)) |
||
261 | new_flows.append(copy.deepcopy(new_int_flow_tbl_0_udp)) |
||
262 | 1 | ||
263 | 1 | # Prepare Flows for Table 0 AFTER proxy. No difference between TCP or UDP |
|
264 | in_port_no = proxy_port.destination.port_number |
||
265 | 1 | ||
266 | 1 | new_int_flow_tbl_0_pos["flow"]["match"]["in_port"] = in_port_no |
|
267 | utils.set_priority(new_int_flow_tbl_0_tcp) |
||
268 | |||
269 | 1 | table_group = self.table_group |
|
270 | new_table_id = table_group[new_int_flow_tbl_x_pos["flow"]["table_group"]] |
||
271 | 1 | instructions = [ |
|
272 | 1 | { |
|
273 | "instruction_type": "apply_actions", |
||
274 | 1 | "actions": [{"action_type": "send_report"}], |
|
275 | 1 | }, |
|
276 | 1 | { |
|
277 | "instruction_type": "goto_table", |
||
278 | "table_id": new_table_id, |
||
279 | }, |
||
280 | ] |
||
281 | new_int_flow_tbl_0_pos["flow"]["instructions"] = instructions |
||
282 | |||
283 | # Prepare Flows for Table X POS proxy |
||
284 | new_int_flow_tbl_x_pos["flow"]["match"]["in_port"] = in_port_no |
||
285 | new_int_flow_tbl_x_pos["flow"]["table_id"] = new_table_id |
||
286 | 1 | ||
287 | for instruction in new_int_flow_tbl_x_pos["flow"]["instructions"]: |
||
288 | if instruction["instruction_type"] == "apply_actions": |
||
289 | 1 | instruction["actions"].insert(0, {"action_type": "pop_int"}) |
|
290 | 1 | ||
291 | new_flows.append(copy.deepcopy(new_int_flow_tbl_0_pos)) |
||
292 | 1 | new_flows.append(copy.deepcopy(new_int_flow_tbl_x_pos)) |
|
293 | 1 | ||
294 | return new_flows |
||
295 |