swaydb.core.build.BuildSerialiser.scala Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of core_2.13 Show documentation
core
The newest version!
/*
 * Copyright (c) 2020 Simer JS Plaha ([email protected] - @simerplaha)
 *
 * This file is a part of SwayDB.
 *
 * SwayDB is free software: you can redistribute it and/or modify
 * it under the terms of the GNU Affero General Public License as
 * published by the Free Software Foundation, either version 3 of the
 * License, or (at your option) any later version.
 *
 * SwayDB 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 Affero General Public License for more details.
 *
 * You should have received a copy of the GNU Affero General Public License
 * along with SwayDB. If not, see .
 *
 * Additional permission under the GNU Affero GPL version 3 section 7:
 * If you modify this Program or any covered work, only by linking or combining
 * it with separate works, the licensors of this Program grant you additional
 * permission to convey the resulting work.
 */

package swaydb.core.build

import java.nio.file.Path

import swaydb.core.util.CRC32
import swaydb.data.DataType
import swaydb.data.slice.Slice
import swaydb.data.util.ByteSizeOf

sealed trait BuildSerialiser {

  def write(build: Build.Info): Slice[Byte]

  def read(bytes: Slice[Byte], file: Path): Build.Info

}

object BuildSerialiser extends BuildSerialiser {
  override def write(buildInfo: Build.Info): Slice[Byte] = {
    val versionBytes = Slice.of[Byte](1 + 1 + ByteSizeOf.int * 3)

    versionBytes add Build.formatId

    versionBytes add buildInfo.dataType.id

    versionBytes addInt buildInfo.version.major
    versionBytes addInt buildInfo.version.minor
    versionBytes addInt buildInfo.version.revision

    val crc = CRC32.forBytes(versionBytes)

    val slice = Slice.of[Byte](ByteSizeOf.long + versionBytes.size)
    slice addLong crc
    slice addAll versionBytes
  }


  override def read(bytes: Slice[Byte], file: Path): Build.Info = {
    val crc = bytes.readLong()
    val versionBytes = bytes.drop(ByteSizeOf.long)
    val versionBytesCRC = CRC32.forBytes(versionBytes)

    assert(versionBytesCRC == crc, s"Invalid CRC. $versionBytesCRC != $crc in file: $file")

    val versionReader = versionBytes.createReader()
    val formatId = versionReader.get()
    assert(formatId == Build.formatId, s"Invalid formatId. $formatId != ${Build.formatId} in file: $file")

    val dataTypeId = versionReader.get()

    DataType(dataTypeId) match {
      case Some(dataType) =>
        val major = versionReader.readInt()
        val minor = versionReader.readInt()
        val revision = versionReader.readInt()
        val version = Build.Version(major = major, minor = minor, revision = revision)
        Build.Info(version, dataType)

      case None =>
        throw new IllegalStateException(s"Invalid data-type id '$dataTypeId' in file: $file.")
    }
  }
}