Add settings for converting float to string in TProto2JsonConfig

init

1 files changed, 22 insertions, 0 deletions

diff --git a/library/cpp/protobuf/json/config.h b/library/cpp/protobuf/json/config.h
index 8b9d60ab4f..2ffda4a065 100644
--- a/library/cpp/protobuf/json/config.h
+++ b/library/cpp/protobuf/json/config.h
@@ -5,6 +5,9 @@
 
 #include <util/generic/vector.h>
 #include <util/generic/yexception.h>
+#include <util/string/cast.h>
+
+#include <library/cpp/json/json_writer.h>
 
 #include <functional>
 
@@ -12,6 +15,10 @@ namespace NProtobufJson {
     struct TProto2JsonConfig {
         using TSelf = TProto2JsonConfig;
 
+        ui32 DoubleNDigits = NJson::TJsonWriterConfig::DefaultDoubleNDigits;
+        ui32 FloatNDigits = NJson::TJsonWriterConfig::DefaultFloatNDigits;
+        EFloatToStringMode FloatToStringMode = NJson::TJsonWriterConfig::DefaultFloatToStringMode;
+
         bool FormatOutput = false;
 
         enum MissingKeyMode {
@@ -91,6 +98,21 @@ namespace NProtobufJson {
 
         bool WriteNanAsString = false;
 
+        TSelf& SetDoubleNDigits(ui32 ndigits) {
+            DoubleNDigits = ndigits;
+            return *this;
+        }
+
+        TSelf& SetFloatNDigits(ui32 ndigits) {
+            FloatNDigits = ndigits;
+            return *this;
+        }
+
+        TSelf& SetFloatToStringMode(EFloatToStringMode mode) {
+            FloatToStringMode = mode;
+            return *this;
+        }
+
         TSelf& SetFormatOutput(bool format) {
             FormatOutput = format;
             return *this;