• Home
• com.github.mjakubowski84
• parquet4s-scalapb_3
• 2.20.0
• source code
• ScalaPBParquetSchemaResolver.scala

×

Project download

Many resources are needed to download a project. Please understand that we have to compensate our server costs. Thank you in advance.
Project price only 1 $

You can buy this project and download/modify it how often you want.

com.github.mjakubowski84.parquet4s.ScalaPBParquetSchemaResolver.scala Maven / Gradle / Ivy

package com.github.mjakubowski84.parquet4s

import com.github.mjakubowski84.parquet4s.ScalaPBImplicits.*
import org.apache.parquet.schema.{Type, Types}
import scalapb.{GeneratedMessage, GeneratedMessageCompanion}

import scala.jdk.CollectionConverters.*

class ScalaPBParquetSchemaResolver[T <: GeneratedMessage: GeneratedMessageCompanion] extends ParquetSchemaResolver[T] {
  private val cmp = implicitly[GeneratedMessageCompanion[T]]

  override def schemaName: Option[String] = Option(cmp.scalaDescriptor.name)

  override def resolveSchema(cursor: Cursor): List[Type] = {
    val md = cmp.scalaDescriptor
    Types
      .buildMessage()
      .addFields(md.fields)
      .named(md.fullName)
      .getFields
      .iterator()
      .asScala
      .toList
  }
}