从 MyBatis 返回值<insert>映射方法

Question

我有一个使用 MyBatis 访问 PostgreSQL 数据库的 Java 项目。 PostgreSQL 允许在

INSERT

语句之后返回新创建行的字段，我想用它来返回新创建记录的自动生成的

BIGSERIAL

id

。因此，我更改了 XML 中的

insert

命令以使用 PostgreSQL 的功能，向

resultType="long"

标记添加

<insert>

属性，并在映射器的 Java 接口中将插入方法设置为返回

long

void

。

当我尝试运行此程序时，我收到一个

org.xml.sax.SAXParseException

表示

Attribute "resultType" must be declared for element type "insert"

。

现在，当我将

<insert>

标签更改为

<select>

时，一切正常，但令我困扰的是我使用

<select>

标签来执行

INSERT

语句。

有没有办法让映射到

<insert>

标签的方法返回结果，或者 MyBatis 不是为此设计的，我应该将它们保留为

<select>

标签？

Answer 1

映射插入方法的返回类型可以是

void

或

int

（在这种情况下它将返回插入的行号）。您可以执行以下机制来返回生成的 id：

<insert id="insert" parameterClass="MyParameter">
  <selectKey order="AFTER" keyProperty="id" resultType="long">
    SELECT currval('my_seq')
  </selectKey>
  INSERT INTO mytable(col1, col2) VALUES (#{val1}, #{val2})
</insert>

这会将生成的

id

列设置为参数类的

id

属性。之后，您作为参数传递的对象将在其属性中生成

id

设置。

Answer 2

有两种方法（至少我知道）可以获取一条插入记录的ID：

比如我们有一个类

EntityDao

:

public class EntityDao {
     private Long id;
     private String name;
     // other fields, getters and setters
}

---

1.使用

insert

标签并返回对象的实例

MyBatis 界面

public interface EntityDaoMapper {
    EntityDao insert(EntityDao entity);
}

MyBatis XML 映射器：

<insert id="insert" parameterType="com.package.EntityDao" useGeneratedKeys="true" keyColumn="entity_id" keyProperty="id">
    INSERT INTO some_table (name, type, other_fields, etc)
    VALUES (#{name}, #{type}, #{other_fields}, #{etc}) 
</insert>

示例代码：

    EntityDao saved = entityDaoMapper.insert(entityToSave);
    System.out.println(saved.getId());

2.使用

select

和

resultType

标签仅返回记录的 ID

MyBatis 界面

public interface EntityDaoMapper {
    Long insert(EntityDao entity);
}

MyBatis XML 映射器：

<select id="insert" parameterType="com.package.EntityDao" resultType="long">
    INSERT INTO some_table (name, type, other_fields, etc)
    VALUES (#{name}, #{type}, #{other_fields}, #{etc}) 
    RETURNING entity_id       <-- id only or many fields
</select>

示例代码：

Long id = entityDaoMapper.insert(entityToSave);
System.out.println(id);

Answer 3

您可以按如下方式使用。在 XML 中

 <insert id="insertNewUser" parameterType="User">
            <selectKey keyProperty="userId" resultType="Integer" order="BEFORE">
                select NEXTVAL('base.user_id_seq')
            </selectKey>
            INSERT INTO base.user(
                user_id, user_name)
            VALUES (#{userId}, #{userName});
    </insert>

在调用插入方法的 Java 类中，可以通过调用

user.getUserId()

来获取值。

基本上下一个 val 存储在对象的变量内。

Here userId inside User.

Answer 4

也可以用Java

Records

和mybatis注解来完成

实体：

record UserEntity (
    int id,
    String name,
){}

地图绘制者：

@Select("INSERT INTO user (name) VALUES (#{name}) returning id")
int create(UserEntity entity);

这里的关键是使用

@Select

代替

@Insert

，并在sql语句末尾添加

returning id

。

Answer 5

您还可以使用生成的密钥：

  <insert id="create" parameterType="Skupina" useGeneratedKeys="true" keyColumn="id" keyProperty="id">
        INSERT INTO ODBOR 
            (NAZEV, POPIS, ZKRATKA, WEBROLE, JEODBOR, AKTIVNI)
        VALUES 
            (#{nazev}, #{popis}, #{webrole}, #{webrole}, false, #{aktivni})
  </insert>

插入后，参数的属性 id 设置为列 id 中的值。

Answer 6

在此示例中使用选项

import org.apache.ibatis.annotations.Insert;
import org.apache.ibatis.annotations.Mapper;
import org.apache.ibatis.annotations.Options;
import org.apache.ibatis.annotations.Param;

@Mapper
public interface POJOCustomMapper {
    @Options(useGeneratedKeys = true, keyProperty = "ID", keyColumn = "ID") 
    @Insert({
        "insert into TABLE_A ( NAME "
        "values (#{NAME,jdbcType=VARCHAR})"
    })
    long insert(POJO record);

Answer 7

使用“返回”返回对象的实例

<select id="insertOrderHeader" resultType="com.***.entity.OrderHeader">
    insert into order_header_t (order_id, delete_flag, organization_code)
    values (#{query.orderId}, #{query.deleteFlag}, #{query.organizationCode})
        returning order_id as orderId, delete_flag as deleteFlag, organization_code as organizationCode
</select>