org.neo4j.shell.kernel.apps.Ls Maven / Gradle / Ivy
Go to download
Show more of this group Show more artifacts with this name
Show all versions of neo4j-shell Show documentation
Show all versions of neo4j-shell Show documentation
A generic command shell with a client and server part.
/*
* 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 \"";
}
@Override
protected Continuation exec( AppCommandParser parser, Session session,
Output out ) throws ShellException, RemoteException
{
boolean brief = parser.options().containsKey( "b" );
boolean verbose = parser.options().containsKey( "v" );
boolean quiet = parser.options().containsKey( "q" );
if ( verbose && quiet )
{
verbose = false;
quiet = false;
}
boolean displayProperties = parser.options().containsKey( "p" );
boolean displayRelationships = parser.options().containsKey( "r" );
boolean caseInsensitiveFilters = parser.options().containsKey( "i" );
boolean looseFilters = parser.options().containsKey( "l" );
Map filterMap = parseFilter( parser.options().get( "f" ), out );
if ( !displayProperties && !displayRelationships )
{
displayProperties = true;
displayRelationships = true;
}
NodeOrRelationship thing = null;
if ( parser.arguments().isEmpty() )
{
thing = this.getCurrent( session );
}
else
{
thing = NodeOrRelationship.wrap( this.getNodeById( Long
.parseLong( parser.arguments().get( 0 ) ) ) );
}
if ( displayProperties )
{
displayLabels( thing, out, filterMap, caseInsensitiveFilters, looseFilters, brief );
displayProperties( thing, out, verbose, quiet, filterMap, caseInsensitiveFilters,
looseFilters, brief );
}
if ( displayRelationships )
{
if ( thing.isNode() )
{
displayRelationships( parser, thing, session, out, verbose, quiet,
filterMap, caseInsensitiveFilters, looseFilters, brief );
}
else
{
displayNodes( parser, thing, session, out );
}
}
return Continuation.INPUT_COMPLETE;
}
private void displayNodes( AppCommandParser parser, NodeOrRelationship thing,
Session session, Output out ) throws RemoteException, ShellException
{
Relationship rel = thing.asRelationship();
out.println( getDisplayName( getServer(), session, rel.getStartNode(), false ) +
" --" + getDisplayName( getServer(), session, rel, true, false ) + "-> " +
getDisplayName( getServer(), session, rel.getEndNode(), false ) );
}
private Iterable sortKeys( Iterable source )
{
List list = new ArrayList();
for ( String item : source )
{
list.add( item );
}
Collections.sort( list, ( item1, item2 ) -> item1.toLowerCase().compareTo( item2.toLowerCase() ) );
return list;
}
private void displayProperties( NodeOrRelationship thing, Output out,
boolean verbose, boolean quiet, Map filterMap,
boolean caseInsensitiveFilters, boolean looseFilters, boolean brief )
throws RemoteException
{
ColumnPrinter columnPrinter = quiet ?
new ColumnPrinter( "*" ) :
new ColumnPrinter( "*", "=" );
int count = 0;
for ( String key : sortKeys( thing.getPropertyKeys() ) )
{
Object value = thing.getProperty( key );
if ( !filterMatches( filterMap, caseInsensitiveFilters, looseFilters, key, value ) )
{
continue;
}
count++;
if ( !brief )
{
if ( quiet )
{
columnPrinter.add( key );
}
else
{
columnPrinter.add( key, verbose ?
format( value, true ) + " (" + getNiceType( value ) + ")" :
format( value, true ) );
}
}
}
columnPrinter.print( out );
if ( brief )
{
out.println( "Property count: " + count );
}
}
private void displayLabels( NodeOrRelationship thing, Output out, Map filterMap,
boolean caseInsensitiveFilters, boolean looseFilters, boolean brief ) throws RemoteException
{
List labelNames = new ArrayList();
for ( Label label : thing.asNode().getLabels() )
labelNames.add( label.name() );
if ( brief )
{
out.println( "Label count: " + labelNames.size() );
}
else
{
for ( String label : sortKeys( labelNames ) )
{
if ( filterMatches( filterMap, caseInsensitiveFilters, looseFilters, label, "" ) )
out.println( ":" + label );
}
}
}
private void displayRelationships( AppCommandParser parser, NodeOrRelationship thing,
Session session, Output out, boolean verbose, boolean quiet,
Map filterMap, boolean caseInsensitiveFilters,
boolean looseFilters, boolean brief ) throws ShellException, RemoteException
{
boolean sortByType = parser.options().containsKey( "s" );
Node node = thing.asNode();
Iterable relationships = getRelationships( node, filterMap,
caseInsensitiveFilters, looseFilters, sortByType|brief );
if ( brief )
{
Iterator iterator = relationships.iterator();
if ( !iterator.hasNext() )
{
return;
}
Relationship sampleRelationship = iterator.next();
RelationshipType lastType = sampleRelationship.getType();
int currentCounter = 1;
while ( iterator.hasNext() )
{
Relationship rel = iterator.next();
if ( !rel.isType( lastType ) )
{
displayBriefRelationships( thing, session, out, sampleRelationship, currentCounter );
sampleRelationship = rel;
lastType = sampleRelationship.getType();
currentCounter = 1;
}
else
{
currentCounter++;
}
}
displayBriefRelationships( thing, session, out, sampleRelationship, currentCounter );
}
else
{
Iterator iterator = relationships.iterator();
if ( parser.options().containsKey( "m" ) )
{
iterator = wrapInLimitingIterator( parser, iterator, filterMap, caseInsensitiveFilters, looseFilters );
}
while ( iterator.hasNext() )
{
Relationship rel = iterator.next();
StringBuffer buf = new StringBuffer( getDisplayName(
getServer(), session, thing, true ) );
String relDisplay = quiet ? "" : getDisplayName( getServer(), session, rel, verbose, true );
buf.append( withArrows( rel, relDisplay, thing.asNode() ) );
buf.append( getDisplayName( getServer(), session, rel.getOtherNode( node ), true ) );
out.println( buf );
}
}
}
private Iterator wrapInLimitingIterator( AppCommandParser parser,
Iterator iterator, Map filterMap, boolean caseInsensitiveFilters,
boolean looseFilters ) throws ShellException
{
final AtomicBoolean handBreak = new AtomicBoolean();
int maxRelsPerType = parser.optionAsNumber( "m", DEFAULT_MAX_RELS_PER_TYPE_LIMIT ).intValue();
Map types = filterMapToTypes( getServer().getDb(),
Direction.BOTH, filterMap, caseInsensitiveFilters, looseFilters );
return new FilteringIterator( iterator,
new LimitPerTypeFilter( maxRelsPerType, types, handBreak ) )
{
@Override
protected Relationship fetchNextOrNull()
{
return handBreak.get() ? null : super.fetchNextOrNull();
}
};
}
private static class LimitPerTypeFilter implements Predicate
{
private final int maxRelsPerType;
private final Map encounteredRelationships = new HashMap();
private int typesMaxedOut = 0;
private final AtomicBoolean iterationHalted;
public LimitPerTypeFilter( int maxRelsPerType, Map types, AtomicBoolean handBreak )
{
this.maxRelsPerType = maxRelsPerType;
this.iterationHalted = handBreak;
for ( String type : types.keySet() )
{
encounteredRelationships.put( type, new AtomicInteger() );
}
}
@Override
public boolean test( Relationship item )
{
AtomicInteger counter = encounteredRelationships.get( item.getType().name() );
int count = counter.get();
if ( count < maxRelsPerType )
{
if ( counter.incrementAndGet() == maxRelsPerType )
{
counter.incrementAndGet();
if ( (++typesMaxedOut) >= encounteredRelationships.size() )
{
iterationHalted.set( true );
}
return true;
}
return true;
}
return false;
}
}
private Iterable getRelationships( final Node node, Map filterMap,
boolean caseInsensitiveFilters, boolean looseFilters, boolean sortByType ) throws ShellException
{
if ( sortByType )
{
Path nodeAsPath = Paths.singleNodePath( node );
return toSortedExpander( getServer().getDb(), Direction.BOTH, filterMap,
caseInsensitiveFilters, looseFilters ).expand( nodeAsPath, BranchState.NO_STATE );
}
else
{
if ( filterMap.isEmpty() )
{
return node.getRelationships();
}
else
{
Path nodeAsPath = Paths.singleNodePath( node );
return toExpander( getServer().getDb(), Direction.BOTH, filterMap, caseInsensitiveFilters, looseFilters ).expand( nodeAsPath, BranchState.NO_STATE );
}
}
}
private void displayBriefRelationships( NodeOrRelationship thing, Session session, Output out,
Relationship sampleRelationship, int count ) throws ShellException,
RemoteException
{
String relDisplay = withArrows( sampleRelationship, getDisplayName( getServer(), session,
sampleRelationship, false, true ), thing.asNode() );
out.println( getDisplayName( getServer(), session, thing, true ) + relDisplay + " x" + count );
}
private static String getNiceType( Object value )
{
return Set.getValueTypeName( value.getClass() );
}
}
© 2015 - 2025 Weber Informatics LLC | Privacy Policy