com.arangodb.internal.InternalArangoDatabaseOps.scala Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of persistence Show documentation
There is a newer version: 0.7.9
/*
 * Copyright 2020 ABSA Group Limited
 *
 * 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.arangodb.internal

import com.arangodb.async.ArangoDatabaseAsync
import com.arangodb.async.internal.{ArangoExecutorAsync, ArangoExecutorAsyncDestructor}
import com.arangodb.internal.velocystream.VstCommunicationDestructor
import com.arangodb.internal.velocystream.VstCommunicationDestructor.ConnectionParams
import org.apache.http.auth.UsernamePasswordCredentials
import za.co.absa.spline.common.rest.{HttpStatusException, RESTClient, RESTClientApacheHttpImpl}

import java.net.URI
import scala.concurrent.{ExecutionContext, Future}

class InternalArangoDatabaseOps(db: ArangoDatabaseAsync)(implicit ec: ExecutionContext) {

  /**
   * @see [[https://github.com/arangodb/arangodb-java-driver/issues/353]]
   */
  def restClient: RESTClient = {
    val asyncExecutable = db.asInstanceOf[ArangoExecuteable[ArangoExecutorAsync]]
    val ArangoExecutorAsyncDestructor(vstComm) = asyncExecutable.executor
    val VstCommunicationDestructor(ConnectionParams(hostDescription, user, password, maybeSslContext)) = vstComm
    val scheme = if (maybeSslContext.isDefined) "https" else "http"
    val host = hostDescription.getHost
    val port = hostDescription.getPort
    val maybeCredentials = Option(user).map(user => new UsernamePasswordCredentials(user, password))

    new RESTClientApacheHttpImpl(new URI(s"$scheme://$host:$port/_db/${db.name}"), maybeCredentials, maybeSslContext)
  }


  /**
   * @see [[https://github.com/arangodb/arangodb-java-driver/issues/353]]
   */
  def adminExecute(script: String)(implicit ec: ExecutionContext): Future[Unit] =
    restClient.post("_admin/execute", script).recover {
      case e: HttpStatusException if e.status == 404 =>
        sys.error("" +
          "'/_admin/execute' endpoint is unreachable. " +
          "Make sure ArangoDB server is running with '--javascript.allow-admin-execute' option. " +
          "See https://www.arangodb.com/docs/stable/programs-arangod-javascript.html#javascript-code-execution")
    }
}