com.jporm.rx.query.save.SaveQueryImpl Maven / Gradle / Ivy

Go to download
/*******************************************************************************
 * Copyright 2013 Francesco Cina'
 *
 * 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
 *
 *   http://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 com.jporm.rx.query.save;

import com.jporm.commons.core.inject.ClassTool;
import com.jporm.commons.core.query.SqlFactory;
import com.jporm.commons.core.query.cache.SqlCache;
import com.jporm.commons.core.query.save.SaveQueryBase;
import com.jporm.persistor.generator.Persistor;
import com.jporm.rx.session.SqlExecutor;
import com.jporm.sql.dialect.DBProfile;
import com.jporm.types.io.GeneratedKeyReader;
import com.jporm.types.io.ResultSet;

import io.reactivex.Single;

/**
 *
 * @author Francesco Cina
 *
 *         10/lug/2011
 */
public class SaveQueryImpl extends SaveQueryBase implements SaveQuery {

	private final BEAN bean;
	private final SqlExecutor sqlExecutor;
	private final ClassTool ormClassTool;

	public SaveQueryImpl(final BEAN bean, final Class clazz, final ClassTool ormClassTool, final SqlCache sqlCache, final SqlExecutor sqlExecutor,
			final SqlFactory sqlFactory, DBProfile dbProfile) {
		super(clazz, sqlCache);
		this.bean = bean;
		this.ormClassTool = ormClassTool;
		this.sqlExecutor = sqlExecutor;
	}


	@Override
	public Single execute() {
		final Persistor persistor = ormClassTool.getPersistor();

		// CHECK IF OBJECT HAS A 'VERSION' FIELD and increase it
		final BEAN clonedBean = persistor.increaseVersion(persistor.clone(bean), true);
		final boolean useGenerator = persistor.useGenerators(clonedBean);
		final String sql = getCacheableQuery(useGenerator);

		if (!useGenerator) {
			final String[] keys = ormClassTool.getDescriptor().getAllColumnJavaNames();
			return sqlExecutor.update(sql, statement -> persistor.setBeanValuesToStatement(keys, clonedBean, statement, 0)).map(result -> clonedBean);
		} else {
			final GeneratedKeyReader generatedKeyExtractor = GeneratedKeyReader.get(ormClassTool.getDescriptor().getAllGeneratedColumnDBNames(),
					(final ResultSet generatedKeyResultSet, Integer affectedRows) -> {
						BEAN result = clonedBean;
						if (generatedKeyResultSet.hasNext()) {
							result = persistor.updateGeneratedValues(generatedKeyResultSet.next(), result);
						}
						return result;
					});
			final String[] keys = ormClassTool.getDescriptor().getAllNotGeneratedColumnJavaNames();
			return sqlExecutor.update(sql, statement -> persistor.setBeanValuesToStatement(keys, clonedBean, statement, 0), generatedKeyExtractor);
		}
	}

}