bookmarks_count.kt

    @Test
    fun countBookmarks() {
        val datastore = initDatastore()

        val queryBuilder = Query.newEntityQueryBuilder()
            .setKind("Bookmarks")
            .setLimit(50)

        var count = 0
        while (true) {
            val result = datastore.run(queryBuilder.build())
            result.forEach {
                if (it.key.toString().contains("kotlinconf2023")) {
                    count += it.names.size
                }
            }

            when (result.moreResults) {
                QueryResultBatch.MoreResultsType.MORE_RESULTS_TYPE_UNSPECIFIED -> TODO()
                QueryResultBatch.MoreResultsType.NOT_FINISHED -> queryBuilder.setStartCursor(result.cursorAfter)
                QueryResultBatch.MoreResultsType.MORE_RESULTS_AFTER_LIMIT -> queryBuilder.setStartCursor(
                    result.cursorAfter
                )

                QueryResultBatch.MoreResultsType.MORE_RESULTS_AFTER_CURSOR -> queryBuilder.setStartCursor(
                    result.cursorAfter
                )

                QueryResultBatch.MoreResultsType.NO_MORE_RESULTS -> break
                QueryResultBatch.MoreResultsType.UNRECOGNIZED -> TODO()
                null -> TODO()
            }
        }

        println("Found: $count bookmarks")
    }