FixDatePrecision.scala

gistfile1.txt

package au.org.ala.biocache.tool

import au.org.ala.biocache.Config
import org.slf4j.LoggerFactory

object FixDatePrecision extends au.org.ala.biocache.cmd.Tool {

  def cmd = "fix-date-precision"

  def desc = "Cleanup old date precision values."

  protected val logger = LoggerFactory.getLogger("SyncLocTable")

  def main(args: Array[String]) {
    new FixDatePrecision().fix
    Config.persistenceManager.shutdown
  }
}

class FixDatePrecision {

  val monthPrecision = List("Year", "Month Range", "Year Range")
  var counter = 0
  var updates = 0

  def fix  {
    Config.persistenceManager.pageOverLocal("occ", (key, map, tokenRangeIdx) => {
      counter += 1
      val datePrecision = map.getOrElse("dateprecision_p", "")
      val eventDate = map.getOrElse("eventdate_p", "")
      val eventDateEnd = map.getOrElse("eventdateend_p", "")
      if(monthPrecision.contains(datePrecision)){
        Config.persistenceManager.putAsync(
          key,
          "occ",
          Map("month_p" -> "", "day_p" -> ""),
          false,
          false
        )
        updates += 1
      } else if(datePrecision == "Day Range") {
        if (eventDate == eventDateEnd && eventDate.length == 7) {
          Config.persistenceManager.putAsync(
            key,
            "occ",
            Map("day_p" -> ""),
            false,
            false
          )
          updates += 1
        } else if (eventDate != eventDateEnd && eventDate.length == 7) {
          Config.persistenceManager.putAsync(
            key,
            "occ",
            Map("month_p" -> "", "day_p" -> ""),
            false,
            false
          )
          updates += 1
        }
      }
      true
    }, 6, Array[String]("rowkey", "dateprecision_p", "eventdate_p", "eventdateend_p"))

    println(s"Rows read: $counter updates: $updates")
  }
}