From aca0d6cf185d193359590f37e62df927242af57e Mon Sep 17 00:00:00 2001
From: fengzero <fengzero@fengzero.com>
Date: Tue, 11 Aug 2026 06:46:06 +0000
Subject: [PATCH] schema support int/bool/float

---
 plugins/mqtt/schema.c        | 31 +++++++++++++++++++++++---
 plugins/mqtt/schema.h        |  3 +++
 tests/ut/mqtt_schema_test.cc | 42 ++++++++++++++++++++++++++++++++++++
 3 files changed, 73 insertions(+), 3 deletions(-)

diff --git a/plugins/mqtt/schema.c b/plugins/mqtt/schema.c
index 69458560..997293db 100644
--- a/plugins/mqtt/schema.c
+++ b/plugins/mqtt/schema.c
@@ -49,10 +49,31 @@ static int mqtt_schema_parse(json_t *root, mqtt_schema_vt_t **vts,
                     str_val[strlen(str_val) - 1] == '}') {
                     return -1;
                 } else {
-                    vt->vt = MQTT_SCHEMA_UD;
+                    vt->vt    = MQTT_SCHEMA_UD;
+                    vt->jtype = NEU_JSON_STR;
                     strncpy(vt->ud, str_val, sizeof(vt->ud) - 1);
                 }
             }
+        } else if (json_is_integer(value) || json_is_real(value) ||
+                   json_is_boolean(value)) {
+            *vts_len += 1;
+            *vts = realloc(*vts, *vts_len * sizeof(mqtt_schema_vt_t));
+            mqtt_schema_vt_t *vt = &(*vts)[*vts_len - 1];
+
+            memset(vt, 0, sizeof(mqtt_schema_vt_t));
+            strncpy(vt->name, key, sizeof(vt->name) - 1);
+            vt->vt = MQTT_SCHEMA_UD;
+
+            if (json_is_integer(value)) {
+                vt->jtype          = NEU_JSON_INT;
+                vt->jvalue.val_int = json_integer_value(value);
+            } else if (json_is_real(value)) {
+                vt->jtype             = NEU_JSON_DOUBLE;
+                vt->jvalue.val_double = json_real_value(value);
+            } else {
+                vt->jtype           = NEU_JSON_BOOL;
+                vt->jvalue.val_bool = json_boolean_value(value);
+            }
         } else if (json_is_object(value)) {
             if (deep >= 3) {
                 return -1;
@@ -280,8 +301,12 @@ static void *schema_encode(char *driver, char *group,
             break;
         }
         case MQTT_SCHEMA_UD:
-            elem.t         = NEU_JSON_STR;
-            elem.v.val_str = vts[i].ud;
+            elem.t = vts[i].jtype;
+            if (vts[i].jtype == NEU_JSON_STR) {
+                elem.v.val_str = vts[i].ud;
+            } else {
+                elem.v = vts[i].jvalue;
+            }
             break;
         case MQTT_SCHEMA_OBJECT: {
             void *sub_root = schema_encode(driver, group, tags, vts[i].sub_vts,
diff --git a/plugins/mqtt/schema.h b/plugins/mqtt/schema.h
index ebeece9a..b51a5082 100644
--- a/plugins/mqtt/schema.h
+++ b/plugins/mqtt/schema.h
@@ -53,6 +53,9 @@ typedef struct mqtt_schema_vt {
 
     char ud[128];
 
+    neu_json_type_e  jtype;
+    neu_json_value_u jvalue;
+
     mqtt_schema_vt_t *sub_vts;
     size_t            n_sub_vts;
 } mqtt_schema_vt_t;
diff --git a/tests/ut/mqtt_schema_test.cc b/tests/ut/mqtt_schema_test.cc
index 5d94dda0..754a84cd 100644
--- a/tests/ut/mqtt_schema_test.cc
+++ b/tests/ut/mqtt_schema_test.cc
@@ -89,6 +89,48 @@ TEST(validate_deep_success, schema)
     free(vts);
 }
 
+TEST(validate_constants, schema)
+{
+    const char *schema =
+        "{\"string\":\"value\",\"integer\":1234,\"float\":11.44,"
+        "\"boolean\":true,\"object\":{\"integer\":-42,\"float\":0.5,"
+        "\"boolean\":false}}";
+    mqtt_schema_vt_t *vts = NULL;
+    size_t            n_vts;
+    int               ret = mqtt_schema_validate(schema, &vts, &n_vts);
+
+    ASSERT_EQ(ret, 0);
+    ASSERT_EQ(n_vts, 5);
+    EXPECT_EQ(vts[0].jtype, NEU_JSON_STR);
+    EXPECT_STREQ(vts[0].ud, "value");
+    EXPECT_EQ(vts[1].jtype, NEU_JSON_INT);
+    EXPECT_EQ(vts[1].jvalue.val_int, 1234);
+    EXPECT_EQ(vts[2].jtype, NEU_JSON_DOUBLE);
+    EXPECT_EQ(vts[2].jvalue.val_double, 11.44);
+    EXPECT_EQ(vts[3].jtype, NEU_JSON_BOOL);
+    EXPECT_TRUE(vts[3].jvalue.val_bool);
+
+    ASSERT_EQ(vts[4].vt, MQTT_SCHEMA_OBJECT);
+    ASSERT_EQ(vts[4].n_sub_vts, 3);
+    EXPECT_EQ(vts[4].sub_vts[0].jvalue.val_int, -42);
+    EXPECT_EQ(vts[4].sub_vts[1].jvalue.val_double, 0.5);
+    EXPECT_FALSE(vts[4].sub_vts[2].jvalue.val_bool);
+
+    char *               result = NULL;
+    neu_json_read_resp_t tags   = { 0 };
+    ret = mqtt_schema_encode((char *) "driver", (char *) "group", &tags, vts,
+                             n_vts, NULL, 0, &result);
+    ASSERT_EQ(ret, 0);
+    EXPECT_STREQ(result,
+                 "{\"string\": \"value\", \"integer\": 1234, \"float\": "
+                 "11.44, \"boolean\": true, \"object\": {\"integer\": -42, "
+                 "\"float\": 0.5, \"boolean\": false}}");
+
+    free(result);
+    free(vts[4].sub_vts);
+    free(vts);
+}
+
 TEST(validate_failed, schema)
 {
     const char *schema =