com.cognite.sdk.scala.v1.resources.dataSets.scala Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of cognite-sdk-scala_2.13 Show documentation
Scala SDK for Cognite Data Fusion.
The newest version!
// Copyright 2020 Cognite AS
// SPDX-License-Identifier: Apache-2.0

package com.cognite.sdk.scala.v1.resources

import com.cognite.sdk.scala.common._
import com.cognite.sdk.scala.v1._
import sttp.client3._
import io.circe.{Decoder, Encoder}
import io.circe.generic.semiauto.{deriveDecoder, deriveEncoder}

class DataSets[F[_]](val requestSession: RequestSession[F])
    extends WithRequestSession[F]
    with Readable[DataSet, F]
    with Create[DataSet, DataSetCreate, F]
    with RetrieveByIdsWithIgnoreUnknownIds[DataSet, F]
    with RetrieveByExternalIdsWithIgnoreUnknownIds[DataSet, F]
    with Filter[DataSet, DataSetFilter, F]
    with Search[DataSet, DataSetQuery, F]
    with UpdateById[DataSet, DataSetUpdate, F]
    with UpdateByExternalId[DataSet, DataSetUpdate, F] {
  import DataSets._
  override val baseUrl = uri"${requestSession.baseUrl}/datasets"

  override private[sdk] def readWithCursor(
      cursor: Option[String],
      limit: Option[Int],
      partition: Option[Partition]
  ): F[ItemsWithCursor[DataSet]] =
    filterWithCursor(DataSetFilter(), cursor, limit, None, None)

  override def retrieveByIds(
      ids: Seq[Long],
      ignoreUnknownIds: Boolean
  ): F[Seq[DataSet]] =
    RetrieveByIdsWithIgnoreUnknownIds.retrieveByIds(
      requestSession,
      baseUrl,
      ids,
      ignoreUnknownIds
    )

  override def retrieveByExternalIds(
      externalIds: Seq[String],
      ignoreUnknownIds: Boolean
  ): F[Seq[DataSet]] =
    RetrieveByExternalIdsWithIgnoreUnknownIds.retrieveByExternalIds(
      requestSession,
      baseUrl,
      externalIds,
      ignoreUnknownIds
    )

  override def createItems(items: Items[DataSetCreate]): F[Seq[DataSet]] =
    Create.createItems[F, DataSet, DataSetCreate](requestSession, baseUrl, items)

  override def updateById(items: Map[Long, DataSetUpdate]): F[Seq[DataSet]] =
    UpdateById.updateById[F, DataSet, DataSetUpdate](requestSession, baseUrl, items)

  override def updateByExternalId(items: Map[String, DataSetUpdate]): F[Seq[DataSet]] =
    UpdateByExternalId.updateByExternalId[F, DataSet, DataSetUpdate](
      requestSession,
      baseUrl,
      items
    )

  private[sdk] def filterWithCursor(
      filter: DataSetFilter,
      cursor: Option[String],
      limit: Option[Int],
      partition: Option[Partition],
      aggregatedProperties: Option[Seq[String]] = None
  ): F[ItemsWithCursor[DataSet]] =
    Filter.filterWithCursor(
      requestSession,
      uri"$baseUrl/list",
      filter,
      cursor,
      limit,
      partition = None,
      Constants.defaultBatchSize
    )

  override def search(searchQuery: DataSetQuery): F[Seq[DataSet]] =
    Search.search(requestSession, baseUrl, searchQuery)
}

object DataSets {
  implicit val dataSetDecoder: Decoder[DataSet] = deriveDecoder
  implicit val dataSetItemsWithCursorDecoder: Decoder[ItemsWithCursor[DataSet]] = deriveDecoder
  implicit val dataSetItemsDecoder: Decoder[Items[DataSet]] = deriveDecoder
  implicit val dataSetCreateEncoder: Encoder[DataSetCreate] = deriveEncoder
  implicit val dataSetCreateItemsEncoder: Encoder[Items[DataSetCreate]] = deriveEncoder
  implicit val dataSetUpdateEncoder: Encoder[DataSetUpdate] = deriveEncoder
  implicit val dataSetUpdateItemsEncoder: Encoder[Items[DataSetUpdate]] = deriveEncoder
  implicit val dataSetFilterEncoder: Encoder[DataSetFilter] = deriveEncoder
  implicit val dataSetFilterRequestEncoder: Encoder[FilterRequest[DataSetFilter]] = deriveEncoder
  implicit val dataSetListQueryEncoder: Encoder[DataSetQuery] = deriveEncoder
}