Skip to content

Commit

Permalink
#1731 Accept missing values in sorts
Browse files Browse the repository at this point in the history
  • Loading branch information
To-om committed Feb 16, 2021
1 parent a323f48 commit c78dcbb
Show file tree
Hide file tree
Showing 6 changed files with 16 additions and 16 deletions.
2 changes: 1 addition & 1 deletion ScalliGraph
Submodule ScalliGraph updated 20 files
+6 −4 core/src/main/scala/org/thp/scalligraph/models/Database.scala
+5 −5 core/src/main/scala/org/thp/scalligraph/models/Model.scala
+1 −1 core/src/main/scala/org/thp/scalligraph/services/EdgeSrv.scala
+9 −9 core/src/main/scala/org/thp/scalligraph/services/IntegrityCheckOps.scala
+1 −1 core/src/main/scala/org/thp/scalligraph/services/VertexSrv.scala
+29 −17 core/src/main/scala/org/thp/scalligraph/traversal/Graph.scala
+0 −26 core/src/main/scala/org/thp/scalligraph/traversal/GraphStrategy.scala
+0 −45 core/src/main/scala/org/thp/scalligraph/traversal/OptimizeIndexStrategy.scala
+0 −46 core/src/main/scala/org/thp/scalligraph/traversal/RewriteOrderGlobalStepStrategy.scala
+2 −33 core/src/main/scala/org/thp/scalligraph/traversal/Traversal.scala
+1 −1 core/src/main/scala/org/thp/scalligraph/traversal/TraversalOps.scala
+117 −102 core/src/main/scala/org/thp/scalligraph/traversal/TraversalPrinter.scala
+1 −26 database/janusgraph/src/main/java/org/thp/scalligraph/janus/strategies/JanusGraphAcceptNullStrategy.java
+1 −0 database/janusgraph/src/main/java/org/thp/scalligraph/janus/strategies/JanusGraphStepAcceptNull.java
+63 −0 database/janusgraph/src/main/java/org/thp/scalligraph/janus/strategies/MultiComparatorAcceptNull.java
+14 −29 ...e/janusgraph/src/main/java/org/thp/scalligraph/janus/strategies/MultiDistinctOrderedIteratorAcceptNull.java
+34 −0 database/janusgraph/src/main/java/org/thp/scalligraph/janus/strategies/OrderAcceptNullStrategy.java
+189 −0 database/janusgraph/src/main/java/org/thp/scalligraph/janus/strategies/OrderGlobalStepAcceptNull.java
+112 −112 database/janusgraph/src/main/java/org/thp/scalligraph/janus/strategies/RewriteOrderGlobalStepStrategy.java
+27 −22 database/janusgraph/src/main/scala/org/thp/scalligraph/janus/JanusDatabase.scala
Original file line number Diff line number Diff line change
Expand Up @@ -11,16 +11,16 @@ import org.thp.scalligraph.auth.{AuthContext, AuthContextImpl, UserSrv => UserDB
import org.thp.scalligraph.janus.JanusDatabase
import org.thp.scalligraph.models._
import org.thp.scalligraph.services._
import org.thp.scalligraph.traversal.Graph
import org.thp.scalligraph.traversal.TraversalOps._
import org.thp.scalligraph.traversal.{Graph, Traversal}
import org.thp.thehive.connector.cortex.models.{CortexSchemaDefinition, TheHiveCortexSchemaProvider}
import org.thp.thehive.connector.cortex.services.{ActionSrv, JobSrv}
import org.thp.thehive.dto.v1.InputCustomFieldValue
import org.thp.thehive.{migration, ClusterSetup}
import org.thp.thehive.migration.IdMapping
import org.thp.thehive.migration.dto._
import org.thp.thehive.models._
import org.thp.thehive.services._
import org.thp.thehive.{migration, ClusterSetup}
import play.api.cache.SyncCacheApi
import play.api.cache.ehcache.EhCacheModule
import play.api.inject.guice.GuiceInjector
Expand Down Expand Up @@ -143,8 +143,8 @@ class Output @Inject() (
val alertsBuilder = Set.newBuilder[(String, String, String)]

db.roTransaction { implicit graph =>
Traversal
.V()
graph
.VV()
.unsafeHas(
"_label",
P.within(
Expand Down Expand Up @@ -265,7 +265,7 @@ class Output @Inject() (
}

def updateMetaData(entity: Entity, metaData: MetaData)(implicit graph: Graph): Unit = {
val vertex = Traversal.V(entity._id).head
val vertex = graph.VV(entity._id).head
UMapping.date.setProperty(vertex, "_createdAt", metaData.createdAt)
UMapping.date.optional.setProperty(vertex, "_updatedAt", metaData.updatedAt)
}
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -94,11 +94,11 @@ class PublicDashboard @Inject() (
Query.init[Traversal.V[Dashboard]](
"listDashboard",
(graph, authContext) =>
Traversal
graph
.union(
organisationSrv.filterTraversal(_).get(authContext.organisation).dashboards,
userSrv.filterTraversal(_).getByName(authContext.userId).dashboards
)(graph)
)
.dedup
)

Expand Down
2 changes: 1 addition & 1 deletion thehive/app/org/thp/thehive/controllers/v0/TaskCtrl.scala
Original file line number Diff line number Diff line change
Expand Up @@ -82,7 +82,7 @@ class TaskCtrl @Inject() (
.visible(organisationSrv)(authContext)
._id
.headOption
.fold[Traversal.V[Task]](Traversal.empty(graph))(c => taskSrv.startTraversal(graph).relatedTo(c))
.fold[Traversal.V[Task]](graph.empty)(c => taskSrv.startTraversal(graph).relatedTo(c))
)
entrypoint("search task in case")
.extract("query", searchParser(query))
Expand Down
12 changes: 6 additions & 6 deletions thehive/app/org/thp/thehive/models/TheHiveSchemaDefinition.scala
Original file line number Diff line number Diff line change
Expand Up @@ -11,8 +11,8 @@ import org.thp.scalligraph.EntityId
import org.thp.scalligraph.auth.AuthContext
import org.thp.scalligraph.janus.JanusDatabase
import org.thp.scalligraph.models._
import org.thp.scalligraph.traversal.Graph
import org.thp.scalligraph.traversal.TraversalOps._
import org.thp.scalligraph.traversal.{Graph, Traversal}
import org.thp.thehive.services.LocalUserSrv
import play.api.Logger

Expand Down Expand Up @@ -101,17 +101,17 @@ class TheHiveSchemaDefinition @Inject() extends Schema with UpdatableSchema {
.addProperty[String]("Taxonomy", "description")
.addProperty[Int]("Taxonomy", "version")
.dbOperation[Database]("Add Custom taxonomy vertex for each Organisation") { db =>
db.tryTransaction { implicit g =>
db.tryTransaction { implicit graph =>
// For each organisation, if there is no custom taxonomy, create it
db.labelFilter("Organisation", Traversal.V()).unsafeHas("name", P.neq("admin")).foreach { o =>
val hasFreetagsTaxonomy = Traversal
.V(EntityId(o.id))
graph.V("Organisation").unsafeHas("name", P.neq("admin")).foreach { o =>
val hasFreetagsTaxonomy = graph
.V("Organisation", EntityId(o.id))
.out[OrganisationTaxonomy]
.v[Taxonomy]
.unsafeHas("namespace", s"_freetags_${o.id()}")
.exists
if (!hasFreetagsTaxonomy) {
val taxoVertex = g.addVertex("Taxonomy")
val taxoVertex = graph.addVertex("Taxonomy")
taxoVertex.property("_label", "Taxonomy")
taxoVertex.property("_createdBy", "[email protected]")
taxoVertex.property("_createdAt", new Date())
Expand Down
2 changes: 1 addition & 1 deletion thehive/app/org/thp/thehive/services/FlowActor.scala
Original file line number Diff line number Diff line change
Expand Up @@ -66,7 +66,7 @@ class FlowActor extends Actor {
.limit(10)
._id
case Some(cid) =>
Traversal
graph
.union(
caseSrv.filterTraversal(_).get(cid).visible(organisationSrv).in[AuditContext],
observableSrv.filterTraversal(_).visible(organisationSrv).relatedTo(caseSrv.caseId(cid)).in[AuditContext],
Expand Down

0 comments on commit c78dcbb

Please sign in to comment.