Skip to content

Commit 5abe338

Browse files
committed
func entity sourcegen support
Co-Authored-By: YunchuWang <[email protected]>
1 parent 861a007 commit 5abe338

File tree

1 file changed

+8
-1
lines changed

1 file changed

+8
-1
lines changed

src/Generators/AzureFunctions/SyntaxNodeUtility.cs

Lines changed: 8 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -61,6 +61,12 @@ public static bool TryGetFunctionKind(MethodDeclarationSyntax method, out Durabl
6161
kind = DurableFunctionKind.Activity;
6262
return true;
6363
}
64+
65+
if (attribute.ToString().Equals("EntityTrigger", StringComparison.Ordinal))
66+
{
67+
kind = DurableFunctionKind.Entity;
68+
return true;
69+
}
6470
}
6571
}
6672

@@ -125,7 +131,8 @@ public static bool TryGetParameter(
125131
{
126132
string attributeName = attribute.Name.ToString();
127133
if ((kind == DurableFunctionKind.Activity && attributeName == "ActivityTrigger") ||
128-
(kind == DurableFunctionKind.Orchestration && attributeName == "OrchestratorTrigger"))
134+
(kind == DurableFunctionKind.Orchestration && attributeName == "OrchestratorTrigger") ||
135+
(kind == DurableFunctionKind.Entity && attributeName == "EntityTrigger"))
129136
{
130137
TypeInfo info = model.GetTypeInfo(methodParam.Type);
131138
if (info.Type is INamedTypeSymbol named)

0 commit comments

Comments
 (0)