graalvm gradle plugin task

Author: dariuszkuc

integration/graalvm/build.gradle.kts

@@ -0,0 +1,20 @@
+import java.util.Properties
+
+allprojects {
+    repositories {
+        mavenCentral()
+        mavenLocal {
+            content {
+                includeGroup("com.expediagroup")
+            }
+        }
+    }
+
+    val properties = Properties()
+    properties.load(File(rootDir.parentFile.parent, "gradle.properties").inputStream())
+    for ((key, value) in properties) {
+        if (!project.ext.has(key.toString())) {
+            project.ext[key.toString()] = value
+        }
+    }
+}

integration/graalvm/gradle.properties

@@ -0,0 +1,6 @@
+group = com.expediagroup.graalvm
+
+org.gradle.jvmargs=-Xmx4g -XX:MaxMetaspaceSize=1g
+org.gradle.caching=true
+org.gradle.parallel=true
+

integration/graalvm/gradle/wrapper

@@ -0,0 +1 @@
+../../../gradle/wrapper

integration/graalvm/gradlew

@@ -0,0 +1 @@
+../../gradlew

integration/graalvm/ktor-graalvm-server/build.gradle.kts

@@ -0,0 +1,55 @@
+import org.jetbrains.kotlin.gradle.tasks.KotlinCompile
+
+@Suppress("DSL_SCOPE_VIOLATION") // TODO: remove once KTIJ-19369 / Gradle#22797 is fixed
+plugins {
+    alias(libs.plugins.kotlin.jvm)
+    application
+    alias(libs.plugins.graalvm.native)
+    id("com.expediagroup.graphql")
+}
+
+dependencies {
+    implementation("com.expediagroup", "graphql-kotlin-ktor-server")
+    implementation(libs.logback)
+    implementation(libs.ktor.server.cio)
+    testImplementation(libs.kotlin.test)
+}
+
+tasks.test {
+    useJUnitPlatform()
+}
+
+tasks.withType<KotlinCompile> {
+    kotlinOptions.jvmTarget = "17"
+}
+
+application {
+    mainClass.set("com.expediagroup.graalvm.ktor.ApplicationKt")
+}
+
+graalvmNative {
+    toolchainDetection.set(false)
+    binaries {
+        named("main") {
+            fallback.set(false)
+            verbose.set(true)
+
+            buildArgs.add("--initialize-at-build-time=io.ktor,kotlin,ch.qos.logback,org.slf4j")
+            buildArgs.add("--trace-object-instantiation=java.util.jar.JarFile")
+
+            buildArgs.add("-H:+InstallExitHandlers")
+            buildArgs.add("-H:+ReportUnsupportedElementsAtRuntime")
+            buildArgs.add("-H:+ReportExceptionStackTraces")
+
+            imageName.set("graal-server")
+//            mainClass.set("org.test.Main")
+        }
+        metadataRepository {
+            enabled.set(true)
+        }
+    }
+}
+
+val graphQLGraalVmMetadata by tasks.getting(com.expediagroup.graphql.plugin.gradle.tasks.GraphQLGraalVmMetadataTask::class) {
+    packages.set(listOf("com.expediagroup.graalvm.ktor"))
+}

integration/graalvm/ktor-graalvm-server/src/main/kotlin/com/expediagroup/graalvm/ktor/Application.kt

