spring-batch HibernateCursorItemReaderBuilder 源码

  • 2022-08-16
  • 浏览 (11)

spring-batch HibernateCursorItemReaderBuilder 代码

文件路径:/spring-batch-infrastructure/src/main/java/org/springframework/batch/item/database/builder/HibernateCursorItemReaderBuilder.java

/*
 * Copyright 2017-2018 the original author or authors.
 *
 * 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
 *
 *      https://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.springframework.batch.item.database.builder;

import java.util.Map;

import org.hibernate.SessionFactory;

import org.springframework.batch.item.database.HibernateCursorItemReader;
import org.springframework.batch.item.database.orm.HibernateNativeQueryProvider;
import org.springframework.batch.item.database.orm.HibernateQueryProvider;
import org.springframework.util.Assert;
import org.springframework.util.StringUtils;

/**
 * This is a builder for the {@link HibernateCursorItemReader}. When configuring, one of
 * the following should be provided (listed in order of precedence):
 * <ul>
 * <li>{@link #queryProvider(HibernateQueryProvider)}</li>
 * <li>{@link #queryName(String)}</li>
 * <li>{@link #queryString(String)}</li>
 * <li>{@link #nativeQuery(String)} and {@link #entityClass(Class)}</li>
 * </ul>
 *
 * @author Michael Minella
 * @author Glenn Renfro
 * @author Mahmoud Ben Hassine
 * @since 4.0
 * @see HibernateCursorItemReader
 */
public class HibernateCursorItemReaderBuilder<T> {

	private Map<String, Object> parameterValues;

	private String queryName;

	private int fetchSize;

	private HibernateQueryProvider<T> queryProvider;

	private String queryString;

	private SessionFactory sessionFactory;

	private boolean useStatelessSession;

	private String nativeQuery;

	private Class<T> nativeClass;

	private boolean saveState = true;

	private String name;

	private int maxItemCount = Integer.MAX_VALUE;

	private int currentItemCount;

	/**
	 * Configure if the state of the
	 * {@link org.springframework.batch.item.ItemStreamSupport} should be persisted within
	 * the {@link org.springframework.batch.item.ExecutionContext} for restart purposes.
	 * @param saveState defaults to true
	 * @return The current instance of the builder.
	 */
	public HibernateCursorItemReaderBuilder<T> saveState(boolean saveState) {
		this.saveState = saveState;

		return this;
	}

	/**
	 * The name used to calculate the key within the
	 * {@link org.springframework.batch.item.ExecutionContext}. Required if
	 * {@link #saveState(boolean)} is set to true.
	 * @param name name of the reader instance
	 * @return The current instance of the builder.
	 * @see org.springframework.batch.item.ItemStreamSupport#setName(String)
	 */
	public HibernateCursorItemReaderBuilder<T> name(String name) {
		this.name = name;

		return this;
	}

	/**
	 * Configure the max number of items to be read.
	 * @param maxItemCount the max items to be read
	 * @return The current instance of the builder.
	 * @see org.springframework.batch.item.support.AbstractItemCountingItemStreamItemReader#setMaxItemCount(int)
	 */
	public HibernateCursorItemReaderBuilder<T> maxItemCount(int maxItemCount) {
		this.maxItemCount = maxItemCount;

		return this;
	}

	/**
	 * Index for the current item. Used on restarts to indicate where to start from.
	 * @param currentItemCount current index
	 * @return this instance for method chaining
	 * @see org.springframework.batch.item.support.AbstractItemCountingItemStreamItemReader#setCurrentItemCount(int)
	 */
	public HibernateCursorItemReaderBuilder<T> currentItemCount(int currentItemCount) {
		this.currentItemCount = currentItemCount;

		return this;
	}

	/**
	 * A map of parameter values to be set on the query. The key of the map is the name of
	 * the parameter to be set with the value being the value to be set.
	 * @param parameterValues map of values
	 * @return this instance for method chaining
	 * @see HibernateCursorItemReader#setParameterValues(Map)
	 */
	public HibernateCursorItemReaderBuilder<T> parameterValues(Map<String, Object> parameterValues) {
		this.parameterValues = parameterValues;

		return this;
	}

	/**
	 * The name of the Hibernate named query to be executed for this reader.
	 * @param queryName name of the query to execute
	 * @return this instance for method chaining
	 * @see HibernateCursorItemReader#setQueryName(String)
	 */
	public HibernateCursorItemReaderBuilder<T> queryName(String queryName) {
		this.queryName = queryName;

		return this;
	}

	/**
	 * The number of items to be returned with each round trip to the database. Used
	 * internally by Hibernate.
	 * @param fetchSize number of records to return per fetch
	 * @return this instance for method chaining
	 * @see HibernateCursorItemReader#setFetchSize(int)
	 */
	public HibernateCursorItemReaderBuilder<T> fetchSize(int fetchSize) {
		this.fetchSize = fetchSize;

		return this;
	}

