1
restaurant-search/server/db/search.js
2014-07-28 16:48:24 +09:00

161 lines
3.8 KiB
JavaScript

'use strict';
var _ = require('underscore');
var db_keywords = require('./keywords.json');
var db_data = require('./data.json');
function innerProduct(values1, values2) {
var result = 0;
for (var feature in values1) {
result += values1[feature] * (values2[feature] || 0.0);
}
return result;
}
function searchData(searchParams, minScore) {
var results = [];
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) {
results.push({
name: record['name'],
url: 'http://www.tripadvisor.com' + record['relativeUrl'],
score: score
});
}
}
results.sort(function(a, b) {
return b.score - a.score;
});
return results;
}
function searchStepper(range, steps, callback) {
var stepSize = (range.max - range.min) / steps;
for (var i = 0; i < steps; ++i) {
var stepMax = range.max - stepSize * i;
var stepMin = stepMax - stepSize;
var stepMid = (stepMin + stepMax) / 2;
callback(stepMid);
}
}
function searchProjection(searchParams, minScore, feature, range, steps) {
var testParams = _.clone(searchParams);
var results = [];
searchStepper(range, steps, function(position) {
testParams[feature] = position;
results.push({
sample: position,
values: searchData(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)
});
});
});
return results;
}
function searchBuildHints(searchParams, minScore, feature, range, steps) {
var projection = searchProjection(
searchParams,
minScore,
feature,
range,
steps
);
var hints = [];
_.each(projection, function(result) {
hints.push({
sample: result.sample,
count: result.values.length
});
});
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
});
});
return hints;
}
module.exports.getKeywords = function() {
return _.keys(db_keywords);
}
module.exports.execQuery = function(query) {
var searchParams = query.searchParams || db_keywords[query.keyword];
var searchResults = searchData(searchParams, query.minScore);
var graphColumns = { };
for (var feature in searchParams) {
var searchHints = searchBuildHints(
searchParams,
query.minScore,
feature,
query.searchRange,
query.hintSteps
);
graphColumns[feature] = {
color: '#607080',
value: searchParams[feature],
hints: searchHints,
steps: query.hintSteps
}
}
return {
columns: graphColumns,
params: searchParams,
items: searchResults
};
}