com.taobao.tdhs.client.easy.impl.SetImpl Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of tdhs-java-client Show documentation
A TDH_SOCKET Client For Java
There is a newer version: 0.4.1
/*
 * Copyright(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.
 *
 *  Authors:
 *    wentong 
 */

package com.taobao.tdhs.client.easy.impl;

import com.taobao.tdhs.client.common.TDHSCommon;
import com.taobao.tdhs.client.easy.Query;
import com.taobao.tdhs.client.easy.Set;
import com.taobao.tdhs.client.request.Get;
import com.taobao.tdhs.client.request.Update;
import org.jetbrains.annotations.Nullable;

/**
 * @author 文通
 * @since 11-12-27 下午4:12
 */
public class SetImpl implements Set {

    private String field = null;

    private final Get get;

    private final Update update;

    private final Query query;

    /**
     * Constructor SetImpl creates a new SetImpl instance.
     *
     * @param get    of type Get
     * @param update of type Update
     * @param query  of type Query
     */
    public SetImpl(Get get, Update update, Query query) {
        this.get = get;
        this.query = query;
        this.update = update;
    }

    /**
     * Method field ...
     *
     * @param field of type String
     *
     * @return Set
     */
    public Set field(String field) {
        if (this.field != null) {
            throw new IllegalArgumentException("can't field twice!");
        }
        this.field = field;
        return this;
    }

    /**
     * Method setIt ...
     *
     * @param value      of type String
     * @param updateFlag of type UpdateFlag
     *
     * @return Query
     */
    private Query setIt(String value, TDHSCommon.UpdateFlag updateFlag) {
        if (this.field == null) {
            throw new IllegalArgumentException("no field!");
        }
        this.get.getTableInfo().getFields().add(this.field);
        this.field = null;
        this.update.addEntry(updateFlag, value);
        return query;
    }

    /**
     * Method setIt ...
     *
     * @param value      of type byte[]
     * @param updateFlag of type UpdateFlag
     *
     * @return Query
     */
    private Query setIt(byte[] value, TDHSCommon.UpdateFlag updateFlag) {
        if (this.field == null) {
            throw new IllegalArgumentException("no field!");
        }
        this.get.getTableInfo().getFields().add(this.field);
        this.field = null;
        this.update.addEntry(updateFlag, value);
        return query;
    }

    /**
     * Method add ...
     *
     * @param value of type long
     *
     * @return Query
     */
    public Query add(long value) {
        return setIt(String.valueOf(value), TDHSCommon.UpdateFlag.TDHS_UPDATE_ADD);
    }

    /**
     * Method sub ...
     *
     * @param value of type long
     *
     * @return Query
     */
    public Query sub(long value) {
        return setIt(String.valueOf(value), TDHSCommon.UpdateFlag.TDHS_UPDATE_SUB);
    }

    /**
     * Method set ...
     *
     * @param value of type String
     *
     * @return Query
     */
    public Query set(@Nullable String value) {
        return setIt(value, TDHSCommon.UpdateFlag.TDHS_UPDATE_SET);
    }

    /**
     * Method set ...
     *
     * @param value of type long
     *
     * @return Query
     */
    public Query set(long value) {
        return set(String.valueOf(value));
    }

    /**
     * Method set ...
     *
     * @param value of type int
     *
     * @return Query
     */
    public Query set(int value) {
        return set(String.valueOf(value));
    }

    /**
     * Method set ...
     *
     * @param value of type short
     *
     * @return Query
     */
    public Query set(short value) {
        return set(String.valueOf(value));
    }

    /**
     * Method set ...
     *
     * @param value of type char
     *
     * @return Query
     */
    public Query set(char value) {
        return set(String.valueOf(value));
    }

    public Query set(byte value) {
        return set(String.valueOf(value));
    }

    public Query set(byte[] value) {
        return setIt(value, TDHSCommon.UpdateFlag.TDHS_UPDATE_SET);
    }

    /**
     * Method setNow ...
     *
     * @return Query
     */
    public Query setNow() {
        return setIt("", TDHSCommon.UpdateFlag.TDHS_UPDATE_NOW);
    }

    /**
     * Method setNull ...
     *
     * @return Query
     */
    public Query setNull() {
        return set((String) null);
    }
}