Page Menu
Home
c4science
Search
Configure Global Search
Log In
Files
F80923266
PhabricatorSearchEngineMySQL.php
No One
Temporary
Actions
Download File
Edit File
Delete File
View Transforms
Subscribe
Mute Notifications
Award Token
Subscribers
None
File Metadata
Details
File Info
Storage
Attached
Created
Tue, Sep 3, 06:53
Size
7 KB
Mime Type
text/x-php
Expires
Thu, Sep 5, 06:53 (2 d)
Engine
blob
Format
Raw Data
Handle
20479665
Attached To
rPH Phabricator
PhabricatorSearchEngineMySQL.php
View Options
<?php
/**
* @group search
*/
final
class
PhabricatorSearchEngineMySQL
extends
PhabricatorSearchEngine
{
public
function
reindexAbstractDocument
(
PhabricatorSearchAbstractDocument
$doc
)
{
$phid
=
$doc
->
getPHID
();
if
(!
$phid
)
{
throw
new
Exception
(
"Document has no PHID!"
);
}
$store
=
new
PhabricatorSearchDocument
();
$store
->
setPHID
(
$doc
->
getPHID
());
$store
->
setDocumentType
(
$doc
->
getDocumentType
());
$store
->
setDocumentTitle
(
$doc
->
getDocumentTitle
());
$store
->
setDocumentCreated
(
$doc
->
getDocumentCreated
());
$store
->
setDocumentModified
(
$doc
->
getDocumentModified
());
$store
->
replace
();
$conn_w
=
$store
->
establishConnection
(
'w'
);
$field_dao
=
new
PhabricatorSearchDocumentField
();
queryfx
(
$conn_w
,
'DELETE FROM %T WHERE phid = %s'
,
$field_dao
->
getTableName
(),
$phid
);
foreach
(
$doc
->
getFieldData
()
as
$field
)
{
list
(
$ftype
,
$corpus
,
$aux_phid
)
=
$field
;
queryfx
(
$conn_w
,
'INSERT INTO %T (phid, phidType, field, auxPHID, corpus) '
.
' VALUES (%s, %s, %s, %ns, %s)'
,
$field_dao
->
getTableName
(),
$phid
,
$doc
->
getDocumentType
(),
$ftype
,
$aux_phid
,
$corpus
);
}
$sql
=
array
();
foreach
(
$doc
->
getRelationshipData
()
as
$relationship
)
{
list
(
$rtype
,
$to_phid
,
$to_type
,
$time
)
=
$relationship
;
$sql
[]
=
qsprintf
(
$conn_w
,
'(%s, %s, %s, %s, %d)'
,
$phid
,
$to_phid
,
$rtype
,
$to_type
,
$time
);
}
$rship_dao
=
new
PhabricatorSearchDocumentRelationship
();
queryfx
(
$conn_w
,
'DELETE FROM %T WHERE phid = %s'
,
$rship_dao
->
getTableName
(),
$phid
);
if
(
$sql
)
{
queryfx
(
$conn_w
,
'INSERT INTO %T'
.
' (phid, relatedPHID, relation, relatedType, relatedTime) '
.
' VALUES %Q'
,
$rship_dao
->
getTableName
(),
implode
(
', '
,
$sql
));
}
}
/**
* Rebuild the PhabricatorSearchAbstractDocument that was used to index
* an object out of the index itself. This is primarily useful for debugging,
* as it allows you to inspect the search index representation of a
* document.
*
* @param phid PHID of a document which exists in the search index.
* @return null|PhabricatorSearchAbstractDocument Abstract document object
* which corresponds to the original abstract document used to
* build the document index.
*/
public
function
reconstructDocument
(
$phid
)
{
$dao_doc
=
new
PhabricatorSearchDocument
();
$dao_field
=
new
PhabricatorSearchDocumentField
();
$dao_relationship
=
new
PhabricatorSearchDocumentRelationship
();
$t_doc
=
$dao_doc
->
getTableName
();
$t_field
=
$dao_field
->
getTableName
();
$t_relationship
=
$dao_relationship
->
getTableName
();
$doc
=
queryfx_one
(
$dao_doc
->
establishConnection
(
'r'
),
'SELECT * FROM %T WHERE phid = %s'
,
$t_doc
,
$phid
);
if
(!
$doc
)
{
return
null
;
}
$fields
=
queryfx_all
(
$dao_field
->
establishConnection
(
'r'
),
'SELECT * FROM %T WHERE phid = %s'
,
$t_field
,
$phid
);
$relationships
=
queryfx_all
(
$dao_relationship
->
establishConnection
(
'r'
),
'SELECT * FROM %T WHERE phid = %s'
,
$t_relationship
,
$phid
);
$adoc
=
id
(
new
PhabricatorSearchAbstractDocument
())
->
setPHID
(
$phid
)
->
setDocumentType
(
$doc
[
'documentType'
])
->
setDocumentTitle
(
$doc
[
'documentTitle'
])
->
setDocumentCreated
(
$doc
[
'documentCreated'
])
->
setDocumentModified
(
$doc
[
'documentModified'
]);
foreach
(
$fields
as
$field
)
{
$adoc
->
addField
(
$field
[
'field'
],
$field
[
'corpus'
],
$field
[
'auxPHID'
]);
}
foreach
(
$relationships
as
$relationship
)
{
$adoc
->
addRelationship
(
$relationship
[
'relation'
],
$relationship
[
'relatedPHID'
],
$relationship
[
'relatedType'
],
$relationship
[
'relatedTime'
]);
}
return
$adoc
;
}
public
function
executeSearch
(
PhabricatorSearchQuery
$query
)
{
$where
=
array
();
$join
=
array
();
$order
=
'ORDER BY documentCreated DESC'
;
$dao_doc
=
new
PhabricatorSearchDocument
();
$dao_field
=
new
PhabricatorSearchDocumentField
();
$t_doc
=
$dao_doc
->
getTableName
();
$t_field
=
$dao_field
->
getTableName
();
$conn_r
=
$dao_doc
->
establishConnection
(
'r'
);
$q
=
$query
->
getQuery
();
if
(
strlen
(
$q
))
{
$join
[]
=
qsprintf
(
$conn_r
,
'%T field ON field.phid = document.phid'
,
$t_field
);
$where
[]
=
qsprintf
(
$conn_r
,
'MATCH(corpus) AGAINST (%s IN BOOLEAN MODE)'
,
$q
);
// When searching for a string, promote user listings above other
// listings.
$order
=
qsprintf
(
$conn_r
,
'ORDER BY
IF(documentType = %s, 0, 1) ASC,
MAX(MATCH(corpus) AGAINST (%s)) DESC'
,
'USER'
,
$q
);
$field
=
$query
->
getParameter
(
'field'
);
if
(
$field
/* && $field != AdjutantQuery::FIELD_ALL*/
)
{
$where
[]
=
qsprintf
(
$conn_r
,
'field.field = %s'
,
$field
);
}
}
$exclude
=
$query
->
getParameter
(
'exclude'
);
if
(
$exclude
)
{
$where
[]
=
qsprintf
(
$conn_r
,
'document.phid != %s'
,
$exclude
);
}
if
(
$query
->
getParameter
(
'type'
))
{
if
(
strlen
(
$q
))
{
// TODO: verify that this column actually does something useful in query
// plans once we have nontrivial amounts of data.
$where
[]
=
qsprintf
(
$conn_r
,
'field.phidType = %s'
,
$query
->
getParameter
(
'type'
));
}
$where
[]
=
qsprintf
(
$conn_r
,
'document.documentType = %s'
,
$query
->
getParameter
(
'type'
));
}
$join
[]
=
$this
->
joinRelationship
(
$conn_r
,
$query
,
'author'
,
PhabricatorSearchRelationship
::
RELATIONSHIP_AUTHOR
);
$join
[]
=
$this
->
joinRelationship
(
$conn_r
,
$query
,
'open'
,
PhabricatorSearchRelationship
::
RELATIONSHIP_OPEN
);
$join
[]
=
$this
->
joinRelationship
(
$conn_r
,
$query
,
'owner'
,
PhabricatorSearchRelationship
::
RELATIONSHIP_OWNER
);
$join
[]
=
$this
->
joinRelationship
(
$conn_r
,
$query
,
'subscribers'
,
PhabricatorSearchRelationship
::
RELATIONSHIP_SUBSCRIBER
);
$join
[]
=
$this
->
joinRelationship
(
$conn_r
,
$query
,
'project'
,
PhabricatorSearchRelationship
::
RELATIONSHIP_PROJECT
);
$join
[]
=
$this
->
joinRelationship
(
$conn_r
,
$query
,
'repository'
,
PhabricatorSearchRelationship
::
RELATIONSHIP_REPOSITORY
);
$join
=
array_filter
(
$join
);
foreach
(
$join
as
$key
=>
$clause
)
{
$join
[
$key
]
=
' JOIN '
.
$clause
;
}
$join
=
implode
(
' '
,
$join
);
if
(
$where
)
{
$where
=
'WHERE '
.
implode
(
' AND '
,
$where
);
}
else
{
$where
=
''
;
}
$offset
=
(
int
)
$query
->
getParameter
(
'offset'
,
0
);
$limit
=
(
int
)
$query
->
getParameter
(
'limit'
,
25
);
$hits
=
queryfx_all
(
$conn_r
,
'SELECT
document.phid
FROM %T document
%Q
%Q
GROUP BY document.phid
%Q
LIMIT %d, %d'
,
$t_doc
,
$join
,
$where
,
$order
,
$offset
,
$limit
);
return
ipull
(
$hits
,
'phid'
);
}
protected
function
joinRelationship
(
AphrontDatabaseConnection
$conn
,
PhabricatorSearchQuery
$query
,
$field
,
$type
)
{
$phids
=
$query
->
getParameter
(
$field
,
array
());
if
(!
$phids
)
{
return
null
;
}
$is_existence
=
false
;
switch
(
$type
)
{
case
PhabricatorSearchRelationship
::
RELATIONSHIP_OPEN
:
$is_existence
=
true
;
break
;
}
$sql
=
qsprintf
(
$conn
,
'%T AS %C ON %C.phid = document.phid AND %C.relation = %s'
,
id
(
new
PhabricatorSearchDocumentRelationship
())->
getTableName
(),
$field
,
$field
,
$field
,
$type
);
if
(!
$is_existence
)
{
$sql
.=
qsprintf
(
$conn
,
' AND %C.relatedPHID in (%Ls)'
,
$field
,
$phids
);
}
return
$sql
;
}
}
Event Timeline
Log In to Comment