@@ -0,0 +1,78 @@
+package com.expediagroup.graalvm.ktor
+
+import com.expediagroup.graalvm.ktor.context.CustomContextFactory
+import com.expediagroup.graalvm.ktor.hooks.CustomHooks
+import com.expediagroup.graalvm.ktor.schema.ArgumentQuery
+import com.expediagroup.graalvm.ktor.schema.AsyncQuery
+import com.expediagroup.graalvm.ktor.schema.BasicMutation
+import com.expediagroup.graalvm.ktor.schema.ContextualQuery
+import com.expediagroup.graalvm.ktor.schema.DataClassQuery
+import com.expediagroup.graalvm.ktor.schema.EnumQuery
+import com.expediagroup.graalvm.ktor.schema.ErrorQuery
+import com.expediagroup.graalvm.ktor.schema.ExampleInterface
+import com.expediagroup.graalvm.ktor.schema.ExampleUnion
+import com.expediagroup.graalvm.ktor.schema.FirstImpl
+import com.expediagroup.graalvm.ktor.schema.FirstUnionMember
+import com.expediagroup.graalvm.ktor.schema.InnerClassQuery
+import com.expediagroup.graalvm.ktor.schema.ListQuery
+import com.expediagroup.graalvm.ktor.schema.PolymorphicQuery
+import com.expediagroup.graalvm.ktor.schema.ScalarQuery
+import com.expediagroup.graalvm.ktor.schema.SecondImpl
+import com.expediagroup.graalvm.ktor.schema.SecondUnionMember
+import com.expediagroup.graalvm.ktor.schema.TypesQuery
+import com.expediagroup.graalvm.ktor.schema.dataloader.ExampleDataLoader
+import com.expediagroup.graphql.dataloader.KotlinDataLoaderRegistryFactory
+import com.expediagroup.graphql.server.ktor.GraphQL
+import com.expediagroup.graphql.server.ktor.graphQLGetRoute
+import com.expediagroup.graphql.server.ktor.graphQLPostRoute
+import com.expediagroup.graphql.server.ktor.graphQLSDLRoute
+import com.expediagroup.graphql.server.ktor.graphiQLRoute
+import io.ktor.server.application.install
+import io.ktor.server.cio.CIO
+import io.ktor.server.engine.embeddedServer
+import io.ktor.server.routing.routing
+
+fun main() {
+    embeddedServer(CIO, port = 8080, host = "0.0.0.0") {
+        install(GraphQL) {
+            schema {
+                packages = listOf("com.expediagroup.graalvm.ktor")
+                queries = listOf(
+                    ArgumentQuery(),
+                    AsyncQuery(),
+                    ContextualQuery(),
+                    DataClassQuery(),
+                    EnumQuery(),
+                    ErrorQuery(),
+                    InnerClassQuery(),
+                    PolymorphicQuery(),
+                    ScalarQuery(),
+                    TypesQuery(),
+                    ListQuery()
+                )
+                mutations = listOf(
+                    BasicMutation()
+                )
+                hooks = CustomHooks()
+                typeHierarchy = mapOf(
+                    ExampleInterface::class to listOf(FirstImpl::class, SecondImpl::class),
+                    ExampleUnion::class to listOf(FirstUnionMember::class, SecondUnionMember::class)
+                )
+            }
+            engine {
+                dataLoaderRegistryFactory = KotlinDataLoaderRegistryFactory(
+                    ExampleDataLoader
+                )
+            }
+            server {
+                contextFactory = CustomContextFactory()
+            }
+        }
+        routing {
+            graphQLGetRoute()
+            graphQLPostRoute()
+            graphQLSDLRoute()
+            graphiQLRoute()
+        }
+    }.start(wait = true)
+}

integration/graalvm/ktor-graalvm-server/src/main/kotlin/com/expediagroup/graalvm/ktor/context/CustomContextFactory.kt

@@ -0,0 +1,13 @@
+package com.expediagroup.graalvm.ktor.context
+
+import com.expediagroup.graphql.generator.extensions.plus
+import com.expediagroup.graphql.server.ktor.DefaultKtorGraphQLContextFactory
+import graphql.GraphQLContext
+import io.ktor.server.request.ApplicationRequest
+import java.util.UUID
+
+class CustomContextFactory : DefaultKtorGraphQLContextFactory() {
+    override suspend fun generateContext(request: ApplicationRequest): GraphQLContext {
+        return super.generateContext(request).plus(mapOf("custom" to UUID.randomUUID().toString()))
+    }
+}

integration/graalvm/ktor-graalvm-server/src/main/kotlin/com/expediagroup/graalvm/ktor/hooks/CustomHooks.kt

@@ -0,0 +1,49 @@
+package com.expediagroup.graalvm.ktor.hooks
+
+import com.expediagroup.graphql.generator.hooks.SchemaGeneratorHooks
+import graphql.language.StringValue
+import graphql.schema.Coercing
+import graphql.schema.CoercingParseLiteralException
+import graphql.schema.CoercingParseValueException
+import graphql.schema.CoercingSerializeException
+import graphql.schema.GraphQLScalarType
+import graphql.schema.GraphQLType
+import java.util.UUID
+import kotlin.reflect.KClass
+import kotlin.reflect.KType
+
+class CustomHooks : SchemaGeneratorHooks {
+    override fun willGenerateGraphQLType(type: KType): GraphQLType? = when (type.classifier as? KClass<*>) {
+        UUID::class -> graphqlUUIDType
+        else -> null
+    }
+}
+
+val graphqlUUIDType: GraphQLScalarType = GraphQLScalarType.newScalar()
+    .name("UUID")
+    .description("A type representing a formatted java.util.UUID")
+    .coercing(UUIDCoercing)
+    .build()
+
+object UUIDCoercing : Coercing<UUID, String> {
+    override fun parseValue(input: Any): UUID = runCatching {
+        UUID.fromString(serialize(input))
+    }.getOrElse {
+        throw CoercingParseValueException("Expected valid UUID but was $input")
+    }
+
+    override fun parseLiteral(input: Any): UUID {
+        val uuidString = (input as? StringValue)?.value
+        return runCatching {
+            UUID.fromString(uuidString)
+        }.getOrElse {
+            throw CoercingParseLiteralException("Expected valid UUID literal but was $uuidString")
+        }
+    }
+
+    override fun serialize(dataFetcherResult: Any): String = runCatching {
+        dataFetcherResult.toString()
+    }.getOrElse {
+        throw CoercingSerializeException("Data fetcher result $dataFetcherResult cannot be serialized to a String")
+    }
+}

