确保延长用作引用的MutexGuard的寿命

huangapple go评论84阅读模式
英文:

Ensure extended lifetime of MutexGuard used as reference

问题

在以下最小代码示例中,使用MutexGuard来访问BindGroup,确保其存在的时间比RenderPass的生命周期更长。然而,显然Rust编译器并不知晓这一点,导致了生命周期错误。

背景信息:RenderPassBindGroup都来自wgpu库。因此,我无法简单地调整它们的方法参数。
RenderPass每帧都会创建,而TextureManager只在程序启动时创建一次。
TextureManager将对BindGroup的引用保持为Arc<Mutex<...>>,以便从多个线程中进行操作。

use core::marker::PhantomData;
use std::ops::Deref;
use std::sync::{Arc, Mutex};

fn main() {
    let texture_manager = TextureManager {
        active_bind_group: Arc::new(Mutex::new(None)),
    };
    texture_manager.activate_texture(0);

    let mut render_pass = RenderPass::new();

    step(&mut render_pass, &texture_manager);
}

fn step<'pass>(render_pass: &mut RenderPass<'pass>, texture_manager: &'pass TextureManager) {
    let guard = texture_manager.active_bind_group.lock().unwrap();
    if let Some(bind_group) = guard.deref() {
        render_pass.set_bind_group(bind_group);
    }
}

struct TextureManager {
    active_bind_group: Arc<Mutex<Option<BindGroup>>>,
}

impl TextureManager {
    fn activate_texture(&self, index: usize) {
        *self.active_bind_group.lock().unwrap() = Some(BindGroup);
    }
}

struct BindGroup;

struct RenderPass<'pass> {
    phantom: PhantomData<&'pass ()>,
}

impl<'pass> RenderPass<'pass> {
    fn new() -> Self {
        Self {
            phantom: PhantomData,
        }
    }

    fn set_bind_group(&mut self, bind_group: &'pass BindGroup) {}
}

错误信息:

error[E0597]: `guard` does not live long enough
  --> src/main.rs:18:31
   |
16 | fn step<'pass>(render_pass: &mut RenderPass<'pass>, texture_manager: &'pass TextureManager) {
   |         ----- lifetime `'pass` defined here
17 |     let guard = texture_manager.active_bind_group.lock().unwrap();
   |         ----- binding `guard` declared here
18 |     if let Some(bind_group) = guard.deref() {
   |                               ^^^^^^^^^^^^^
   |                               |
   |                               borrowed value does not live long enough
   |                               argument requires that `guard` is borrowed for `'pass`
...
21 | }
   | - `guard` dropped here while still borrowed

我理解错误的根源:MutexGuard只在step方法的范围内存在。一旦它被释放,就不能再保证对原始值的引用。然而,我还没有找到解决这个问题的方法。在这种情况下,是否可能有不同的构造来替代Mutex

英文:

In the following minimal code example, a MutexGuard is used for accessing a BindGroup that is sure to exist longer than the lifetime of the RenderPass. However, this is obviously not known to the Rust compiler, resulting in a lifetime error.

For context: both RenderPass and BindGroup stem from the wgpu crate. Thus, it is not possible for me to simply adjust their method parameters.
The RenderPass is created every frame, whereas the TextureManager is only created once the program starts.
The TextureManager keeps the reference to the BindGroup as Arc&lt;Mutex&lt;…&gt;&gt; in order to manipulate it from multiple threads.

use core::marker::PhantomData;
use std::ops::Deref;
use std::sync::{Arc, Mutex};

fn main() {
    let texture_manager = TextureManager {
        active_bind_group: Arc::new(Mutex::new(None)),
    };
    texture_manager.activate_texture(0);

    let mut render_pass = RenderPass::new();

    step(&amp;mut render_pass, &amp;texture_manager);
}

fn step&lt;&#39;pass&gt;(render_pass: &amp;mut RenderPass&lt;&#39;pass&gt;, texture_manager: &amp;&#39;pass TextureManager) {
    let guard = texture_manager.active_bind_group.lock().unwrap();
    if let Some(bind_group) = guard.deref() {
        render_pass.set_bind_group(bind_group);
    }
}

struct TextureManager {
    active_bind_group: Arc&lt;Mutex&lt;Option&lt;BindGroup&gt;&gt;&gt;,
    // textures: Arc&lt;Mutex&lt;Vec&lt;Texture&gt;&gt;&gt;,
}

