如何利用 PostgreSQL 的 JSONB API 作为扩展的轻量级 JSON 剖析器 ...

前言

在基于 C 语言的 PostgreSQL 扩睁开发中，您可能会遇到必要处置处罚 JSON 等结构化数据的情况。通常，您可能会在扩展中引入第三方 JSON 剖析库，例如 cJSON 或 libjansson。这些库功能强盛、易于使用且提供了丰富的特性，但如果我们并未充分利用这些库的高级功能，引入它们则会显得多余。很多时间，我们只是希望从 JSON 中读取某个特定值或简单地遍历它。PostgreSQL 本身已经具备了处置处罚 JSON 数据的足够能力，尽管这些功能可能不如第三方库那样直观。如果您可以或许充分把握 PostgreSQL 已有的功能，或允许以避免引入第三方 JSON 库的成本。
在本文中，将向您展示如何使用 PostgreSQL 的 JSONB API 来剖析、提取和遍历 JSON 结构。
剖析和获取

要使用 PostgreSQL 的 JSONB API，您必要在 C 扩展中包含其头文件：

1. #include "utils/jsonb.h"

复制代码

现在，我们可以开始使用 JSONB 了。假设我们有一个 char * 指针，指向一个完整的 JSON 结构内容。我们必要将其转换为 Jsonb *，然后才能对其进行操作。

1. /* myjson points to a complete JSON content */
2. void jsonb_example(const char * myjson)
3. {
4.     Datum jsonb_datum;
5.     Jsonb * jb;
7.     /* we first convert char * to datum representation */
8.     jsonb_datum = DirectFunctionCall1(jsonb_in, CStringGetDatum(myjson));
10.     /* then, we convert it to Jsonb * */
11.     jb = DatumGetJsonbP(jsonb_datum);
12. }

复制代码

假设我们的 JSON 如下所示：

1. {
2.   "version": "1.0",
3.   "payload": {
4.     "name": "exampleapp",
5.     "ts_ms": 1720811216000,
6.     "db": "postgresql",
7.     "table": "mytable",
8.     "schema": "myschema"
9.   },
10.   "queries": [
11.     {
12.       "query": "select * from mytable"
13.     },
14.     {
15.       "query": "update mytable set a = 1"
16.     }
17.   ]
18. }

复制代码

为了获取 payload 组下 db 的值，我们可以使用 JSONB 的 jsonb_get_element() 函数，函数原型如下：

1. Datum jsonb_get_element(Jsonb *jb, Datum *path, int npath,
2. bool *isnull, bool as_text);

复制代码

该函数担当一个 JSONB 指针（即我们之前创建的表示整个 JSON 消息的指针），以及一个Datum 数组和 npath，用于表示 JSON 元素的路径。请注意，此路径不必一直指向标量值，它可以停在另一个内部组或数组，具体取决于您的用例。它还担当一个 isnull 布尔指针，如果找不到元素，函数会将其设置为 false。最后，as_text 布尔值指示函数是否将结果作为 Text Datum 或 Jsonb Datum 返回。我倾向于将其设置为 false，以便返回 JSONB Datum，从而可以进一步操作。将其转换为字符串表示也很简单（通过 stringinfo 结构）。请参见以下示例。

1. /* myjson points to a complete JSON content */
2. void jsonb_example(const char * myjson)
3. {
4.     Datum jsonb_datum;
5.     Jsonb * jb;
7.     /* variables needed for fetching element */
8.     Datum datum_elems[2];
9.     Datum res;
10.     int numpath = 2;
11.     bool isnull;
12.     StringInfoData strinfo;
14.     /* we first convert char * to datum representation */
15.     jsonb_datum = DirectFunctionCall1(jsonb_in, CStringGetDatum(myjson));
17.     /* then, we convert it to Jsonb * */
18.     jb = DatumGetJsonbP(jsonb_datum);
20.     /* prepare element paths to fetch, from outer to inner */
21.     initStringInfo(&strinfo);
22.     datum_elems[0] = CStringGetTextDatum("payload");
23.     datum_elems[1] = CStringGetTextDatum("db");
25.     /* fetch it */
26.     res = jsonb_get_element(jb, datum_elems, numPaths, &isnull, false);
27.     if (isnull)
28.     {
29.         /* write NULL if element does not exist */
30.         resetStringInfo(&strinfoo);
31.         appendStringInfoString(&strinfoo, "NULL");
32.     }
33.     else
34.     {
35.         Jsonb *resjb = DatumGetJsonbP(res);
36.         resetStringInfo(strinfoout);
37.         JsonbToCString(&strinfo, &resjb->root, VARSIZE(resjb));
38.     }
40.     /* strinfo contains the value of the element at this point. Print it */
41.     elog(WARNING, "data = %s", strinfo.data);
42. }

复制代码

现在，如果我们想从数组中的特定索引处获取特定值。例如，queries 数组下索引为 1 的 query 值（update mytable set a = 1）。我们只必要修改描述此路径的 datum_elems。我们可以在数组元素后的 datum_elems 中直接放入一个数字（作为字符串），以告诉函数我们要获取特定索引。请参见以下示例：

