From 6207917e10872f123eb70feebb30f5b130722181 Mon Sep 17 00:00:00 2001
From: Kenjiro Nakayama <nakayamakenjiro@gmail.com>
Date: Fri, 20 May 2022 10:48:52 +0900
Subject: [PATCH] Use map()

---
 aya/src/programs/cgroup_skb.rs | 9 ++++-----
 1 file changed, 4 insertions(+), 5 deletions(-)

diff --git a/aya/src/programs/cgroup_skb.rs b/aya/src/programs/cgroup_skb.rs
index fdde66740..070812de9 100644
--- a/aya/src/programs/cgroup_skb.rs
+++ b/aya/src/programs/cgroup_skb.rs
@@ -62,11 +62,10 @@ pub struct CgroupSkb {
 impl CgroupSkb {
     /// Loads the program inside the kernel.
     pub fn load(&mut self) -> Result<(), ProgramError> {
-        self.data.expected_attach_type = match self.expected_attach_type {
-            Some(CgroupSkbAttachType::Ingress) => Some(BPF_CGROUP_INET_INGRESS),
-            Some(CgroupSkbAttachType::Egress) => Some(BPF_CGROUP_INET_EGRESS),
-            None => None,
-        };
+        self.data.expected_attach_type = self.expected_attach_type.map(|attach_type| match attach_type {
+            CgroupSkbAttachType::Ingress => BPF_CGROUP_INET_INGRESS,
+            CgroupSkbAttachType::Egress => BPF_CGROUP_INET_EGRESS,
+        });
         load_program(BPF_PROG_TYPE_CGROUP_SKB, &mut self.data)
     }