Some final 0.6.0 updates.

clockworklabs · Aug 1, 2023 · c70676d · c70676d
1 parent 083ba3d
commit c70676d
Show file tree

Hide file tree

Showing 3 changed files with 223 additions and 156 deletions.
diff --git a/Scripts/ClientCache.cs b/Scripts/ClientCache.cs
@@ -62,6 +62,8 @@ public Type ClientTableType
             // TODO: Consider renaming this one, this kind of implies that its a callback for the Update operation
             public Action<SpacetimeDBClient.TableOp, object, object, ClientApi.Event> RowUpdatedCallback;
             public Func<AlgebraicType, AlgebraicValue, AlgebraicValue, bool> ComparePrimaryKeyFunc;
+            public Func<AlgebraicValue, AlgebraicValue> GetPrimaryKeyValueFunc;
+            public Func<AlgebraicType, AlgebraicType> GetPrimaryKeyTypeFunc;
 
             public string Name
             {
@@ -90,6 +92,10 @@ public TableCache(Type clientTableType, AlgebraicType rowSchema, Func<AlgebraicV
                     ?.CreateDelegate(typeof(Action<SpacetimeDBClient.TableOp, object, object, ClientApi.Event>));
                 ComparePrimaryKeyFunc = (Func<AlgebraicType, AlgebraicValue, AlgebraicValue, bool>)clientTableType.GetMethod("ComparePrimaryKey", BindingFlags.Static | BindingFlags.Public)
                     ?.CreateDelegate(typeof(Func<AlgebraicType, AlgebraicValue, AlgebraicValue, bool>));
+                GetPrimaryKeyValueFunc = (Func<AlgebraicValue, AlgebraicValue>)clientTableType.GetMethod("GetPrimaryKeyValue", BindingFlags.Static | BindingFlags.Public)
+                    ?.CreateDelegate(typeof(Func<AlgebraicValue, AlgebraicValue>));
+                GetPrimaryKeyTypeFunc = (Func<AlgebraicType, AlgebraicType>)clientTableType.GetMethod("GetPrimaryKeyType", BindingFlags.Static | BindingFlags.Public)
+                    ?.CreateDelegate(typeof(Func<AlgebraicType, AlgebraicType>));
                 entries = new Dictionary<byte[], (AlgebraicValue, object)>(new ByteArrayComparer());
                 decodedValues = new ConcurrentDictionary<byte[], (AlgebraicValue, object)>(new ByteArrayComparer());
             }
@@ -212,6 +218,16 @@ public bool ComparePrimaryKey(AlgebraicValue v1, AlgebraicValue v2)
                 return (bool)ComparePrimaryKeyFunc.Invoke(rowSchema, v1, v2);
             }
 
+            public AlgebraicValue GetPrimaryKeyValue(AlgebraicValue row)
+            {
+                return GetPrimaryKeyValueFunc != null ? GetPrimaryKeyValueFunc.Invoke(row) : null;
+            }
+
+            public AlgebraicType GetPrimaryKeyType(AlgebraicType row)
+            {
+                return GetPrimaryKeyTypeFunc != null ? GetPrimaryKeyTypeFunc.Invoke(row) : null;
+            }
+
             public bool ComparePrimaryKey(byte[] rowPk1, byte[] rowPk2)
             {
                 if (!decodedValues.TryGetValue(rowPk1, out var v1))

diff --git a/Scripts/SATS/AlgebraicValue.cs b/Scripts/SATS/AlgebraicValue.cs
@@ -1,4 +1,5 @@
 using System;
+using System.Linq;
 using System.Collections.Generic;
 using System.IO;
 
@@ -475,5 +476,33 @@ public void Serialize(AlgebraicType type, BinaryWriter writer)
                     throw new NotImplementedException();
             }
         }
+
+        public class AlgebraicValueComparer : IEqualityComparer<AlgebraicValue>
+        {
+            private AlgebraicType type;
+            public AlgebraicValueComparer(AlgebraicType type)
+            {
+                this.type = type;
+            }
+
+            public bool Equals(AlgebraicValue l, AlgebraicValue r)
+            {
+                return AlgebraicValue.Compare(type, l, r);
+            }
+
+            public int GetHashCode(AlgebraicValue value)
+            {
+                var stream = new MemoryStream();
+                var writer = new BinaryWriter(stream);
+                value.Serialize(type, writer);
+                var s = stream.ToArray();
+                if (s.Length >= 4)
+                {
+                    return BitConverter.ToInt32(s, 0);
+                }
+                return s.Sum(b => b);
+            }
+        }
+
     }
 }