1. /* myjson points to a complete JSON content */
2. void jsonb_example(const char * myjson)
3. {
4.     Datum jsonb_datum;
5.     Jsonb * jb;
7.     /* variables needed for fetching element */
8.     Datum datum_elems[3];
9.     Datum res;
10.     int numpath = 3;
11.     bool isnull;
12.     StringInfoData strinfo;
14.     /* we first convert char * to datum representation */
15.     jsonb_datum = DirectFunctionCall1(jsonb_in, CStringGetDatum(myjson));
17.     /* then, we convert it to Jsonb * */
18.     jb = DatumGetJsonbP(jsonb_datum);
20.     /* prepare element paths to fetch, from outer to inner */
21.     initStringInfo(&strinfo);
22.     datum_elems[0] = CStringGetTextDatum("queries");
23.     datum_elems[1] = CStringGetTextDatum("1");
24.     datum_elems[2] = CStringGetTextDatum("query");
26.     /* fetch it */
27.     res = jsonb_get_element(jb, datum_elems, numPaths, &isnull, false);
28.     if (isnull)
29.     {
30.         /* write NULL if element does not exist */
31.         resetStringInfo(&strinfoo);
32.         appendStringInfoString(&strinfoo, "NULL");
33.     }
34.     else
35.     {
36.         Jsonb *resjb = DatumGetJsonbP(res);
37.         resetStringInfo(strinfoout);
38.         JsonbToCString(&strinfo, &resjb->root, VARSIZE(resjb));
39.     }
41.     /* strinfo contains the value of the element at this point. Print it */
42.     elog(WARNING, "data = %s", strinfo.data);
43. }

复制代码

如你所见，获取特定元素非常简单。我们只必要准备正确的 datum_elems 数组来描述通向某个值的路径，其他部分保持稳固。我们可以编写一个辅助函数，通过从单个字符串自动创建 datum_elems 来简化此过程，该字符串用点分隔每个层次结构（例如：“payload.name”，“queries.0.query” 等）。

1. void getPathElementString(Jsonb * jb, char * path, StringInfoData strinfoout)
2. {
3.     Datum * datum_elems = NULL;
4.     char * str_elems = NULL, * p = path;
5.     int numPaths = 0, curr = 0;
6.     char * pathcopy = pstrdup(path);
7.     Datum res;
8.     bool isnull;
10.     if (!strinfoout)
11.     {
12.         elog(WARNING, "strinfo is null");
13.         return -1;
14.     }
16.     /* Count the number of elements in the path */
17.     if (strstr(pathcopy, "."))
18.     {
19.         while (*p != '\0')
20.         {
21.             if (*p == '.')
22.             {
23.                  numPaths++;
24.             }
25.             p++;
26.         }
27.         numPaths++; /* Add the last one */
28.     }
29.     else
30.     {
31.         numPaths = 1;
32.     }
34.     datum_elems = palloc0(sizeof(Datum) * numPaths);
35.     /* Parse the path into elements */
36.     if (strstr(pathcopy, "."))
37.     {
39.         str_elems= strtok(pathcopy, ".");
40.         if (str_elems)
41.         {
42.             datum_elems[curr] = CStringGetTextDatum(str_elems);
43.             curr++;
44.             while ((str_elems = strtok(NULL, ".")))
45.             {
46.                 datum_elems[curr] = CStringGetTextDatum(str_elems);
47.                 curr++;
48.             }
49.         }
50.     }
51.     else
52.     {
53.         /* only one level, just use pathcopy*/
54.         datum_elems[curr] = CStringGetTextDatum(pathcopy);
55.     }
57.     /* Get the element from JSONB */
58.     res = jsonb_get_element(jb, datum_elems, numPaths, &isnull, false);
59.     if (isnull)
60.     {
61.         resetStringInfo(strinfoout);
62.         appendStringInfoString(strinfoout, "NULL");
63.     }
64.     else
65.     {
66.         Jsonb *resjb = DatumGetJsonbP(res);
67.         resetStringInfo(strinfoout);
68.         JsonbToCString(strinfoout, &resjb->root, VARSIZE(resjb));
69.     }
71.     pfree(datum_elems);
72.     pfree(pathcopy);
73. }
75. /* myjson points to a complete JSON content */
76. void jsonb_example(const char * myjson)
77. {
78.     Datum jsonb_datum;
79.     Jsonb * jb;
80.     StringInfoData strinfo;
82.     /* we first convert char * to datum representation */
83.     jsonb_datum = DirectFunctionCall1(jsonb_in, CStringGetDatum(myjson));
85.     /* then, we convert it to Jsonb * */
86.     jb = DatumGetJsonbP(jsonb_datum);
88.     initStringInfo(&strinfo);
90.     getPathElementString(jb, "payload.db", &strinfo);
91.     elog(WARNING, "payload.db= %s", strinfo.data);
93.     getPathElementString(jb, "queries.0.query", &strinfo);
94.     elog(WARNING, "queries.0.query= %s", strinfo.data);
95. }

复制代码

遍历整个 JSON 结构

