From d7dee1d0016b02b10db48e4c091637dcfbc4cee4 Mon Sep 17 00:00:00 2001
From: Gokhan Gulbiz <ggulbiz@gmail.com>
Date: Thu, 9 Mar 2023 11:20:32 +0300
Subject: [PATCH] Add ExtractFieldInt32(..) to jsonbutils

---
 src/backend/distributed/utils/jsonbutils.c | 19 +++++++++++++++++++
 src/include/distributed/jsonbutils.h       |  1 +
 2 files changed, 20 insertions(+)

diff --git a/src/backend/distributed/utils/jsonbutils.c b/src/backend/distributed/utils/jsonbutils.c
index 22fa4f568..4855ee004 100644
--- a/src/backend/distributed/utils/jsonbutils.c
+++ b/src/backend/distributed/utils/jsonbutils.c
@@ -83,6 +83,25 @@ ExtractFieldBoolean(Datum jsonbDoc, const char *fieldName, bool defaultValue)
 }
 
 
+/*
+ * ExtractFieldInt32 gets value of fieldName from jsonbDoc, or returns
+ * defaultValue if it doesn't exist.
+ */
+int32
+ExtractFieldInt32(Datum jsonbDoc, const char *fieldName, int32 defaultValue)
+{
+	Datum jsonbDatum = 0;
+	bool found = ExtractFieldJsonb(jsonbDoc, fieldName, &jsonbDatum, false);
+	if (!found)
+	{
+		return defaultValue;
+	}
+
+	Datum int32Datum = DirectFunctionCall1(jsonb_int4, jsonbDatum);
+	return DatumGetInt32(int32Datum);
+}
+
+
 /*
  * ExtractFieldTextP gets value of fieldName as text* from jsonbDoc, or
  * returns NULL if it doesn't exist.
diff --git a/src/include/distributed/jsonbutils.h b/src/include/distributed/jsonbutils.h
index 3e37fa38e..d44044fcb 100644
--- a/src/include/distributed/jsonbutils.h
+++ b/src/include/distributed/jsonbutils.h
@@ -16,5 +16,6 @@
 bool ExtractFieldJsonbDatum(Datum jsonbDoc, const char *fieldName, Datum *result);
 text * ExtractFieldTextP(Datum jsonbDoc, const char *fieldName);
 bool ExtractFieldBoolean(Datum jsonbDoc, const char *fieldName, bool defaultValue);
+int32 ExtractFieldInt32(Datum jsonbDoc, const char *fieldName, int32 defaultValue);
 
 #endif /* CITUS_JSONBUTILS_H */