/*
 * To change this license header, choose License Headers in Project Properties.
 * To change this template file, choose Tools | Templates
 * and open the template in the editor.
 */
package org.redkalex.cache.redis;

import org.redkale.net.AsyncConnection;
import org.redkale.net.AsyncGroup;
import org.redkale.net.client.Client;
import org.redkale.net.client.ClientAddress;
import org.redkale.util.Traces;

/**
 *
 * @author zhangjx
 */
public class RedisCacheClient extends Client<RedisCacheConnection, RedisCacheRequest, RedisCacheResult> {

    public RedisCacheClient(String appName, String name, AsyncGroup group, String key,
        ClientAddress address, int maxConns, int maxPipelines, RedisCacheReqAuth authReq, RedisCacheReqDB dbReq) {
        super(name, group, true, address, maxConns, maxPipelines, () -> new RedisCacheReqPing(), () -> new RedisCacheReqClose(), null); //maxConns
        RedisCacheReqClientName clientNameReq = new RedisCacheReqClientName(appName, name);
        if (authReq != null && dbReq != null) {
            this.authenticate = traceid -> {
                Traces.currentTraceid(traceid);
                return conn -> writeChannelBatch(conn, authReq.createTime(), dbReq.createTime(), clientNameReq.createTime())
                    .thenApply(v -> conn);
            };
        } else if (authReq != null) {
            this.authenticate = traceid -> {
                Traces.currentTraceid(traceid);
                return conn -> writeChannelBatch(conn, authReq.createTime(), clientNameReq.createTime())
                    .thenApply(v -> conn);
            };
        } else if (dbReq != null) {
            this.authenticate = traceid -> {
                Traces.currentTraceid(traceid);
                return conn -> writeChannelBatch(conn, dbReq.createTime(), clientNameReq.createTime())
                    .thenApply(v -> conn);
            };
        } else {
            this.authenticate = traceid -> {
                Traces.currentTraceid(traceid);
                return conn -> writeChannel(conn, clientNameReq.createTime())
                    .thenApply(v -> conn);
            };
        }
        this.readTimeoutSeconds = 3;
        this.writeTimeoutSeconds = 3;
    }

    @Override
    protected RedisCacheConnection createClientConnection(final int index, AsyncConnection channel) {
        return new RedisCacheConnection(this, index, channel);
    }

}