com.github.huxinghai1988.AbstractClient Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of sidekiq-client-plugin Show documentation
sidekiq queue client support java language
There is a newer version: 1.5.2
Show newest version
package com.github.huxinghai1988;

import redis.clients.jedis.Jedis;

import java.util.Date;
import java.util.Set;

/**
 * Created by huxinghai on 15/4/25.
 */
public abstract class AbstractClient implements Client {

    private String namespace;
    private final Jedis redis;

    protected AbstractClient(String namespace, Jedis redis) {
        this.namespace = namespace;
        this.redis = redis;
    }

    protected AbstractClient(Jedis redis){
        this.redis = redis;
    }
    
    public String enqueue(Worker w){
        if(w.getEnqueued_at() > 0){
            joinSchedule(w);
        }else{
            joinQueue(w);
        }
        return w.getJid();
    }

    public Worker find(String jid){
        Set list = this.redis.zrange(this.namespaceKey("schedule"), 0, -1);
        for(String json : list){
            Worker w = Worker.parse(json);
            if(w != null && w.getJid().equals(jid))
                return w;
        }
        return null;
    }
    
    public long delete(String jid){
        String json = findJson("schedule", jid);
        String key = this.namespaceKey("schedule");
        System.out.println("delete(String) - key: " + key);
        System.out.println("delete(String) - json: " + json);
        if(json != null && json != "") {
            return this.redis.zrem(key, json);
        } else {
            return deleteRetry(jid);
        }        
    }
    
    public RetryWork findRetry(String jid) {
        if(jid == null || jid.isEmpty()) return null;
        Set list = this.redis.zrange(this.namespaceKey("retry"), 0, -1);
        for(String json : list){
            if(json != null && !json.isEmpty()) {
                RetryWork w = RetryWork.parse(json);
                if(w != null && w.getJid().equals(jid))
                    return w;
            }
        }
        return null;
    }
    
    public String findJson(String name, String jid) {
        if(jid == null || jid.isEmpty()) return null;
        Set list = this.redis.zrange(this.namespaceKey(name), 0, -1);
        for(String json : list){
            if(json != null && !json.isEmpty()) {
                Worker w = Worker.parse(json);
                if(w != null && w.getJid().equals(jid))
                    return json;
            }
        }
        return null;
    }
    
    public long deleteRetry(String jid){
        String json = findJson("retry", jid);
        String key = this.namespaceKey("retry");
        System.out.println("deleteRetry(String) - key: " + key);
        System.out.println("deleteRetry(String) - json: " + json);
        if(json != null) {
            return this.redis.zrem(key, json);
        }
        return this.redis.zcard(key);
    }

    private void joinQueue(Worker w){
        w.setEnqueued_at(new Date().getTime() / 1000);
        this.redis.lpush(this.namespaceKey("queue:"+ w.getQueue()), w.toJSON());
        this.redis.sadd(this.namespaceKey("queues"), w.getQueue());
    }

    private void joinSchedule(Worker w){
        w.setEnqueued_at(w.getEnqueued_at() / 1000);
        this.redis.zadd(this.namespaceKey("schedule"), w.getEnqueued_at(), w.toJSON());
    }

    private String namespaceKey(String name){
        if(this.namespace == null || this.namespace.isEmpty()){
            return name;
        }
        return this.namespace + ":"+ name;
    }
    
}