org.apache.shardingsphere.transaction.annotation.ShardingTransactionType Java Examples

/**
 * Execute XA.
 *
 * @param count insert record count
 * @return transaction type
 */
@Transactional
@ShardingTransactionType(TransactionType.BASE)
public TransactionType insert(final int count) {
    return jdbcTemplate.execute("INSERT INTO t_order (user_id, status) VALUES (?, ?)", (PreparedStatementCallback<TransactionType>) preparedStatement -> {
        doInsert(count, preparedStatement);
        return TransactionTypeHolder.get();
    });
}
 

private ShardingTransactionType getAnnotation(final JoinPoint joinPoint) {
    Method method = ((MethodSignature) joinPoint.getSignature()).getMethod();
    ShardingTransactionType result = method.getAnnotation(ShardingTransactionType.class);
    if (null == result) {
        result = method.getDeclaringClass().getAnnotation(ShardingTransactionType.class);
    }
    return result;
}
 

private ShardingTransactionType getAnnotation(final JoinPoint joinPoint) {
    Method method = ((MethodSignature) joinPoint.getSignature()).getMethod();
    ShardingTransactionType result = method.getAnnotation(ShardingTransactionType.class);
    if (null == result) {
        result = method.getDeclaringClass().getAnnotation(ShardingTransactionType.class);
    }
    return result;
}
 

/**
 * Execute XA.
 *
 * @param count insert record count
 * @return transaction type
 */
@Transactional
@ShardingTransactionType(TransactionType.XA)
public TransactionType insert(final int count) {
    return jdbcTemplate.execute("INSERT INTO t_order (user_id, status) VALUES (?, ?)", (PreparedStatementCallback<TransactionType>) preparedStatement -> {
        doInsert(count, preparedStatement);
        return TransactionTypeHolder.get();
    });
}
 

/**
 * Execute XA with exception.
 *
 * @param count insert record count
 */
@Transactional
@ShardingTransactionType(TransactionType.XA)
public void insertFailed(final int count) {
    jdbcTemplate.execute("INSERT INTO t_order (user_id, status) VALUES (?, ?)", (PreparedStatementCallback<TransactionType>) preparedStatement -> {
        doInsert(count, preparedStatement);
        throw new SQLException("mock transaction failed");
    });
}
 

/**
 * process failure, XA transaction will be rollback.
 */
@Override
@ShardingTransactionType(TransactionType.XA)
@Transactional
public void processFailure() {
    springPojoService.processFailure();
}
 

/**
 * process success, XA transaction will be committed.
 */
@Override
@ShardingTransactionType(TransactionType.XA)
@Transactional
public void processSuccess() {
    springPojoService.processSuccess();
}
 

/**
 * process failure, XA transaction will be rollback.
 */
@Override
@ShardingTransactionType(TransactionType.BASE)
@Transactional
public void processFailure() {
    springPojoService.processFailure();
}
 

/**
 * process success, XA transaction will be committed.
 */
@Override
@ShardingTransactionType(TransactionType.BASE)
@Transactional
public void processSuccess() {
    springPojoService.processSuccess();
}
 

/**
 * Execute XA with exception.
 *
 * @param count insert record count
 */
@Transactional
@ShardingTransactionType(TransactionType.BASE)
public void insertFailed(final int count) {
    jdbcTemplate.execute("INSERT INTO t_order (user_id, status) VALUES (?, ?)", (PreparedStatementCallback<TransactionType>) preparedStatement -> {
        doInsert(count, preparedStatement);
        throw new SQLException("mock transaction failed");
    });
}
 

/**
 * Execute XA.
 *
 * @param count insert record count
 * @return transaction type
 */
@Transactional
@ShardingTransactionType(TransactionType.XA)
public TransactionType insert(final int count) {
    return jdbcTemplate.execute("INSERT INTO t_order (user_id, status) VALUES (?, ?)", (PreparedStatementCallback<TransactionType>) preparedStatement -> {
        doInsert(count, preparedStatement);
        return TransactionTypeHolder.get();
    });
}
 

/**
 * Execute XA with exception.
 *
 * @param count insert record count
 */
@Transactional
@ShardingTransactionType(TransactionType.XA)
public void insertFailed(final int count) {
    jdbcTemplate.execute("INSERT INTO t_order (user_id, status) VALUES (?, ?)", (PreparedStatementCallback<TransactionType>) preparedStatement -> {
        doInsert(count, preparedStatement);
        throw new SQLException("mock transaction failed");
    });
}
 