现在，我们已经知道如何在知道目标的情况下从 JSON 中获取特定值。有时，我们必要遍历整个 JSON 以构建内部数据结构以满足某些需求。在这种情况下，我们可以利用 JSONB 的迭代函数。以下示例代码将创建一个 JSONB 迭代器，然后实验遍历其中的每个元素。它会在迭代器即将进入组或数组时以及即将退出组或数组时进行指示。您可以根据必要生存键和值。JSONB 读取的值可以表示为差别的数据范例，例如字符串、二进制、数字等。以下示例实验将它们转换为字符串（二进制除外）以进行输出。

1. /* myjson points to a complete JSON content */
2. void jsonb_iterate_example(const char * myjson)
3. {
4.     Datum jsonb_datum;
5.     Jsonb * jb;
7.     /* iterator related */
8.     JsonbIterator *it;
9.     JsonbValue v;
10.     JsonbIteratorToken r;
11.     char * key = NULL;
12.     char * value = NULL;
14.     /* we first convert char * to datum representation */
15.     jsonb_datum = DirectFunctionCall1(jsonb_in, CStringGetDatum(myjson));
17.     /* then, we convert it to Jsonb * */
18.     jb = DatumGetJsonbP(jsonb_datum);
20.     it = JsonbIteratorInit(&jb->root);
21.     while ((r = JsonbIteratorNext(&it, &v, false)) != WJB_DONE)
22.     {
23.         switch (r)
24.         {
25.             case WJB_BEGIN_OBJECT:
26.                 elog(WARNING, "begin group --------------------");
27.                 break;
28.             case WJB_END_OBJECT:
29.                 elog(WARNING, "end group --------------------");
30.                 break;
31.             case WJB_BEGIN_ARRAY:
32.                 elog(WARNING, "begin array --------------------");
33.                 break;
34.             case WJB_END_ARRAY:
35.                 elog(WARNING, "end array --------------------");
36.                 break;
37.             case WJB_KEY:
38.                 key = pnstrdup(v.val.string.val, v.val.string.len);
39.                 elog(WARNING, "key: %s", key);
40.                 break;
41.             case WJB_VALUE:
42.             case WJB_ELEM:
43.                 switch (v.type)
44.                 {
45.                     case jbvNull:
46.                         elog(WARNING, "value: NULL");
47.                         break;
48.                     case jbvString:
49.                         value = pnstrdup(v.val.string.val, v.val.string.len);
50.                         elog(WARNING, "string value: %s", value);
51.                         break;
52.                     case jbvNumeric:
53.                     {
54.                         value = DatumGetCString(DirectFunctionCall1(numeric_out, PointerGetDatum(v.val.numeric)));
55.                         elog(WARNING, "numeric value: %s", value);
56.                         break;
57.                     }
58.                     case jbvBool:
59.                         elog(WARNING, "boolean value: %s", v.val.boolean ? "true" : "false");
60.                         if (v.val.boolean)
61.                             value = pnstrdup("true", strlen("true"));
62.                         else
63.                             value = pnstrdup("false", strlen("false"));
64.                         break;
65.                     case jbvBinary:
66.                         elog(WARNING, "binary value");
67.                         break;
68.                     default:
69.                         elog(WARNING, "unknown value type: %d", v.type);
70.                         break;
71.                 }
72.                 break;
73.             default:
74.                 elog(WARNING, "Unknown token: %d", r);
75.                 break;
76.         }
78.         if (key != NULL && value != NULL)
79.         {
80.             pfree(key);
81.             pfree(value);
82.             key = NULL;
83.             value = NULL;
84.         }
85. }

复制代码

总结

PostgreSQL 中的 JSONB API 能做的不但仅是简单的获取和遍历。例如，它还可以将额外的值推送到现有的 JSONB 结构中。今天我们主要关注获取和遍历，在我看来，这是处置处罚 JSON 时最常见的用例。我希望这里分享的代码示例能对你有所帮助，并防止你在扩睁开发中引入第三方 JSON 剖析器，由于那可能是多余的。
关于 IvorySQL

lvorySQL 是由瀚高股份主导研发的一款开源的兼容 Oracle 的 PostgreSQL。IvorySQL 与 PostgreSQL 国际社区紧密合作，保持与最新 PG 版本内核同步，为用户提供便捷的升级体验。基于双 Parser 架构设计，100% 与原生 PostgreSQL 兼容，支持丰富的 PostgreSQL 周边工具和扩展，并根据用户需求提供定制化工具。同时，IvorySQL 4.0 提供更全面机动的 Oracle 兼容功能，具备高度的 SQL 和 PL/SQL 兼容性可以或许为企业构建更加高效、稳固和机动的数据库办理方案。

• 官网：https://www.ivorysql.org
  
• GitHub（欢迎点击 star 收藏哦）：https://github.com/IvorySQL/IvorySQL
  

本文由博客一文多发平台 OpenWrite 发布！

免责声明：如果侵犯了您的权益，请联系站长，我们会及时删除侵权内容，谢谢合作！更多信息从访问主页：qidao123.com:ToB企服之家，中国第一个企服评测及商务社交产业平台。