Skip to content

Commit 7aa1163

Browse files
authored
Merge pull request #60 from cnblogs/fix-fetch-same-builder
fix: fetch same builder when calling Entity<T> multiple times
2 parents 70bd39a + 0021848 commit 7aa1163

File tree

1 file changed

+9
-3
lines changed

1 file changed

+9
-3
lines changed

src/Cnblogs.Architecture.Ddd.Infrastructure.Dapper.Clickhouse/ClickhouseModelCollectionBuilder.cs

+9-3
Original file line numberDiff line numberDiff line change
@@ -15,9 +15,15 @@ public class ClickhouseModelCollectionBuilder
1515
public ClickhouseModelBuilder<T> Entity<T>()
1616
where T : class
1717
{
18-
var builder = new ClickhouseModelBuilder<T>();
19-
_builders.Add(typeof(T), builder);
20-
return builder;
18+
var type = typeof(T);
19+
if (_builders.TryGetValue(type, out var builder))
20+
{
21+
return (ClickhouseModelBuilder<T>)builder;
22+
}
23+
24+
var modelBuilder = new ClickhouseModelBuilder<T>();
25+
_builders.Add(typeof(T), modelBuilder);
26+
return modelBuilder;
2127
}
2228

2329
internal void Build(ClickhouseContextOptions options)

0 commit comments

Comments
 (0)