jp.co.cyberagent.aeromock.protobuf.ParsedProto.scala Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of aeromock-server_2.11 Show documentation

Aeromock server daemon.

There is a newer version: 0.2.5

package jp.co.cyberagent.aeromock.protobuf

import com.google.protobuf.CodedOutputStream
import jp.co.cyberagent.aeromock.{AeromockProtoDataViolationException, AeromockInvalidDataFileException, AeromockProtoTypeNotFoundException}

import scalaz._
import Scalaz._

/**
 *
 * @author stormcat24
 */
case class ParsedProto(
  types: Map[String, List[ProtoField]],
  dependencyTypes: Map[String, List[ProtoField]]
) {

  def buildData(targetTypeName: String, data: Map[Any, Any]): Array[Byte] = {

    val result = (types.get(targetTypeName) match {
      case None => throw new AeromockProtoTypeNotFoundException(targetTypeName)
      case Some(value) => value.map(f => {
        f.toValue(data, dependencyTypes)
      })
    }).sequenceU match {
      case Success(value) => value.flatten
      case Failure(errors) => {
        val errorLocations = errors.map(e => s"${e}").toList
        throw new AeromockProtoDataViolationException(errorLocations.mkString("\n"))
      }
    }

    val totalSize = result.foldLeft(0)((left, right) => left + right.serializedSize)
    val bytes = new Array[Byte](totalSize)
    val output = CodedOutputStream.newInstance(bytes)
    result.map(_.write(output))
    bytes
  }


}