public void ReplaceMethod(string fileName, string methodName, NewMethodInfo newMethodInfo)
{
ConstructedFile file;
if (_files.Where(cf => cf.FileName == fileName).Count() != 0)
{
file = _files.Where(cf => cf.FileName == fileName).First();
}
else
{
file = new ConstructedFile(fileName);
}
file.AddMethod(methodName, newMethodInfo);
}
//Constructor
public InMemoryWorkspace(String name, string contextNamespace, string contextTypeName)
: base(DataLayerProviderKind.InMemoryLinq, name, contextNamespace, contextTypeName)
{
this.DataGenerator = new GraphBasedDataGenerator(this, new CodeGeneratingDataInserter(this));
if (AstoriaTestProperties.IsRemoteClient)
{
BeforeServiceCreation.Add(() => WorkspaceLibrary.AddSilverlightHostFiles(this));
}
this.RequiredFrameworkSources.Add("InMemoryContext.cs");
this.RequiredFrameworkSources.Add("InMemoryDataService.cs");
this.RequiredFrameworkSources.Add("InMemoryEntitySetDictionary.cs");
ConstructedFile inMemoryContextFile = new ConstructedFile("InMemoryContext.cs");
ServiceModifications.Files.Add(inMemoryContextFile);
ServiceModifications.Interfaces.IUpdatable.ImplementingFile = inMemoryContextFile;
ServiceModifications.Interfaces.IDataServiceUpdateProvider.ImplementingFile = inMemoryContextFile;
}
//Constructor
public NonClrWorkspace(String name, string contextNamespace, string contextTypeName)
: base(DataLayerProviderKind.NonClr, name, contextNamespace, contextTypeName)
{
_resources = new NonClrEntitySetDictionary();
this.DataGenerator = new GraphBasedDataGenerator(this, new CodeGeneratingDataInserter(this));
this.DataGenerator.DataInserter.OnAddingEntity += this.DataInserter_HandleAddingEntity;
this.DataGenerator.DataInserter.OnAddingAssociation += this.DataInserter_HandleAddingAssociation;
BeforeServiceCreation.Insert(0, DefineClrProperties);
if (AstoriaTestProperties.IsRemoteClient)
BeforeServiceCreation.Add(() => WorkspaceLibrary.AddSilverlightHostFiles(this));
RequiredFrameworkSources.Add("NonClrContext.cs");
RequiredFrameworkSources.Add("NonClrDataService.cs");
RequiredFrameworkSources.Add("NonClrEntitySetDictionary.cs");
RequiredFrameworkSources.Add("NonClrExpressionTreeToXmlSerializer.cs");
RequiredFrameworkSources.Add("NonClrQueryable.cs");
RequiredFrameworkSources.Add("DSPMethodsImplementation.cs");
RequiredFrameworkSources.Add("LateBoundBaseExpressionVisitor.cs");
RequiredFrameworkSources.Add("LateBoundExpressionVisitor.cs");
RequiredFrameworkSources.Add("LateBoundToClrConverter.cs");
RequiredFrameworkSources.Add("OpenTypeMethodsImplementation.cs");
ConstructedFile nonClrContextFile = new ConstructedFile("NonClrContext.cs");
ServiceModifications.Files.Add(nonClrContextFile);
ServiceModifications.Interfaces.IUpdatable.ImplementingFile = nonClrContextFile;
ServiceModifications.Interfaces.IDataServiceUpdateProvider.ImplementingFile = nonClrContextFile;
ServiceModifications.Interfaces.IDataServiceMetadataProvider.ImplementingFile = nonClrContextFile;
ServiceModifications.Interfaces.IDataServiceQueryProvider.ImplementingFile = nonClrContextFile;
ServiceModifications.Interfaces.IServiceProvider.Services[typeof(Microsoft.OData.Service.Providers.IDataServiceQueryProvider)]
= "new QueryProviderWrapper(ContextInstance)";
ServiceModifications.Interfaces.IServiceProvider.Services[typeof(Microsoft.OData.Service.Providers.IDataServiceMetadataProvider)]
= "new MetadataProviderWrapper(ContextInstance)";
ServiceModifications.Interfaces.IServiceProvider.Services[typeof(Microsoft.OData.Service.Providers.IDataServiceUpdateProvider)]
= "new UpdateProviderWrapper(ContextInstance)";
}
protected override void WorkspaceCallback(Workspace workspace)
{
Microsoft.OData.Client.ODataProtocolVersion maxProtocolVersion = Microsoft.OData.Client.ODataProtocolVersion.V4;
if (workspace.Settings.MaxProtocolVersion.HasValue)
{
maxProtocolVersion = workspace.Settings.MaxProtocolVersion.Value;
}
workspace.GenerateCallOrderInterceptors = true;
// use custom concurrency provider for workspaces that can
if(workspace.DataLayerProviderKind == DataLayerProviderKind.InMemoryLinq)
{
if (workspace.Settings.UpdatableImplementation == UpdatableImplementation.DataServiceUpdateProvider)
{
workspace.ServiceModifications.Interfaces.IServiceProvider.Services[typeof(Microsoft.OData.Service.Providers.IDataServiceUpdateProvider)]
= "new UpdateProviderWrapper(this.CurrentDataSource)";
}
else
{
workspace.ServiceModifications.Interfaces.IServiceProvider.Services[typeof(Microsoft.OData.Service.IUpdatable)]
= "new UpdatableWrapper(this.CurrentDataSource)";
}
}
else if (workspace.DataLayerProviderKind == DataLayerProviderKind.NonClr)
{
workspace.Settings.UpdatableImplementation = UpdatableImplementation.DataServiceUpdateProvider;
workspace.Settings.UseLazyPropertyLoading = false;
// do this NOW instead of later, so that we can refer to these settings
(workspace as NonClrWorkspace).DefineClrProperties();
if (AstoriaTestProperties.UseOpenTypes)
{
OpenTypesUtil.SetupDefaultOpenTypeAttributes(workspace);
}
}
workspace.ServiceContainer.ResourceContainers.Remove("Invoices");
#region set up etags
if (workspace.DataLayerProviderKind != DataLayerProviderKind.LinqToSql)
{
HashSet<ResourceType> typesWithETags = new HashSet<ResourceType>();
foreach (ResourceContainer container in workspace.ServiceContainer.ResourceContainers)
{
ResourceType baseType = container.BaseType;
if (!typesWithETags.Add(baseType))
continue;
// set up ETags
NodeType[] etagTypes = new NodeType[] { Clr.Types.String, Clr.Types.Int16, Clr.Types.Int32, Clr.Types.Int64, Clr.Types.Guid };
List<ResourceProperty> possibleETagProperties = baseType.Properties.OfType<ResourceProperty>()
.Where(p => etagTypes.Contains(p.Type) && !p.IsComplexType && !p.IsNavigation && p.PrimaryKey == null
&& p.Facets.IsDeclaredProperty && !p.Facets.FixedLength && p.ResourceType == baseType
&& p.Facets.UnderlyingType == UnderlyingType.Same && !p.Facets.IsStoreBlob
&& (p.Type != Clr.Types.String || (p.Facets.MaxSize.HasValue && p.Facets.MaxSize.Value < 512)))
.ToList();
if (possibleETagProperties.Any())
baseType.Facets.Add(NodeFacet.Attribute(new ConcurrencyAttribute(baseType, possibleETagProperties.Choose(2).Select(p => p.Name).ToArray())));
}
}
#endregion
#region set up blobs
if (Versioning.Server.SupportsLiveFeatures && workspace.DataLayerProviderKind != DataLayerProviderKind.LinqToSql)
{
var edmWorkspace = workspace as EdmWorkspace;
if (edmWorkspace != null)
{
edmWorkspace.CsdlCallbacks.Add(delegate(XmlDocument doc)
{
doc.InnerXml = doc.InnerXml.Replace(TestUtil.TestNamespaceManager.LookupNamespace("csdl"), TestUtil.TestNamespaceManager.LookupNamespace("csdl2"));
});
edmWorkspace.SsdlCallbacks.Add(delegate(XmlDocument doc)
{
doc.InnerXml = doc.InnerXml.Replace("http://schemas.microsoft.com/ado/2006/04/edm/ssdl", "http://schemas.microsoft.com/ado/2009/02/edm/ssdl");
});
edmWorkspace.MslCallbacks.Add(delegate(XmlDocument doc)
{
doc.InnerXml = doc.InnerXml.Replace("urn:schemas-microsoft-com:windows:storage:mapping:CS", "http://schemas.microsoft.com/ado/2008/09/mapping/cs");
});
}
foreach(ResourceContainer container in workspace.ServiceContainer.ResourceContainers.Where(rc => !(rc is ServiceOperation)))
{
if (workspace.DataLayerProviderKind == DataLayerProviderKind.Edm)
{
// After named-stream redesign, adding streams to EF is non-trivial
continue;
}
if (container.ResourceTypes.Any(rt => rt.Facets.NamedStreams.Any()))
{
continue;
}
// because we don't represent streams as properties, having streams appear on anything other than the absolute base type
// makes call order prediction very hard
if (container.BaseType.BaseTypes.Any())
{
continue;
}
var type = container.BaseType;
switch(AstoriaTestProperties.Random.Next(3))
{
case 0:
type.Facets.Add(NodeFacet.Attribute(new NamedStreamResourceAttribute(type, "PhotoStream")));
break;
case 1:
type.Facets.Add(NodeFacet.Attribute(new NamedStreamResourceAttribute(type, "PhotoStream")));
type.Facets.Add(NodeFacet.Attribute(new NamedStreamResourceAttribute(type, "ThumbnailStream")));
type.Facets.Add(NodeFacet.Attribute(new NamedStreamResourceAttribute(type, "HighResolutionStream")));
break;
}
}
}
if (workspace.DataLayerProviderKind != DataLayerProviderKind.LinqToSql)
{
string streamProviderComparerTypeName = typeof(ReferenceEqualityComparer).FullName;
if (workspace.DataLayerProviderKind == DataLayerProviderKind.Edm)
{
streamProviderComparerTypeName = workspace.Name + "EqualityComparer";
List<string> equalityComparerCode = new List<string>()
{
"public class " + streamProviderComparerTypeName + " : System.Data.Test.Astoria.KeyBasedEqualityComparerBase",
"{",
" protected override string[] GetKeyPropertyNames(Type entityType)",
" {",
" entityType = ObjectContext.GetObjectType(entityType);",
" var fullTypeName = entityType.FullName;",
};
foreach (var type in workspace.ServiceContainer.ResourceTypes)
{
equalityComparerCode.Add(string.Join(Environment.NewLine, new string[]
{
" if(fullTypeName == \"" + type.FullName + "\")",
" {",
" return new string[] { " + string.Join(", ", type.Properties.Where(p => p.PrimaryKey != null).Select(p => '"' + p.Name + '"').ToArray()) + " };",
" }",
}));
}
equalityComparerCode.Add(" throw new Exception(\"Unrecognized type\");");
equalityComparerCode.Add(" }");
equalityComparerCode.Add("}");
workspace.GlobalAdditionalCode += string.Join(Environment.NewLine, equalityComparerCode);
}
#region for inmemory/nonclr, make all int key properties server generated
if (workspace.DataLayerProviderKind == DataLayerProviderKind.NonClr || workspace.DataLayerProviderKind == DataLayerProviderKind.InMemoryLinq)
{
foreach (ResourceType type in workspace.ServiceContainer.ResourceTypes)
foreach (NodeProperty property in type.Key.Properties)
if (property.Type == Clr.Types.Int32)
property.Facets.ServerGenerated = true;
}
#endregion
#region find possible blob types, and set up stream provider
List<ResourceType> possibleBlobTypes = new List<ResourceType>();
foreach (ResourceType type in workspace.ServiceContainer.ResourceContainers.Where(rc => !(rc is ServiceOperation)).Select(rc => rc.BaseType))
{
// need to find one that has only generated keys, all other propeties nullable, and no nav props
if (type.Key.Properties.Any(p => !p.Facets.ServerGenerated))
continue;
if (type.Properties.OfType<ResourceProperty>().Any(p => p.PrimaryKey == null && !p.IsNavigation && !p.Facets.Nullable))
continue;
if (type.Properties.OfType<ResourceProperty>().Any(p => p.IsNavigation))
continue;
possibleBlobTypes.Add(type);
}
if (possibleBlobTypes.Any())
{
List<ResourceType> blobTypes = new List<ResourceType>();
// want one with etags, and one without
ResourceType blobType = possibleBlobTypes.Where(rt => rt.Properties.Any(p => p.Facets.ConcurrencyModeFixed)).Choose();
if (blobType != null)
{
blobType.Facets.Add(NodeFacet.Attribute(new BlobsAttribute(blobType)));
blobTypes.Add(blobType);
}
blobType = possibleBlobTypes.Where(rt => !rt.Properties.Any(p => p.Facets.ConcurrencyModeFixed)).Choose();
if (blobType != null)
{
blobType.Facets.Add(NodeFacet.Attribute(new BlobsAttribute(blobType)));
blobTypes.Add(blobType);
}
workspace.ServiceModifications.Interfaces.IServiceProvider.Services[typeof(Microsoft.OData.Service.Providers.IDataServiceStreamProvider)]
= "new StreamProviderWrapper(new System.Data.Test.Astoria.InMemoryStreamProvider<" + streamProviderComparerTypeName + ">())";
workspace.Settings.StreamProviderImplementation = StreamProviderImplementation.DataServiceStreamProvider;
if (Versioning.Server.SupportsLiveFeatures && workspace.DataLayerProviderKind != DataLayerProviderKind.Edm)
{
workspace.ServiceModifications.Interfaces.IServiceProvider.Services[typeof(Microsoft.OData.Service.Providers.IDataServiceStreamProvider2)]
= "new StreamProvider2Wrapper(new System.Data.Test.Astoria.InMemoryStreamProvider<" + streamProviderComparerTypeName + ">())";
workspace.Settings.StreamProviderImplementation = StreamProviderImplementation.DataServiceStreamProvider2;
var blobTypesWithoutNamedStreams = blobTypes.Where(t => !t.Facets.NamedStreams.Any()).ToList();
if (blobTypesWithoutNamedStreams.Count != 0)
{
var blobTypeWithNamedStreams = blobTypesWithoutNamedStreams.Choose();
blobTypeWithNamedStreams.Facets.Add(NodeFacet.Attribute(new NamedStreamResourceAttribute(blobTypeWithNamedStreams, "Thumbnail")));
}
}
List<string> typeResolverLines = new List<string>();
foreach (ResourceType rt in blobTypes)
{
// safe because we know the types are disjoint (either do or do not have etags) and that they were the basetypes of their containers
IEnumerable<ResourceContainer> containers = workspace.ServiceContainer.ResourceContainers.Where(rc => rc.BaseType == rt);
foreach (ResourceContainer rc in containers)
{
typeResolverLines.Add(" if(entitySetName == \"" + rc.Name + "\")");
typeResolverLines.Add(" return \"" + rt.Namespace + "." + rt.Name + "\";");
}
}
typeResolverLines.Add(" return null;");
string typeResolverCode = string.Join(Environment.NewLine, typeResolverLines.ToArray());
workspace.RequiredFrameworkSources.Add("InMemoryStreamProvider.cs");
workspace.RequiredFrameworkSources.Add("KeyBasedEqualityComparerBase.cs");
workspace.RequiredFrameworkSources.Add("ReferenceEqualityComparer.cs");
var streamProviderFile = new ConstructedFile("InMemoryStreamProvider.cs");
workspace.ServiceModifications.Files.Add(streamProviderFile);
streamProviderFile.AddMethod("ResolveType", new NewMethodInfo()
{
MethodSignature = "string ResolveType(string entitySetName, DataServiceOperationContext operationContext)",
BodyText = typeResolverCode,
});
}
#endregion
}
#endregion
#region set up service operations
List<ServiceOperation> serviceOperations = new List<ServiceOperation>();
foreach(ResourceContainer container in workspace.ServiceContainer.ResourceContainers)
{
// no MEST yet for service ops!!
if (!workspace.ServiceContainer.ResourceContainers.Any(c => c != container && c.BaseType == container.BaseType))
{
foreach (RequestVerb verb in new RequestVerb[] { RequestVerb.Get, RequestVerb.Post })
{
ServiceOperation serviceOp = Resource.ServiceOperation(verb.ToString() + "_" + container.Name, container, container.BaseType, container.ResourceTypes.ToArray());
serviceOp.Verb = verb;
string fullTypeName = container.BaseType.Namespace + "." + container.BaseType.Name;
if (workspace.DataLayerProviderKind == DataLayerProviderKind.NonClr && !container.BaseType.Facets.IsClrType)
fullTypeName = typeof(NonClr.RowEntityType).FullName;
serviceOp.ServiceOpCode = string.Join(Environment.NewLine, new string[]
{
(verb == RequestVerb.Get ? "[WebGet]" : "[WebInvoke(Method = \"POST\")]"),
"public IQueryable<" + fullTypeName + "> " + serviceOp.Name + "()",
"{",
" return GetEntitySet<" + fullTypeName + ">(\"" + container.Name + "\");",
"}"
});
serviceOp.ServiceOperationResultKind = Microsoft.OData.Service.Providers.ServiceOperationResultKind.QueryWithMultipleResults;
serviceOp.ExpectedTypeName = fullTypeName;
serviceOperations.Add(serviceOp);
}
}
}
workspace.ServiceContainer.AddNodes(serviceOperations);
workspace.AddServiceOperationCode();
#endregion
}
protected override void WorkspaceCallback(Workspace workspace)
{
Microsoft.OData.Client.ODataProtocolVersion maxProtocolVersion = Microsoft.OData.Client.ODataProtocolVersion.V4;
if (workspace.Settings.MaxProtocolVersion.HasValue)
{
maxProtocolVersion = workspace.Settings.MaxProtocolVersion.Value;
}
workspace.GenerateCallOrderInterceptors = true;
// use custom concurrency provider for workspaces that can
if (workspace.DataLayerProviderKind == DataLayerProviderKind.InMemoryLinq)
{
if (workspace.Settings.UpdatableImplementation == UpdatableImplementation.DataServiceUpdateProvider)
{
workspace.ServiceModifications.Interfaces.IServiceProvider.Services[typeof(Microsoft.OData.Service.Providers.IDataServiceUpdateProvider)]
= "new UpdateProviderWrapper(this.CurrentDataSource)";
}
else
{
workspace.ServiceModifications.Interfaces.IServiceProvider.Services[typeof(Microsoft.OData.Service.IUpdatable)]
= "new UpdatableWrapper(this.CurrentDataSource)";
}
}
else if (workspace.DataLayerProviderKind == DataLayerProviderKind.NonClr)
{
workspace.Settings.UpdatableImplementation = UpdatableImplementation.DataServiceUpdateProvider;
workspace.Settings.UseLazyPropertyLoading = false;
// do this NOW instead of later, so that we can refer to these settings
(workspace as NonClrWorkspace).DefineClrProperties();
if (AstoriaTestProperties.UseOpenTypes)
{
OpenTypesUtil.SetupDefaultOpenTypeAttributes(workspace);
}
}
workspace.ServiceContainer.ResourceContainers.Remove("Invoices");
#region set up etags
if (workspace.DataLayerProviderKind != DataLayerProviderKind.LinqToSql)
{
HashSet <ResourceType> typesWithETags = new HashSet <ResourceType>();
foreach (ResourceContainer container in workspace.ServiceContainer.ResourceContainers)
{
ResourceType baseType = container.BaseType;
if (!typesWithETags.Add(baseType))
{
continue;
}
// set up ETags
NodeType[] etagTypes = new NodeType[] { Clr.Types.String, Clr.Types.Int16, Clr.Types.Int32, Clr.Types.Int64, Clr.Types.Guid };
List <ResourceProperty> possibleETagProperties = baseType.Properties.OfType <ResourceProperty>()
.Where(p => etagTypes.Contains(p.Type) && !p.IsComplexType && !p.IsNavigation && p.PrimaryKey == null &&
p.Facets.IsDeclaredProperty && !p.Facets.FixedLength && p.ResourceType == baseType &&
p.Facets.UnderlyingType == UnderlyingType.Same && !p.Facets.IsStoreBlob &&
(p.Type != Clr.Types.String || (p.Facets.MaxSize.HasValue && p.Facets.MaxSize.Value < 512)))
.ToList();
if (possibleETagProperties.Any())
{
baseType.Facets.Add(NodeFacet.Attribute(new ConcurrencyAttribute(baseType, possibleETagProperties.Choose(2).Select(p => p.Name).ToArray())));
}
}
}
#endregion
#region set up blobs
if (Versioning.Server.SupportsLiveFeatures && workspace.DataLayerProviderKind != DataLayerProviderKind.LinqToSql)
{
var edmWorkspace = workspace as EdmWorkspace;
if (edmWorkspace != null)
{
edmWorkspace.CsdlCallbacks.Add(delegate(XmlDocument doc)
{
doc.InnerXml = doc.InnerXml.Replace(TestUtil.TestNamespaceManager.LookupNamespace("csdl"), TestUtil.TestNamespaceManager.LookupNamespace("csdl2"));
});
edmWorkspace.SsdlCallbacks.Add(delegate(XmlDocument doc)
{
doc.InnerXml = doc.InnerXml.Replace("http://schemas.microsoft.com/ado/2006/04/edm/ssdl", "http://schemas.microsoft.com/ado/2009/02/edm/ssdl");
});
edmWorkspace.MslCallbacks.Add(delegate(XmlDocument doc)
{
doc.InnerXml = doc.InnerXml.Replace("urn:schemas-microsoft-com:windows:storage:mapping:CS", "http://schemas.microsoft.com/ado/2008/09/mapping/cs");
});
}
foreach (ResourceContainer container in workspace.ServiceContainer.ResourceContainers.Where(rc => !(rc is ServiceOperation)))
{
if (workspace.DataLayerProviderKind == DataLayerProviderKind.Edm)
{
// After named-stream redesign, adding streams to EF is non-trivial
continue;
}
if (container.ResourceTypes.Any(rt => rt.Facets.NamedStreams.Any()))
{
continue;
}
// because we don't represent streams as properties, having streams appear on anything other than the absolute base type
// makes call order prediction very hard
if (container.BaseType.BaseTypes.Any())
{
continue;
}
var type = container.BaseType;
switch (AstoriaTestProperties.Random.Next(3))
{
case 0:
type.Facets.Add(NodeFacet.Attribute(new NamedStreamResourceAttribute(type, "PhotoStream")));
break;
case 1:
type.Facets.Add(NodeFacet.Attribute(new NamedStreamResourceAttribute(type, "PhotoStream")));
type.Facets.Add(NodeFacet.Attribute(new NamedStreamResourceAttribute(type, "ThumbnailStream")));
type.Facets.Add(NodeFacet.Attribute(new NamedStreamResourceAttribute(type, "HighResolutionStream")));
break;
}
}
}
if (workspace.DataLayerProviderKind != DataLayerProviderKind.LinqToSql)
{
string streamProviderComparerTypeName = typeof(ReferenceEqualityComparer).FullName;
if (workspace.DataLayerProviderKind == DataLayerProviderKind.Edm)
{
streamProviderComparerTypeName = workspace.Name + "EqualityComparer";
List <string> equalityComparerCode = new List <string>()
{
"public class " + streamProviderComparerTypeName + " : System.Data.Test.Astoria.KeyBasedEqualityComparerBase",
"{",
" protected override string[] GetKeyPropertyNames(Type entityType)",
" {",
" entityType = ObjectContext.GetObjectType(entityType);",
" var fullTypeName = entityType.FullName;",
};
foreach (var type in workspace.ServiceContainer.ResourceTypes)
{
equalityComparerCode.Add(string.Join(Environment.NewLine, new string[]
{
" if(fullTypeName == \"" + type.FullName + "\")",
" {",
" return new string[] { " + string.Join(", ", type.Properties.Where(p => p.PrimaryKey != null).Select(p => '"' + p.Name + '"').ToArray()) + " };",
" }",
}));
}
equalityComparerCode.Add(" throw new Exception(\"Unrecognized type\");");
equalityComparerCode.Add(" }");
equalityComparerCode.Add("}");
workspace.GlobalAdditionalCode += string.Join(Environment.NewLine, equalityComparerCode);
}
#region for inmemory/nonclr, make all int key properties server generated
if (workspace.DataLayerProviderKind == DataLayerProviderKind.NonClr || workspace.DataLayerProviderKind == DataLayerProviderKind.InMemoryLinq)
{
foreach (ResourceType type in workspace.ServiceContainer.ResourceTypes)
{
foreach (NodeProperty property in type.Key.Properties)
{
if (property.Type == Clr.Types.Int32)
{
property.Facets.ServerGenerated = true;
}
}
}
}
#endregion
#region find possible blob types, and set up stream provider
List <ResourceType> possibleBlobTypes = new List <ResourceType>();
foreach (ResourceType type in workspace.ServiceContainer.ResourceContainers.Where(rc => !(rc is ServiceOperation)).Select(rc => rc.BaseType))
{
// need to find one that has only generated keys, all other propeties nullable, and no nav props
if (type.Key.Properties.Any(p => !p.Facets.ServerGenerated))
{
continue;
}
if (type.Properties.OfType <ResourceProperty>().Any(p => p.PrimaryKey == null && !p.IsNavigation && !p.Facets.Nullable))
{
continue;
}
if (type.Properties.OfType <ResourceProperty>().Any(p => p.IsNavigation))
{
continue;
}
possibleBlobTypes.Add(type);
}
if (possibleBlobTypes.Any())
{
List <ResourceType> blobTypes = new List <ResourceType>();
// want one with etags, and one without
ResourceType blobType = possibleBlobTypes.Where(rt => rt.Properties.Any(p => p.Facets.ConcurrencyModeFixed)).Choose();
if (blobType != null)
{
blobType.Facets.Add(NodeFacet.Attribute(new BlobsAttribute(blobType)));
blobTypes.Add(blobType);
}
blobType = possibleBlobTypes.Where(rt => !rt.Properties.Any(p => p.Facets.ConcurrencyModeFixed)).Choose();
if (blobType != null)
{
blobType.Facets.Add(NodeFacet.Attribute(new BlobsAttribute(blobType)));
blobTypes.Add(blobType);
}
workspace.ServiceModifications.Interfaces.IServiceProvider.Services[typeof(Microsoft.OData.Service.Providers.IDataServiceStreamProvider)]
= "new StreamProviderWrapper(new System.Data.Test.Astoria.InMemoryStreamProvider<" + streamProviderComparerTypeName + ">())";
workspace.Settings.StreamProviderImplementation = StreamProviderImplementation.DataServiceStreamProvider;
if (Versioning.Server.SupportsLiveFeatures && workspace.DataLayerProviderKind != DataLayerProviderKind.Edm)
{
workspace.ServiceModifications.Interfaces.IServiceProvider.Services[typeof(Microsoft.OData.Service.Providers.IDataServiceStreamProvider2)]
= "new StreamProvider2Wrapper(new System.Data.Test.Astoria.InMemoryStreamProvider<" + streamProviderComparerTypeName + ">())";
workspace.Settings.StreamProviderImplementation = StreamProviderImplementation.DataServiceStreamProvider2;
var blobTypesWithoutNamedStreams = blobTypes.Where(t => !t.Facets.NamedStreams.Any()).ToList();
if (blobTypesWithoutNamedStreams.Count != 0)
{
var blobTypeWithNamedStreams = blobTypesWithoutNamedStreams.Choose();
blobTypeWithNamedStreams.Facets.Add(NodeFacet.Attribute(new NamedStreamResourceAttribute(blobTypeWithNamedStreams, "Thumbnail")));
}
}
List <string> typeResolverLines = new List <string>();
foreach (ResourceType rt in blobTypes)
{
// safe because we know the types are disjoint (either do or do not have etags) and that they were the basetypes of their containers
IEnumerable <ResourceContainer> containers = workspace.ServiceContainer.ResourceContainers.Where(rc => rc.BaseType == rt);
foreach (ResourceContainer rc in containers)
{
typeResolverLines.Add(" if(entitySetName == \"" + rc.Name + "\")");
typeResolverLines.Add(" return \"" + rt.Namespace + "." + rt.Name + "\";");
}
}
typeResolverLines.Add(" return null;");
string typeResolverCode = string.Join(Environment.NewLine, typeResolverLines.ToArray());
workspace.RequiredFrameworkSources.Add("InMemoryStreamProvider.cs");
workspace.RequiredFrameworkSources.Add("KeyBasedEqualityComparerBase.cs");
workspace.RequiredFrameworkSources.Add("ReferenceEqualityComparer.cs");
var streamProviderFile = new ConstructedFile("InMemoryStreamProvider.cs");
workspace.ServiceModifications.Files.Add(streamProviderFile);
streamProviderFile.AddMethod("ResolveType", new NewMethodInfo()
{
MethodSignature = "string ResolveType(string entitySetName, DataServiceOperationContext operationContext)",
BodyText = typeResolverCode,
});
}
#endregion
}
#endregion
#region set up service operations
List <ServiceOperation> serviceOperations = new List <ServiceOperation>();
foreach (ResourceContainer container in workspace.ServiceContainer.ResourceContainers)
{
// no MEST yet for service ops!!
if (!workspace.ServiceContainer.ResourceContainers.Any(c => c != container && c.BaseType == container.BaseType))
{
foreach (RequestVerb verb in new RequestVerb[] { RequestVerb.Get, RequestVerb.Post })
{
ServiceOperation serviceOp = Resource.ServiceOperation(verb.ToString() + "_" + container.Name, container, container.BaseType, container.ResourceTypes.ToArray());
serviceOp.Verb = verb;
string fullTypeName = container.BaseType.Namespace + "." + container.BaseType.Name;
if (workspace.DataLayerProviderKind == DataLayerProviderKind.NonClr && !container.BaseType.Facets.IsClrType)
{
fullTypeName = typeof(NonClr.RowEntityType).FullName;
}
serviceOp.ServiceOpCode = string.Join(Environment.NewLine, new string[]
{
(verb == RequestVerb.Get ? "[WebGet]" : "[WebInvoke(Method = \"POST\")]"),
"public IQueryable<" + fullTypeName + "> " + serviceOp.Name + "()",
"{",
" return GetEntitySet<" + fullTypeName + ">(\"" + container.Name + "\");",
"}"
});
serviceOp.ServiceOperationResultKind = Microsoft.OData.Service.Providers.ServiceOperationResultKind.QueryWithMultipleResults;
serviceOp.ExpectedTypeName = fullTypeName;
serviceOperations.Add(serviceOp);
}
}
}
workspace.ServiceContainer.AddNodes(serviceOperations);
workspace.AddServiceOperationCode();
#endregion
}
public IDataServiceUpdateProviderInterface(ConstructedFile file)
: base(file)
{
}
public IUpdatableInterface(ConstructedFile file)
: base(file)
{
}
public ConstructedInterface(ConstructedFile file)
{
_file = file;
this.Implemented = true;
}
public void ReplaceMethod(string fileName, string methodName, NewMethodInfo newMethodInfo)
{
ConstructedFile file;
if (_files.Where(cf => cf.FileName == fileName).Count() != 0)
file = _files.Where(cf => cf.FileName == fileName).First();
else
file = new ConstructedFile(fileName);
file.AddMethod(methodName, newMethodInfo);
}
public IDataServiceUpdateProviderInterface(ConstructedFile file)
: base(file)
{ }
public IUpdatableInterface(ConstructedFile file)
: base(file)
{ }
public ConstructedInterface(ConstructedFile file)
{
_file = file;
this.Implemented = true;
}
