io.quarkus.funqy.gcp.functions.FunqyBackgroundFunction Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of quarkus-funqy-google-cloud-functions Show documentation

Google Cloud Functions Binding for Quarkus Funqy framework

There is a newer version: 3.17.5

Show newest version

package io.quarkus.funqy.gcp.functions;

import java.io.PrintWriter;
import java.io.StringWriter;

import com.google.cloud.functions.Context;
import com.google.cloud.functions.RawBackgroundFunction;

import io.quarkus.runtime.Application;

public class FunqyBackgroundFunction implements RawBackgroundFunction {
    protected static final String deploymentStatus;
    protected static boolean started = false;

    static {
        StringWriter error = new StringWriter();
        PrintWriter errorWriter = new PrintWriter(error, true);
        if (Application.currentApplication() == null) { // were we already bootstrapped?  Needed for mock unit testing.
            // For GCP functions, we need to set the TCCL to the QuarkusHttpFunction classloader then restore it.
            // Without this, we have a lot of classloading issues (ClassNotFoundException on existing classes)
            // during static init.
            ClassLoader currentCl = Thread.currentThread().getContextClassLoader();
            try {
                Thread.currentThread().setContextClassLoader(FunqyBackgroundFunction.class.getClassLoader());
                Class appClass = Class.forName("io.quarkus.runner.ApplicationImpl");
                String[] args = {};
                Application app = (Application) appClass.getConstructor().newInstance();
                Runtime.getRuntime().addShutdownHook(new Thread() {
                    @Override
                    public void run() {
                        app.stop();
                    }
                });
                app.start(args);
                errorWriter.println("Quarkus bootstrapped successfully.");
                started = true;
            } catch (Exception ex) {
                errorWriter.println("Quarkus bootstrap failed.");
                ex.printStackTrace(errorWriter);
            } finally {
                Thread.currentThread().setContextClassLoader(currentCl);
            }
        } else {
            errorWriter.println("Quarkus bootstrapped successfully.");
            started = true;
        }
        deploymentStatus = error.toString();
    }

    @Override
    public void accept(String event, Context context) {
        if (!started) {
            throw new RuntimeException(deploymentStatus);
        }
        FunqyCloudFunctionsBindingRecorder.handle(event, context);
    }
}