NEW BOT Телеграм, страница

Forwarded from From Junior to CTO (Ivan Osipov)

Кодогенерация

Два года назад я узнал о проекте jooq. Это генерация кода с помощью которого можно работать с БД из Java. Автоматически создаются классы на основе схемы бд. Тогда это мне показалось чем-то новым, но с тех пор прошло время и генерация кода стала ощущаться чем-то странным. Тем не менее сама по себе задача интересная и отчасти уже решена. На днях открыл для себя библиотеку JavaPoet и её разновидность KotlinPoet, которая сейчас в Early-Access Release, но это нас не остановит, не так ли?

Создадим пустой проект на Kotlin и подцепим gradle зависимость

compile 'com.squareup:kotlinpoet:0.6.0'

внутри функции main опишем генерируемый класс:

val userClass = ClassName("example", "User")

класс находится в каком-то файле, по этому нам нужно этот файл описать, тут куча всяких билдеров и появилось острое желание сделать для этого Kotlin DSL ибо "слишком много шумного кода". Итак, к файлу:

val file = FileSpec.builder("example", "model")
    ... 
    .build()
file.writeTo(System.out)

Здесь мы говорим о том, что будем генерировать файл model c пакетом example, а затем распечатаем его на экран. Между инициализацией билдера и его терминальным вызовом build() происходит основной экшен. Например, мы можем добавить класс User в этот файл:

.addType(TypeSpec.classBuilder(userClass)
      ...
    .build())
.build()

А в класс можно добавить проперти, но важно понимать, что библиотека не гарантирует компилируемости результата, так что сразу защищайтесь от вещей типа lateinit val, которые не скомпилируются. Итак, добавим к нашему классу свойство name:

.addProperty(PropertySpec
    .varBuilder("name", String::class, KModifier.LATEINIT)
    .build())
.build()

Так мы описали спецификацию var переменной типа String с модификатором lateinit

В результате описанного выше примера у нас сгенерируется следующий код:

package example

import kotlin.String

class User {
    lateinit var name: String
}

Библиотека на GitHub здесь: https://github.com/square/kotlinpoet
Код, который мы рассмотрели выше для удобочитаемости выложил вот сюда: https://goo.gl/yuY7jd

GitHub

GitHub - square/kotlinpoet: A Kotlin API for generating .kt source files.

A Kotlin API for generating .kt source files. Contribute to square/kotlinpoet development by creating an account on GitHub.

31 viewsIvan Osipov, 18:42