alex/restaurant-search
1
Fork 0
Code Releases Activity

Search optimizaiton

Browse Source
This commit is contained in:
Alex Yatskov 2014-07-28 22:00:01 +09:00
parent 0d49b9d6a0
commit fc865ea2f8
1 changed files with 24 additions and 50 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

72
server/db/search.js
View File

@ -15,17 +15,32 @@ function innerProduct(values1, values2) {
return result; return result;
} }
function searchData(searchParams, minScore) { function countData(searchParams, minScore) {
var dataCount = 0;
for (var i = 0, count = db_data.length; i < count; ++i) {
var record = db_data[i];
var score = innerProduct(searchParams, record.rating);
if (score >= minScore) {
++dataCount;
}
}
return dataCount;
}
function findData(searchParams, minScore, maxResults) {
var results = []; var results = [];
for (var i = 0, count = db_data.length; i < count; ++i) { for (var i = 0, count = db_data.length; i < count; ++i) {
var record = db_data[i]; var record = db_data[i];
var score = innerProduct(searchParams, record['rating']); var score = innerProduct(searchParams, record.rating);
if (score >= minScore) { if (score >= minScore) {
results.push({ results.push({
name: record['name'], name: record.name,
url: 'http://www.tripadvisor.com' + record['relativeUrl'], url: 'http://www.tripadvisor.com' + record.relativeUrl,
score: score score: score
}); });
} }
@ -35,7 +50,7 @@ function searchData(searchParams, minScore) {
return b.score - a.score; return b.score - a.score;
}); });
return results; return results.slice(0, maxResults);
} }
function searchStepper(range, steps, callback) { function searchStepper(range, steps, callback) {
@ -58,26 +73,7 @@ function searchProjection(searchParams, minScore, feature, range, steps) {
testParams[feature] = position; testParams[feature] = position;
results.push({ results.push({
sample: position, sample: position,
values: searchData(testParams, minScore) count: countData(testParams, minScore)
});
});
return results;
}
function searchProjection2d(searchParams, minScore, feature1, feature2, range, steps) {
var testParams = _.clone(searchParams);
var results = [];
searchStepper(range, steps, function(sampleX) {
testParams[feature1] = sampleX;
searchStepper(range, steps, function(sampleY) {
testParams[feature2] = sampleY;
results.push({
sampleX: sampleX,
sampleY: sampleY,
values: searchData(testParams, minScore)
});
}); });
}); });
@ -97,29 +93,7 @@ function searchBuildHints(searchParams, minScore, feature, range, steps) {
_.each(projection, function(result) { _.each(projection, function(result) {
hints.push({ hints.push({
sample: result.sample, sample: result.sample,
count: result.values.length count: result.count
});
});
return hints;
}
function searchBuildHints2d(searchParams, minScore, feature1, feature2, range, steps) {
var projection = searchProjection2d(
searchParams,
minScore,
feature1,
feature2,
range,
steps
);
var hints = [];
_.each(projection, function(result) {
hints.push({
sampleX: result.sampleX,
sampleY: result.sampleY,
count: result.values.length
}); });
}); });
@ -132,7 +106,7 @@ module.exports.getKeywords = function() {
module.exports.execQuery = function(query) { module.exports.execQuery = function(query) {
var searchParams = query.searchParams || db_keywords[query.keyword]; var searchParams = query.searchParams || db_keywords[query.keyword];
var searchResults = searchData(searchParams, query.minScore); var searchResults = findData(searchParams, query.minScore, query.maxResults);
var graphColumns = {}; var graphColumns = {};
for (var feature in searchParams) { for (var feature in searchParams) {