Storm/Trident integration for JDBC. This package includes the core bolts and trident states that allows a storm topology to either insert storm tuples in a database table or to execute select queries against a database and enrich tuples in a storm topology.
Note: Throughout the examples below, we make use of com.google.common.collect.Lists and com.google.common.collect.Maps.
The bolt and trident state included in this package for inserting data into a database tables are tied to a single table.
An interface that should be implemented by different connection pooling mechanism org.apache.storm.jdbc.common.ConnectionProvider
public interface ConnectionProvider extends Serializable {
/**
* method must be idempotent.
*/
void prepare();
/**
*
* @return a DB connection over which the queries can be executed.
*/
Connection getConnection();
/**
* called once when the system is shutting down, should be idempotent.
*/
void cleanup();
}
Out of the box we support org.apache.storm.jdbc.common.HikariCPConnectionProvider
which is an implementation that uses HikariCP.
The main API for inserting data in a table using JDBC is the org.apache.storm.jdbc.mapper.JdbcMapper
interface:
public interface JdbcMapper extends Serializable {
List<Column> getColumns(ITuple tuple);
}
The getColumns()
method defines how a storm tuple maps to a list of columns representing a row in a database.
The order of the returned list is important. The place holders in the supplied queries are resolved in the same order as returned list.
For example if the user supplied insert query is insert into user(user_id, user_name, create_date) values (?,?, now())
the 1st item
of the returned list of getColumns
method will map to the 1st place holder and the 2nd to the 2nd and so on. We do not parse
the supplied queries to try and resolve place holder by column names. Not making any assumptions about the query syntax allows this connector
to be used by some non-standard sql frameworks like Pheonix which only supports upsert into.
To use the JdbcInsertBolt
, you construct an instance of it by specifying a ConnectionProvider
implementation
and a JdbcMapper
implementation that converts storm tuple to DB row. In addition, you must either supply
a table name using withTableName
method or an insert query using withInsertQuery
.
If you specify a insert query you should ensure that your JdbcMapper
implementation will return a list of columns in the same order as in your insert query.
You can optionally specify a query timeout seconds param that specifies max seconds an insert query can take.
The default is set to value of topology.message.timeout.secs and a value of -1 will indicate not to set any query timeout.
You should set the query timeout value to be <= topology.message.timeout.secs.
Map hikariConfigMap = Maps.newHashMap();
hikariConfigMap.put("dataSourceClassName","com.mysql.jdbc.jdbc2.optional.MysqlDataSource");
hikariConfigMap.put("dataSource.url", "jdbc:mysql://localhost/test");
hikariConfigMap.put("dataSource.user","root");
hikariConfigMap.put("dataSource.password","password");
ConnectionProvider connectionProvider = new HikariCPConnectionProvider(hikariConfigMap);
String tableName = "user_details";
JdbcMapper simpleJdbcMapper = new SimpleJdbcMapper(tableName, connectionProvider);
JdbcInsertBolt userPersistenceBolt = new JdbcInsertBolt(connectionProvider, simpleJdbcMapper)
.withTableName("user")
.withQueryTimeoutSecs(30);
Or
JdbcInsertBolt userPersistenceBolt = new JdbcInsertBolt(connectionProvider, simpleJdbcMapper)
.withInsertQuery("insert into user values (?,?)")
.withQueryTimeoutSecs(30);
storm-jdbc
includes a general purpose JdbcMapper
implementation called SimpleJdbcMapper
that can map Storm
tuple to a Database row. SimpleJdbcMapper
assumes that the storm tuple has fields with same name as the column name in
the database table that you intend to write to.
To use SimpleJdbcMapper
, you simply tell it the tableName that you want to write to and provide a connectionProvider instance.
The following code creates a SimpleJdbcMapper
instance that:
Map hikariConfigMap = Maps.newHashMap();
hikariConfigMap.put("dataSourceClassName","com.mysql.jdbc.jdbc2.optional.MysqlDataSource");
hikariConfigMap.put("dataSource.url", "jdbc:mysql://localhost/test");
hikariConfigMap.put("dataSource.user","root");
hikariConfigMap.put("dataSource.password","password");
ConnectionProvider connectionProvider = new HikariCPConnectionProvider(hikariConfigMap);
String tableName = "user_details";
JdbcMapper simpleJdbcMapper = new SimpleJdbcMapper(tableName, connectionProvider);
The mapper initialized in the example above assumes a storm tuple has value for all the columns of the table you intend to insert data into and its getColumn
method will return the columns in the order in which Jdbc connection instance's connection.getMetaData().getColumns();
method returns them.
If you specified your own insert query to JdbcInsertBolt
you must initialize SimpleJdbcMapper
with explicit columnschema such that the schema has columns in the same order as your insert queries.
For example if your insert query is Insert into user (user_id, user_name) values (?,?)
then your SimpleJdbcMapper
should be initialized with the following statements:
List<Column> columnSchema = Lists.newArrayList(
new Column("user_id", java.sql.Types.INTEGER),
new Column("user_name", java.sql.Types.VARCHAR));
JdbcMapper simpleJdbcMapper = new SimpleJdbcMapper(columnSchema);
If your storm tuple only has fields for a subset of columns i.e. if some of the columns in your table have default values and you want to only insert values for columns with no default values you can enforce the behavior by initializing the
SimpleJdbcMapper
with explicit columnschema. For example, if you have a user_details table create table if not exists user_details (user_id integer, user_name varchar(100), dept_name varchar(100), create_time TIMESTAMP DEFAULT CURRENT_TIMESTAMP);
In this table the create_time column has a default value. To ensure only the columns with no default values are inserted
you can initialize the jdbcMapper
as below:
List<Column> columnSchema = Lists.newArrayList(
new Column("user_id", java.sql.Types.INTEGER),
new Column("user_name", java.sql.Types.VARCHAR),
new Column("dept_name", java.sql.Types.VARCHAR));
JdbcMapper simpleJdbcMapper = new SimpleJdbcMapper(columnSchema);
We also support a trident persistent state that can be used with trident topologies. To create a jdbc persistent trident state you need to initialize it with the table name or an insert query, the JdbcMapper instance and connection provider instance. See the example below:
JdbcState.Options options = new JdbcState.Options()
.withConnectionProvider(connectionProvider)
.withMapper(jdbcMapper)
.withTableName("user_details")
.withQueryTimeoutSecs(30);
JdbcStateFactory jdbcStateFactory = new JdbcStateFactory(options);
similar to JdbcInsertBolt
you can specify a custom insert query using withInsertQuery
instead of specifying a table name.
We support select
queries from databases to allow enrichment of storm tuples in a topology. The main API for
executing select queries against a database using JDBC is the org.apache.storm.jdbc.mapper.JdbcLookupMapper
interface:
void declareOutputFields(OutputFieldsDeclarer declarer);
List<Column> getColumns(ITuple tuple);
List<Values> toTuple(ITuple input, List<Column> columns);
The declareOutputFields
method is used to indicate what fields will be emitted as part of output tuple of processing a storm
tuple.
The getColumns
method specifies the place holder columns in a select query and their SQL type and the value to use.
For example in the user_details table mentioned above if you were executing a query select user_name from user_details where
user_id = ? and create_time > ?
the getColumns
method would take a storm input tuple and return a List containing two items.
The first instance of Column
type's getValue()
method will be used as the value of user_id
to lookup for and the
second instance of Column
type's getValue()
method will be used as the value of create_time
.
Note: the order in the returned list determines the place holder's value. In other words the first item in the list maps
to first ?
in select query, the second item to second ?
in query and so on.
The toTuple
method takes in the input tuple and a list of columns representing a DB row as a result of the select query
and returns a list of values to be emitted.
Please note that it returns a list of Values
and not just a single instance of Values
.
This allows a for a single DB row to be mapped to multiple output storm tuples.
storm-jdbc
includes a general purpose JdbcLookupMapper
implementation called SimpleJdbcLookupMapper
.
To use SimpleJdbcMapper
, you have to initialize it with the fields that will be outputted by your bolt and the list of
columns that are used in your select query as place holder. The following example shows initialization of a SimpleJdbcLookupMapper
that declares user_id,user_name,create_date
as output fields and user_id
as the place holder column in select query.
SimpleJdbcMapper assumes the field name in your tuple is equal to the place holder column name, i.e. in our example
SimpleJdbcMapper
will look for a field use_id
in the input tuple and use its value as the place holder's value in the
select query. For constructing output tuples, it looks for fields specified in outputFields
in the input tuple first,
and if it is not found in input tuple then it looks at select query's output row for a column with same name as field name.
So in the example below if the input tuple had fields user_id, create_date
and the select query was
select user_name from user_details where user_id = ?
, For each input tuple SimpleJdbcLookupMapper.getColumns(tuple)
will return the value of tuple.getValueByField("user_id")
which will be used as the value in ?
of select query.
For each output row from DB, SimpleJdbcLookupMapper.toTuple()
will use the user_id, create_date
from the input tuple as
is adding only user_name
from the resulting row and returning these 3 fields as a single output tuple.
Fields outputFields = new Fields("user_id", "user_name", "create_date");
List<Column> queryParamColumns = Lists.newArrayList(new Column("user_id", Types.INTEGER));
this.jdbcLookupMapper = new SimpleJdbcLookupMapper(outputFields, queryParamColumns);
To use the JdbcLookupBolt
, construct an instance of it using a ConnectionProvider
instance, JdbcLookupMapper
instance and the select query to execute.
You can optionally specify a query timeout seconds param that specifies max seconds the select query can take.
The default is set to value of topology.message.timeout.secs. You should set this value to be <= topology.message.timeout.secs.
String selectSql = "select user_name from user_details where user_id = ?";
SimpleJdbcLookupMapper lookupMapper = new SimpleJdbcLookupMapper(outputFields, queryParamColumns)
JdbcLookupBolt userNameLookupBolt = new JdbcLookupBolt(connectionProvider, selectSql, lookupMapper)
.withQueryTimeoutSecs(30);
We also support a trident query state that can be used with trident topologies.
JdbcState.Options options = new JdbcState.Options()
.withConnectionProvider(connectionProvider)
.withJdbcLookupMapper(new SimpleJdbcLookupMapper(new Fields("user_name"), Lists.newArrayList(new Column("user_id", Types.INTEGER))))
.withSelectQuery("select user_name from user_details where user_id = ?");
.withQueryTimeoutSecs(30);
A runnable example can be found in the src/test/java/topology
directory.
create table if not exists user (user_id integer, user_name varchar(100), dept_name varchar(100), create_date date);
create table if not exists department (dept_id integer, dept_name varchar(100));
create table if not exists user_department (user_id integer, dept_id integer);
insert into department values (1, 'R&D');
insert into department values (2, 'Finance');
insert into department values (3, 'HR');
insert into department values (4, 'Sales');
insert into user_department values (1, 1);
insert into user_department values (2, 2);
insert into user_department values (3, 3);
insert into user_department values (4, 4);
select dept_name from department, user_department where department.dept_id = user_department.dept_id and user_department.user_id = ?;
Run the org.apache.storm.jdbc.topology.UserPersistenceTopology
class using storm jar command. The class expects 5 args
storm jar org.apache.storm.jdbc.topology.UserPersistenceTopology
To make it work with Mysql, you can add the following to the pom.xml
<dependency>
<groupId>mysql</groupId>
<artifactId>mysql-connector-java</artifactId>
<version>5.1.31</version>
</dependency>
You can generate a single jar with dependencies using mvn assembly plugin. To use the plugin add the following to your pom.xml and execute
mvn clean compile assembly:single
<plugin>
<artifactId>maven-assembly-plugin</artifactId>
<configuration>
<archive>
<manifest>
<mainClass>fully.qualified.MainClass</mainClass>
</manifest>
</archive>
<descriptorRefs>
<descriptorRef>jar-with-dependencies</descriptorRef>
</descriptorRefs>
</configuration>
</plugin>
Mysql Example:
storm jar ~/repo/incubator-storm/external/storm-jdbc/target/storm-jdbc-0.10.0-SNAPSHOT-jar-with-dependencies.jar org.apache.storm.jdbc.topology.UserPersistenceTopology com.mysql.jdbc.jdbc2.optional.MysqlDataSource jdbc:mysql://localhost/test root password UserPersistenceTopology
You can execute a select query against the user table which should show newly inserted rows:
select * from user;
For trident you can view org.apache.storm.jdbc.topology.UserPersistenceTridentTopology
.