com.taobao.common.tedis.commands.DefaultStringCommands Maven / Gradle / Ivy

Go to download
/**
 * (C) 2011-2012 Alibaba Group Holding Limited.
 *
 * This program is free software; you can redistribute it and/or
 * modify it under the terms of the GNU General Public License
 * version 2 as published by the Free Software Foundation.
 *
 */
package com.taobao.common.tedis.commands;

import java.util.Collection;
import java.util.Collections;
import java.util.LinkedHashMap;
import java.util.List;
import java.util.Map;
import java.util.concurrent.TimeUnit;

import com.taobao.common.tedis.TedisException;
import com.taobao.common.tedis.binary.RedisCommands;
import com.taobao.common.tedis.core.BaseCommands;
import com.taobao.common.tedis.core.StringCommands;

/**
 * @author juxin.zj E-mail:[email protected]
 * @since 2011-7-25 上午11:34:39
 * @version 1.0
 */
public class DefaultStringCommands extends BaseCommands implements StringCommands {

    private RedisCommands redisCommands;

    public DefaultStringCommands() {
        setKeySerializer(stringSerializer);
        setValueSerializer(stringSerializer);
    }

    public DefaultStringCommands(RedisCommands redisCommands) {
        this.redisCommands = redisCommands;
        setKeySerializer(stringSerializer);
        setValueSerializer(stringSerializer);
    }

    public RedisCommands getRedisCommands() {
        return redisCommands;
    }

    public void setRedisCommands(RedisCommands redisCommands) {
        this.redisCommands = redisCommands;
    }

    public void init() {
        if (commandsProvider == null) {
            throw new TedisException("commandsProvider is null.please set a commandsProvider first.");
        }
        this.redisCommands = commandsProvider.getTedis();
    }

    @Override
    public Long append(final int namespace, final String key, final String value) {
        return (Long) doInTedis(namespace, new TedisBlock(redisCommands) {
            @Override
            public Object execute() {
                return commands.append(rawKey(namespace, key), rawString(value));
            }
        });
    }

    @Override
    public String get(final int namespace, final Object key) {
        return deserializeString((byte[]) doInTedis(namespace, new TedisBlock(redisCommands) {
            @Override
            public Object execute() {
                return commands.get(rawKey(namespace, key));
            }
        }));
    }

    @Override
    public String get(final int namespace, final String key, final long start, final long end) {
        return deserializeString((byte[]) doInTedis(namespace, new TedisBlock(redisCommands) {
            @Override
            public Object execute() {
                return commands.getRange(rawKey(namespace, key), (int) start, (int) end);
            }
        }));
    }

    @Override
    public String getAndSet(final int namespace, final String key, final String value) {
        return deserializeString((byte[]) doInTedis(namespace, new TedisBlock(redisCommands) {
            @Override
            public Object execute() {
                return commands.getSet(rawKey(namespace, key), rawString(value));
            }
        }));
    }

    @SuppressWarnings("unchecked")
    @Override
    public List multiGet(final int namespace, final Collection keys) {
        if (keys.isEmpty()) {
            return Collections.emptyList();
        }
        final byte[][] rawKeys = new byte[keys.size()][];
        int counter = 0;
        for (String hashKey : keys) {
            rawKeys[counter++] = rawKey(namespace, hashKey);
        }
        return deserializeStrings((List) doInTedis(namespace, new TedisBlock(redisCommands) {
            @Override
            public Object execute() {
                return commands.mGet(rawKeys);
            }
        }));
    }

    @Override
    public void multiSet(final int namespace, final Map m) {
        if (m.isEmpty()) {
            return;
        }
        final Map rawKeys = new LinkedHashMap(m.size());
        for (Map.Entry entry : m.entrySet()) {
            rawKeys.put(rawKey(namespace, entry.getKey()), rawString(entry.getValue()));
        }
        doInTedis(namespace, new TedisBlock(redisCommands) {
            @Override
            public Object execute() {
                commands.mSet(rawKeys);
                return null;
            }
        });
    }

    @Override
    public void multiSetIfAbsent(final int namespace, final Map m) {
        if (m.isEmpty()) {
            return;
        }

        final Map rawKeys = new LinkedHashMap(m.size());

        for (Map.Entry entry : m.entrySet()) {
            rawKeys.put(rawKey(namespace, entry.getKey()), rawString(entry.getValue()));
        }
        doInTedis(namespace, new TedisBlock(redisCommands) {
            @Override
            public Object execute() {
                commands.mSetNX(rawKeys);
                return null;
            }
        });

    }

    @Override
    public void set(final int namespace, final String key, final String value) {
        doInTedis(namespace, new TedisBlock(redisCommands) {
            @Override
            public Object execute() {
                commands.set(rawKey(namespace, key), rawString(value));
                return null;
            }
        });
    }

    @Override
    public void set(final int namespace, final String key, final String value, final long timeout, final TimeUnit unit) {
        doInTedis(namespace, new TedisBlock(redisCommands) {
            @Override
            public Object execute() {
                commands.setEx(rawKey(namespace, key), (int) unit.toSeconds(timeout), rawString(value));
                return null;
            }
        });
    }

    @Override
    public void set(final int namespace, final String key, final String value, final long offset) {
        doInTedis(namespace, new TedisBlock(redisCommands) {
            @Override
            public Object execute() {
                commands.setRange(rawKey(namespace, key), rawString(value), offset);
                return null;
            }
        });
    }

    @Override
    public Boolean setIfAbsent(final int namespace, final String key, final String value) {
        return (Boolean) doInTedis(namespace, new TedisBlock(redisCommands) {
            @Override
            public Object execute() {
                return commands.setNX(rawKey(namespace, key), rawString(value));
            }
        });
    }

    @Override
    public Long size(final int namespace, final String key) {
        return (Long) doInTedis(namespace, new TedisBlock(redisCommands) {
            @Override
            public Object execute() {
                return commands.strLen(rawKey(namespace, key));
            }
        });
    }

}