integration/graalvm/ktor-graalvm-server/src/main/kotlin/com/expediagroup/graalvm/ktor/schema/ArgumentQuery.kt

@@ -0,0 +1,40 @@
+package com.expediagroup.graalvm.ktor.schema
+
+import com.expediagroup.graphql.generator.scalars.ID
+import com.expediagroup.graphql.server.operations.Query
+import java.util.UUID
+
+class ArgumentQuery : Query {
+    fun idArg(arg: ID): ID = ID(arg.value.reversed())
+    fun optionalIdArg(arg: ID? = null): ID? = arg?.let { ID(it.value.reversed()) }
+    fun stringArg(arg: String): String = arg.reversed()
+    fun optionalStringArg(arg: String? = null): String? = arg?.reversed()
+    fun intArg(arg: Int): Int = arg.inc()
+    fun optionalIntArg(arg: Int? = null): Int? = arg?.inc()
+    fun doubleArg(arg: Double): Double = arg.inc()
+    fun optionalDoubleArg(arg: Double? = null): Double? = arg?.inc()
+    fun booleanArg(arg: Boolean): Boolean = arg.not()
+    fun optionalBooleanArg(arg: Boolean? = null): Boolean? = arg?.not()
+//    fun customScalarArg(arg: UUID): UUID = arg
+//    fun optionalCustomScalarArg(arg: UUID? = null): UUID? = arg
+    fun manyOptionalArgs(
+        idArg: ID? = null,
+        stringArg: String? = null,
+        intArg: Int? = null,
+        doubleArg: Double? = null,
+        booleanArg: Boolean? = null
+    ): String? = if (idArg != null || stringArg != null || intArg != null || doubleArg != null || booleanArg != null) {
+        "${idArg?.value}|$stringArg|$intArg|$doubleArg|$booleanArg"
+    } else {
+        null
+    }
+//    fun listArg(arg: List<String>): List<String> = arg.reversed()
+//    fun optionalListArg(arg: List<String>? = null): List<String>? = arg?.reversed()
+    fun objectArg(arg: SimpleType): SimpleType = arg
+    fun optionalObjectArg(arg: SimpleType? = null): SimpleType? = arg
+//    fun listObjectArg(arg: List<SimpleType>): String = UUID.randomUUID().toString()
+}
+
+data class SimpleType(val id: Int, val description: String) {
+    fun outputOnly(): String = UUID.randomUUID().toString()
+}

integration/graalvm/ktor-graalvm-server/src/main/kotlin/com/expediagroup/graalvm/ktor/schema/AsyncQuery.kt

@@ -0,0 +1,25 @@
+package com.expediagroup.graalvm.ktor.schema
+
+import com.expediagroup.graalvm.ktor.schema.dataloader.EXAMPLE_LOADER
+import com.expediagroup.graphql.generator.scalars.ID
+import com.expediagroup.graphql.server.operations.Query
+import graphql.schema.DataFetchingEnvironment
+import kotlinx.coroutines.coroutineScope
+import kotlinx.coroutines.delay
+import java.util.concurrent.CompletableFuture
+import kotlin.random.Random
+
+class AsyncQuery : Query {
+    private val random: Random = Random(100)
+
+    fun future(): CompletableFuture<Int> = CompletableFuture.completedFuture(random.nextInt())
+
+    fun dataLoader(env: DataFetchingEnvironment, id: ID): CompletableFuture<String> = env.getDataLoader<ID, String>(EXAMPLE_LOADER).load(id)
+
+    suspend fun coroutine(): Int = coroutineScope {
+        delay(10)
+        random.nextInt()
+    }
+
+    // TODO reactor monads
+}

