hmily-master.zip

/* * Copyright 2017-2021 Dromara.org * * 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 org.dromara.hmily.repository.database.manager; import com.google.common.base.Splitter; import com.google.common.collect.Maps; import com.zaxxer.hikari.HikariDataSource; import lombok.extern.slf4j.Slf4j; import org.dromara.hmily.common.exception.HmilyRuntimeException; import org.dromara.hmily.common.utils.CollectionUtils; import org.dromara.hmily.config.api.ConfigEnv; import org.dromara.hmily.config.api.entity.HmilyConfig; import org.dromara.hmily.config.api.entity.HmilyDatabaseConfig; import org.dromara.hmily.repository.spi.HmilyRepository; import org.dromara.hmily.repository.spi.entity.HmilyDataSnapshot; import org.dromara.hmily.repository.spi.entity.HmilyInvocation; import org.dromara.hmily.repository.spi.entity.HmilyLock; import org.dromara.hmily.repository.spi.entity.HmilyParticipant; import org.dromara.hmily.repository.spi.entity.HmilyParticipantUndo; import org.dromara.hmily.repository.spi.entity.HmilyTransaction; import org.dromara.hmily.repository.spi.exception.HmilyRepositoryException; import org.dromara.hmily.serializer.spi.HmilySerializer; import org.dromara.hmily.serializer.spi.exception.HmilySerializerException; import javax.sql.DataSource; import java.sql.Connection; import java.sql.PreparedStatement; import java.sql.ResultSet; import java.sql.ResultSetMetaData; import java.sql.SQLException; import java.util.ArrayList; import java.util.Arrays; import java.util.Collection; import java.util.Collections; import java.util.Date; import java.util.LinkedList; import java.util.List; import java.util.Map; import java.util.Objects; import java.util.Optional; import java.util.stream.Collectors; /** * The type Abstract hmily database. */ @Slf4j public abstract class AbstractHmilyDatabase implements HmilyRepository { /** * The constant INSERT_HMILY_TRANSACTION. */ protected static final String INSERT_HMILY_TRANSACTION = "INSERT INTO hmily_transaction_global (trans_id, app_name, status, trans_type, " + "retry, version, create_time, update_time) VALUES(?, ?, ?, ?, ?, ?, ?, ?)"; /** * The constant SELECT_HMILY_TRANSACTION_COMMON. */ protected static final String SELECT_HMILY_TRANSACTION_COMMON = "select trans_id, app_name, status, trans_type, retry, version from hmily_transaction_global "; /** * The constant SELECT_HMILY_TRANSACTION_DELAY. */ protected static final String SELECT_HMILY_TRANSACTION_DELAY = SELECT_HMILY_TRANSACTION_COMMON + " where update_time < ? and app_name = ?"; /** * The constant SELECT_HMILY_TRANSACTION_WITH_TRANS_ID. */ protected static final String SELECT_HMILY_TRANSACTION_WITH_TRANS_ID = SELECT_HMILY_TRANSACTION_COMMON + " where trans_id = ?"; /** * The constant UPDATE_HMILY_TRANSACTION_STATUS. */ protected static final String UPDATE_HMILY_TRANSACTION_STATUS = "update hmily_transaction_global set status=? where trans_id = ? "; /** * The constant UPDATE_HMILY_TRANSACTION_RETRY_LOCK. */ protected static final String UPDATE_HMILY_TRANSACTION_RETRY_LOCK = "update hmily_transaction_global set version =?, retry =? where trans_id = ? and version = ? "; /** * The constant UPDATE_HMILY_PARTICIPANT_LOCK. */ protected static final String UPDATE_HMILY_PARTICIPANT_LOCK = "update hmily_transaction_participant set version =?, retry =? where participant_id = ? and version = ? "; /** * The constant DELETE_HMILY_TRANSACTION. */ protected static final String DELETE_HMILY_TRANSACTION = "delete from hmily_transaction_global where trans_id = ? "; /** * The constant DELETE_HMILY_TRANSACTION_WITH_DATA. */ protected static final String DELETE_HMILY_TRANSACTION_WITH_DATA = "delete from hmily_transaction_global where update_time < ? and status = 4"; /** * The constant INSERT_HMILY_PARTICIPANT. */ protected static final String INSERT_HMILY_PARTICIPANT = "INSERT INTO hmily_transaction_participant (participant_id, participant_ref_id, trans_id, trans_type, status, app_name," + "role, retry, target_class, target_method, confirm_method, cancel_method, confirm_invocation, cancel_invocation, version, create_time, update_time)" + " VALUES(?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ? ,? , ? , ?, ?)"; /** * The constant SELECTOR_HMILY_PARTICIPANT_COMMON. */ protected static final String SELECTOR_HMILY_PARTICIPANT_COMMON = "select participant_id, participant_ref_id, trans_id, trans_type, status, app_name," + "role, retry, target_class, target_method, confirm_method, cancel_method, confirm_invocation, cancel_invocation, version from hmily_transaction_participant "; /** * The constant SELECTOR_HMILY_PARTICIPANT_WITH_KEY. */ protected static final String SELECTOR_HMILY_PARTICIPANT_WITH_KEY = SELECTOR_HMILY_PARTICIPANT_COMMON + " where participant_id = ?"; /** * The constant SELECTOR_HMILY_PARTICIPANT_WITH_PARTICIPANT_REF_ID. */ protected static final String SELECTOR_HMILY_PARTICIPANT_WITH_PARTICIPANT_REF_ID = SELECTOR_HMILY_PARTICIPANT_COMMON + " where participant_ref_id = ?"; /** * The constant SELECTOR_HMILY_PARTICIPANT_WITH_TRANS_ID. */ protected static final String SELECTOR_HMILY_PARTICIPANT_WITH_TRANS_ID = SELECTOR_HMILY_PARTICIPANT_COMMON + " where trans_id = ?"; /** * The constant EXIST_HMILY_PARTICIPANT_WITH_TRANS_ID. */ protected static final String EXIST_HMILY_PARTICIPANT_WITH_TRANS_ID = " select count(*) as count_total from hmily_transaction_participant where trans_id = ? "; /** * The constant SELECTOR_HMILY_PARTICIPANT_WITH_DELAY_AND_APP_NAME_TRANS_TYPE. */ protected static final String SELECTOR_HMILY_PARTICIPANT_WITH_DELAY_AND_APP_NAME_TRANS_TYPE = SELECTOR_HMILY_PARTICIPANT_COMMON + " where update_time < ? and app_name = ? and trans_type = ? and status not in (4, 8) "; /** * The constant UPDATE_HMILY_PARTICIPANT_STATUS. */ protected static final String UPDATE_HMILY_PARTICIPANT_STATUS = "update hmily_transaction_participant set status=? where participant_id = ? "; /** * The constant DELETE_HMILY_PARTICIPANT. */ protected static final String DELETE_HMILY_PARTICIPANT = "delete from hmily_transaction_participant where participant_id = ? "; /** * The constant DELETE_HMILY_PARTICIPANT_WITH_DATA. */ protected static final String DELETE_HMILY_PARTICIPANT_WITH_DATA = "delete from hmily_transaction_participant where update_time < ? and status = 4"; /** * The constant INSERT_HMILY_PARTICIPANT_UNDO. */ protected static final String INSERT_HMILY_PARTICIPANT_UNDO = "INSERT INTO hmily_participant_undo" + "(undo_id, participant_id, trans_id, resource_id, data_snapshot, status, create_time, update_time) " + " VALUES (?, ?, ?, ?, ?, ?, ?, ?)"; /** * The constant SELECTOR_HMILY_PARTICIPANT_UNDO_WITH_PARTICIPANT_ID. */ protected static final String SELECTOR_HMILY_PARTICIPANT_UNDO_WITH_PARTICIPANT_ID = " select undo_id, participant_id, trans_id, resource_id, data_snapshot, status " + "from hmily_participant_undo where participant_id =? "; /** * The constant REMOVE_HMILY_PARTICIPANT_UNDO. */