<!DOCTYPE html><html lang="en"><head><meta charset="utf-8"><meta name="viewport" content="width=device-width, initial-scale=1.0"><meta name="generator" content="rustdoc"><meta name="description" content="Source of the Rust file `src\channels\queue.rs`."><title>queue.rs - source</title><script>if(window.location.protocol!=="file:")document.head.insertAdjacentHTML("beforeend","SourceSerif4-Regular-6b053e98.ttf.woff2,FiraSans-Italic-81dc35de.woff2,FiraSans-Regular-0fe48ade.woff2,FiraSans-MediumItalic-ccf7e434.woff2,FiraSans-Medium-e1aa3f0a.woff2,SourceCodePro-Regular-8badfe75.ttf.woff2,SourceCodePro-Semibold-aa29a496.ttf.woff2".split(",").map(f=>`<link rel="preload" as="font" type="font/woff2"href="../../../static.files/${f}">`).join(""))</script><link rel="stylesheet" href="../../../static.files/normalize-9960930a.css"><link rel="stylesheet" href="../../../static.files/rustdoc-ca0dd0c4.css"><script id="default-settings"
data-use_system_theme="false"
data-theme="trash"></script><meta name="rustdoc-vars" data-root-path="../../../" data-static-root-path="../../../static.files/" data-current-crate="trash_utilities" data-themes="trash" data-resource-suffix="" data-rustdoc-version="1.92.0-nightly (b925a865e 2025-10-09)" data-channel="nightly" data-search-js="search-8d3311b9.js" data-stringdex-js="stringdex-828709d0.js" data-settings-js="settings-c38705f0.js" ><script src="../../../static.files/storage-e2aeef58.js"></script><script defer src="../../../static.files/src-script-813739b1.js"></script><script defer src="../../../src-files.js"></script><script defer src="../../../static.files/main-ce535bd0.js"></script><noscript><link rel="stylesheet" href="../../../static.files/noscript-263c88ec.css"></noscript><link rel="alternate icon" type="image/png" href="../../../static.files/favicon-32x32-eab170b8.png"><link rel="icon" type="image/svg+xml" href="../../../static.files/favicon-044be391.svg"></head><body class="rustdoc src"><!--[if lte IE 11]><div class="warning">This old browser is unsupported and will most likely display funky things.</div><![endif]--><nav class="sidebar"><div class="src-sidebar-title"><h2>Files</h2></div></nav><div class="sidebar-resizer" title="Drag to resize sidebar"></div><main><section id="main-content" class="content"><div class="main-heading"><h1><div class="sub-heading">trash_utilities\channels/</div>queue.rs</h1><rustdoc-toolbar></rustdoc-toolbar></div><div class="example-wrap digits-2"><pre class="rust"><code><a href=#1 id=1 data-nosnippet>1</a><span class="comment">// Standard library imports
<a href=#2 id=2 data-nosnippet>2</a>
<a href=#3 id=3 data-nosnippet>3</a>// External crate imports
<a href=#4 id=4 data-nosnippet>4</a></span><span class="kw">use </span>parking_lot::Mutex;
<a href=#5 id=5 data-nosnippet>5</a>
<a href=#6 id=6 data-nosnippet>6</a><span class="doccomment">/// High-throughput work queue with load balancing
<a href=#7 id=7 data-nosnippet>7</a></span><span class="kw">pub struct </span>WorkQueue<T, R> {
<a href=#8 id=8 data-nosnippet>8</a> workers: Vec<<span class="kw">crate</span>::channels::core::TxFuture<T>>,
<a href=#9 id=9 data-nosnippet>9</a> result_rx: <span class="kw">crate</span>::channels::core::RxFuture<R>,
<a href=#10 id=10 data-nosnippet>10</a> next_worker: Mutex<usize>,
<a href=#11 id=11 data-nosnippet>11</a>}
<a href=#12 id=12 data-nosnippet>12</a>
<a href=#13 id=13 data-nosnippet>13</a><span class="kw">impl</span><T: Send + <span class="lifetime">'static</span>, R: Send + <span class="lifetime">'static</span>> WorkQueue<T, R> {
<a href=#14 id=14 data-nosnippet>14</a> <span class="doccomment">/// Create a new work queue with N workers
<a href=#15 id=15 data-nosnippet>15</a> </span><span class="attr">#[must_use]
<a href=#16 id=16 data-nosnippet>16</a> </span><span class="kw">pub fn </span>new(num_workers: usize) -> <span class="self">Self </span>{
<a href=#17 id=17 data-nosnippet>17</a> <span class="kw">let </span><span class="kw-2">mut </span>workers = Vec::new();
<a href=#18 id=18 data-nosnippet>18</a> <span class="kw">let </span>_result_txs: Vec<<span class="kw">crate</span>::channels::core::TxFuture<R>> = Vec::new();
<a href=#19 id=19 data-nosnippet>19</a> <span class="kw">let </span>(result_tx, result_rx) = <span class="kw">crate</span>::channels::core::bounded_queue_3(num_workers * <span class="number">10</span>);
<a href=#20 id=20 data-nosnippet>20</a>
<a href=#21 id=21 data-nosnippet>21</a> <span class="kw">for _ in </span><span class="number">0</span>..num_workers {
<a href=#22 id=22 data-nosnippet>22</a> <span class="kw">let </span>(task_tx, task_rx) = <span class="kw">crate</span>::channels::core::bounded_queue_3(<span class="number">100</span>);
<a href=#23 id=23 data-nosnippet>23</a> workers.push(task_tx);
<a href=#24 id=24 data-nosnippet>24</a>
<a href=#25 id=25 data-nosnippet>25</a> <span class="kw">let </span>_result_tx = result_tx.clone();
<a href=#26 id=26 data-nosnippet>26</a> smol::spawn(<span class="kw">async move </span>{
<a href=#27 id=27 data-nosnippet>27</a> <span class="kw">let </span>rx = task_rx;
<a href=#28 id=28 data-nosnippet>28</a> <span class="kw">while let </span><span class="prelude-val">Ok</span>(_task) = rx.recv().<span class="kw">await </span>{
<a href=#29 id=29 data-nosnippet>29</a> <span class="comment">// Task processing handled externally via submit_with_processor
<a href=#30 id=30 data-nosnippet>30</a> </span>}
<a href=#31 id=31 data-nosnippet>31</a> })
<a href=#32 id=32 data-nosnippet>32</a> .detach();
<a href=#33 id=33 data-nosnippet>33</a> }
<a href=#34 id=34 data-nosnippet>34</a>
<a href=#35 id=35 data-nosnippet>35</a> <span class="self">Self </span>{
<a href=#36 id=36 data-nosnippet>36</a> workers,
<a href=#37 id=37 data-nosnippet>37</a> result_rx,
<a href=#38 id=38 data-nosnippet>38</a> next_worker: Mutex::new(<span class="number">0</span>),
<a href=#39 id=39 data-nosnippet>39</a> }
<a href=#40 id=40 data-nosnippet>40</a> }
<a href=#41 id=41 data-nosnippet>41</a>
<a href=#42 id=42 data-nosnippet>42</a> <span class="doccomment">/// Submit a task to the queue (non-blocking)
<a href=#43 id=43 data-nosnippet>43</a> ///
<a href=#44 id=44 data-nosnippet>44</a> /// # Errors
<a href=#45 id=45 data-nosnippet>45</a> ///
<a href=#46 id=46 data-nosnippet>46</a> /// Returns an error if the channel is closed or full.
<a href=#47 id=47 data-nosnippet>47</a> </span><span class="kw">pub async fn </span>submit(<span class="kw-2">&</span><span class="self">self</span>, task: T) -> <span class="prelude-ty">Result</span><(), smol::channel::SendError<T>> {
<a href=#48 id=48 data-nosnippet>48</a> <span class="kw">let </span>worker_index = {
<a href=#49 id=49 data-nosnippet>49</a> <span class="kw">let </span><span class="kw-2">mut </span>next = <span class="self">self</span>.next_worker.lock();
<a href=#50 id=50 data-nosnippet>50</a> <span class="kw">let </span>index = <span class="kw-2">*</span>next % <span class="self">self</span>.workers.len();
<a href=#51 id=51 data-nosnippet>51</a> <span class="kw-2">*</span>next += <span class="number">1</span>;
<a href=#52 id=52 data-nosnippet>52</a> index
<a href=#53 id=53 data-nosnippet>53</a> };
<a href=#54 id=54 data-nosnippet>54</a> <span class="kw">let </span>worker = <span class="kw-2">&</span><span class="self">self</span>.workers[worker_index];
<a href=#55 id=55 data-nosnippet>55</a> worker.send(task).<span class="kw">await
<a href=#56 id=56 data-nosnippet>56</a> </span>}
<a href=#57 id=57 data-nosnippet>57</a>
<a href=#58 id=58 data-nosnippet>58</a> <span class="doccomment">/// Collect a result (non-blocking)
<a href=#59 id=59 data-nosnippet>59</a> ///
<a href=#60 id=60 data-nosnippet>60</a> /// # Errors
<a href=#61 id=61 data-nosnippet>61</a> ///
<a href=#62 id=62 data-nosnippet>62</a> /// Returns an error if the channel is closed or empty.
<a href=#63 id=63 data-nosnippet>63</a> </span><span class="kw">pub async fn </span>collect(<span class="kw-2">&</span><span class="self">self</span>) -> <span class="prelude-ty">Result</span><R, smol::channel::RecvError> {
<a href=#64 id=64 data-nosnippet>64</a> <span class="self">self</span>.result_rx.recv().<span class="kw">await
<a href=#65 id=65 data-nosnippet>65</a> </span>}
<a href=#66 id=66 data-nosnippet>66</a>}</code></pre></div></section></main></body></html>