Skip to content

Commit cfdae4d

Browse files
committed
minor cleanup to test
1 parent 4ed753e commit cfdae4d

File tree

1 file changed

+3
-16
lines changed

1 file changed

+3
-16
lines changed

hibernate-core/src/test/java/org/hibernate/orm/test/idgen/userdefined/SequenceOrAssignedGeneratorTest.java

Lines changed: 3 additions & 16 deletions
Original file line numberDiff line numberDiff line change
@@ -257,15 +257,8 @@ public void setName(String name) {
257257
public static class SequenceOrAssignedGenerator extends SequenceStyleGenerator {
258258
@Override
259259
public Object generate(SharedSessionContractImplementor session, Object owner) throws HibernateException {
260-
final Long id;
261-
if ( owner instanceof MyEntity ) {
262-
id = ( (MyEntity) owner ).getId();
263-
}
264-
else {
265-
id = null;
266-
}
267-
268-
return id != null ? id : super.generate( session, owner );
260+
final Long id = owner instanceof MyEntity myEntity ? myEntity.getId() : null;
261+
return id == null ? super.generate( session, owner ) : id;
269262
}
270263

271264
@Override
@@ -293,13 +286,7 @@ public Object generate(
293286
Object owner,
294287
Object currentValue,
295288
EventType eventType) {
296-
final Long id;
297-
if ( owner instanceof MyVersionedEntity ) {
298-
id = ( (MyVersionedEntity) owner ).getId();
299-
}
300-
else {
301-
id = null;
302-
}
289+
final Long id = owner instanceof MyVersionedEntity myVersionedEntity ? myVersionedEntity.getId() : null;
303290
return id != null ? id : count++;
304291
}
305292

0 commit comments

Comments
 (0)