org.neo4j.shell.kernel.apps.IndexProviderShellApp 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 commandsToRun = new ArrayList();
boolean specialCommand = false;
if ( doCd || doLs )
{
specialCommand = true;
if ( doCd )
{
commandsToRun.add( "cd -a $i" );
}
else if ( doLs )
{
commandsToRun.add( "ls $i" );
}
}
else if ( commandToRun != null )
{
commandsToRun.addAll( Arrays.asList( commandToRun.split( Pattern.quote( "&&" ) ) ) );
}
if ( getIndex( getIndexName( parser ), getEntityType( parser ), out ) == null )
{
return Continuation.INPUT_COMPLETE;
}
IndexHits result = query ? query( parser, out ) : get( parser, out );
try
{
for ( PropertyContainer hit : result )
{
printAndInterpretTemplateLines( commandsToRun, false, !specialCommand, NodeOrRelationship.wrap( hit ),
getServer(), session, out );
}
}
finally
{
result.close();
}
}
else if ( index )
{
index( parser, session, out );
}
else if ( remove )
{
if ( getIndex( getIndexName( parser ), Node.class, out ) == null )
{
return null;
}
remove( parser, session, out );
}
else if ( getConfig )
{
displayConfig( parser, out );
}
else if ( create )
{
createIndex( parser, out );
}
else if ( setConfig )
{
setConfig( parser, out );
}
else if ( delete )
{
deleteIndex( parser, out );
}
if ( indexes )
{
listIndexes( out );
}
return Continuation.INPUT_COMPLETE;
}
private String getIndexName( AppCommandParser parser ) throws ShellException
{
return parser.argument( 0, "Index name not supplied" );
}
private void listIndexes( Output out ) throws RemoteException
{
out.println( "Node indexes:" );
for ( String name : getServer().getDb().index().nodeIndexNames() )
{
out.println( " " + name );
}
out.println( "" );
out.println( "Relationship indexes:" );
for ( String name : getServer().getDb().index().relationshipIndexNames() )
{
out.println( " " + name );
}
}
private void deleteIndex( AppCommandParser parser, Output out ) throws RemoteException, ShellException
{
Index index = getIndex( getIndexName( parser ), getEntityType( parser ), out );
if ( index != null )
{
index.delete();
}
}
private void setConfig( AppCommandParser parser, Output out ) throws ShellException, RemoteException
{
String indexName = getIndexName( parser );
String key = parser.argument( 1, "Key not supplied" );
String value = parser.arguments().size() > 2 ? parser.arguments().get( 2 ) : null;
Class entityType = getEntityType( parser );
Index index = getIndex( indexName, entityType, out );
if ( index == null )
{
return;
}
String oldValue = value != null ?
getServer().getDb().index().setConfiguration( index, key, value ) :
getServer().getDb().index().removeConfiguration( index, key );
printWarning( out );
}
private void printWarning( Output out ) throws RemoteException
{
out.println( "INDEX CONFIGURATION CHANGED, INDEX DATA MAY BE INVALID" );
}
private void createIndex( AppCommandParser parser, Output out ) throws RemoteException, ShellException
{
String indexName = getIndexName( parser );
Class entityType = getEntityType( parser );
if ( getIndex( indexName, entityType, null ) != null )
{
out.println( entityType.getClass().getSimpleName() + " index '" + indexName + "' already exists" );
return;
}
Map config;
try
{
config = parser.arguments().size() >= 2 ? parseJSONMap( parser.arguments().get( 1 ) ) : null;
}
catch ( JSONException e )
{
throw ShellException.wrapCause( e );
}
if ( entityType.equals( Node.class ) )
{
Index index = config != null ? getServer().getDb().index().forNodes( indexName, config ) :
getServer().getDb().index().forNodes( indexName );
}
else
{
Index index = config != null ? getServer().getDb().index().forRelationships( indexName, config ) :
getServer().getDb().index().forRelationships( indexName );
}
}
private Index getIndex( String indexName, Class type, Output out )
throws RemoteException
{
IndexManager index = getServer().getDb().index();
boolean exists = (type.equals( Node.class ) && index.existsForNodes( indexName )) ||
(type.equals( Relationship.class ) && index.existsForRelationships( indexName ));
if ( !exists )
{
if ( out != null )
{
out.println( "No such " + type.getSimpleName().toLowerCase() + " index '" + indexName + "'" );
}
return null;
}
return (Index) (type.equals( Node.class ) ? index.forNodes( indexName ) : index.forRelationships( indexName ));
}
private void displayConfig( AppCommandParser parser, Output out )
throws RemoteException, ShellException
{
String indexName = getIndexName( parser );
Index index = getIndex( indexName, getEntityType( parser ), out );
if ( index == null )
{
return;
}
try
{
out.println( new JSONObject( getServer().getDb().index().getConfiguration( index ) ).toString( 4 ) );
}
catch ( JSONException e )
{
throw ShellException.wrapCause( e );
}
}
private Class getEntityType( AppCommandParser parser ) throws ShellException
{
String type = parser.options().get( "t" );
type = type != null ? type.toLowerCase() : null;
if ( type == null || type.equals( "node" ) )
{
return Node.class;
}
else if ( type.equals( "relationship" ) )
{
return Relationship.class;
}
throw new ShellException( "'type' expects one of [Node, Relationship]" );
}
private int boolCount( boolean... bools )
{
int count = 0;
for ( boolean bool : bools )
{
if ( bool )
{
count++;
}
}
return count;
}
private IndexHits get( AppCommandParser parser, Output out ) throws ShellException, RemoteException
{
String index = getIndexName( parser );
String key = parser.argument( 1, "Key not supplied" );
String value = parser.argument( 2, "Value not supplied" );
Index theIndex = getIndex( index, getEntityType( parser ), out );
return theIndex.get( key, value );
}
private IndexHits query( AppCommandParser parser, Output out ) throws RemoteException, ShellException
{
String index = getIndexName( parser );
String query1 = parser.argument( 1, "Key not supplied" );
String query2 = parser.argumentWithDefault( 2, null );
Index theIndex = getIndex( index, getEntityType( parser ), out );
return query2 != null ? theIndex.query( query1, query2 ) : theIndex.query( query1 );
}
private void index( AppCommandParser parser, Session session, Output out ) throws ShellException, RemoteException
{
NodeOrRelationship current = getCurrent( session );
String index = getIndexName( parser );
String key = parser.argument( 1, "Key not supplied" );
Object value = parser.arguments().size() > 2 ? parser.arguments().get( 2 ) : current.getProperty( key, null );
if ( value == null )
{
throw new ShellException( "No value to index" );
}
Index theIndex = current.isNode() ? getServer().getDb().index().forNodes( index ) :
getServer().getDb().index().forRelationships( index );
theIndex.add( current.asPropertyContainer(), key, value );
}
private void remove( AppCommandParser parser, Session session, Output out ) throws ShellException, RemoteException
{
NodeOrRelationship current = getCurrent( session );
String index = getIndexName( parser );
String key = parser.argumentWithDefault( 1, null );
Object value = null;
if ( key != null )
{
value = parser.argumentWithDefault( 2, null );
}
Index theIndex;
if ( current.isNode() )
{
theIndex = getIndex( index, Node.class, out );
}
else
{
theIndex = getIndex( index, Relationship.class, out );
}
if ( theIndex != null )
{
if ( key != null && value != null )
{
theIndex.remove( current.asPropertyContainer(), key, value );
}
else if ( key != null )
{
theIndex.remove( current.asPropertyContainer(), key );
}
else
{
theIndex.remove( current.asPropertyContainer() );
}
}
}
}
© 2015 - 2025 Weber Informatics LLC | Privacy Policy