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

Go to download
Show more of this group Show more artifacts with this name
Show all versions of beangle-serializer-text_2.13 Show documentation
The newest version!
/*
 * 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 org.beangle.commons.collection.page.Page
import org.beangle.commons.lang.Primitives
import org.beangle.commons.lang.reflect.BeanInfos
import org.beangle.serializer.text.io.StreamWriter
import org.beangle.serializer.text.mapper.Mapper
import org.beangle.serializer.text.marshal.Type.Type

class PageMarshaller(val mapper: Mapper) extends Marshaller[Page[Object]] {

  def marshal(source: Page[Object], writer: StreamWriter, context: MarshallingContext): Unit = {
    val sourceType = source.getClass
    val properties = context.getProperties(sourceType)
    val getters = BeanInfos.get(sourceType).readables
    properties foreach { property =>
      val itemType = Primitives.wrap(getters(property).clazz)
      writer.startNode(mapper.serializedMember(source.getClass, property), itemType)
      property match {
        case "pageIndex" =>
          context.marshal(Integer.valueOf(source.pageIndex))
        case "pageSize" =>
          context.marshal(Integer.valueOf(source.pageSize))
        case "totalPages" =>
          context.marshal(Integer.valueOf(source.totalPages))
        case "totalItems" =>
          context.marshal(Integer.valueOf(source.totalItems))
        case "items" =>
          context.marshal(source.items)
        case other: String => context.marshal(getters(other).getter.get.invoke(source))
      }
      writer.endNode()
    }
  }

  protected def writeItem(item: Object, writer: StreamWriter, context: MarshallingContext): Unit = {
    val realitem = extractOption(item)
    if (realitem == null) {
      writer.startNode(mapper.serializedClass(classOf[Null]), classOf[Null])
    } else {
      val name = mapper.serializedClass(realitem.getClass)
      writer.startNode(name, realitem.getClass)
      context.marshal(realitem)
    }
    writer.endNode()
  }

  override def support(clazz: Class[_]): Boolean = {
    classOf[Page[_]].isAssignableFrom(clazz)
  }

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