snapshot of https://github.com/ent/ent.git (master, 2026-09-15; blobs > 131072B omitted)

entc/gen/bench_test.go

@@ -0,0 +1,41 @@
+// Copyright 2019-present Facebook Inc. All rights reserved.
+// This source code is licensed under the Apache 2.0 license found
+// in the LICENSE file in the root directory of this source tree.
+
+package gen_test
+
+import (
+	"os"
+	"path/filepath"
+	"testing"
+
+	"entgo.io/ent/entc"
+	"entgo.io/ent/entc/gen"
+	"entgo.io/ent/schema/field"
+	"github.com/stretchr/testify/require"
+)
+
+func BenchmarkGraph_Gen(b *testing.B) {
+	target := filepath.Join(os.TempDir(), "ent")
+	require.NoError(b, os.MkdirAll(target, os.ModePerm), "creating tmpdir")
+	defer os.RemoveAll(target)
+	storage, err := gen.NewStorage("sql")
+	require.NoError(b, err)
+	graph, err := entc.LoadGraph("../integration/ent/schema", &gen.Config{
+		Storage: storage,
+		IDType:  &field.TypeInfo{Type: field.TypeInt},
+		Target:  target,
+		Package: "entgo.io/ent/entc/integration/ent",
+		Templates: []*gen.Template{
+			gen.MustParse(gen.NewTemplate("template").
+				Funcs(gen.Funcs).
+				ParseGlob("../integration/ent/template/*.tmpl")),
+		},
+	})
+	require.NoError(b, err)
+	b.ResetTimer()
+	for i := 0; i < b.N; i++ {
+		err := graph.Gen()
+		require.NoError(b, err)
+	}
+}