質問編集履歴

tokio::sync::Mutex::try_lock を使った実装を追記

2020/03/16 16:19

投稿

techno-tanoC

スコア24

test CHANGED Viewed

File without changes

test CHANGED Viewed

@@ -16,11 +16,11 @@
-### 発生している問題・エラーメッセージ
+### 発生している問題
-`tokio::sync::Mutex` はロックを取得するのに `mutex.lock().await` する必要がありますが、 `await` は `AsyncWrite::poll_write` の中では使えないので困っています。上手く実装を与えることはできないのでしょうか？
+`tokio::sync::Mutex` はロックを取得するのに `mutex.lock().await` する必要がありますが、 `await` は `AsyncWrite::poll_write` の中では使えないので困っています。上手く実装を与えることはできないのでしょうか？(追記した `try_lock` を使った実装は適切なのでしょうか？ )
@@ -218,6 +218,64 @@
+### 追記 2020-03-17 01:20
+`tokio::sync::Mutex::try_lock()` を使うことで実装できました(これが適切な方針なのかは自信が無いのでアドバイスお願いします)
+```rust
+impl<T: AsyncWrite + Unpin> AsyncWrite for Progress<T> {
+    fn poll_write(
+        self: Pin<&mut Self>,
+        cx: &mut Context,
+        buf: &[u8]
+    ) -> Poll<Result<usize, Error>> {
+        match self.0.try_lock() {
+            Ok(mut s) => {
+                let poll = Pin::new(&mut s.inner).poll_write(cx, buf);
+                if let Poll::Ready(Ok(n)) = poll {
+                    s.size += n;
+                }
+                poll
+            },
+            Err(_e) => {
+                Poll::Pending
+            }
+        }
+    }
+    // poll_flush, poll_shutdown 略
+}
+```
 ### 補足情報（FW/ツールのバージョンなど）
 rustc 1.41.1

タスクのリンクを変更

2020/03/16 16:19

投稿

techno-tanoC

スコア24

test CHANGED Viewed

File without changes

test CHANGED Viewed

@@ -4,7 +4,7 @@
-ある[タスク](https://docs.rs/tokio/0.2.13/tokio/task/index.html)で書き込みを行い、他のタスクから進捗を見る感じです。
+ある[タスク](https://docs.rs/tokio/0.2.13/tokio/task/index.html#what-are-tasks)で書き込みを行い、他のタスクから進捗を見る感じです。