content: /*
   * Licensed to the Apache Software Foundation (ASF) under one
   * or more contributor license agreements.  See the NOTICE file
   * distributed with this work for additional information
   * regarding copyright ownership.  The ASF licenses this file
   * to you 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.datastax.dse.driver.internal.core.cql.reactive;
  
  import com.datastax.dse.driver.api.core.cql.reactive.ReactiveResultSet;
  import com.datastax.oss.driver.api.core.cql.Statement;
  import com.datastax.oss.driver.api.core.session.Request;
  import com.datastax.oss.driver.api.core.type.reflect.GenericType;
  import com.datastax.oss.driver.internal.core.context.InternalDriverContext;
  import com.datastax.oss.driver.internal.core.cql.CqlRequestAsyncProcessor;
  import com.datastax.oss.driver.internal.core.session.DefaultSession;
  import com.datastax.oss.driver.internal.core.session.RequestProcessor;
  import net.jcip.annotations.ThreadSafe;
  
  @ThreadSafe
  public class CqlRequestReactiveProcessor
      implements RequestProcessor<Statement<?>, ReactiveResultSet> {
  
    public static final GenericType<ReactiveResultSet> REACTIVE_RESULT_SET =
        GenericType.of(ReactiveResultSet.class);
  
    private final CqlRequestAsyncProcessor asyncProcessor;
  
    public CqlRequestReactiveProcessor(CqlRequestAsyncProcessor asyncProcessor) {
      this.asyncProcessor = asyncProcessor;
    }
  
    @Override
    public boolean canProcess(Request request, GenericType<?> resultType) {
      return request instanceof Statement && resultType.equals(REACTIVE_RESULT_SET);
    }
  
    @Override
    public ReactiveResultSet process(
        Statement<?> request,
        DefaultSession session,
        InternalDriverContext context,
        String sessionLogPrefix) {
      return new DefaultReactiveResultSet(
          () -> asyncProcessor.process(request, session, context, sessionLogPrefix));
    }
  
    @Override
    public ReactiveResultSet newFailure(RuntimeException error) {
      return new FailedReactiveResultSet(error);
    }
  }
  