如何在JDBI中使用IN运算符?

Ken*_*eth 10 java mysql jdbi dropwizard

我正在尝试使用Dropwizard上的MYSQL JDBI进行IN查询(不相关,我假设).

@SqlQuery("SELECT id FROM table where field in (<list>)")
List<Integer> findSomething(@BindIn("list") List<String> someList);
Run Code Online (Sandbox Code Playgroud)

正如这里所建议的那样,我也用类别注释了这个类

@UseStringTemplate3StatementLocator
Run Code Online (Sandbox Code Playgroud)

但是当我启动应用程序时,我收到以下错误:

Exception in thread "main" java.lang.annotation.AnnotationFormatError: Invalid default: public abstract java.lang.Class org.skife.jdbi.v2.sqlobject.stringtemplate.UseStringTemplate3StatementLocator.errorListener()
Run Code Online (Sandbox Code Playgroud)

有没有人对如何解决这个问题有个好主意?

小智 13

我想,你用的是StringTemplate 4.你需要使用StringTemplate 3而不是StringTemplate 4.添加此依赖项:

<dependency>
    <groupId>org.antlr</groupId>
    <artifactId>stringtemplate</artifactId>
    <version>3.2.1</version>
</dependency>
Run Code Online (Sandbox Code Playgroud)

  • 谢谢,如果我没有添加此依赖项,我的 Dropwizard 应用程序将无法以 `AnnotationFormatError` 启动:`Invalid default: public abstract java.lang.Class org.skife.jdbi.v2.sqlobject.stringtemplate.UseStringTemplate3StatementLocator.errorListener() `. (2认同)

Siv*_*iva 8

有两种方法可以实现它.

1.运用UseStringTemplate3StatementLocator

此批注要求StringTemplate中的Group Files with SQL语句

说我有这个文件PersonExternalizedSqlDAO

// PersonExternalizedSqlDAO.java

package com.daoexp.dao;

@@ExternalizedSqlViaStringTemplate3
@RegisterMapper(PersonMapper.class)
public interface PersonExternalizedSqlDAO {
    @SqlQuery
    List<Person> getPersonByNames(@BindIn("names") List<String> names);
}
Run Code Online (Sandbox Code Playgroud)

由于我们正在使用,UseStringTemplate3StatementLocator我们必须*.sql.stg在相同的类路径中创建文件. 例如: inresources/com/daoexp/dao/PersonExternalizedSqlDAO.sql.stg

group PersonExternalizedSqlDAO;

getPersonByNames(names) ::= <<
  select * from person where name in (<names>)
>>
Run Code Online (Sandbox Code Playgroud)

现在您应该可以查询而不会出现任何问题.


2.另一种方法是使用ArgumentFactory它来处理JDBI的自定义数据类型(在本例中为List)@Bind.这是最优选的方法.

所以创建这个列表参数工厂

public class ListArgumentFactory implements ArgumentFactory<List> {
    @Override
    public boolean accepts(Class<?> expectedType, Object value, StatementContext ctx) {
        return value instanceof List;
    }

    @Override
    public Argument build(Class<?> expectedType, final List value, StatementContext ctx) {
        return new Argument() {
            @Override
            public void apply(int position, PreparedStatement statement, StatementContext ctx) throws SQLException {
                String type = null;
                if(value.get(0).getClass() == String.class){
                    type = "varchar";
                } else if(value.get(0).getClass() == Integer.class){
                    // For integer and so on...
                } else {
                    // throw error.. type not handled
                }
                Array array = ctx.getConnection().createArrayOf(type, value.toArray());
                statement.setArray(position, array);
            }
        };
    }
}
Run Code Online (Sandbox Code Playgroud)

这堂课做什么?

  • 接受List的实例
  • 将整数/字符串列表转换为数组并使用预准备语句绑定

确保将此参数factory注册到DBI实例.

final DBIFactory factory = new DBIFactory();
final DBI jdbi = factory.build(environment, configuration.getDataSourceFactory(), "h2");
jdbi.registerArgumentFactory(new ListArgumentFactory());
Run Code Online (Sandbox Code Playgroud)

现在,您应该能够以List更简单的方式(即)使用查询@Bind.而已.

@RegisterMapper(PersonMapper.class)
public interface PersonDAO {
    @SqlQuery("select * from person where name = any(:names)")
    List<Person> getPersonByNames(@Bind("names") List<String> names);
}
Run Code Online (Sandbox Code Playgroud)

参考:

附加信息:

// PersonMapper.java
public class PersonMapper implements ResultSetMapper<Person> {

    public Person map(int index, ResultSet r, StatementContext ctx) throws SQLException {
        Person person = new Person();
        person.setId(r.getInt("id"));
        person.setName(r.getString("name"));

        return person;
    }
}
Run Code Online (Sandbox Code Playgroud)