no.digipost.function.ThrowingSupplier Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of digg Show documentation

Some stellar general purpose utils.

The newest version!

/*
 * Copyright (C) Posten Norge AS
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *         http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
package no.digipost.function;

import no.digipost.DiggExceptions;

import java.util.Optional;
import java.util.function.Consumer;
import java.util.function.Function;
import java.util.function.Supplier;

@FunctionalInterface
public interface ThrowingSupplier {
    T get() throws X;

    default Supplier asUnchecked() {
        return ifExceptionThrow(DiggExceptions::asUnchecked);
    }

    default Supplier ifExceptionThrow(Function exceptionMapper) {
        return () -> ifException(e -> { throw exceptionMapper.apply(e); }).get().orElse(null);
    }

    default Supplier> ifException(Consumer exceptionHandler) {
        return () -> {
            return Optional.ofNullable(ifExceptionGet(e -> {
                exceptionHandler.accept(e);
                return null;
            }).get());
        };
    }

    default Supplier ifExceptionGet(Function recoveryFunction) {
        return () -> {
            try {
                return ThrowingSupplier.this.get();
            } catch (Exception e) {
                return recoveryFunction.apply(e);
            } catch (Error e) {
                throw e;
            } catch (Throwable e) {
                throw DiggExceptions.asUnchecked(e);
            }
        };
    }
}