unroll producer result decode loop

jet · Jul 14, 2017 · 32d17fc · 32d17fc
1 parent 36ded55
commit 32d17fc
Showing 1 changed file with 6 additions and 7 deletions.
diff --git a/src/kafunk/Producer.fs b/src/kafunk/Producer.fs
@@ -367,15 +367,14 @@ module Producer =
           |> Option.iter (fun res -> IVar.tryPut res b.rep |> ignore)
 
       if oks.Length > 0 then
-        let oks = 
-          oks 
-          |> Seq.map (fun (p,o) -> 
-            let pmc = ps.[p].Count
-            p, Success (ProducerResult(p,o,pmc))) 
-          |> Dict.ofSeq
+        let oks' = Dict.empty
+        for i = 0 to oks.Length - 1 do
+          let p,o = oks.[i] in
+          let pmc = ps.[p].Count in
+          oks'.[p] <- Success (ProducerResult(p,o,pmc))
         for b in batch do
           let mutable res = Unchecked.defaultof<_>
-          if (oks.TryGetValue(b.partition, &res)) then
+          if (oks'.TryGetValue(b.partition, &res)) then
             IVar.tryPut res b.rep |> ignore
 
     | Failure errs ->