sources/hv-collector-utils/src/main/kotlin/org/onap/dcae/collectors/veshv/utils/arrow/core.kt

   1 /*
   2  * ============LICENSE_START=======================================================
   3  * dcaegen2-collectors-veshv
   4  * ================================================================================
   5  * Copyright (C) 2018 NOKIA
   6  * ================================================================================
   7  * Licensed under the Apache License, Version 2.0 (the "License");
   8  * you may not use this file except in compliance with the License.
   9  * You may obtain a copy of the License at
  10  *
  11  *      http://www.apache.org/licenses/LICENSE-2.0
  12  *
  13  * Unless required by applicable law or agreed to in writing, software
  14  * distributed under the License is distributed on an "AS IS" BASIS,
  15  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  16  * See the License for the specific language governing permissions and
  17  * limitations under the License.
  18  * ============LICENSE_END=========================================================
  19  */
  20 package org.onap.dcae.collectors.veshv.utils.arrow
  21
  22 import arrow.core.Either
  23 import arrow.core.Option
  24 import arrow.core.Try
  25 import arrow.core.identity
  26 import arrow.syntax.collections.firstOption
  27 import java.util.concurrent.atomic.AtomicReference
  28
  29 /**
  30  * @author Piotr Jaszczyk <piotr.jaszczyk@nokia.com>
  31  * @since July 2018
  32  */
  33
  34 fun <A> Either<A, A>.flatten() = fold(::identity, ::identity)
  35
  36 fun <B> Either<Throwable, B>.rightOrThrow() = fold({ throw it }, ::identity)
  37
  38 fun <A, B> Either<A, B>.rightOrThrow(mapper: (A) -> Throwable) = fold({ throw mapper(it) }, ::identity)
  39
  40 fun <A> AtomicReference<A>.getOption() = Option.fromNullable(get())
  41
  42 fun <A> Option.Companion.fromNullablesChain(firstValue: A?, vararg nextValues: () -> A?): Option<A> =
  43         if (firstValue != null)
  44             Option.just(firstValue)
  45         else nextValues.asSequence()
  46                 .map { it() }
  47                 .filter { it != null }
  48                 .firstOption()
  49
  50
  51 fun <A, B> Either<A, B>.doOnLeft(action: () -> Unit): Either<A, B> = apply { if (isLeft()) action() }
  52
  53 fun <A> Option<A>.doOnEmpty(action: () -> Unit): Option<A> = apply { if (isEmpty()) action() }
  54
  55 fun <A> Try<A>.doOnFailure(action: () -> Unit): Try<A> = apply { if (isFailure()) action() }