All Downloads are FREE. Search and download functionalities are using the official Maven repository.
  • Log In
  • Register

    • org.elasticsearch.xpack.esql.action.EsqlResponseImpl Maven / Gradle / Ivy

    There is a newer version: 8.16.1
    Show newest version
    /*
 * Copyright Elasticsearch B.V. and/or licensed to Elasticsearch B.V. under one
 * or more contributor license agreements. Licensed under the Elastic License
 * 2.0; you may not use this file except in compliance with the Elastic License
 * 2.0.
 */

package org.elasticsearch.xpack.esql.action;

import org.elasticsearch.xpack.core.esql.action.ColumnInfo;
import org.elasticsearch.xpack.core.esql.action.EsqlResponse;

import java.util.Iterator;
import java.util.List;

/** View over the response, that supports the xpack core transport API. */
public class EsqlResponseImpl implements EsqlResponse {

    private final EsqlQueryResponse queryResponse;
    private boolean closed;

    EsqlResponseImpl(EsqlQueryResponse queryResponse) {
        this.queryResponse = queryResponse;
    }

    @Override
    public List columns() {
        return queryResponse.columns();
    }

    @Override
    public Iterable> rows() {
        ensureOpen();
        return () -> {
            ensureOpen();
            return new DelegatingIterator<>(queryResponse.rows().iterator());
        };
    }

    @Override
    public Iterable column(int columnIndex) {
        ensureOpen();
        return () -> {
            ensureOpen();
            return new DelegatingIterator<>(queryResponse.column(columnIndex));
        };
    }

    @Override
    public void close() {
        setClosedState();
    }

    public void setClosedState() {
        closed = true;
    }

    private void ensureOpen() {
        if (closed || queryResponse.hasReferences() == false) {
            throw new IllegalStateException("closed");
        }
    }

    @Override
    public String toString() {
        return "EsqlResponse[response=" + queryResponse + "]";
    }

    /** A delegating iterator, that first checks the closed state before delegating. */
    final class DelegatingIterator implements Iterator {
        final Iterator delegate;

        DelegatingIterator(Iterator delegate) {
            this.delegate = delegate;
        }

        @Override
        public boolean hasNext() {
            ensureOpen();
            return delegate.hasNext();
        }

        @Override
        public T next() {
            ensureOpen();
            return delegate.next();
        }
    }
}
    
    
    
    

    




    
    
    © 2015 - 2024 Weber Informatics LLC | Privacy Policy