com.quincy.core.aspect.SynchronizedAop Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of quincy-redis Show documentation

There is a newer version: 1.1.1

package com.quincy.core.aspect;

import org.aspectj.lang.JoinPoint;
import org.aspectj.lang.annotation.Aspect;
import org.aspectj.lang.annotation.Pointcut;
import org.springframework.stereotype.Component;

import com.quincy.sdk.annotation.Synchronized;
import com.quincy.sdk.helper.AopHelper;

import redis.clients.jedis.Jedis;
import redis.clients.jedis.JedisPubSub;

@Aspect
@Component
public class SynchronizedAop extends JedisNeededBaseAop {
	private final static String SET_KEY = "SYNCHRONIZED";

	@Pointcut("@annotation(com.quincy.sdk.annotation.Synchronized)")
    public void pointCut() {}

	@Override
	protected Object handle(JoinPoint joinPoint, Jedis jedis) throws NoSuchMethodException, SecurityException, InterruptedException {
		Synchronized annotation = AopHelper.getAnnotation(joinPoint, Synchronized.class);
		String key = annotation.value();
		String channels = SET_KEY+"_"+key;
		while(true) {
			if(jedis.sismember(SET_KEY, key)||jedis.sadd(SET_KEY, key)==0) {
				jedis.subscribe(new MyListener(), channels);
			} else {//Successfully held the distributed lock.
				break;
			}
		}
		return key;
	}

	@Override
	protected void destroy(JoinPoint joinPoint, Jedis jedis, Object obj) {
		String key = obj.toString();
		jedis.srem(SET_KEY, key);
		jedis.publish(SET_KEY+"_"+key, "Finished");
	}

	public class MyListener extends JedisPubSub {
		public void onMessage(String channel, String message) {
			this.unsubscribe();
		}
	}

}