@Override
public Object invoke(final MethodInvocation methodInvocation) throws Throwable {
    ShardingTransactionType shardingTransactionType = getAnnotation(methodInvocation);
    Objects.requireNonNull(shardingTransactionType, "could not found sharding transaction type annotation");
    TransactionType preTransactionType = TransactionTypeHolder.get();
    TransactionTypeHolder.set(shardingTransactionType.value());
    try {
        return methodInvocation.proceed();
    } finally {
        TransactionTypeHolder.clear();
        if (null != preTransactionType) {
            TransactionTypeHolder.set(preTransactionType);
        }
    }
}
 

ShardingTransactionTypeAdvisor() {
    Pointcut classPointcut = new ComposablePointcut(AnnotationMatchingPointcut.forClassAnnotation(ShardingTransactionType.class));
    Pointcut methodPointcut = new ComposablePointcut(AnnotationMatchingPointcut.forMethodAnnotation(ShardingTransactionType.class));
    transactionTypePointcut = new ComposablePointcut(classPointcut).union(methodPointcut);
    transactionTypeInterceptor = new ShardingTransactionTypeInterceptor();
    setOrder(Ordered.LOWEST_PRECEDENCE - 1);
}
 

/**
 * process failure, XA transaction will be rollback.
 */
@ShardingTransactionType(TransactionType.BASE)
@Transactional
public void processFailure() {
    springPojoService.processFailure();
}
 

private ShardingTransactionType getAnnotation(final MethodInvocation invocation) {
    Objects.requireNonNull(invocation.getThis());
    Class<?> targetClass = AopUtils.getTargetClass(invocation.getThis());
    ShardingTransactionType result = getMethodAnnotation(invocation, targetClass);
    return null != result ? result : targetClass.getAnnotation(ShardingTransactionType.class);
}
 

@ShardingTransactionType(TransactionType.BASE)
public void testChangeTransactionTypeToBASE() {
}
 

@ShardingTransactionType(TransactionType.XA)
public void testChangeTransactionTypeToXA() {
}
 

@ShardingTransactionType
public void testChangeTransactionTypeToLOCAL() {
}
 

private ShardingTransactionType getMethodAnnotation(final MethodInvocation invocation, final Class<?> targetClass) {
    Method specificMethod = ClassUtils.getMostSpecificMethod(invocation.getMethod(), targetClass);
    final Method userDeclaredMethod = BridgeMethodResolver.findBridgedMethod(specificMethod);
    return userDeclaredMethod.getAnnotation(ShardingTransactionType.class);
}
 

/**
 * Set transaction type before transaction begin.
 *
 * @param joinPoint join point
 */
@Before(value = "shardingTransactionalProxyPointCut()")
public void setTransactionTypeBeforeTransaction(final JoinPoint joinPoint) {
    ShardingTransactionType shardingTransactionType = getAnnotation(joinPoint);
    transactionManagerHandler.switchTransactionType(shardingTransactionType.value());
}
 

@ShardingTransactionType(TransactionType.BASE)
public void testChangeTransactionTypeToBASE() {
    assertThat(TransactionTypeHolder.get(), is(TransactionType.BASE));
}
 

@ShardingTransactionType(TransactionType.XA)
public void testChangeTransactionTypeToXA() {
    assertThat(TransactionTypeHolder.get(), is(TransactionType.XA));
}
 

@ShardingTransactionType
public void testChangeTransactionTypeToLOCAL() {
    assertThat(TransactionTypeHolder.get(), is(TransactionType.LOCAL));
}
 

@ShardingTransactionType(TransactionType.BASE)
public void testChangeTransactionTypeToBASE() {
    assertThat(TransactionTypeHolder.get(), is(TransactionType.BASE));
}
 

/**
 * process success, XA transaction will be committed.
 */
@ShardingTransactionType(TransactionType.BASE)
@Transactional
public void processSuccess() {
    springPojoService.processSuccess();
}
 

/**
 * process success.
 */
@ShardingTransactionType(TransactionType.BASE)
@Transactional
public void processSuccess() {
    jpaCommonService.processSuccess();
}
 

/**
 * process failure.
 */
@ShardingTransactionType(TransactionType.BASE)
@Transactional
public void processFailure() {
    jpaCommonService.processFailure();
}
 

/**
 * process success.
 */
@ShardingTransactionType(TransactionType.BASE)
@Transactional
public void processSuccess() {
    jpaCommonService.processSuccess();
}
 

/**
 * process failure.
 */
@ShardingTransactionType(TransactionType.BASE)
@Transactional
public void processFailure() {
    jpaCommonService.processFailure();
}