From 5a631be1ca8d699acd699f36b8dfd17b6f915a3f Mon Sep 17 00:00:00 2001
From: Vitaliy Kirsanov <krokoziabla@yandex-team.ru>
Date: Wed, 5 Apr 2023 15:33:56 +0300
Subject: [PATCH] Issue #1020 Infinity numeric support

---
 codecs/numeric.pyx | 19 +++++++++++++++++--
 1 file changed, 17 insertions(+), 2 deletions(-)

diff --git a/codecs/numeric.pyx b/codecs/numeric.pyx
index 16b0083..b75d096 100644
--- a/codecs/numeric.pyx
+++ b/codecs/numeric.pyx
@@ -16,6 +16,8 @@ DEF MAX_DSCALE = 0x3FFF
 DEF NUMERIC_POS = 0x0000
 DEF NUMERIC_NEG = 0x4000
 DEF NUMERIC_NAN = 0xC000
+DEF NUMERIC_PINF = 0xD000
+DEF NUMERIC_NINF = 0xF000
 
 _Dec = decimal.Decimal
 
@@ -51,8 +53,6 @@ cdef numeric_encode_binary(CodecContext settings, WriteBuffer buf, obj):
         dec = _Dec(obj)
 
     dt = dec.as_tuple()
-    if dt.exponent == 'F':
-        raise ValueError('numeric type does not support infinite values')
 
     if dt.exponent == 'n' or dt.exponent == 'N':
         # NaN
@@ -60,6 +60,15 @@ cdef numeric_encode_binary(CodecContext settings, WriteBuffer buf, obj):
         num_pgdigits = 0
         weight = 0
         dscale = 0
+    elif dt.exponent == 'F':
+        # Infinity
+        if dt.sign:
+            sign = NUMERIC_NINF
+        else:
+            sign = NUMERIC_PINF
+        num_pgdigits = 0
+        weight = 0
+        dscale = 0
     else:
         exponent = dt.exponent
         if exponent < 0 and -exponent > MAX_DSCALE:
@@ -160,6 +169,12 @@ cdef numeric_decode_binary_ex(
     if sign == NUMERIC_NAN:
         # Not-a-number
         return _Dec('NaN')
+    elif sign == NUMERIC_PINF:
+        # +Infinity
+        return _Dec('Infinity')
+    elif sign == NUMERIC_NINF:
+        # -Infinity
+        return _Dec('-Infinity')
 
     if num_pgdigits == 0:
         # Zero