/*

 * Copyright 2011 Tomas Zeman <tzeman@volny.cz>

 *

 * 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 fis.base.model

import org.squeryl.{Session, Table}

import org.squeryl.adapters.PostgreSqlAdapter

import org.squeryl.internals.StatementWriter

import scala.collection.mutable.HashSet

/**

 * Postgres adapter which enables multiple entities to share the same

 * sequence. 

 *

 * The implementation is based on Postgres adapter which is

 *

 *    Copyright 2010 Maxime Lévesque

 *

 * and licensed under the same license (Apache 2).

 */

class SeqIdPostgreSqlAdapter extends PostgreSqlAdapter {

  private val seqs = new HashSet[String]

  override def postCreateTable(t: Table[_],

    printSinkWhenWriteOnlyMode: Option[String => Unit]) = {

    val autoIncrementedFields = t.posoMetaData.fieldsMetaData.

      filter(_.isAutoIncremented).

      filter(f => !seqs.contains(f.sequenceName))

    for(fmd <-autoIncrementedFields) {

      val sw = new StatementWriter(false, this)

      sw.write("create sequence ", quoteName(fmd.sequenceName))

      seqs += fmd.sequenceName

      if(printSinkWhenWriteOnlyMode == None) {

        val st = Session.currentSession.connection.createStatement

        st.execute(sw.statement)

      }

      else

        printSinkWhenWriteOnlyMode.get.apply(sw.statement + ";")

    }

  }

}

// vim: set ts=2 sw=2 et: