Added function_score to Sense KB

Both on 0.90 and 1.0 apis. Closes elastic#177
spalger · Apr 24, 2014 · f4d578a · f4d578a
1 parent 3c26c52
commit f4d578a
Show file tree

Hide file tree

Showing 2 changed files with 140 additions and 5 deletions.
diff --git a/sense/app/kb/api_0_90/query.js b/sense/app/kb/api_0_90/query.js
@@ -1,4 +1,4 @@
-define(function () {
+define(["_"], function (_) {
   'use strict';
 
   var SPAN_QUERIES = {
@@ -20,6 +20,51 @@ define(function () {
     }
   };
 
+  var DECAY_FUNC_DESC = {
+      __template: {
+        "FIELD": {
+          "origin": "",
+          "scale": ""
+        }
+      },
+      "{field}": {
+        "origin": "",
+        "scale": "",
+        "offset": "",
+        "decay": 0.5
+      }
+    },
+    SCORING_FUNCS = {
+      "script_score": {
+        __template: {
+          "script": "_score * doc['f'].value"
+        },
+        "script": "",
+        "lang": "mvel",
+        "params": {
+
+        }
+      },
+      "boost_factor": 2.0,
+      "random_score": {
+        "seed": 314159265359
+      },
+      "linear": DECAY_FUNC_DESC,
+      "exp": DECAY_FUNC_DESC,
+      "gauss": DECAY_FUNC_DESC,
+      "field_value_factor": {
+        __template: {
+          "field": ""
+        },
+        "field": "{field}",
+        "factor": 1.2,
+        "modifier": {
+          __one_of: ["none", "log", "log1p", "log2p", "ln", "ln1p", "ln2p", "square", "sqrt", "reciprocal"]
+        }
+      }
+    };
+
+
   return function init(api) {
     api.addGlobalAutocompleteRules('query', {
       match: {
@@ -495,9 +540,31 @@ define(function () {
           relation: 'within'
         },
         __scope_link: '.filter.geo_shape'
-      }
+      },
+      function_score: _.defaults({
+        __template: {
+          query: {
 
+          },
+          functions: [
+            {}
+          ]
+        },
+        query: {},
+        filter: {},
+        functions: [
+          _.defaults(
+            {
+              filter: {}
+            },
+            SCORING_FUNCS
+          )
+        ],
+        boost: 1.0,
+        boost_mode: { __one_of: ["multiply", "replace", "sum", "avg", "max", "min"]},
+        score_mode: { __one_of: ["multiply", "sum", "first", "avg", "max", "min"]},
+        max_boost: 10
+      }, SCORING_FUNCS)
     });
   };
-
 });
diff --git a/sense/app/kb/api_1_0/query.js b/sense/app/kb/api_1_0/query.js
@@ -1,4 +1,4 @@
-define(function () {
+define(["_"], function (_) {
   'use strict';
 
   var SPAN_QUERIES = {
@@ -20,6 +20,50 @@ define(function () {
     }
   };
 
+  var DECAY_FUNC_DESC = {
+      __template: {
+        "FIELD": {
+          "origin": "",
+          "scale": ""
+        }
+      },
+      "{field}": {
+        "origin": "",
+        "scale": "",
+        "offset": "",
+        "decay": 0.5
+      }
+    },
+    SCORING_FUNCS = {
+      "script_score": {
+        __template: {
+          "script": "_score * doc['f'].value"
+        },
+        "script": "",
+        "lang": "mvel",
+        "params": {
+
+        }
+      },
+      "boost_factor": 2.0,
+      "random_score": {
+        "seed": 314159265359
+      },
+      "linear": DECAY_FUNC_DESC,
+      "exp": DECAY_FUNC_DESC,
+      "gauss": DECAY_FUNC_DESC,
+      "field_value_factor": {
+        __template: {
+          "field": ""
+        },
+        "field": "{field}",
+        "factor": 1.2,
+        "modifier": {
+          __one_of: ["none", "log", "log1p", "log2p", "ln", "ln1p", "ln2p", "square", "sqrt", "reciprocal"]
+        }
+      }
+    };
+
   return function init(api) {
     api.addGlobalAutocompleteRules('query', {
       match: {
@@ -495,7 +539,31 @@ define(function () {
           relation: 'within'
         },
         __scope_link: '.filter.geo_shape'
-      }
+      },
+      function_score: _.defaults({
+        __template: {
+          query: {
+
+          },
+          functions: [
+            {}
+          ]
+        },
+        query: {},
+        filter: {},
+        functions: [
+          _.defaults(
+            {
+              filter: {}
+            },
+            SCORING_FUNCS
+          )
+        ],
+        boost: 1.0,
+        boost_mode: { __one_of: ["multiply", "replace", "sum", "avg", "max", "min"]},
+        score_mode: { __one_of: ["multiply", "sum", "first", "avg", "max", "min"]},
+        max_boost: 10
+      }, SCORING_FUNCS)
 
     });
   };