build(copy) to unity

Cysharp · Sep 9, 2019 · 1bb8a70 · 1bb8a70
1 parent cbf33db
commit 1bb8a70
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 3 deletions.
diff --git a/src/MagicOnion.Client.Unity/Assets/Scripts/MagicOnion/Client/ClientFilter.cs b/src/MagicOnion.Client.Unity/Assets/Scripts/MagicOnion/Client/ClientFilter.cs
@@ -21,6 +21,7 @@ public abstract class RequestContext
         public Type ResponseType { get; }
         public abstract Type RequestType { get; }
         public Func<byte[], byte[]> RequestMutator { get; private set; }
+        public Func<byte[], byte[]> ResponseMutator { get; private set; }
 
         Dictionary<string, object> items;
         public IDictionary<string, object> Items
@@ -49,12 +50,18 @@ internal RequestContext(MagicOnionClientBase client, string methodPath, CallOpti
             this.Filters = filters;
             this.RequestMethod = requestMethod;
             this.RequestMutator = DefaultMutator;
+            this.ResponseMutator = DefaultMutator;
         }
 
         public void SetRequestMutator(Func<byte[], byte[]> mutator)
         {
             this.RequestMutator = mutator;
         }
+
+        public void SetResponseMutator(Func<byte[], byte[]> mutator)
+        {
+            this.ResponseMutator = mutator;
+        }
     }
 
     public class RequestContext<T> : RequestContext
@@ -100,7 +107,7 @@ public ResponseContext()
             this.ResponseMutator = DefaultMutator;
         }
 
-        public void SetRequestMutator(Func<byte[], byte[]> mutator)
+        public void SetResponseMutator(Func<byte[], byte[]> mutator)
         {
             this.ResponseMutator = mutator;
         }

diff --git a/src/MagicOnion.Client.Unity/Assets/Scripts/MagicOnion/Client/MagicOnionClientBase.cs b/src/MagicOnion.Client.Unity/Assets/Scripts/MagicOnion/Client/MagicOnionClientBase.cs
@@ -18,15 +18,19 @@ static protected ResponseContext CreateResponseContext<TResponse>(RequestContext
         {
             var self = context.Client;
             var callResult = self.callInvoker.AsyncUnaryCall(method, self.host, context.CallOptions, context.RequestMutator(MagicOnionMarshallers.UnsafeNilBytes));
-            return new ResponseContext<TResponse>(callResult, self.resolver);
+            var response = new ResponseContext<TResponse>(callResult, self.resolver);
+            response.SetResponseMutator(context.ResponseMutator);
+            return response;
         }
 
         static protected ResponseContext CreateResponseContext<TRequest, TResponse>(RequestContext context, Method<byte[], byte[]> method)
         {
             var self = context.Client;
             var message = LZ4MessagePackSerializer.Serialize<TRequest>(((RequestContext<TRequest>)context).Request, self.resolver);
             var callResult = self.callInvoker.AsyncUnaryCall(method, self.host, context.CallOptions, context.RequestMutator(message));
-            return new ResponseContext<TResponse>(callResult, self.resolver);
+            var response = new ResponseContext<TResponse>(callResult, self.resolver);
+            response.SetResponseMutator(context.ResponseMutator);
+            return response;
         }
     }