neo4j cypher - many inserts / published by https://github.com/dacr/code-examples-manager #43425fdc-b629-48de-9f3c-2b52835ee43a/753510acfe3c67c33c98d2a1001506f736962b1a

neo4j-neotypes-creates.sc

// summary : neo4j cypher - many inserts
// keywords : scala, scalatest, neo4j, neotypes, cypher, dsl, @testable
// publish : gist
// authors : David Crosson
// license : Apache License Version 2.0 (https://www.apache.org/licenses/LICENSE-2.0.txt)
// id : 43425fdc-b629-48de-9f3c-2b52835ee43a
// created-on : 2024-06-26T14:49:20+02:00
// managed-by : https://github.com/dacr/code-examples-manager
// run-with : scala-cli $file

// ---------------------
//> using scala "3.4.2"
//> using dep io.github.neotypes::neotypes-core:1.1.0
//> using dep org.neo4j.test:neo4j-harness:5.20.0
// ---------------------

import neotypes.GraphDatabase
import neotypes.mappers.ResultMapper
import neotypes.syntax.all.*
import org.neo4j.driver.AuthTokens

import scala.util.Using
import scala.concurrent.{Await, Future}
import scala.concurrent.ExecutionContext.Implicits.global
import scala.concurrent.duration._

case class Record(name: String, age: Int)

val builder =
  org.neo4j.harness.Neo4jBuilders
    .newInProcessBuilder()

Using(builder.build()) { embedded =>
  val driver = GraphDatabase.asyncDriver[Future](embedded.boltURI(), AuthTokens.none())

  val insertsQueries = for {
    num <- LazyList.from(1).take(1000)
    name = s"joe$num"
    age  = scala.util.Random.between(1, 130)
    qry  = c"CREATE (:Person {name: $name, age: $age})"
  } yield qry

  val insertsQueriesBatchFuture = for {
    batch <- insertsQueries.grouped(100)
    batchFuture  = batch.reduce( _ + _).execute.void(driver)
  } yield batchFuture

  Await.ready(Future.sequence(insertsQueriesBatchFuture), 20.seconds)

  val countFuture =
    c"MATCH (p: Person) return count(p)"
      .query(ResultMapper.fromFunction(Record.apply))
      .single(driver)

  val dump = for {
    count <- countFuture
  } yield println(s"Found $count records")

  Await.result(dump, 5.seconds)
}