org.graylog2.indexer.searches.ChunkCommand Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of graylog2-server Show documentation
Graylog
There is a newer version: 6.1.4
/*
 * Copyright (C) 2020 Graylog, Inc.
 *
 * This program is free software: you can redistribute it and/or modify
 * it under the terms of the Server Side Public License, version 1,
 * as published by MongoDB, Inc.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
 * Server Side Public License for more details.
 *
 * You should have received a copy of the Server Side Public License
 * along with this program. If not, see
 * .
 */
package org.graylog2.indexer.searches;

import com.fasterxml.jackson.annotation.JsonAutoDetect;
import com.google.auto.value.AutoValue;
import com.google.common.base.Preconditions;
import org.graylog.plugins.views.search.searchfilters.model.UsedSearchFilter;
import org.graylog2.plugin.indexer.searches.timeranges.TimeRange;

import javax.annotation.Nullable;
import java.util.Collections;
import java.util.List;
import java.util.Optional;
import java.util.OptionalInt;
import java.util.OptionalLong;
import java.util.Set;

@AutoValue
@JsonAutoDetect
public abstract class ChunkCommand {
    public static final int NO_BATCHSIZE = -1;
    public static final int NO_LIMIT = -1;

    public abstract String query();

    public abstract Set indices();

    public abstract Optional> streams();

    public abstract Optional sorting();

    public abstract Optional filter();

    public abstract List filters();

    public abstract Optional range();

    public abstract OptionalInt limit();

    public abstract OptionalInt offset();

    public abstract List fields();

    public abstract OptionalLong batchSize();

    public abstract Optional sliceParams();

    public abstract boolean highlight();

    public abstract Builder toBuilder();

    public record SliceParams(int id, int max) {
        public SliceParams {
            Preconditions.checkArgument(id >= 0 && id <= 9, "slice id must be between 0 and 9");
            Preconditions.checkArgument(max >= 2 && max <= 10, "slice max must be between 2 and 10");
            Preconditions.checkArgument(max > id, "max must be greater than id");
        }
    }

    public static Builder builder() {
        return new AutoValue_ChunkCommand.Builder()
                .query("")
                .fields(Collections.emptyList())
                .filters(Collections.emptyList())
                .highlight(false);
    }

    @AutoValue.Builder
    public static abstract class Builder {
        public abstract Builder query(String query);

        public abstract Builder indices(Set indices);

        public abstract Builder streams(Set streams);

        public abstract Builder sorting(Sorting sorting);

        public abstract Builder filter(@Nullable String filter);

        public abstract Builder filters(List filters);

        public abstract Builder range(TimeRange range);

        public abstract Builder limit(int limit);

        public abstract Builder offset(int offset);

        public abstract Builder fields(List fields);

        public abstract Builder batchSize(int batchSize);

        public abstract Builder sliceParams(@Nullable SliceParams sliceParams);

        public Builder sliceParams(int id, int max) {
            return sliceParams(new SliceParams(id, max));
        }

        public abstract Builder highlight(boolean highlight);

        public abstract ChunkCommand build();
    }
}