com.convergencelabs.convergence.server.backend.datastore.ODocumentMapper.scala Maven / Gradle / Ivy

Go to download
/*
 * Copyright (c) 2019 - Convergence Labs, Inc.
 *
 * This file is part of the Convergence Server, which is released under
 * the terms of the GNU General Public License version 3 (GPLv3). A copy
 * of the GPLv3 should have been provided along with this file, typically
 * located in the "LICENSE" file, which is part of this source code package.
 * Alternatively, see  for the
 * full text of the GPLv3 license, if it was not provided.
 */

package com.convergencelabs.convergence.server.backend.datastore

import com.orientechnologies.orient.core.record.impl.ODocument

// scalastyle:off null
trait ODocumentMapper {
  protected[datastore] def valueOrNull[T >: Null](option: Option[T]): T = {
    option match {
      case Some(value) => value
      case None => null
    }
  }

  protected[datastore] def toOption[T >: Null](value: T): Option[T] = {
    value match {
      case value: Any => Some(value.asInstanceOf[T])
      case null => None
    }
  }

  protected[datastore] def validateDocumentClass(doc: ODocument, validClassNames: String*): Unit = {
    if (!validClassNames.contains(doc.getClassName)) {
      throw new IllegalArgumentException(s"The ODocument class must be one of '${validClassNames}': ${doc.getClassName}")
    }
  }

  protected[datastore] def mapOrNull[V >: Null, T](option: Option[T])(m: T => V): V = {
    option match {
      case Some(value) => m(value)
      case None => null
    }
  }
}