Tools/Castle.DynamicProxy2/Castle.DynamicProxy/CacheMappingsAttribute.cs

   1 using System;
   2 using System.Collections;
   3 using System.Collections.Generic;
   4 using System.IO;
   5 using System.Reflection;
   6 using System.Reflection.Emit;
   7 using System.Runtime.Serialization.Formatters.Binary;
   8 using Castle.DynamicProxy.Generators;
   9
  10 namespace Castle.DynamicProxy
  11 {
  12         /// <summary>
  13         /// Applied to the assemblies saved by <see cref="ModuleScope"/> in order to persist the cache data included in the persisted assembly.
  14         /// </summary>
  15         [AttributeUsage (AttributeTargets.Assembly, AllowMultiple = false)]
  16         [CLSCompliant (false)]
  17         public class CacheMappingsAttribute : Attribute
  18         {
  19                 private static readonly ConstructorInfo constructor = typeof (CacheMappingsAttribute).GetConstructor (new Type[] { typeof (byte[]) });
  20
  21                 public static void ApplyTo (AssemblyBuilder assemblyBuilder, Dictionary<CacheKey, string> mappings)
  22                 {
  23                         using (MemoryStream stream = new MemoryStream ())
  24                         {
  25                                 BinaryFormatter formatter = new BinaryFormatter();
  26                                 formatter.Serialize (stream, mappings);
  27                                 byte[] bytes = stream.ToArray();
  28                                 CustomAttributeBuilder attributeBuilder = new CustomAttributeBuilder (constructor, new object[] {bytes});
  29                                 assemblyBuilder.SetCustomAttribute (attributeBuilder);
  30                         }
  31                 }
  32
  33                 private readonly byte[] _serializedCacheMappings;
  34
  35                 public CacheMappingsAttribute (byte[] serializedCacheMappings)
  36                 {
  37                         _serializedCacheMappings = serializedCacheMappings;
  38                 }
  39
  40                 public byte[] SerializedCacheMappings
  41                 {
  42                         get { return _serializedCacheMappings; }
  43                 }
  44
  45                 public Dictionary<CacheKey, string> GetDeserializedMappings ()
  46                 {
  47                         using (MemoryStream stream = new MemoryStream (SerializedCacheMappings))
  48                         {
  49                                 BinaryFormatter formatter = new BinaryFormatter ();
  50                                 return (Dictionary<CacheKey, string>) formatter.Deserialize (stream);
  51                         }
  52                 }
  53         }
  54 }