improve

Author: digoal

201802/20180202_01.md

@@ -797,11 +797,25 @@ postgres=# select set_limit(0.9);
 例子  
 
 ```  
-create or replace function get_res(text, int8) returns setof record as $$  
+create or replace function get_res(
+  text,     -- 要按相似搜的文本
+  int8,     -- 限制返回多少条
+  float4 default 0.3,   -- 相似度阈值，低于这个值不再搜搜
+  float4 default 0.1    -- 相似度递减步长，直至阈值
+) returns setof record as $$  
 declare  
   lim float4 := 1;  
 begin  
-    
+  -- 判定
+  if not ($3 <= 1 and $3 > 0) then 
+    raise notice '$3 must >0 and <=1';
+    return;
+  end if;
+  
+  if not ($4 > 0 and $4 < 1) then
+    raise notice '$4 must >0 and <=1';
+    return;
+  end if;
   loop  
     -- 设置相似度阈值  
     perform set_limit(lim);  
@@ -817,18 +831,18 @@ begin
   
     -- 否则继续，降低阈值  
     -- 当阈值小于0.3时，不再降阈值搜索，认为没有相似。  
-    if lim < 0.3 then  
+    if lim < $3 then  
       return;  
     else  
-      lim := lim - 0.1;  
+      lim := lim - $4;  
     end if;  
   end loop;  
 end;  
 $$ language plpgsql strict;  
 ```  
 
 ```  
-select * from get_res('输入搜索文本', 输入限制条数) as t(sml float4, id int, info text);  
+select * from get_res('输入搜索文本', 输入限制条数, 输入阈值, 输入步长) as t(sml float4, id int, info text);  
 ```  
 
 使用这个UDF搜索，既快又准。