From 6ae87d9d661e721ee3b3f249817b7f20b4f4d652 Mon Sep 17 00:00:00 2001 From: Zhongtao Hu Date: Mon, 31 Jan 2022 11:25:13 +0800 Subject: [PATCH] libs/types: use contains to make code more readable use contains to when validate hypervisor block_device_driver Signed-off-by: Zhongtao Hu --- src/libs/kata-types/src/config/hypervisor/mod.rs | 14 ++++++++------ 1 file changed, 8 insertions(+), 6 deletions(-) diff --git a/src/libs/kata-types/src/config/hypervisor/mod.rs b/src/libs/kata-types/src/config/hypervisor/mod.rs index da8e20650..4aa4ceba7 100644 --- a/src/libs/kata-types/src/config/hypervisor/mod.rs +++ b/src/libs/kata-types/src/config/hypervisor/mod.rs @@ -165,12 +165,14 @@ impl BlockDeviceInfo { if self.disable_block_device_use { return Ok(()); } - if self.block_device_driver != VIRTIO_BLK - && self.block_device_driver != VIRTIO_BLK_CCW - && self.block_device_driver != VIRTIO_BLK_MMIO - && self.block_device_driver != VIRTIO_SCSI - && self.block_device_driver != VIRTIO_PMEM - { + let l = [ + VIRTIO_BLK, + VIRTIO_BLK_CCW, + VIRTIO_BLK_MMIO, + VIRTIO_SCSI, + VIRTIO_PMEM, + ]; + if !l.contains(&self.block_device_driver.as_str()) { return Err(eother!( "{} is unsupported block device type.", self.block_device_driver