com.databend.client.QueryAffect Maven / Gradle / Ivy

Go to download
/*
 * 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.databend.client;

import com.fasterxml.jackson.annotation.JsonCreator;
import com.fasterxml.jackson.annotation.JsonProperty;
import com.fasterxml.jackson.annotation.JsonSubTypes;
import com.fasterxml.jackson.annotation.JsonTypeInfo;

import java.util.List;

import static com.google.common.base.MoreObjects.toStringHelper;

@JsonTypeInfo(use = JsonTypeInfo.Id.NAME, property = "type")
@JsonSubTypes({
        @JsonSubTypes.Type(value = QueryAffect.Create.class, name = "Create"),
        @JsonSubTypes.Type(value = QueryAffect.UseDB.class, name = "UseDB"),
        @JsonSubTypes.Type(value = QueryAffect.ChangeSettings.class, name = "ChangeSettings")
})
public abstract class QueryAffect {

    // TODO(zhihanz): currently create affect have not been implemented
    public static class Create extends QueryAffect {
        private final String kind;
        private final String name;
        private final boolean success;

        @JsonCreator
        public Create(
                @JsonProperty("kind") String kind,
                @JsonProperty("name") String name,
                @JsonProperty("success") boolean success
        ) {
            this.kind = kind;
            this.name = name;
            this.success = success;
        }

        // add builder

        @JsonProperty
        public String getKind() {
            return kind;
        }

        @JsonProperty
        public String getName() {
            return name;
        }

        @JsonProperty
        public boolean isSuccess() {
            return success;
        }

        @Override
        public String toString() {
            return toStringHelper(this)
                    .add("kind", kind)
                    .add("name", name)
                    .add("success", success)
                    .toString();
        }
    }

    public static class UseDB extends QueryAffect {
        private final String name;

        @JsonCreator
        public UseDB(@JsonProperty("name") String name) {
            this.name = name;
        }

        // add builder

        @JsonProperty
        public String getName() {
            return name;
        }

        @Override
        public String toString() {
            return toStringHelper(this)
                    .add("name", name)
                    .toString();
        }


    }

    public static class ChangeSettings extends QueryAffect {
        private final List keys;
        private final List values;
        private final List isGlobals;

        @JsonCreator
        public ChangeSettings(
                @JsonProperty("keys") List keys,
                @JsonProperty("values") List values,
                @JsonProperty("is_globals") List isGlobals
        ) {
            this.keys = keys;
            this.values = values;
            this.isGlobals = isGlobals;
        }
        @JsonProperty
        public List getKeys() {
            return keys;
        }

        @JsonProperty
        public List getValues() {
            return values;
        }

        @JsonProperty
        public List getIsGlobals() {
            return isGlobals;
        }

        @Override
        public String toString() {
            return toStringHelper(this)
                    .add("keys", keys)
                    .add("values", values)
                    .add("isGlobals", isGlobals)
                    .toString();
        }
    }

}