tech.ytsaurus.spyt.wrapper.table.YtTableAttributes.scala Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of spark-yt-yt-wrapper_2.12 Show documentation
Spark over YTsaurus
The newest version!
package tech.ytsaurus.spyt.wrapper.table

import tech.ytsaurus.spyt.wrapper.cypress.{YtAttributes, YtCypressUtils}
import tech.ytsaurus.client.CompoundClient
import tech.ytsaurus.ysontree.YTreeNode

trait YtTableAttributes {
  self: YtCypressUtils =>

  def rowCount(path: String, transaction: Option[String] = None)(implicit yt: CompoundClient): Long = {
    attribute(path, YtAttributes.rowCount, transaction).longValue()
  }

  def rowCount(attrs: Map[String, YTreeNode]): Long = {
    attrs(YtAttributes.rowCount).longValue()
  }

  def chunkRowCount(path: String, transaction: Option[String] = None)(implicit yt: CompoundClient): Long = {
    attribute(path, YtAttributes.chunkRowCount, transaction).longValue()
  }

  def chunkRowCount(attrs: Map[String, YTreeNode]): Long = {
    attrs(YtAttributes.chunkRowCount).longValue()
  }

  def chunkCount(path: String, transaction: Option[String] = None)(implicit yt: CompoundClient): Int = {
    attribute(path, YtAttributes.chunkCount, transaction).longValue().toInt
  }

  def chunkCount(attrs: Map[String, YTreeNode]): Int = {
    attrs(YtAttributes.chunkCount).longValue().toInt
  }

  def optimizeMode(path: String, transaction: Option[String] = None)(implicit yt: CompoundClient): OptimizeMode = {
    optimizeMode(attribute(path, YtAttributes.optimizeFor, transaction))
  }

  def optimizeMode(node: YTreeNode): OptimizeMode = {
    OptimizeMode.fromName(node.stringValue())
  }

  def optimizeMode(attrs: Map[String, YTreeNode]): OptimizeMode = {
    optimizeMode(attrs(YtAttributes.optimizeFor))
  }

  def tableType(path: String, transaction: Option[String] = None)(implicit yt: CompoundClient): TableType = {
    tableType(attribute(path, YtAttributes.dynamic, transaction))
  }

  def tableType(node: YTreeNode): TableType = {
    if (node.boolValue()) TableType.Dynamic else TableType.Static
  }

  def tableType(attrs: Map[String, YTreeNode]): TableType = {
    tableType(attrs(YtAttributes.dynamic))
  }

  def dataWeight(attrs: Map[String, YTreeNode]): Long = {
   attrs(YtAttributes.dataWeight).longValue()
  }

  def tabletCount(attrs: Map[String, YTreeNode]): Long = {
    attrs(YtAttributes.tabletCount).longValue()
  }

  def tabletCount(path: String)(implicit yt: CompoundClient): Long = {
    attribute(path, YtAttributes.tabletCount).longValue()
  }
}