[Rust] fix generated comparison when value is NAN (#895) (#896)

mward · Michael Ward · web-flow · commit d9600d4656a2 · 2022-05-24T15:25:16.000+01:00
Co-authored-by: Michael Ward &lt;mward@drw.com&gt;
diff --git a/sbe-tool/src/main/java/uk/co/real_logic/sbe/generation/rust/RustGenerator.java b/sbe-tool/src/main/java/uk/co/real_logic/sbe/generation/rust/RustGenerator.java
@@ -746,8 +746,18 @@ private static void generatePrimitiveOptionalDecoder(
         indent(sb, level + 1, "let value = self.get_buf().get_%s_at(self.%s);\n",
             rustPrimitiveType,
             getBufOffset(fieldToken));
-        indent(sb, level + 1, "if value == %s {\n",
-            generateRustLiteral(primitiveType, encoding.applicableNullValue().toString()));
+
+
+        final String literal = generateRustLiteral(primitiveType, encoding.applicableNullValue().toString());
+        if (literal.endsWith("::NAN"))
+        {
+            indent(sb, level + 1, "if value.is_nan() {\n");
+        }
+        else
+        {
+            indent(sb, level + 1, "if value == %s {\n", literal);
+        }
+
         indent(sb, level + 2, "None\n");
         indent(sb, level + 1, "} else {\n");
         indent(sb, level + 2, "Some(value)\n");
