Lift Framework example source code file (MongoPk.scala)

This example Lift Framework source code file (MongoPk.scala) is included in the DevDaily.com "Java Source Code Warehouse" project. The intent of this project is to help you "Learn Java by Example" ^TM.

Java - Lift Framework tags/keywords

any, intpk, longpk, mongopk, mongopk, mongorecord, mongorecord, objectidpk, ownertype, ownertype, pktype, stringpk, util, uuidpk, uuidpk

The Lift Framework MongoPk.scala source code

/*
 * Copyright 2010 WorldWide Conferencing, LLC
 *
 * 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 net.liftweb
package mongodb
package record
package field

import common.{Box, Empty, Full}
import util.StringHelpers

import scala.util.Random
import java.util.UUID

import org.bson.types.ObjectId
import net.liftweb.record.field.{IntField, LongField, StringField}

/*
 * Trait for creating a "Primary Key" Field. These are all an id field
 * that is saved as _id in the database. Mix one of these into your
 * MongoRecord.
 */
trait MongoPk[PkType] {
  def id: PkType
  /** Override this to set default value of id field */
  def defaultIdValue: Any
}

trait ObjectIdPk[OwnerType <: MongoRecord[OwnerType]]
  extends MongoPk[ObjectIdField[OwnerType]]
{
  self: OwnerType =>

  def defaultIdValue = ObjectId.get

  object id extends ObjectIdField(this.asInstanceOf[OwnerType]) {
    override def name = "_id"
    override def defaultValue = defaultIdValue
    override def shouldDisplay_? = false
  }
}

trait UUIDPk[OwnerType <: MongoRecord[OwnerType]]
  extends MongoPk[UUIDField[OwnerType]]
{
  self: OwnerType =>

  def defaultIdValue = UUID.randomUUID

  object id extends UUIDField(this.asInstanceOf[OwnerType]) {
    override def name = "_id"
    override def defaultValue = defaultIdValue
    override def shouldDisplay_? = false
  }
}

trait StringPk[OwnerType <: MongoRecord[OwnerType]]
  extends MongoPk[StringField[OwnerType]]
{
  self: OwnerType =>

  def defaultIdValue = StringHelpers.randomString(32)

  object id extends StringField(this.asInstanceOf[OwnerType], 12) {
    override def name = "_id"
    override def defaultValue = defaultIdValue
    override def shouldDisplay_? = false
  }
}

trait IntPk[OwnerType <: MongoRecord[OwnerType]]
  extends MongoPk[IntField[OwnerType]]
{
  self: OwnerType =>

  def defaultIdValue = Random.nextInt

  object id extends IntField(this.asInstanceOf[OwnerType]) {
    override def name = "_id"
    override def defaultValue = defaultIdValue
    override def shouldDisplay_? = false
  }
}

trait LongPk[OwnerType <: MongoRecord[OwnerType]]
  extends MongoPk[LongField[OwnerType]]
{
  self: OwnerType =>

  def defaultIdValue = Random.nextLong

  object id extends LongField(this.asInstanceOf[OwnerType]) {
    override def name = "_id"
    override def defaultValue = defaultIdValue
    override def shouldDisplay_? = false
  }
}