org.beangle.serializer.text.marshal.MapMarshaller.scala Maven / Gradle / Ivy

Go to download
/*
 * Beangle, Agile Development Scaffold and Toolkits.
 *
 * Copyright © 2005, The Beangle Software.
 *
 * This program is free software: you can redistribute it and/or modify
 * it under the terms of the GNU Lesser General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU Lesser General Public License for more details.
 *
 * You should have received a copy of the GNU Lesser General Public License
 * along with this program.  If not, see .
 */
package org.beangle.serializer.text.marshal

import java.{ util => ju }

import org.beangle.commons.collection.Properties
import org.beangle.serializer.text.io.StreamWriter
import org.beangle.serializer.text.mapper.Mapper

import Type.Type

abstract class AbstractMapMarshaller[T] extends Marshaller[T] {
  protected def writeItem(key: Boolean, item: Object, writer: StreamWriter, context: MarshallingContext): Unit = {
    val realitem = extractOption(item)
    if (realitem == null) {
      val name = if (key) "key" else "value"
      writer.startNode(name, classOf[Null])
    } else {
      val name = if (key) "key" else "value"
      writer.startNode(name, realitem.getClass)
      context.marshal(realitem)
    }
    writer.endNode()
  }

  override def targetType: Type = {
    Type.Collection
  }
}

class MapMarshaller(mapper: Mapper) extends AbstractMapMarshaller[collection.Map[Object, Object]] {

  def marshal(source: collection.Map[Object, Object], writer: StreamWriter, context: MarshallingContext): Unit = {
    val entryName = "entry"
    source.foreach { item =>
      writer.startNode(entryName, item.getClass)
      writeItem(true, item._1, writer, context)
      writeItem(false, item._2, writer, context)
      writer.endNode();
    }
  }

  override def support(clazz: Class[_]): Boolean = {
    clazz != classOf[Properties]
  }

}

class JavaMapMarshaller(mapper: Mapper) extends AbstractMapMarshaller[ju.Map[Object, Object]] {

  def marshal(source: ju.Map[Object, Object], writer: StreamWriter, context: MarshallingContext): Unit = {
    val entryName = "entry"
    val iterator = source.entrySet().iterator()
    while (iterator.hasNext) {
      val item = iterator.next()
      writer.startNode(entryName, item.getClass)
      writeItem(true, item.getKey, writer, context)
      writeItem(false, item.getValue, writer, context)
      writer.endNode()
    }
  }
}

class JavaMapEntryMarshaller(mapper: Mapper) extends AbstractMapMarshaller[ju.Map.Entry[Object, Object]] {

  def marshal(source: ju.Map.Entry[Object, Object], writer: StreamWriter, context: MarshallingContext): Unit = {
    writer.startNode("entry", source.getClass)
    writeItem(true, source.getKey, writer, context)
    writeItem(false, source.getValue, writer, context)
    writer.endNode()
  }

}