impl TextureManager {
    // Only uses &amp;self instead of &amp;mut self for providing immutable interface
    // so that a `Arc&lt;TextureManager&gt;` suffices for use in multithreaded code
    fn activate_texture(&amp;self, index: usize) {
        // let texture = textures.lock().unwrap()[index];
        // update bind group using the texture
        *self.active_bind_group.lock().unwrap() = Some(BindGroup);
    }
}

struct BindGroup;

struct RenderPass&lt;&#39;pass&gt; {
    phantom: PhantomData&lt;&amp;&#39;pass ()&gt;,
}

impl&lt;&#39;pass&gt; RenderPass&lt;&#39;pass&gt; {
    fn new() -&gt; Self {
        Self {
            phantom: PhantomData,
        }
    }

    fn set_bind_group(&amp;mut self, bind_group: &amp;&#39;pass BindGroup) {}
}

Rust Playground

The resulting error:

error[E0597]: `guard` does not live long enough
  --&gt; src/main.rs:18:31
   |
16 | fn step&lt;&#39;pass&gt;(render_pass: &amp;mut RenderPass&lt;&#39;pass&gt;, texture_manager: &amp;&#39;pass TextureManager) {
   |         ----- lifetime `&#39;pass` defined here
17 |     let guard = texture_manager.active_bind_group.lock().unwrap();
   |         ----- binding `guard` declared here
18 |     if let Some(bind_group) = guard.deref() {
   |                               ^^^^^^^^^^^^^
   |                               |
   |                               borrowed value does not live long enough
   |                               argument requires that `guard` is borrowed for `&#39;pass`
...
21 | }
   | - `guard` dropped here while still borrowed

I understand the origin of the error: the MutexGuard only lives for the scope of the step method. As soon as it is dropped, a reference to the original value can no longer be ensured.
Still, I haven't found any solution as to how to solve this problem.
Are there potentially different constructs than Mutex to use in this scenario?

答案1

得分: 2

根本上,你这里不只是要写下正确的生命周期;你必须向编译器证明你借用的绑定组在渲染过程中使用时不会被丢弃或变异

一种方法是将MutexGuard移到与RenderPass相同的作用域中(而不是step),以便它能够足够长时间存在。这基本上意味着将step()内联到main()中。

另一种更具可组合性的方法是利用共享所有权。将BindGroup放入Arc中,并进行克隆,这样即使互斥体中的值发生变化,旧值仍然可用 - 不可变且不会丢弃 - 给渲染过程使用。然而,你仍然需要一个地方来存储克隆的Arc以在所需的时间段内保持其有效。直接而高效的做法是在创建渲染过程之前设置要使用的绑定组的某个变量中:

fn main() {
    let texture_manager = TextureManager {
        active_bind_group: Arc::new(Mutex::new(None)),
    };
    texture_manager.activate_texture(0);

    let bind_group_to_use = get_bind_group(&texture_manager); // 在渲染过程之前
    let mut render_pass = RenderPass::new();
    if let Some(g) = bind_group_to_use {                      // 在渲染过程中
        render_pass.set_bind_group(&g);
    }
}

/// 这是以前的step()函数
fn get_bind_group(texture_manager: &TextureManager) -> Option<Arc<BindGroup>> {
    Option::clone(&texture_manager.active_bind_group.lock().unwrap())
}

struct TextureManager {
    active_bind_group: Arc<Mutex<Option<Arc<BindGroup>>>>,
}

impl TextureManager {
    fn activate_texture(&self, index: usize) {
        // 创建Arc
        *self.active_bind_group.lock().unwrap() = Some(Arc::new(BindGroup));
    }
}

但也许这不够。也许你实际上需要在构建RenderPass时可能创建许多绑定组,而且要在此过程中进行。在这种情况下,typed_arena::Arena 可以帮助你 - 它为你提供了一个地方来存放具有相同生命周期的可借用的事物,即使其中一些在你开始借用其他事物之前不存在也可以。

使用 arena 的劣势是它必须为其元素分配内存。

use typed_arena::Arena;

pub fn main() {
    let texture_manager = TextureManager {
        active_bind_group: Arc::new(Mutex::new(None)),
    };
    texture_manager.activate_texture(0);

    let arena = Arena::new();
    let mut render_pass = RenderPass::new();

    step(&arena, &mut render_pass, &texture_manager);
}

