io.datarouter.client.mysql.op.write.MysqlIncrementOp Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of datarouter-mysql Show documentation
datarouter-mysql
There is a newer version: 0.0.125
/*
 * Copyright © 2009 HotPads ([email protected])
 *
 * 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 io.datarouter.client.mysql.op.write;

import java.sql.Connection;
import java.sql.PreparedStatement;
import java.sql.SQLException;
import java.time.Duration;
import java.util.List;

import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import io.datarouter.client.mysql.field.codec.factory.MysqlFieldCodecFactory;
import io.datarouter.client.mysql.op.BaseMysqlOp;
import io.datarouter.client.mysql.op.Isolation;
import io.datarouter.client.mysql.sql.MysqlSqlFactory;
import io.datarouter.client.mysql.util.MysqlTool;
import io.datarouter.model.databean.Databean;
import io.datarouter.model.field.Field;
import io.datarouter.model.key.primary.PrimaryKey;
import io.datarouter.model.serialize.fielder.DatabeanFielder;
import io.datarouter.storage.client.DatarouterClients;
import io.datarouter.storage.config.Config;
import io.datarouter.storage.serialize.fieldcache.PhysicalDatabeanFieldInfo;
import io.datarouter.storage.tally.Tally;
import io.datarouter.storage.tally.Tally.TallyFielder;
import io.datarouter.storage.tally.TallyKey;

public class MysqlIncrementOp<
		PK extends PrimaryKey,
		D extends Databean,
		F extends DatabeanFielder>
extends BaseMysqlOp{
	private static final Logger logger = LoggerFactory.getLogger(MysqlIncrementOp.class);

	private final PhysicalDatabeanFieldInfo fieldInfo;
	private final MysqlFieldCodecFactory fieldCodecFactory;
	private final MysqlSqlFactory mysqlSqlFactory;
	private final String key;
	private final Long incrementAmount;
	private final Config config;

	public MysqlIncrementOp(
			DatarouterClients datarouterClients,
			PhysicalDatabeanFieldInfo fieldInfo,
			MysqlFieldCodecFactory fieldCodecFactory,
			MysqlSqlFactory mysqlSqlFactory,
			String key,
			Long incrementAmount,
			Config config){
		super(datarouterClients, fieldInfo.getClientId(), Isolation.readCommitted, false);
		this.fieldInfo = fieldInfo;
		this.fieldCodecFactory = fieldCodecFactory;
		this.mysqlSqlFactory = mysqlSqlFactory;
		this.key = key;
		this.incrementAmount = incrementAmount;
		this.config = config;
	}

	@Override
	public Long runOnce(){
		Long currentMillis = System.currentTimeMillis();
		Long expiresAt = config.findTtl()
				.map(Duration::toMillis)
				.map(ttlMs -> currentMillis + ttlMs)
				.orElse(null);
		Connection connection = getConnection();
		Tally databean = new Tally(key, incrementAmount, expiresAt);
		List> fields = fieldInfo.getFieldsWithValues(
				databean);
		var incrementSql = mysqlSqlFactory
				.createSql(getClientId(), fieldInfo.getTableName(), fieldInfo.getDisableIntroducer())
				.incrementTally(fieldInfo.getTableName(), key, incrementAmount, expiresAt);
		var selectSql = mysqlSqlFactory
				.createSql(getClientId(), fieldInfo.getTableName(), fieldInfo.getDisableIntroducer())
				.addSelectFromClause(fieldInfo.getTableName(), fields)
				.appendWhereClauseDisjunction(List.of(databean.getKey()));
		PreparedStatement incrementStatement = incrementSql.prepare(connection);
		PreparedStatement selectStatement = selectSql.prepare(connection);
		try{
			incrementStatement.execute();
			selectStatement.execute();
			connection.commit();
			List tallies = MysqlTool.getDatabeansFromSelectResult(
					fieldCodecFactory,
					fieldInfo.getDatabeanSupplier(),
					fieldInfo.getFields(),
					selectStatement);
			return tallies.isEmpty() ? null : tallies.getFirst().getTally();
		}catch(SQLException e){
			try{
				connection.rollback();
			}catch(SQLException exception){
				logger.error("error with rolling back sql.", exception);
			}
		}
		return null;
	}

}