MyBatis-Plus 实现PostgreSQL数据库jsonb类型的保存与查询

文章目录

• 一、在 handle 包下新建 Jsonb 处理类
• 二、PostgreSQL jsonb 类型保存
• • 1. 新建数据库表含有 jsonb 类型
  • 2. 创建实体类
  • 3. Control 创建保存数据库方法
  • 4. 发起请求
• 三、PostgreSQL jsonb 类型查询
• • 1. Control 创建查询数据库方法
  • 2. 发起请求

一、在 handle 包下新建 Jsonb 处理类

1. 方式一

import com.alibaba.fastjson.JSON;
import com.alibaba.fastjson.serializer.SerializerFeature;
import com.baomidou.mybatisplus.extension.handlers.AbstractJsonTypeHandler;
import org.apache.commons.lang3.StringUtils;
import org.apache.ibatis.type.JdbcType;
import org.apache.ibatis.type.MappedJdbcTypes;
import org.apache.ibatis.type.MappedTypes;
import org.postgresql.util.PGobject;

import java.sql.CallableStatement;
import java.sql.PreparedStatement;
import java.sql.ResultSet;
import java.sql.SQLException;


@MappedTypes({Object.class})
@MappedJdbcTypes({JdbcType.VARCHAR})
public class JsonbTypeHandler extends AbstractJsonTypeHandlerObject> {
    private static final PGobject jsonObject = new PGobject();
    private final Class?> type;

    public JsonbTypeHandler(Class?> type) {
        this.type = type;
    }

    
    @Override
    public void setNonNullParameter(PreparedStatement ps, int i, Object parameter, JdbcType jdbcType) throws SQLException {
        if (ps != null) {
            jsonObject.setType("jsonb");
            jsonObject.setValue(JSON.toJSONString(parameter));
            ps.setObject(i, jsonObject);
        }
    }

    
    @Override
    public Object getNullableResult(ResultSet rs, String columnName) throws SQLException {
        Object v = rs.getObject(columnName);
        return toFill(v);
    }

    
    @Override
    public Object getNullableResult(ResultSet rs, int columnIndex) throws SQLException {
        Object v = rs.getObject(columnIndex);
        return toFill(v);
    }

    @Override
    public Object getNullableResult(CallableStatement cs, int columnIndex) throws SQLException {
        Object v = cs.getObject(columnIndex);
        return toFill(v);
    }

    @Override
    protected Object parse(String json) {
        return JSON.parseObject(json, this.type);
    }

    
    private Object toFill(Object v) {
        if (v != null && v instanceof PGobject) {
            PGobject p = (PGobject) v;
            String pv = p.getValue();
            if (StringUtils.isNotEmpty(pv) && ("jsonb".equals(p.getType()) || "json".equals(p.getType()))) {
                return parse(p.getValue());
            }
        }
        return v;
    }

    @Override
    protected String toJson(Object obj) {
        return JSON.toJSONString(obj, SerializerFeature.WriteMapNullValue, SerializerFeature.WriteNullListAsEmpty, SerializerFeature.WriteNullStringAsEmpty);
    }
}

2. 方式二

package com.hlta.map.utils;
 
import com.alibaba.fastjson.JSON;
import org.apache.ibatis.type.BaseTypeHandler;
import org.apache.ibatis.type.JdbcType;
import org.apache.ibatis.type.MappedTypes;
import org.postgresql.util.PGobject;
 
import java.sql.CallableStatement;
import java.sql.PreparedStatement;
import java.sql.ResultSet;
import java.sql.SQLException;


@MappedTypes({Object.class})
public class JsonbTypeHandler extends BaseTypeHandlerObject> {
    private static final PGobject jsonObject = new PGobject();
 