fn step<'mutex: 'pass, 'pass>(
    arena: &'pass Arena<MutexGuard<'mutex, Option<BindGroup>>>,
    render_pass: &mut RenderPass<'pass>,
    texture_manager: &'mutex TextureManager,
) {
    let guard = texture_manager.active_bind_group.lock().unwrap();
    let guard = arena.alloc(guard);
    if let Some(bind_group) = &**guard {
        render_pass.set_bind_group(bind_group);
    }
}

你也可以结合上述两种方法,将Arc<BindGroup>存储在Arena而不是MutexGuard中。这将提供更多的灵活性 - 你可以使用来自任何源的渲染过程(只要它们是Arc),而且 arena 不会在其类型中具有 ‘mutex 生命周期。

英文:

Fundamentally, what you have here is not just a matter of writing down the right lifetime; you have to demonstrate to the compiler that the bind group you're borrowing will not be dropped or mutated while it is in use by the render pass.

One way to do this is to move the MutexGuard into the same scope as the RenderPass (instead of step) so that it is known to live long enough. This basically means inlining step() into main().

Another way, that is more composable, is to make use of shared ownership. Put the BindGroup into an Arc, and clone it, so that even if the value in the mutex is changed, the old value is still available — immutable and not dropped — to the render pass. However, you'll still need a place to stash the cloned Arc to hold it alive for the desired period. The straightforward and efficient way to do this is to set up the bind groups you want to use in some variable before creating the render pass:

fn main() {
    let texture_manager = TextureManager {
        active_bind_group: Arc::new(Mutex::new(None)),
    };
    texture_manager.activate_texture(0);

    let bind_group_to_use = get_bind_group(&amp;texture_manager); // before pass
    let mut render_pass = RenderPass::new();
    if let Some(g) = bind_group_to_use {                      // within pass
        render_pass.set_bind_group(&amp;g);
    }
}

/// This is what used to be the step() function
fn get_bind_group(texture_manager: &amp;TextureManager) -&gt; Option&lt;Arc&lt;BindGroup&gt;&gt; {
    Option::clone(&amp;texture_manager.active_bind_group.lock().unwrap())
}

struct TextureManager {
    active_bind_group: Arc&lt;Mutex&lt;Option&lt;Arc&lt;BindGroup&gt;&gt;&gt;&gt;,
}

impl TextureManager {
    fn activate_texture(&amp;self, index: usize) {
        // creates the Arc
        *self.active_bind_group.lock().unwrap() = Some(Arc::new(BindGroup));
    }
}

Playground

But perhaps this won't do. Perhaps you actually need to potentially create many bind groups, and do it while you're building the RenderPass. In that case, typed_arena::Arena can help you — it gives you a place to stash borrowable things that all have the same lifetime even if some of them didn't exist at the time you start borrowing other ones.

The disadvantage of using an arena is that the arena must allocate memory for its elements.

use typed_arena::Arena;

pub fn main() {
    let texture_manager = TextureManager {
        active_bind_group: Arc::new(Mutex::new(None)),
    };
    texture_manager.activate_texture(0);

    let arena = Arena::new();
    let mut render_pass = RenderPass::new();

    step(&amp;arena, &amp;mut render_pass, &amp;texture_manager);
}

fn step&lt;&#39;mutex: &#39;pass, &#39;pass&gt;(
    arena: &amp;&#39;pass Arena&lt;MutexGuard&lt;&#39;mutex, Option&lt;BindGroup&gt;&gt;&gt;,
    render_pass: &amp;mut RenderPass&lt;&#39;pass&gt;,
    texture_manager: &amp;&#39;mutex TextureManager,
) {
    let guard = texture_manager.active_bind_group.lock().unwrap();
    let guard = arena.alloc(guard);
    if let Some(bind_group) = &amp;**guard {
        render_pass.set_bind_group(bind_group);
    }
}

You could also combine both of the above, by storing Arc&lt;BindGroup&gt;s in the Arena instead of MutexGuards. This would give even more flexibility — you can use render passes from any source (as long as they are Arced), and the arena won't have the &#39;mutex lifetime in its type.

huangapple
  • 本文由 发表于 2023年8月9日 17:08:58
  • 转载请务必保留本文链接:https://go.coder-hub.com/76866204-2.html
匿名

发表评论

匿名网友

:?: :razz: :sad: :evil: :!: :smile: :oops: :grin: :eek: :shock: :???: :cool: :lol: :mad: :twisted: :roll: :wink: :idea: :arrow: :neutral: :cry: :mrgreen:

确定