Skip to content

Commit 990f88a

Browse files
bahusoidbahusoid
committed
Convert EntityKey to struct
1 parent 14c5cdb commit 990f88a

14 files changed

+61
-53
lines changed

‎src/NHibernate/Action/EntityIdentityInsertAction.cs‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -19,7 +19,7 @@ public EntityIdentityInsertAction(object[] state, object instance, IEntityPersis
1919
: base(null, state, instance, persister, session)
2020
{
2121
this.isDelayed = isDelayed;
22-
delayedEntityKey = this.isDelayed ? GenerateDelayedEntityKey() : null;
22+
delayedEntityKey = this.isDelayed ? GenerateDelayedEntityKey() : EntityKey.Null;
2323
}
2424

2525
public object GeneratedId

‎src/NHibernate/Async/Event/Default/AbstractSaveEventListener.cs‎

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -150,7 +150,7 @@ protected virtual async Task<object> PerformSaveAsync(object entity, object id,
150150
}
151151
else
152152
{
153-
key = null;
153+
key = EntityKey.Null;
154154
}
155155

156156
if (InvokeSaveLifecycle(entity, persister, source))
@@ -184,7 +184,7 @@ protected virtual async Task<object> PerformSaveOrReplicateAsync(object entity,
184184
cancellationToken.ThrowIfCancellationRequested();
185185
Validate(entity, persister, source);
186186

187-
object id = key == null ? null : key.Identifier;
187+
object id = key.IsNull ? null : key.Identifier;
188188

189189
bool shouldDelayIdentityInserts = !requiresImmediateIdAccess;
190190

‎src/NHibernate/Async/Event/Default/DefaultReplicateEventListener.cs‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -102,7 +102,7 @@ public virtual async Task OnReplicateAsync(ReplicateEvent @event, CancellationTo
102102
}
103103

104104
bool regenerate = persister.IsIdentifierAssignedByInsert; // prefer re-generation of identity!
105-
EntityKey key = regenerate ? null : source.GenerateEntityKey(id, persister);
105+
EntityKey key = regenerate ? EntityKey.Null : source.GenerateEntityKey(id, persister);
106106

107107
await (PerformSaveOrReplicateAsync(entity, key, persister, regenerate, replicationMode, source, true, cancellationToken)).ConfigureAwait(false);
108108
}

‎src/NHibernate/Async/Impl/MultiCriteriaImpl.cs‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -184,7 +184,7 @@ private async Task GetResultsFromDatabaseAsync(IList results, CancellationToken
184184

185185
object o =
186186
await (loader.GetRowFromResultSetAsync(reader, session, queryParameters, loader.GetLockModes(queryParameters.LockModes),
187-
null, hydratedObjects[i], keys, true, cancellationToken)).ConfigureAwait(false);
187+
EntityKey.Null, hydratedObjects[i], keys, true, cancellationToken)).ConfigureAwait(false);
188188
if (createSubselects[i])
189189
{
190190
subselectResultKeys[i].Add(keys);

‎src/NHibernate/Async/Loader/Loader.cs‎

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -113,7 +113,7 @@ protected async Task<object> LoadSingleRowAsync(DbDataReader resultSet, ISession
113113
try
114114
{
115115
result =
116-
await (GetRowFromResultSetAsync(resultSet, session, queryParameters, GetLockModes(queryParameters.LockModes), null,
116+
await (GetRowFromResultSetAsync(resultSet, session, queryParameters, GetLockModes(queryParameters.LockModes), EntityKey.Null,
117117
hydratedObjects, new EntityKey[entitySpan], returnProxies, cancellationToken)).ConfigureAwait(false);
118118
}
119119
catch (OperationCanceledException) { throw; }
@@ -178,7 +178,7 @@ internal async Task<object> GetRowFromResultSetAsync(DbDataReader resultSet, ISe
178178
{
179179
object entity = row[i];
180180
var key = keys[i];
181-
if (entity == null && key != null && IsChildFetchEntity(i))
181+
if (entity == null && key.IsNotNull && IsChildFetchEntity(i))
182182
{
183183
// The entity was missing in the session, fallback on internal load (which will just yield a
184184
// proxy if the persister supports it).
@@ -554,7 +554,7 @@ private async Task<EntityKey> GetKeyFromResultSetAsync(int i, IEntityPersister p
554554
}
555555
}
556556

557-
return resultId == null ? null : session.GenerateEntityKey(resultId, persister);
557+
return resultId == null ? EntityKey.Null : session.GenerateEntityKey(resultId, persister);
558558
}
559559

560560
/// <summary>
@@ -610,7 +610,7 @@ private async Task<object[]> GetRowAsync(DbDataReader rs, ILoadable[] persisters
610610
object obj = null;
611611
EntityKey key = keys[i];
612612

613-
if (key == null)
613+
if (key.IsNull)
614614
{
615615
// do nothing
616616
/* TODO NH-1001 : if (persisters[i]...EntityType) is an OneToMany or a ManyToOne and
@@ -736,7 +736,7 @@ private async Task<object> InstanceNotYetLoadedAsync(DbDataReader dr, int i, ILo
736736

737737
string instanceClass = await (GetInstanceClassAsync(dr, i, persister, key.Identifier, session, cancellationToken)).ConfigureAwait(false);
738738

739-
if (optionalObjectKey != null && key.Equals(optionalObjectKey))
739+
if (optionalObjectKey.IsNotNull && key.Equals(optionalObjectKey))
740740
{
741741
// its the given optional object
742742
obj = optionalObject;

‎src/NHibernate/Engine/EntityEntry.cs‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -192,7 +192,7 @@ public EntityKey EntityKey
192192
{
193193
get
194194
{
195-
if (cachedEntityKey == null)
195+
if (cachedEntityKey.IsNull)
196196
{
197197
if (id == null)
198198
throw new InvalidOperationException("cannot generate an EntityKey when id is null.");

‎src/NHibernate/Engine/EntityKey.cs‎

Lines changed: 27 additions & 24 deletions
Original file line numberDiff line numberDiff line change
@@ -3,7 +3,6 @@
33
using System.Security;
44
using NHibernate.Impl;
55
using NHibernate.Persister.Entity;
6-
using NHibernate.Type;
76

87
namespace NHibernate.Engine
98
{
@@ -12,28 +11,41 @@ namespace NHibernate.Engine
1211
/// and the identifier space (eg. tablename)
1312
/// </summary>
1413
[Serializable]
15-
public sealed class EntityKey : IDeserializationCallback, ISerializable, IEquatable<EntityKey>
14+
public struct EntityKey : ISerializable, IEquatable<EntityKey>
1615
{
16+
public static EntityKey Null { get; } = new EntityKey();
17+
18+
public bool IsNull => identifier == null;
19+
public bool IsNotNull => !IsNull;
20+
1721
private readonly object identifier;
1822
private readonly IEntityPersister _persister;
1923
// hashcode may vary among processes, they cannot be stored and have to be re-computed after deserialization
20-
[NonSerialized]
21-
private int? _hashCode;
24+
private readonly int _hashCode;
2225

2326
/// <summary> Construct a unique identifier for an entity class instance</summary>
2427
public EntityKey(object id, IEntityPersister persister)
2528
{
2629
identifier = id ?? throw new AssertionFailure("null identifier");
2730
_persister = persister;
28-
_hashCode = GenerateHashCode();
31+
_hashCode = GenerateHashCode(persister, id);
2932
}
3033

3134
private EntityKey(SerializationInfo info, StreamingContext context)
3235
{
3336
identifier = info.GetValue(nameof(Identifier), typeof(object));
37+
if (identifier == null)
38+
{
39+
_hashCode = 0;
40+
_persister = null;
41+
return;
42+
}
43+
3444
var factory = (ISessionFactoryImplementor) info.GetValue(nameof(_persister.Factory), typeof(ISessionFactoryImplementor));
35-
var entityName = (string) info.GetValue(nameof(EntityName), typeof(string));
45+
var entityName = info.GetString(nameof(EntityName));
46+
3647
_persister = factory.GetEntityPersister(entityName);
48+
_hashCode = GenerateHashCode(_persister, identifier);
3749
}
3850

3951
public bool IsBatchLoadable => _persister.IsBatchLoadable;
@@ -54,10 +66,8 @@ public override bool Equals(object other)
5466

5567
public bool Equals(EntityKey other)
5668
{
57-
if (other == null)
58-
{
59-
return false;
60-
}
69+
if (other.IsNull)
70+
return IsNull;
6171

6272
return
6373
other.RootEntityName.Equals(RootEntityName)
@@ -66,26 +76,16 @@ public bool Equals(EntityKey other)
6676

6777
public override int GetHashCode()
6878
{
69-
// If the object is put in a set or dictionary during deserialization, the hashcode will not yet be
70-
// computed. Compute the hashcode on the fly. So long as this happens only during deserialization, there
71-
// will be no thread safety issues. For the hashcode to be always defined after deserialization, the
72-
// deserialization callback is used.
73-
return _hashCode ?? GenerateHashCode();
74-
}
75-
76-
/// <inheritdoc />
77-
public void OnDeserialization(object sender)
78-
{
79-
_hashCode = GenerateHashCode();
79+
return _hashCode;
8080
}
8181

82-
private int GenerateHashCode()
82+
private static int GenerateHashCode(IEntityPersister persister, object id)
8383
{
8484
int result = 17;
8585
unchecked
8686
{
87-
result = 37 * result + RootEntityName.GetHashCode();
88-
result = 37 * result + _persister.IdentifierType.GetHashCode(identifier, _persister.Factory);
87+
result = 37 * result + persister.RootEntityName.GetHashCode();
88+
result = 37 * result + persister.IdentifierType.GetHashCode(id, persister.Factory);
8989
}
9090
return result;
9191
}
@@ -99,6 +99,9 @@ public override string ToString()
9999
public void GetObjectData(SerializationInfo info, StreamingContext context)
100100
{
101101
info.AddValue(nameof(Identifier), identifier);
102+
if (identifier == null)
103+
return;
104+
102105
info.AddValue(nameof(_persister.Factory), _persister.Factory);
103106
info.AddValue(nameof(EntityName), EntityName);
104107
}

‎src/NHibernate/Engine/StatefulPersistenceContext.cs‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -753,7 +753,7 @@ public object NarrowProxy(INHibernateProxy proxy, IEntityPersister persister, En
753753
/// </summary>
754754
public object ProxyFor(IEntityPersister persister, EntityKey key, object impl)
755755
{
756-
if (!persister.HasProxy || key == null)
756+
if (!persister.HasProxy || key.IsNull)
757757
return impl;
758758

759759
INHibernateProxy proxy;

‎src/NHibernate/Event/Default/AbstractSaveEventListener.cs‎

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -177,7 +177,7 @@ protected virtual object PerformSave(object entity, object id, IEntityPersister
177177
}
178178
else
179179
{
180-
key = null;
180+
key = EntityKey.Null;
181181
}
182182

183183
if (InvokeSaveLifecycle(entity, persister, source))
@@ -209,7 +209,7 @@ protected virtual object PerformSaveOrReplicate(object entity, EntityKey key, IE
209209
{
210210
Validate(entity, persister, source);
211211

212-
object id = key == null ? null : key.Identifier;
212+
object id = key.IsNull ? null : key.Identifier;
213213

214214
bool shouldDelayIdentityInserts = !requiresImmediateIdAccess;
215215

‎src/NHibernate/Event/Default/DefaultReplicateEventListener.cs‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -95,7 +95,7 @@ public virtual void OnReplicate(ReplicateEvent @event)
9595
}
9696

9797
bool regenerate = persister.IsIdentifierAssignedByInsert; // prefer re-generation of identity!
98-
EntityKey key = regenerate ? null : source.GenerateEntityKey(id, persister);
98+
EntityKey key = regenerate ? EntityKey.Null : source.GenerateEntityKey(id, persister);
9999

100100
PerformSaveOrReplicate(entity, key, persister, regenerate, replicationMode, source, true);
101101
}

0 commit comments

Comments
 (0)