org.neo4j.kernel.impl.transaction.state.Loaders Maven / Gradle / Ivy
/*
* Copyright (c) 2002-2016 "Neo Technology,"
* Network Engine for Objects in Lund AB [http://neotechnology.com]
*
* This file is part of Neo4j.
*
* Neo4j is free software: you can redistribute it and/or modify
* it under the terms of the GNU General Public License as published by
* the Free Software Foundation, either version 3 of the License, or
* (at your option) any later version.
*
* This program is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU General Public License for more details.
*
* You should have received a copy of the GNU General Public License
* along with this program. If not, see nodeLoader;
private final Loader propertyLoader;
private final Loader relationshipLoader;
private final Loader relationshipGroupLoader;
private final Loader schemaRuleLoader;
private final Loader propertyKeyTokenLoader;
private final Loader labelTokenLoader;
private final Loader relationshipTypeTokenLoader;
public Loaders( NeoStores neoStores )
{
nodeLoader = nodeLoader( neoStores.getNodeStore() );
propertyLoader = propertyLoader( neoStores.getPropertyStore() );
relationshipLoader = relationshipLoader( neoStores.getRelationshipStore() );
relationshipGroupLoader = relationshipGroupLoader( neoStores.getRelationshipGroupStore() );
schemaRuleLoader = schemaRuleLoader( neoStores.getSchemaStore() );
propertyKeyTokenLoader = propertyKeyTokenLoader( neoStores.getPropertyKeyTokenStore() );
labelTokenLoader = labelTokenLoader( neoStores.getLabelTokenStore() );
relationshipTypeTokenLoader = relationshipTypeTokenLoader( neoStores.getRelationshipTypeTokenStore() );
}
public Loader nodeLoader()
{
return nodeLoader;
}
public Loader propertyLoader()
{
return propertyLoader;
}
public Loader relationshipLoader()
{
return relationshipLoader;
}
public Loader relationshipGroupLoader()
{
return relationshipGroupLoader;
}
public Loader schemaRuleLoader()
{
return schemaRuleLoader;
}
public Loader propertyKeyTokenLoader()
{
return propertyKeyTokenLoader;
}
public Loader labelTokenLoader()
{
return labelTokenLoader;
}
public Loader relationshipTypeTokenLoader()
{
return relationshipTypeTokenLoader;
}
public static Loader nodeLoader( final NodeStore store )
{
return new Loader()
{
@Override
public NodeRecord newUnused( Long key, Void additionalData )
{
return andMarkAsCreated( new NodeRecord( key ) );
}
@Override
public NodeRecord load( Long key, Void additionalData )
{
return store.getRecord( key, store.newRecord(), NORMAL );
}
@Override
public void ensureHeavy( NodeRecord record )
{
store.ensureHeavy( record );
}
@Override
public NodeRecord clone(NodeRecord nodeRecord)
{
return nodeRecord.clone();
}
};
}
public static Loader propertyLoader( final PropertyStore store )
{
return new Loader()
{
@Override
public PropertyRecord newUnused( Long key, PrimitiveRecord additionalData )
{
PropertyRecord record = new PropertyRecord( key );
setOwner( record, additionalData );
return andMarkAsCreated( record );
}
private void setOwner( PropertyRecord record, PrimitiveRecord owner )
{
if ( owner != null )
{
owner.setIdTo( record );
}
}
@Override
public PropertyRecord load( Long key, PrimitiveRecord additionalData )
{
PropertyRecord record = store.getRecord( key, store.newRecord(), NORMAL );
setOwner( record, additionalData );
return record;
}
@Override
public void ensureHeavy( PropertyRecord record )
{
for ( PropertyBlock block : record )
{
store.ensureHeavy( block );
}
}
@Override
public PropertyRecord clone(PropertyRecord propertyRecord)
{
return propertyRecord.clone();
}
};
}
public static Loader relationshipLoader( final RelationshipStore store )
{
return new Loader()
{
@Override
public RelationshipRecord newUnused( Long key, Void additionalData )
{
return andMarkAsCreated( new RelationshipRecord( key ) );
}
@Override
public RelationshipRecord load( Long key, Void additionalData )
{
return store.getRecord( key, store.newRecord(), NORMAL );
}
@Override
public void ensureHeavy( RelationshipRecord record )
{ // Nothing to load
}
@Override
public RelationshipRecord clone(RelationshipRecord relationshipRecord)
{
return relationshipRecord.clone();
}
};
}
public static Loader relationshipGroupLoader(
final RecordStore store )
{
return new Loader()
{
@Override
public RelationshipGroupRecord newUnused( Long key, Integer type )
{
RelationshipGroupRecord record = new RelationshipGroupRecord( key );
record.setType( type );
return andMarkAsCreated( record );
}
@Override
public RelationshipGroupRecord load( Long key, Integer type )
{
return store.getRecord( key, store.newRecord(), NORMAL );
}
@Override
public void ensureHeavy( RelationshipGroupRecord record )
{ // Not needed
}
@Override
public RelationshipGroupRecord clone( RelationshipGroupRecord record )
{
return record.clone();
}
};
}
public static Loader schemaRuleLoader( final SchemaStore store )
{
return new Loader()
{
@Override
public SchemaRecord newUnused( Long key, SchemaRule additionalData )
{
// Don't blindly mark as created here since some records may be reused.
return new SchemaRecord( store.allocateFrom( additionalData ) );
}
@Override
public SchemaRecord load( Long key, SchemaRule additionalData )
{
return new SchemaRecord( store.getRecords( key, RecordLoad.NORMAL ) );
}
@Override
public void ensureHeavy( SchemaRecord records )
{
for ( DynamicRecord record : records)
{
store.ensureHeavy(record);
}
}
@Override
public SchemaRecord clone( SchemaRecord records )
{
return records.clone();
}
};
}
public static Loader propertyKeyTokenLoader(
final TokenStore store )
{
return new Loader()
{
@Override
public PropertyKeyTokenRecord newUnused( Integer key, Void additionalData )
{
return andMarkAsCreated( new PropertyKeyTokenRecord( key ) );
}
@Override
public PropertyKeyTokenRecord load( Integer key, Void additionalData )
{
return store.getRecord( key, store.newRecord(), NORMAL );
}
@Override
public void ensureHeavy( PropertyKeyTokenRecord record )
{
store.ensureHeavy( record );
}
@Override
public PropertyKeyTokenRecord clone( PropertyKeyTokenRecord record )
{
return record.clone();
}
};
}
public static Loader labelTokenLoader(
final TokenStore store )
{
return new Loader()
{
@Override
public LabelTokenRecord newUnused( Integer key, Void additionalData )
{
return andMarkAsCreated( new LabelTokenRecord( key ) );
}
@Override
public LabelTokenRecord load( Integer key, Void additionalData )
{
return store.getRecord( key, store.newRecord(), NORMAL );
}
@Override
public void ensureHeavy( LabelTokenRecord record )
{
store.ensureHeavy( record );
}
@Override
public LabelTokenRecord clone( LabelTokenRecord record )
{
return record.clone();
}
};
}
public static Loader relationshipTypeTokenLoader(
final TokenStore store )
{
return new Loader()
{
@Override
public RelationshipTypeTokenRecord newUnused( Integer key, Void additionalData )
{
return andMarkAsCreated( new RelationshipTypeTokenRecord( key ) );
}
@Override
public RelationshipTypeTokenRecord load( Integer key, Void additionalData )
{
return store.getRecord( key, store.newRecord(), NORMAL );
}
@Override
public void ensureHeavy( RelationshipTypeTokenRecord record )
{
store.ensureHeavy( record );
}
@Override
public RelationshipTypeTokenRecord clone( RelationshipTypeTokenRecord record )
{
return record.clone();
}
};
}
protected static RECORD andMarkAsCreated( RECORD record )
{
record.setCreated();
return record;
}
}
© 2015 - 2025 Weber Informatics LLC | Privacy Policy