您现在的位置是：网站首页> 编程资料编程资料

Mybatis调用PostgreSQL存储过程实现数组入参传递_PostgreSQL_

2023-05-27 459人已围观

简介 Mybatis调用PostgreSQL存储过程实现数组入参传递_PostgreSQL_

前言

项目中用到了Mybatis调用PostgreSQL存储过程（自定义函数）相关操作，由于PostgreSQL自带数组类型，所以有一个自定义函数的入参就是一个int数组，形如：

复制代码 代码如下:
CREATE OR REPLACE FUNCTION "public"."func_arr_update"(ids _int4)...

如上所示，参数是一个int数组，Mybatis提供了对调用存储过程的支持，那么PostgreSQL独有的数组类型作为存储过程的参数又将如何处理呢？其实很简单，mybatis提供了typeHandlers可以创建一个数组类型的类型处理器，具体做法为：实现 org.apache.ibatis.type.TypeHandler 接口， 或继承一个很便利的类 org.apache.ibatis.type.BaseTypeHandler， 然后可以选择性地将它映射到一个 JDBC 类型，先稍作了解，后面再做详细说明，接下来依旧结合一个示例来看看。

创建自定义函数

如图，第一步首先是创建一个用于调用的自定义函数，功能也很简单，遍历参数数组的每一个元素和t_student表的stuid做比较，若一致，则修改那条记录的stuname（在其后拼接一段字符串），该自定义函数的DLL语句如下：

 CREATE OR REPLACE FUNCTION "public"."func_arr_update"(ids _int4) RETURNS "pg_catalog"."void" AS $BODY$ DECLARE scount INTEGER; rownum integer := 1; BEGIN scount:=array_length(ids,1); while rownum <= scount LOOP update t_student set stuname = stuname || ' has been modified. ' where stuid = ids[rownum]; rownum := rownum + 1; END LOOP; RETURN; END $BODY$ LANGUAGE 'plpgsql' VOLATILE COST 100 ; ALTER FUNCTION "public"."func_arr_update"(ids _int4) OWNER TO "postgres";

很简单，获取到参数数组的长度后开始循环，匹配stuid并更新stuname，直接在数据库调用一下看看结果：

如上图，可以看到成功修改了stuid为101，102和103的stuname，自定义函数已经没问题了，接下来就具体看一下如何通过mybatis调用。

调用自定义函数

mybatis中调用自定义函数很简单，Mapper XML文件中的select元素直接提供了属性支持——statementType，在官方文档中可以看到：

如上图，statementType的值默认是PREPARED，也就是说底层默认会使用jdbc的PreparedStatement，而我们都知道jdbc调用存储过程时需要用CallableStatement，所以在这里我们需要将statementType的值设置为CALLABLE。

mybatis默认的ArrayTypeHandler

