Skip to content

Commit 37fabfb

Browse files
committed
Merge branch 'illumen-master'
2 parents aad4413 + a31231e commit 37fabfb

File tree

3 files changed

+13
-3
lines changed

3 files changed

+13
-3
lines changed

src/EntityFrameworkCore.Scaffolding.Handlebars/HbsCSharpDbContextGenerator.cs

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -343,8 +343,12 @@ private void InitializeEntityTypeBuilder(IEntityType entityType, IndentedStringB
343343

344344
private string GetEntityTypeName(IEntityType entityType, string entityTypeName)
345345
{
346+
var schema = !string.IsNullOrEmpty(entityType.GetTableName())
347+
? entityType.GetSchema()
348+
: entityType.GetViewSchema();
349+
346350
return _options?.Value?.EnableSchemaFolders == true
347-
? $"{entityType.GetSchema()}.{entityTypeName}" : entityTypeName;
351+
? $"{schema}.{entityTypeName}" : entityTypeName;
348352
}
349353

350354
private void GenerateEntityType(IEntityType entityType, bool useDataAnnotations, IndentedStringBuilder sb)

src/EntityFrameworkCore.Scaffolding.Handlebars/HbsCSharpEntityTypeGenerator.cs

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -139,7 +139,10 @@ public override string WriteCode(IEntityType entityType, string @namespace, bool
139139

140140
private string GetNamespaceForEntity(IEntityType entityType, string defaultNamespace)
141141
{
142-
return $"{defaultNamespace}.{CSharpHelper.Namespace(entityType.GetSchema())}";
142+
var schema = !string.IsNullOrEmpty(entityType.GetTableName())
143+
? entityType.GetSchema()
144+
: entityType.GetViewSchema();
145+
return $"{defaultNamespace}.{CSharpHelper.Namespace(schema)}";
143146
}
144147

145148
/// <summary>

src/EntityFrameworkCore.Scaffolding.Handlebars/HbsCSharpModelGenerator.cs

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -149,8 +149,11 @@ public override ScaffoldedModel GenerateModel(IModel model, ModelCodeGenerationO
149149
options.UseDataAnnotations);
150150

151151
var transformedFileName = EntityTypeTransformationService.TransformEntityFileName(entityType.DisplayName());
152+
var schema = !string.IsNullOrEmpty(entityType.GetTableName())
153+
? entityType.GetSchema()
154+
: entityType.GetViewSchema();
152155
var entityTypeFileName = _options?.Value?.EnableSchemaFolders == true
153-
? Path.Combine(CSharpHelper.Namespace(entityType.GetSchema()), transformedFileName + FileExtension)
156+
? Path.Combine(CSharpHelper.Namespace(schema), transformedFileName + FileExtension)
154157
: transformedFileName + FileExtension;
155158
resultingFiles.AdditionalFiles.Add(
156159
new ScaffoldedFile

0 commit comments

Comments
 (0)