    @Override
    public void setNonNullParameter(PreparedStatement preparedStatement, int i, Object o, JdbcType jdbcType) throws SQLException {
        if (preparedStatement != null) {
            jsonObject.setType("jsonb");
            jsonObject.setValue(JSON.toJSONString(o));
            preparedStatement.setObject(i, jsonObject);
        }
    }
 
    @Override
    public Object getNullableResult(ResultSet resultSet, String s) throws SQLException {
        return JSON.parse(resultSet.getString(s));
    }
 
    @Override
    public Object getNullableResult(ResultSet resultSet, int i) throws SQLException {
        return JSON.parse(resultSet.getString(i));
    }
 
    @Override
    public Object getNullableResult(CallableStatement callableStatement, int i) throws SQLException {
        return JSON.parse(callableStatement.getString(i));
    }
}

二、PostgreSQL jsonb 类型保存

1. 新建数据库表含有 jsonb 类型

2. 创建实体类

• 在实体类上加上 @TableName(value = "表名", autoResultMap = true)
• 在 jsonb 属性上加上 @TableField(value = "字段", typeHandler = JsonbTypeHandler.class)

import com.baomidou.mybatisplus.annotation.*;
import com.fasterxml.jackson.annotation.JsonFormat;
import com.chh.handle.JsonbTypeHandler;
import lombok.Data;
import lombok.EqualsAndHashCode;
import lombok.experimental.Accessors;

import java.io.Serializable;
import java.util.Date;


@Data
@EqualsAndHashCode(callSuper = false)
@Accessors(chain = true)
@TableName(value = "app_control_manage", autoResultMap = true)
public class AppControlManage implements Serializable {

    private static final long serialVersionUID = 1L;

    @TableId(value = "id", type = IdType.AUTO)
    private Integer id;

    
    @TableField("name")
    private String name;

    
    @TableField("channel")
    private String channel;

    
    @TableField(value = "content", typeHandler = JsonbTypeHandler.class)
    private Object content;

    
    @TableField("description")
    private String description;

    
    @JsonFormat(pattern = "yyyy-MM-dd HH:mm:ss")
    @TableField(fill = FieldFill.INSERT)
    private Date createTime;
}

3. Control 创建保存数据库方法

以下方法用于接收 post 请求，将数据保存到数据库

@PostMapping("/test")
public Object test(@RequestBody JSONObject object, final HttpServletResponse response, final HttpServletRequest request) {
    AppControlManage appControlManage = object.toJavaObject(AppControlManage.class);
    boolean save = appControlManageService.save(appControlManage);
    if (save) {
        System.out.println("保存成功");
    } else {
        System.out.println("保存失败");
    }
    return appControlManage;
}

4. 发起请求

可以看到，json 数据被保存进数据库

三、PostgreSQL jsonb 类型查询

1. Control 创建查询数据库方法

以下方法用于接收 get 请求，查询数据 content 字段，其中 #>> 为以文本形式获取在指定路径的 JSON 对象，'{test,0,name}' 可以得到“test”JSON 对象下的下标为 0 的数组，其中 key 为 name 所对应的 value 的文本。以下包含了排序和分页可供参考。

不懂如何操作 json/jsonb，可参考博主的另一篇博文：
PostgreSQL 操作 json/jsonb

@GetMapping("/select")
public Object select(@RequestParam("mpage") Integer mpage,
                     @RequestParam("msize") Integer msize,
                     @RequestParam("msort") String msort,
                     @RequestParam("name") String name) {
    IPageAppControlManage> iPage = new LambdaQueryChainWrapper>(appControlManageMapper)
            .apply(StringUtil.isNotEmpty(name), "content #>>'{test,0,name}'= {0}", name)
            .last(StringUtil.isNotEmpty(msort), "order by" + msort)
            .page(new Page>(mpage + 1, msize));
    return new InfoResult(iPage.getTotal(), iPage.getRecords());
}

2. 发起请求

可以看到，查询到了 content 字段下的 test JSON 对象下的第一个数组，key 为 name，value 为 apple 的数据。