integration/graalvm/ktor-graalvm-server/src/main/kotlin/com/expediagroup/graalvm/ktor/schema/BasicMutation.kt

@@ -0,0 +1,7 @@
+package com.expediagroup.graalvm.ktor.schema
+
+import com.expediagroup.graphql.server.operations.Mutation
+
+class BasicMutation : Mutation {
+    fun mutate(arg: String): String = arg.reversed()
+}

integration/graalvm/ktor-graalvm-server/src/main/kotlin/com/expediagroup/graalvm/ktor/schema/ContextualQuery.kt

@@ -0,0 +1,8 @@
+package com.expediagroup.graalvm.ktor.schema
+
+import com.expediagroup.graphql.server.operations.Query
+import graphql.schema.DataFetchingEnvironment
+
+class ContextualQuery : Query {
+    fun contextualQuery(env: DataFetchingEnvironment): String = env.graphQlContext.getOrDefault("custom", "defaultValue")
+}

integration/graalvm/ktor-graalvm-server/src/main/kotlin/com/expediagroup/graalvm/ktor/schema/DataClassQuery.kt

@@ -0,0 +1,66 @@
+package com.expediagroup.graalvm.ktor.schema
+
+import com.expediagroup.graphql.generator.scalars.ID
+import com.expediagroup.graphql.server.operations.Query
+import java.util.UUID
+import kotlin.random.Random
+
+private val random: Random = Random(100)
+
+class DataClassQuery : Query {
+    fun dataClassQuery(): ExampleDataClass = ExampleDataClass(
+        id = ID(UUID.randomUUID().toString()),
+        nullableId = null,
+        idList = listOf(ID(UUID.randomUUID().toString())),
+        nullableIdList = null,
+        stringValue = UUID.randomUUID().toString(),
+        nullableString = null,
+        stringList = listOf(UUID.randomUUID().toString()),
+        nullableStringList = null,
+        intValue = random.nextInt(),
+        nullableInt = null,
+        doubleValue = random.nextDouble(),
+        nullableDouble = null,
+        booleanValue = random.nextBoolean(),
+        nullableBoolean = null
+    )
+    fun dataClassWithDefaultsQuery(): DefaultsDataClass = DefaultsDataClass()
+
+    fun dataClassList(): List<ExampleDataClass> = listOf(dataClassQuery())
+
+    fun dataClassArgQuery(arg: List<DefaultsDataClass>? = null) = arg
+}
+
+data class ExampleDataClass(
+    val id: ID,
+    val nullableId: ID?,
+    val idList: List<ID>,
+    val nullableIdList: List<ID>?,
+    val stringValue: String,
+    val nullableString: String?,
+    val stringList: List<String>,
+    val nullableStringList: List<String>?,
+    val intValue: Int,
+    val nullableInt: Int?,
+    val doubleValue: Double,
+    val nullableDouble: Double?,
+    val booleanValue: Boolean,
+    val nullableBoolean: Boolean?,
+)
+
+data class DefaultsDataClass(
+    val id: ID = ID(UUID.randomUUID().toString()),
+    val nullableId: ID? = null,
+    val idList: List<ID> = listOf(ID(UUID.randomUUID().toString())),
+    val nullableIdList: List<ID>? = null,
+    val stringValue: String = UUID.randomUUID().toString(),
+    val stringList: List<String> = listOf(UUID.randomUUID().toString()),
+    val nullableStringList: List<String>? = null,
+    val nullableString: String? = null,
+    val intValue: Int = random.nextInt(),
+    val nullableInt: Int? = null,
+    val doubleValue: Double = random.nextDouble(),
+    val nullableDouble: Double? = null,
+    val booleanValue: Boolean = random.nextBoolean(),
+    val nullableBoolean: Boolean? = null
+)

integration/graalvm/ktor-graalvm-server/src/main/kotlin/com/expediagroup/graalvm/ktor/schema/EnumQuery.kt

@@ -0,0 +1,14 @@
+package com.expediagroup.graalvm.ktor.schema
+
+import com.expediagroup.graphql.server.operations.Query
+
+class EnumQuery : Query {
+    fun enumQuery(): BinaryEnum = BinaryEnum.ONE
+    fun enumArgQuery(arg: BinaryEnum): BinaryEnum = arg
+    fun optionalEnum(arg: BinaryEnum? = null): BinaryEnum? = arg
+}
+
+enum class BinaryEnum {
+    ZERO,
+    ONE
+}