-
Notifications
You must be signed in to change notification settings - Fork 2.3k
/
BuiltInDbmsProcedures.java
113 lines (101 loc) · 4.24 KB
/
BuiltInDbmsProcedures.java
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
/*
* Copyright (c) 2002-2017 "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 <http://www.gnu.org/licenses/>.
*/
package org.neo4j.kernel.builtinprocs;
import java.util.Comparator;
import java.util.stream.Stream;
import org.neo4j.graphdb.security.AuthorizationViolationException;
import org.neo4j.kernel.api.proc.ProcedureSignature;
import org.neo4j.kernel.api.proc.UserFunctionSignature;
import org.neo4j.kernel.api.security.SecurityContext;
import org.neo4j.kernel.configuration.Config;
import org.neo4j.kernel.impl.proc.Procedures;
import org.neo4j.kernel.internal.GraphDatabaseAPI;
import org.neo4j.procedure.Context;
import org.neo4j.procedure.Description;
import org.neo4j.procedure.Name;
import org.neo4j.procedure.Procedure;
import static org.neo4j.graphdb.security.AuthorizationViolationException.PERMISSION_DENIED;
import static org.neo4j.procedure.Mode.DBMS;
@SuppressWarnings( "unused" )
public class BuiltInDbmsProcedures
{
@Context
public GraphDatabaseAPI graph;
@Context
public SecurityContext securityContext;
@Description( "List the currently active config of Neo4j." )
@Procedure( name = "dbms.listConfig", mode = DBMS )
public Stream<ConfigResult> listConfig( @Name( value = "searchString", defaultValue = "" ) String searchString )
{
securityContext.assertCredentialsNotExpired();
if ( !securityContext.isAdmin() )
{
throw new AuthorizationViolationException( PERMISSION_DENIED );
}
Config config = graph.getDependencyResolver().resolveDependency( Config.class );
return config.getConfigValues().values().stream()
.filter( c -> !c.internal() )
.map( ConfigResult::new )
.filter( c -> c.name.toLowerCase().contains( searchString.toLowerCase() ) )
.sorted( Comparator.comparing( c -> c.name ) );
}
@Description( "List all procedures in the DBMS." )
@Procedure( name = "dbms.procedures", mode = DBMS )
public Stream<ProcedureResult> listProcedures()
{
securityContext.assertCredentialsNotExpired();
return graph.getDependencyResolver().resolveDependency( Procedures.class ).getAllProcedures().stream()
.sorted( Comparator.comparing( a -> a.name().toString() ) )
.map( ProcedureResult::new );
}
@Description( "List all user functions in the DBMS." )
@Procedure(name = "dbms.functions", mode = DBMS)
public Stream<FunctionResult> listFunctions()
{
securityContext.assertCredentialsNotExpired();
return graph.getDependencyResolver().resolveDependency( Procedures.class ).getAllFunctions().stream()
.sorted( Comparator.comparing( a -> a.name().toString() ) )
.map( FunctionResult::new );
}
public static class FunctionResult
{
public final String name;
public final String signature;
public final String description;
private FunctionResult( UserFunctionSignature signature )
{
this.name = signature.name().toString();
this.signature = signature.toString();
this.description = signature.description().orElse( "" );
}
}
public static class ProcedureResult
{
public final String name;
public final String signature;
public final String description;
private ProcedureResult( ProcedureSignature signature )
{
this.name = signature.name().toString();
this.signature = signature.toString();
this.description = signature.description().orElse( "" );
}
}
}