com.wandoulabs.nedis.NedisClientPoolBuilder Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of nedis-client Show documentation
There is a newer version: 0.1.1
/**
 * Copyright (c) 2015 Wandoujia Inc.
 *
 * 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 com.wandoulabs.nedis;

import static com.wandoulabs.nedis.util.NedisUtils.toBytes;
import io.netty.channel.Channel;
import io.netty.channel.EventLoopGroup;

import java.lang.management.ManagementFactory;
import java.net.InetSocketAddress;
import java.net.SocketAddress;

import com.wandoulabs.nedis.util.AbstractNedisBuilder;

/**
 * @author Apache9
 */
public class NedisClientPoolBuilder extends AbstractNedisBuilder {

    private byte[] password;

    private int database;

    private byte[] clientName;

    private int maxPooledConns = Math.max(2, 2 * ManagementFactory.getOperatingSystemMXBean()
            .getAvailableProcessors());

    private boolean exclusive;

    private SocketAddress remoteAddress;

    @Override
    public NedisClientPoolBuilder group(EventLoopGroup group) {
        super.group(group);
        return this;
    }

    @Override
    public NedisClientPoolBuilder channel(Class channelClass) {
        super.channel(channelClass);
        return this;
    }

    @Override
    public NedisClientPoolBuilder timeoutMs(long timeoutMs) {
        super.timeoutMs(timeoutMs);
        return this;
    }

    public NedisClientPoolBuilder password(String password) {
        this.password = toBytes(password);
        return this;
    }

    public NedisClientPoolBuilder database(int database) {
        this.database = database;
        return this;
    }

    public NedisClientPoolBuilder clientName(String clientName) {
        this.clientName = toBytes(clientName);
        return this;
    }

    public NedisClientPoolBuilder maxPooledConns(int maxPooledConns) {
        this.maxPooledConns = maxPooledConns;
        return this;
    }

    public NedisClientPoolBuilder exclusive(boolean exclusive) {
        this.exclusive = exclusive;
        return this;
    }

    public boolean exclusive() {
        return exclusive;
    }

    public NedisClientPoolBuilder remoteAddress(String host) {
        return remoteAddress(host, 6379);
    }

    public NedisClientPoolBuilder remoteAddress(String inetHost, int inetPort) {
        return remoteAddress(new InetSocketAddress(inetHost, inetPort));
    }

    public NedisClientPoolBuilder remoteAddress(SocketAddress remoteAddress) {
        this.remoteAddress = remoteAddress;
        return this;
    }

    private void validate() {
        validateGroupConfig();
        if (remoteAddress == null) {
            throw new IllegalArgumentException("remoteAddress is not set");
        }
    }

    public NedisClientPoolImpl build() {
        validate();
        return new NedisClientPoolImpl(group, channelClass, timeoutMs, remoteAddress, password,
                database, clientName, maxPooledConns, exclusive);
    }

    public static NedisClientPoolBuilder create() {
        return new NedisClientPoolBuilder();
    }
}