-
Notifications
You must be signed in to change notification settings - Fork 2
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
- Loading branch information
Showing
6 changed files
with
145 additions
and
10 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
104 changes: 104 additions & 0 deletions
104
src/com/komanov/serialization/converters/KryoMacrosConverter.scala
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,104 @@ | ||
| package com.komanov.serialization.converters | ||
|
|
||
| import java.util.UUID | ||
|
|
||
| import com.esotericsoftware.kryo.io.{FastInput, FastOutput, Input, Output} | ||
| import com.esotericsoftware.kryo.{Kryo, Serializer => EsotericSerializer} | ||
| import com.komanov.serialization.converters.api.MyConverter | ||
| import com.komanov.serialization.domain._ | ||
|
|
||
| /** https://github.com/evolution-gaming/kryo-macros */ | ||
| object KryoMacrosConverter extends MyConverter { | ||
|
|
||
| import Implicits._ | ||
|
|
||
| private val pool = new ThreadLocal[KryoMacrosConverter] { | ||
| override def initialValue(): KryoMacrosConverter = | ||
| KryoMacrosConverter(new Kryo, new FastInput(), new FastOutput(131072, 131072)) | ||
| } | ||
|
|
||
| def toByteArray(site: Site): Array[Byte] = pool.get().write(site) | ||
|
|
||
| def fromByteArray(bytes: Array[Byte]): Site = pool.get().read[Site](bytes) | ||
|
|
||
| def toByteArray(event: SiteEvent): Array[Byte] = pool.get().write(event) | ||
|
|
||
| def siteEventFromByteArray(clazz: Class[_], bytes: Array[Byte]): SiteEvent = pool.get().read[SiteEvent](bytes) | ||
|
|
||
| private case class KryoMacrosConverter(kryo: Kryo, in: FastInput, out: FastOutput) { | ||
| def read[A <: AnyRef](bs: Array[Byte])(implicit s: EsotericSerializer[A], m: Manifest[A]): A = { | ||
| in.setBuffer(bs) | ||
| kryo.readObject(in, m.runtimeClass.asInstanceOf[Class[A]], s) | ||
| } | ||
|
|
||
| def write[A <: AnyRef](a: A)(implicit s: EsotericSerializer[A]): Array[Byte] = { | ||
| out.clear() | ||
| kryo.writeObject(out, a, s) | ||
| out.toBytes | ||
| } | ||
| } | ||
|
|
||
| private object Implicits { | ||
|
|
||
| import com.evolutiongaming.kryo.Serializer | ||
|
|
||
| implicit val uuidSerializer: EsotericSerializer[UUID] = new EsotericSerializer[UUID] { | ||
| override def write(kryo: Kryo, output: Output, t: UUID): Unit = { | ||
| output.writeLong(t.getMostSignificantBits) | ||
| output.writeLong(t.getLeastSignificantBits) | ||
| } | ||
| override def read(kryo: Kryo, input: Input, aClass: Class[UUID]): UUID = { | ||
| new UUID(input.readLong(), input.readLong()) | ||
| } | ||
| } | ||
|
|
||
| implicit val domainSerializer: EsotericSerializer[Domain] = Serializer.make[Domain] | ||
| implicit val metaTagSerializer: EsotericSerializer[MetaTag] = Serializer.make[MetaTag] | ||
| implicit val entryPointSerializer: EsotericSerializer[EntryPoint] = Serializer.makeCommon[EntryPoint] { | ||
| case 0 => Serializer.inner[DomainEntryPoint] | ||
| case 1 => Serializer.inner[FreeEntryPoint] | ||
| } | ||
| implicit val pageComponentDataSerializer: EsotericSerializer[PageComponentData] = Serializer.makeCommon[PageComponentData] { | ||
| case 0 => Serializer.inner[TextComponentData] | ||
| case 1 => Serializer.inner[ButtonComponentData] | ||
| case 2 => Serializer.inner[BlogComponentData] | ||
| } | ||
| implicit val pageComponentPositionSerializer: EsotericSerializer[PageComponentPosition] = Serializer.make[PageComponentPosition] | ||
| implicit val pageComponentSerializer: EsotericSerializer[PageComponent] = Serializer.make[PageComponent] | ||
| implicit val pageSerializer: EsotericSerializer[Page] = Serializer.make[Page] | ||
|
|
||
| implicit val siteSerializer: EsotericSerializer[Site] = Serializer.make[Site] | ||
|
|
||
| implicit val siteEventSerializer: EsotericSerializer[SiteEvent] = Serializer.makeCommon[SiteEvent] { | ||
| case 0 => Serializer.inner[SiteCreated] | ||
| case 1 => Serializer.inner[SiteNameSet] | ||
| case 2 => Serializer.inner[SiteDescriptionSet] | ||
| case 3 => Serializer.inner[SiteRevisionSet] | ||
| case 4 => Serializer.inner[SitePublished] | ||
| case 5 => Serializer.inner[SiteUnpublished] | ||
| case 6 => Serializer.inner[SiteFlagAdded] | ||
| case 7 => Serializer.inner[SiteFlagRemoved] | ||
| case 8 => Serializer.inner[DomainAdded] | ||
| case 9 => Serializer.inner[DomainRemoved] | ||
| case 10 => Serializer.inner[PrimaryDomainSet] | ||
| case 11 => Serializer.inner[DefaultMetaTagAdded] | ||
| case 12 => Serializer.inner[DefaultMetaTagRemoved] | ||
| case 13 => Serializer.inner[PageAdded] | ||
| case 14 => Serializer.inner[PageRemoved] | ||
| case 15 => Serializer.inner[PageNameSet] | ||
| case 16 => Serializer.inner[PageMetaTagAdded] | ||
| case 17 => Serializer.inner[PageMetaTagRemoved] | ||
| case 18 => Serializer.inner[PageComponentAdded] | ||
| case 19 => Serializer.inner[PageComponentRemoved] | ||
| case 20 => Serializer.inner[PageComponentPositionSet] | ||
| case 21 => Serializer.inner[PageComponentPositionReset] | ||
| case 22 => Serializer.inner[TextComponentDataSet] | ||
| case 23 => Serializer.inner[ButtonComponentDataSet] | ||
| case 24 => Serializer.inner[BlogComponentDataSet] | ||
| case 25 => Serializer.inner[DomainEntryPointAdded] | ||
| case 26 => Serializer.inner[FreeEntryPointAdded] | ||
| case 27 => Serializer.inner[EntryPointRemoved] | ||
| case 28 => Serializer.inner[PrimaryEntryPointSet] | ||
| } | ||
| } | ||
| } |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters