io.fsq.rogue.util.QueryLogger.scala Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of rogue-util_2.11 Show documentation

Foursquare's Scala DSL for constructing and executing find and modify commands against MongoDB

There is a newer version: 2.3.0

// Copyright 2017 Foursquare Labs Inc. All Rights Reserved.

package io.fsq.rogue.util

import io.fsq.rogue.{IndexCheckerLogger, Query}
import io.fsq.rogue.index.UntypedMongoIndex


trait QueryLogger[Result[_]] extends IndexCheckerLogger {
  def logCounter(name: String, count: Int = 1): Unit

  def log(
    query: Query[_, _, _],
    instanceName: String,
    msg: => String,
    timeMillis: Long
  ): Unit

  def onExecuteQuery[T](
    query: Query[_, _, _],
    instanceName: String,
    msg: => String,
    f: => Result[T]
  ): Result[T]

  def onExecuteWriteCommand[T](
    operationName: String,
    collectionName: String,
    instanceName: String,
    msg: => String,
    f: => Result[T]
  ): Result[T]

  def warn(query: Query[_, _, _], msg: => String): Unit
}

class DefaultQueryLogger[Result[_]] extends QueryLogger[Result] {
  override def logCounter(name: String, count: Int = 1): Unit = ()

  override def log(
    query: Query[_, _, _],
    instanceName: String,
    msg: => String,
    timeMillis: Long
  ): Unit = ()

  override def onExecuteQuery[T](
    query: Query[_, _, _],
    instanceName: String,
    msg: => String,
    f: => Result[T]
  ): Result[T] = f

  override def onExecuteWriteCommand[T](
    operationName: String,
    collectionName: String,
    instanceName: String,
    msg: => String,
    f: => Result[T]
  ): Result[T] = f

  override def logIndexMismatch(query: Query[_, _, _], msg: => String): Unit = ()

  override def logIndexHit(query: Query[_, _, _], index: UntypedMongoIndex): Unit = ()

  override def warn(query: Query[_, _, _], msg: => String): Unit = ()
}