从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})
</isnert>

这会将生成的id列设置为参数类的id属性。之后，您作为参数传递的对象将在其属性中生成id。

Answer 2

您可以使用如下。在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 3

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

  <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 4

有两种方法（至少我知道）来获取一条插入记录的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 5

在此示例中，使用选项

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);