src/storage/stress-tests/blobfs/instance_actor.rs

// Copyright 2021 The Fuchsia Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

use {
    async_trait::async_trait,
    fs_management::filesystem::ServingSingleVolumeFilesystem,
    storage_stress_test_utils::fvm::FvmInstance,
    stress_test::actor::{Actor, ActorError},
};

/// An actor that kills blobfs and destroys the ramdisk
pub struct InstanceActor {
    pub instance: Option<(ServingSingleVolumeFilesystem, FvmInstance)>,
}

impl InstanceActor {
    pub fn new(fvm: FvmInstance, blobfs: ServingSingleVolumeFilesystem) -> Self {
        Self { instance: Some((blobfs, fvm)) }
    }
}

#[async_trait]
impl Actor for InstanceActor {
    async fn perform(&mut self) -> Result<(), ActorError> {
        if let Some((blobfs, _)) = self.instance.take() {
            // TODO(https://fxbug.dev/42057166): Make termination more abrupt.
            blobfs.shutdown().await.expect("Could not kill blobfs");
        } else {
            panic!("Instance was already killed!")
        }
        Err(ActorError::ResetEnvironment)
    }
}