pg存储过程 字符串转数组后交叉比对

dragon76

最近因工作需要，碰到一个临时对帐的需求，有两列数据，数据中包含以空格分隔的客户名称字符串，

比对要求：只要字符串中有一个客户名称在另一列中存在即为真，否则返回否;当两列数据生成数组元素只有一个时，数据只要被相互包含，则返回真。

例1：

数据列一"AAA BBB CCC",数据列二"DDD CCC EEE"，返回真

例2：

数据列一"AAA BBB CCC",数据列二"DDD EEE FFF"，返回否

例3：

数据列一"AA",数据列二"BBBAABBB",返回真

想在数据库查询中直接返回结果，所以在postgresql中通过PL实现

 

1. -- Function: mydiff(character varying, character varying)
   
2. 
   
3. -- DROP FUNCTION mydiff(character varying, character varying);
   
4. 
   
5. CREATE OR REPLACE FUNCTION mydiff(character varying, character varying)
   
6.   RETURNS boolean AS
   
7. $BODY$
   
8. declare
   
9. sz1 alias for $1;
   
10. sz2 alias for $2;
    
11. 
    
12. bdjg boolean;
    
13. sza character varying[];
    
14. szb character varying[];
    
15. max integer;
    
16. begin
    
17.    sz1:=trim(sz1);
    
18.    sz2:=trim(sz2);
    
19.    IF length(sz1)<1 then
    
20.        return false;
    
21.    else
    
22.        if length(sz2)<1 then
    
23.            return false;
    
24.        else
    
25.            sza:=regexp_split_to_array(sz1,E'\\s+');
    
26.            szb:=regexp_split_to_array(sz2,E'\\s+');
    
27.            bdjg:=false;
    
28.            
    
29.            if array_upper(sza,1)>array_upper(szb,1) then
    
30.               for i in 1..ARRAY_upper(szb,1) LOOP
    
31.                   bdjg:=(szb[i]=any(sza));
    
32.                   if bdjg then
    
33.                      EXIT;
    
34.                   end if;
    
35.               end loop;
    
36.            else
    
37.               for i in 1..ARRAY_upper(sza,1) LOOP
    
38.                 bdjg:=(sza[i]=any(szb));
    
39.                 if bdjg then
    
40.                    EXIT;
    
41.                 end if;
    
42.               end loop;
    
43.            end if;
    
44. 
    
45.            if array_upper(sza,1)=1 and array_upper(szb,1)=1 then
    
46.               if position(sza[1] in szb[1])>0 or position(szb[1] in sza[1])>0 then
    
47.                  bdjg:=true;
    
48.               end if;
    
49.            end if;
    
50.            
    
51.           return bdjg;
    
52.        end if;
    
53. 
    
54.    end if;
    
55. end;
    
56. $BODY$
    
57.   LANGUAGE plpgsql VOLATILE
    
58.   COST 100;
    
59. ALTER FUNCTION mydiff(character varying, character varying) OWNER TO postgres;