	/**
	 * A query provider. This should be set only if {@link #queryString(String)} and
	 * {@link #queryName(String)} have not been set.
	 * @param queryProvider the query provider
	 * @return this instance for method chaining
	 * @see HibernateCursorItemReader#setQueryProvider(HibernateQueryProvider)
	 */
	public HibernateCursorItemReaderBuilder<T> queryProvider(HibernateQueryProvider<T> queryProvider) {
		this.queryProvider = queryProvider;

		return this;
	}

	/**
	 * The HQL query string to execute. This should only be set if
	 * {@link #queryProvider(HibernateQueryProvider)} and {@link #queryName(String)} have
	 * not been set.
	 * @param queryString the HQL query
	 * @return this instance for method chaining
	 * @see HibernateCursorItemReader#setQueryString(String)
	 */
	public HibernateCursorItemReaderBuilder<T> queryString(String queryString) {
		this.queryString = queryString;

		return this;
	}

	/**
	 * The Hibernate {@link SessionFactory} to execute the query against.
	 * @param sessionFactory the session factory
	 * @return this instance for method chaining
	 * @see HibernateCursorItemReader#setSessionFactory(SessionFactory)
	 */
	public HibernateCursorItemReaderBuilder<T> sessionFactory(SessionFactory sessionFactory) {
		this.sessionFactory = sessionFactory;

		return this;
	}

	/**
	 * Indicator for whether to use a {@link org.hibernate.StatelessSession}
	 * (<code>true</code>) or a {@link org.hibernate.Session} (<code>false</code>).
	 * @param useStatelessSession Defaults to false
	 * @return this instance for method chaining
	 * @see HibernateCursorItemReader#setUseStatelessSession(boolean)
	 */
	public HibernateCursorItemReaderBuilder<T> useStatelessSession(boolean useStatelessSession) {
		this.useStatelessSession = useStatelessSession;

		return this;
	}

	/**
	 * Used to configure a {@link HibernateNativeQueryProvider}. This is ignored if
	 * @param nativeQuery {@link String} containing the native query.
	 * @return this instance for method chaining
	 */
	public HibernateCursorItemReaderBuilder<T> nativeQuery(String nativeQuery) {
		this.nativeQuery = nativeQuery;

		return this;
	}

	public HibernateCursorItemReaderBuilder<T> entityClass(Class<T> nativeClass) {
		this.nativeClass = nativeClass;

		return this;
	}

	/**
	 * Returns a fully constructed {@link HibernateCursorItemReader}.
	 * @return a new {@link HibernateCursorItemReader}
	 */
	public HibernateCursorItemReader<T> build() {
		Assert.state(this.fetchSize >= 0, "fetchSize must not be negative");
		Assert.state(this.sessionFactory != null, "A SessionFactory must be provided");

		if (this.saveState) {
			Assert.state(StringUtils.hasText(this.name), "A name is required when saveState is set to true.");
		}

		HibernateCursorItemReader<T> reader = new HibernateCursorItemReader<>();

		reader.setFetchSize(this.fetchSize);
		reader.setParameterValues(this.parameterValues);

		if (this.queryProvider != null) {
			reader.setQueryProvider(this.queryProvider);
		}
		else if (StringUtils.hasText(this.queryName)) {
			reader.setQueryName(this.queryName);
		}
		else if (StringUtils.hasText(this.queryString)) {
			reader.setQueryString(this.queryString);
		}
		else if (StringUtils.hasText(this.nativeQuery) && this.nativeClass != null) {
			HibernateNativeQueryProvider<T> provider = new HibernateNativeQueryProvider<>();
			provider.setSqlQuery(this.nativeQuery);
			provider.setEntityClass(this.nativeClass);

			try {
				provider.afterPropertiesSet();
			}
			catch (Exception e) {
				throw new IllegalStateException("Unable to initialize the HibernateNativeQueryProvider", e);
			}

			reader.setQueryProvider(provider);
		}
		else {
			throw new IllegalStateException("A HibernateQueryProvider, queryName, queryString, "
					+ "or both the nativeQuery and entityClass must be configured");
		}

		reader.setSessionFactory(this.sessionFactory);
		reader.setUseStatelessSession(this.useStatelessSession);
		reader.setCurrentItemCount(this.currentItemCount);
		reader.setMaxItemCount(this.maxItemCount);
		reader.setName(this.name);
		reader.setSaveState(this.saveState);

		return reader;
	}

}

相关信息

spring-batch 源码目录

相关文章

spring-batch HibernateItemWriterBuilder 源码

spring-batch HibernatePagingItemReaderBuilder 源码

spring-batch JdbcBatchItemWriterBuilder 源码

spring-batch JdbcCursorItemReaderBuilder 源码

spring-batch JdbcPagingItemReaderBuilder 源码

spring-batch JpaCursorItemReaderBuilder 源码

spring-batch JpaItemWriterBuilder 源码

spring-batch JpaPagingItemReaderBuilder 源码

spring-batch StoredProcedureItemReaderBuilder 源码

spring-batch package-info 源码

0  赞