PostgreSQL 務實應用（四/5）JSON

JSON 可謂風靡互聯網，在數據交換使用上，其優勢特別明顯，其結構簡潔、可讀易讀、形式靈活。很多 API 介面的數據都採用 JSON 來表示。

PostgreSQL 對 JSON 提供了良好的支持。具體的相關函數可參考：JSON類型和函數

從使用的角度而言，個人覺得常見的應用場景為：

1. 讀取單個 JSON 的屬性值
2. 遍歷單個 JSON 的所有屬性
3. 遍歷一個 JSON 數組
4. 創建一個 JSON 作為返回值

之所以僅這些簡單的場景，原因在於，在應用中使用高級語言處理 JSON 與在資料庫中使用那些高級的 JSON 函數相比，從操作上和可讀性上均爽很多。在不支持 JSON 的資料庫中，我們也常使用單個文本欄位存儲 JSON 字元串，然後在應用中加以解析處理。

四個場景

我們以以下的 JSON 字元串作為輸入，來瞭解 PostgreSQL 在各場景中的應用實現。

{
    "label": {
        "names": ["Amy", "Kala", "Lily"]
    },
    "color": "red",
    "count": 3
}

// 寫成一行即是
{"label":{"names":["Amy","Kala","Lily"]},"color":"red","count":3}

1.讀取屬性

首先，通過下麵的表格，感受一下 JavaScript 與 PostgreSQL 中讀取 color 屬性與 label 屬性中 names 的第二個值的形式。

在 PostgreSQL 中我們可以使用以下語句逐層指定屬性路徑（屬性名稱需要使用字元串需單引號，數組索引使用數字）來獲取值：

-- 取得 color 屬性
SELECT '{"label":{"names":["Amy","Kala","Lily"]},"color":"red","count":3}'::json 
       -> 'color';

-- 取得 label 屬性下的 names 的第二個值
SELECT '{"label":{"names":["Amy","Kala","Lily"]},"color":"red","count":3}'::json 
       -> 'label' -> 'names' -> 1;

 此時取得的值仍然為 json 類型，如果需要取得值的文本形式，則把最後一個 "->" 變成 "->>" 即可。

當然，路徑的表示，也可以通過 #> '{label,names,1}' 的形式表示：

SELECT '{"label":{"names":["Amy","Kala","Lily"]},"color":"red","count":3}'::json 
       #> '{label,names,1}';

 2.遍歷屬性

使用 json_each 函數，即可返回屬性鍵值對的數據集，數據集包括兩列，key 表示屬性，value 表示屬性值。如下語句輸出所有結果：

DO $$
DECLARE 
    lv_row record; 
    jsonObj json := '{"label":{"names":["Amy","Kala","Lily"]},"color":"red","count":3}'::json;
BEGIN   
    FOR lv_row IN SELECT * FROM json_each(jsonObj) LOOP  
        raise notice 'key is %, value is %', lv_row.key, lv_row.value;
    END LOOP;  
END $$;

輸出

NOTICE: key is label, value is {"names":["Amy","Kala","Lily"]}

NOTICE: key is color, value is "red"

NOTICE: key is count, value is 3

3.遍曆數組

通過使用 json_array_length 函數獲取數組的長度，然後根據索引遍歷整個數組即可。

DO $$
DECLARE 
    lv_row record; 
    lv_size int;
    jsonObj json := '{"label":{"names":["Amy","Kala","Lily"]},"color":"red","count":3}'::json;
BEGIN
    -- 取得label 下names 這個json數組
    jsonObj := jsonObj #> '{label,names}';
    -- 取得數組的長度
    lv_size := json_array_length(jsonObj);
    -- 按索引遍歷整個數組
    FOR i IN 0..lv_size-1 LOOP
        raise notice '%', jsonObj -> i;
    END LOOP;
END $$;

輸出：

NOTICE: "Amy"

NOTICE: "Kala"

NOTICE: "Lily" 

4.創建一個 JSON

使用 json_build_object 函數，傳遞  key, value 成對的參數即可創建一個 json，如以下語句形成一個 api 常用的返回執行情況的 json。 

SELECT json_build_object('code', 200, 'err_msg', 'run success!');

應用示例

我們以填寫學生地址為例，傳遞給存儲過程的是一個 json 數組，每個數組中的 json 對象包括了學生標識與地址信息。

以下語句創建數據表

-- student_id 學生標識, address 地址
CREATE TABLE student_address (student_id varchar(10) PRIMARY KEY, address varchar(100));

以下為處理過程

CREATE OR REPLACE FUNCTION save_student_addresses_json( 
    v_array_json json)
    RETURNS json
    LANGUAGE 'plpgsql' 
AS $$
DECLARE 
    lv_row_json json; 
    lv_length int;
    lv_field_student_id varchar;
    lv_field_address    varchar;
BEGIN   
    -- 取得數組的長度
    lv_length := json_array_length(v_array_json);
    FOR i IN 0..lv_length-1 LOOP
        -- 取得第 i 行的 json 值
        lv_row_json := v_array_json -> i;
        lv_field_student_id := lv_row_json ->> 'student_id';
        lv_field_address := lv_row_json ->> 'address';

        -- 插入學生地址信息，如果存在則更新地址
        INSERT INTO student_address (student_id, address)
        VALUES (lv_field_student_id, lv_field_address)
            ON CONFLICT (student_id) 
        DO UPDATE SET address = excluded.address;
    END LOOP; 

    RETURN json_build_object(
        'err_code', 200, 
        'err_msg', '保存或更新 ' || lv_length || ' 條記錄'
    ); 
end 
$$

我們執行以下操作

SELECT save_student_addresses_json(
    '[
     {"student_id":"01","address":"街道A"},
     {"student_id":"02","address":"街道B"}
    ]'
);

運行結果：{"err_code":200,"err_msg":"保存或更新 2 條記錄"}

小結一下

PostgreSQL 對 JSON 的操作支持特性很豐富，但文檔中那麼多函數一下映入眼帘，讓人覺得複雜凌亂。本文從簡單易理解的幾個應用場景出發，希望能先爽上一把，而後再細細深入。I love PostgreSQL!