quasar.physical.mongodb.package.scala Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of quasar-mongodb-internal_2.11 Show documentation
quasar-mongodb-internal
The newest version!
/*
 * Copyright 2014–2017 SlamData Inc.
 *
 * 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 quasar.physical

import slamdata.Predef._
import quasar.common.SortDir
import quasar.javascript.Js
import quasar.fs.PhysicalError
import quasar.namegen._
import quasar.qscript._
import quasar.contrib.pathy.AFile

import com.mongodb.async.AsyncBatchCursor
import org.bson.BsonValue
import scalaz._

package object mongodb {
  type BsonCursor         = AsyncBatchCursor[BsonValue]

  type MongoErrT[F[_], A] = EitherT[F, PhysicalError, A]

  type WorkflowExecErrT[F[_], A] = EitherT[F, WorkflowExecutionError, A]

  type JavaScriptPrg    = Vector[Js.Stmt]
  type JavaScriptLog[A] = Writer[JavaScriptPrg, A]

  // TODO: parameterize over label (SD-512)
  def freshName: State[NameGen, BsonField.Name] =
    quasar.namegen.freshName("tmp").map(BsonField.Name(_))

  // TODO use implicit class
  def sortDirToBson(sort: SortDir): Bson = sort match {
    case SortDir.Ascending => Bson.Int32(1)
    case SortDir.Descending => Bson.Int32(-1)
  }

  implicit def qScriptToQScriptTotal[T[_[_]]]: Injectable.Aux[fs.MongoQScriptCP[T]#M, QScriptTotal[T, ?]] =
    ::\::[QScriptCore[T, ?]](::/::[T, EquiJoin[T, ?], Const[ShiftedRead[AFile], ?]])

  implicit def qScriptCoreToQScript[T[_[_]]]: Injectable.Aux[QScriptCore[T, ?], fs.MongoQScriptCP[T]#M] =
    Injectable.inject[QScriptCore[T, ?], fs.MongoQScriptCP[T]#M]

  implicit def equiJoinToQScript[T[_[_]]]: Injectable.Aux[EquiJoin[T, ?], fs.MongoQScriptCP[T]#M] =
    Injectable.inject[EquiJoin[T, ?], fs.MongoQScriptCP[T]#M]

  implicit def shiftedReadToQScript[T[_[_]]]: Injectable.Aux[Const[ShiftedRead[AFile], ?], fs.MongoQScriptCP[T]#M] =
    Injectable.inject[Const[ShiftedRead[AFile], ?], fs.MongoQScriptCP[T]#M]

}