调用存储过程很简单，那么接下来的问题是如何在mybatis中传一个数组参数到存储过程中呢？这里就要用到另外一个概念——TypeHandler，这是mybatis提供的自定义类型转换器，mybatis在预编译语句对象（PreparedStatement）设置参数时或是从结果集中取值时都会用类型处理器将获取的值以合适的方式转换成Java类型，mybatis默认实现了一部分TypeHandler供我们使用，当我们没有指定TypeHandler时（大多数情况都不会指定），mybatis会根据参数或者返回结果的不同，默认为我们选择合适的TypeHandler处理，下面可以通过查看源码大概看一下默认的TypeHandler，导入源码后可以在org.apache.ibatis.type包下找到一个TypeHandlerRegistry类，typeHandler正是通过这个类管理的，先看一下它的构造方法：

 public TypeHandlerRegistry() { register(Boolean.class, new BooleanTypeHandler()); register(boolean.class, new BooleanTypeHandler()); register(JdbcType.BOOLEAN, new BooleanTypeHandler()); register(JdbcType.BIT, new BooleanTypeHandler()); register(Byte.class, new ByteTypeHandler()); register(byte.class, new ByteTypeHandler()); register(JdbcType.TINYINT, new ByteTypeHandler()); register(Short.class, new ShortTypeHandler()); register(short.class, new ShortTypeHandler()); register(JdbcType.SMALLINT, new ShortTypeHandler()); register(Integer.class, new IntegerTypeHandler()); register(int.class, new IntegerTypeHandler()); register(JdbcType.INTEGER, new IntegerTypeHandler()); register(Long.class, new LongTypeHandler()); register(long.class, new LongTypeHandler()); register(Float.class, new FloatTypeHandler()); register(float.class, new FloatTypeHandler()); register(JdbcType.FLOAT, new FloatTypeHandler()); register(Double.class, new DoubleTypeHandler()); register(double.class, new DoubleTypeHandler()); register(JdbcType.DOUBLE, new DoubleTypeHandler()); register(String.class, new StringTypeHandler()); register(String.class, JdbcType.CHAR, new StringTypeHandler()); register(String.class, JdbcType.CLOB, new ClobTypeHandler()); register(String.class, JdbcType.VARCHAR, new StringTypeHandler()); register(String.class, JdbcType.LONGVARCHAR, new ClobTypeHandler()); register(String.class, JdbcType.NVARCHAR, new NStringTypeHandler()); register(String.class, JdbcType.NCHAR, new NStringTypeHandler()); register(String.class, JdbcType.NCLOB, new NClobTypeHandler()); register(JdbcType.CHAR, new StringTypeHandler()); register(JdbcType.VARCHAR, new StringTypeHandler()); register(JdbcType.CLOB, new ClobTypeHandler()); register(JdbcType.LONGVARCHAR, new ClobTypeHandler()); register(JdbcType.NVARCHAR, new NStringTypeHandler()); register(JdbcType.NCHAR, new NStringTypeHandler()); register(JdbcType.NCLOB, new NClobTypeHandler()); register(Object.class, JdbcType.ARRAY, new ArrayTypeHandler()); register(JdbcType.ARRAY, new ArrayTypeHandler()); register(BigInteger.class, new BigIntegerTypeHandler()); register(JdbcType.BIGINT, new LongTypeHandler()); register(BigDecimal.class, new BigDecimalTypeHandler()); register(JdbcType.REAL, new BigDecimalTypeHandler()); register(JdbcType.DECIMAL, new BigDecimalTypeHandler()); register(JdbcType.NUMERIC, new BigDecimalTypeHandler()); register(Byte[].class, new ByteObjectArrayTypeHandler()); register(Byte[].class, JdbcType.BLOB, new BlobByteObjectArrayTypeHandler()); register(Byte[].class, JdbcType.LONGVARBINARY, new BlobByteObjectArrayTypeHandler()); register(byte[].class, new ByteArrayTypeHandler()); register(byte[].class, JdbcType.BLOB, new BlobTypeHandler()); register(byte[].class, JdbcType.LONGVARBINARY, new BlobTypeHandler()); register(JdbcType.LONGVARBINARY, new BlobTypeHandler()); register(JdbcType.BLOB, new BlobTypeHandler()); register(Object.class, UNKNOWN_TYPE_HANDLER); register(Object.class, JdbcType.OTHER, UNKNOWN_TYPE_HANDLER); register(JdbcType.OTHER, UNKNOWN_TYPE_HANDLER); register(Date.class, new DateTypeHandler()); register(Date.class, JdbcType.DATE, new DateOnlyTypeHandler()); register(Date.class, JdbcType.TIME, new TimeOnlyTypeHandler()); register(JdbcType.TIMESTAMP, new DateTypeHandler()); register(JdbcType.DATE, new DateOnlyTypeHandler()); register(JdbcType.TIME, new TimeOnlyTypeHandler()); register(java.sql.Date.class, new SqlDateTypeHandler()); register(java.sql.Time.class, new SqlTimeTypeHandler()); register(java.sql.Timestamp.class, new SqlTimestampTypeHandler()); // issue #273 register(Character.class, new CharacterTypeHandler()); register(char.class, new CharacterTypeHandler()); }

如上所示，这就是全部默认的typeHandler了，注意一下46，47行可以看到默认有一个ArrayTypeHandler，顺便看一下它的源码：

 /* * Copyright 2009-2012 The MyBatis Team * * Licensed under the Apache License, Version 2.0 (the "License"); * you may not use this file except in compliance with the License. * You may obtain a copy of the License at * * http://www.apache.org/licenses/LICENSE-2.0 * * Unless required by applicable law or agreed to in writing, software * distributed under the License is distributed on an "AS IS" BASIS, * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. * See the License for the specific language governing permissions and * limitations under the License. */ package org.apache.ibatis.type; import java.sql.Array; import java.sql.CallableStatement; import java.sql.PreparedStatement; import java.sql.ResultSet; import java.sql.SQLException; public class ArrayTypeHandler extends BaseTypeHandler { public ArrayTypeHandler() { super(); } @Override public void setNonNullParameter(PreparedStatement ps, int i, Object parameter, JdbcType jdbcType) throws SQLException { ps.setArray(i, (Array) parameter); } @Override public Object getNullableResult(ResultSet rs, String columnName) throws SQLException { Array array = rs.getArray(columnName); return array == null ? null : array.getArray(); } @Override public Object getNullableResult(ResultSet rs, int columnIndex) throws SQLException { Array array = rs.getArray(columnIndex); return array == null ? null : array.getArray(); } @Override public Object getNullableResult(CallableStatement cs, int columnIndex) throws SQLException { Array array = cs.getArray(columnIndex); return array == null ? null : array.getArray(); } } 
那它能否识别PostgreSQL的数组类型并将它自动转换成Java数组类型呢？按官方的说法，既然这是默认的typeHandler，那么我们无需做任何配置mybatis会自动尝试适配，所以直接写测试代码看看：
 @Test public void testFunc1() { SqlSession session = sqlSessionFactory.openSession(); try { Map map = new HashMap(); map.put("ids", new Integer[] { 101, 102, 103 }); session.update("com.wl.entity.StudentMapper.testFuncUpdate2", map); session.commit(); } catch (Exception e) { e.printStackTrace(); } finally { session.close(); } } 
 {call func_arr_update (#{ids,mode=IN})} 
如上所示，参数传的是一个Integer[]，直接运行一下junit看看测试结果： 
Can't infer the SQL type to use for an instance of [Ljava.lang.Integer;. Use setObject() with an explicit Types value to specify the type to use.
异常log如上所示，在调用AbstractJdbc2Statement类的setObject方法时抛出异常，那么再看看这个方法的源码：
 /* * This stores an Object into a parameter. */ public void setObject(int parameterIndex, Object x) throws SQLException { checkClosed(); if (x == null) setNull(parameterIndex, Types.OTHER); else if (x instanceof String) setString(parameterIndex, (String)x); else if (x instanceof BigDecimal) setBigDecimal(parameterIndex, (BigDecimal)x); else if (x instanceof Short) setShort(parameterIndex, ((Short)x).shortValue()); else if (x instanceof Integer) setInt(parameterIndex, ((Integer)x).intValue()); else if (x instanceof Long) setLong(parameterIndex, ((Long)x).longValue()); else if (x instanceof Float) setFloat(parameterIndex, ((Float)x).floatValue()); else if (x instanceof Double) setDouble(parameterIndex, ((Double)x).doubleValue()); else if (x instanceof byte[]) se
                
                

                

                                            
上一篇：Linux CentOS 7源码编译安装PostgreSQL9.5_PostgreSQL_

                                                                
下一篇：PostgreSQL分区表（partitioning）应用实例详解_PostgreSQL_

                                    

            

        
        

            
相关内容

            
        

    
    

        

            
点击排行

            
        

        

            
本栏推荐

            
        


        

            
猜你喜欢