Czy można wywołać funkcję z unaryPlus w kotlin?

Dec 01 2020

To jest pytanie uzupełniające do innego pytania, które zadałem wczoraj.

Jak zbudować zagnieżdżoną listę przy użyciu wzorca konstruktora?

Podziękowania dla : Pelocho za udzielenie miłej odpowiedzi.

Użyłem tego samouczka, aby utworzyć bezpieczny typ konstruktora zapytań GraphQL:

Chcę teraz uprościć to, co zrobiłem. Wiem, że kotlin musi mieć do tego kilka fajnych funkcji.

Teraz muszę wywoływać funkcje, gdy chcę dodać jednostkę do mojego zapytania:

fun main() {
    events{
        title() // I don't like to do () when it is an edge case
    }
}

Zastanawiam się, czy da się zrobić coś innego i prostszego, na przykład:

fun main() {
    events{
        title // this is much nicer
    }
}

Ale jeśli nie jest to możliwe, wystarczy zmniejszyć liczbę znaków za pomocą przeciążenia operatora, takiego jak np.unaryPlus

fun main() {
    events{
        +title // this is also nicer
    }
}

Mój kod, który mam teraz, ma klasę o nazwie, w Queryktórej umieściłem metodę:

    operator fun Query.unaryPlus() {
        visitEntity(this,{})
    }

Ale wydaje mi się, że to nie działa.

Cały mój kod jest tutaj.

interface Element {
    fun render(builder: StringBuilder, indent: String)
}

@DslMarker //Domain Specific Language
annotation class GraphQLMarker


@GraphQLMarker
abstract class Query(val name: String) : Element {
    val children = arrayListOf<Element>()
    protected fun <T : Element> visitEntity(entity: T, visit: T.() -> Unit = {}): T {
        entity.visit()
        children.add(entity)
        return entity
    }

    override fun render(builder: StringBuilder, indent: String) {
        builder.append("$indent$name")
        if (children.isNotEmpty()) {
            builder.append("{\n")
            for (c in children) {
                c.render(builder, "$indent ") } builder.append("$indent}")
        }
        builder.append("\n")
    }

    operator fun Query.unaryPlus() {
        visitEntity(this,{})
    }

    override fun toString(): String {
        val builder = StringBuilder()
        render(builder, "")
        return builder.toString()
    }
}

Następnie stworzyłem zajęcia odpowiednie dla mojego przypadku


class Filter private constructor(val filters: MutableMap<FilterType, Any>) {
    class Builder {
        private val filters = mutableMapOf<FilterType, Any>()
        fun filters(key: FilterType, value: Any) = apply {
            this.filters[key] = value
        }

        fun build(): Filter {
            return Filter(filters)
        }
    }
}

class EVENTS(private val filter: Filter) : Query("events") {
    override fun render(builder: StringBuilder, indent: String) {
        builder.append("{$name") if (filter.filters.isNotEmpty()) { builder.append("(" + filter.filters.map { if (it.value is Int || it.value is Long) { it.key.str + ":" + it.value + "," } else { it.key.str + ":\"" + it.value + "\"," } }.joinToString(" ").dropLast(1) + ")") } if (children.isNotEmpty()) { builder.append("{\n") for (c in children) { c.render(builder, "$indent  ")
            }
            builder.append("$indent}")
        }
        builder.append("\n}")
    }

    fun title() = visitEntity(TITLE())
    fun genre() = visitEntity(GENRE())
    fun image() = visitEntity(IMAGE())
    fun link() = visitEntity(LINK())
    fun other() = visitEntity(OTHER())
    fun price() = visitEntity(PRICE())
    fun text() = visitEntity(TEXT())
    fun tickets() = visitEntity(TICKETS())
    fun time() = visitEntity(TIME())
    fun location(visit: LOCATION.() -> Unit) = visitEntity(LOCATION(), visit)
}

class TITLE : Query("title")
class GENRE : Query("genre")
class IMAGE : Query("image")
class LINK : Query("link")
class OTHER : Query("other")
class PRICE : Query("price")
class TEXT : Query("text")
class TICKETS : Query("tickets")
class TIME : Query("time")
class LOCATION : Query("location") {
    fun area() = visitEntity(AREA())
    fun place() = visitEntity(PLACE())
    fun address(visit: ADDRESS.() -> Unit) = visitEntity(ADDRESS(), visit)
    fun coordinates(visit: COORDINATES.() -> Unit) = visitEntity(COORDINATES(), visit)
}

class AREA : Query("area")
class PLACE : Query("place")
class ADDRESS : Query("address") {
    fun city() = visitEntity(CITY())
    fun street() = visitEntity(STREET())
    fun no() = visitEntity(NO())
    fun state() = visitEntity(STATE())
    fun zip() = visitEntity(ZIP())
}

class CITY : Query("city")
class STREET : Query("street")
class NO : Query("no")
class STATE : Query("state")
class ZIP : Query("zip")
class COORDINATES : Query("coordinates") {
    fun longitude() = visitEntity(LONGITUDE())
    fun latitude() = visitEntity(LATITUDE())
}

class LONGITUDE : Query("longitude")
class LATITUDE : Query("latitude")

enum class FilterType(val str: String) {
    PLACE("place"),
    PRICELT("priceLT"),
    PRICEGT("priceGT"),
    TIMELT("timestampLT"),
    TIMEGT("timestampGT"),
    AREA("area"),
    TITLE("title"),
    GENRE("genre")
}

Niektóre z klas są na krawędziach i nie zagnieżdżą się dalej. Zastanawiałem się więc, czy można to trochę uprościć. I jeśli mogę użyć a, unaryPlusaby je wywołać, gdy wywołuję je z main()funkcji, więc nie muszę pisać {}po jednym przypadku, gdy nie mają żadnego zagnieżdżenia

fun main(){
    events(filter) {
        title()
        genre()
        image()
        link()
        tickets()
        other()
        price()
        text()
        time()
        location {
            area()
            place()
            address {
                city()
                street()
                no()
                state()
                zip()
            }
            coordinates {
                longitude()
                latitude()
            }
        }
    }
}

Z góry dziękuję. Miłego dnia!

Odpowiedzi

2 Pelocho Dec 01 2020 at 10:18

Tak, możesz używać unaryPlusdo wszystkiego, co chcesz

Sprawdź dokumentację dotyczącą przeciążenia operatora

Tutaj masz działający przykład:

fun main() {
    val events = events {
        +title
        +genre
    }

    println(events.render())
}

interface Element {
    fun render() : String
}

@DslMarker //Domain Specific Language
annotation class GraphQLMarker

@GraphQLMarker
fun events(init: EventBody.() -> Unit): Element {
    val events = EventBody()
    events.init()
    return events
}


class EventBody: Element {
    override fun render() =
        "events:${elements.joinToString { it.render() }}"

    private val elements = mutableListOf<Element>()

    operator fun Element.unaryPlus() = elements.add(this)
}

@GraphQLMarker
object title: Element {
    override fun render() = "title"
}

@GraphQLMarker
object genre: Element {
    override fun render() = "genre"
}

które wyjścia

events:title, genre