kingbase的数据类型与函数

2024-07-11 10:52:36 阅读:1 编辑
-- CREATE TYPE tablestruct;

CREATE TYPE tablestruct AS (
    "fields_key_name" varchar(100),
    "fields_name" varchar(200),
    "fields_type" varchar(20),
    "fields_length" int8,
    "fields_not_null" varchar(10),
    "fields_default" varchar(500),
    "fields_comment" varchar(1000));

CREATE OR REPLACE FUNCTION public.date_format(indate timestamp with time zone, intext varchar)
 RETURNS integer
 LANGUAGE plpgsql
AS $function$ 
BEGIN
IF upper(inText) = upper('%Y%m%d_%H%i') THEN
return to_char(inDate,'YYYYMMDD_HH24MI');
END IF;
IF upper(inText) = upper('%Y%m%d%H%i%s') THEN
return to_char(inDate,'YYYYMMDDHH24MISS');
END IF;
IF upper(inText) = upper('%Y-%m-%d %H') THEN
return to_char(inDate,'YYYY-MM-DD HH24');
END IF;
IF upper(inText) = upper('%Y-%m-%d') THEN
return to_char(inDate,'YYYY-MM-DD');
END IF;
IF upper(inText) = upper('%Y%M') THEN
return to_char(inDate,'YYYYMM');
END IF;
IF upper(inText) = upper('%Y-%m') THEN
return to_char(inDate,'YYYY-MM');
end if;
IF upper(inText) = upper('%m%d') THEN
return to_char(inDate,'MMDD');
END IF;
return 0;
END;
$function$
;

CREATE OR REPLACE FUNCTION public.find_in_set(str integer, strlist text)
 RETURNS integer
 LANGUAGE plpgsql
AS $function$  
DECLARE
     v_type integer;
BEGIN
    return text = ANY(string_to_array(strlist,','));

END;
$function$
;

CREATE OR REPLACE FUNCTION public.find_in_set(str text, strlist text)
 RETURNS integer
 LANGUAGE plpgsql
AS $function$  
DECLARE
     v_type integer;
BEGIN
    return text = ANY(string_to_array(strlist,','));

END;
$function$
;

CREATE OR REPLACE FUNCTION public.from_unixtime(c1 integer)
 RETURNS timestamp without time zone
 LANGUAGE plpgsql
AS $function$ 
DECLARE
     v_type datetime;
begin
     v_type:= to_timestamp(c1)::timestamp;
    RETURN  v_type;
END;
$function$
;

CREATE OR REPLACE FUNCTION public.from_unixtime(c1 integer, c2 varchar)
 RETURNS varchar
 LANGUAGE plpgsql
AS $function$
DECLARE
     v_type datetime;
BEGIN
     v_type:= to_timestamp(c1)::timestamp;
    RETURN  to_char(v_type,c2);
END;
$function$
;

CREATE OR REPLACE FUNCTION public.generate_subscripts(c1 anyarray, c2 integer)
 RETURNS integer
 LANGUAGE plpgsql
AS $function$  
DECLARE
     v_type integer;
BEGIN
      SELECT generate_series(array_lower(c1,c2), array_upper(c1,c2));

END;
$function$
;

CREATE OR REPLACE FUNCTION public.pgsql_type(a_type varchar)
 RETURNS varchar
 LANGUAGE plpgsql
AS $function$
DECLARE
     v_type varchar;
BEGIN
     IF a_type='int8' THEN
          v_type:='bigint';
     ELSIF a_type='int4' THEN
          v_type:='integer';
     ELSIF a_type='int2' THEN
          v_type:='smallint';
     ELSIF a_type='bpchar' THEN
          v_type:='char';
     ELSE
          v_type:=a_type;
     END IF;
     RETURN v_type;
END;
$function$
;

CREATE OR REPLACE FUNCTION public.radians(str varchar)
 RETURNS numeric
 LANGUAGE plpgsql
AS $function$  
DECLARE
     v_type decimal;
BEGIN
    return radians(cast(str as decimal));

END;
$function$
;

CREATE OR REPLACE FUNCTION public.table_msg(a_table_name varchar)
 RETURNS SETOF tablestruct
 LANGUAGE plpgsql
AS $function$
DECLARE
    v_ret tablestruct;
BEGIN
    FOR v_ret IN SELECT * FROM table_msg('public',a_table_name) LOOP
        RETURN NEXT v_ret;
    END LOOP;
    RETURN;
END;
$function$
;

CREATE OR REPLACE FUNCTION public.table_msg(a_schema_name varchar, a_table_name varchar)
 RETURNS SETOF tablestruct
 LANGUAGE plpgsql
AS $function$
DECLARE
     v_ret tablestruct;
     v_oid oid;
     v_sql varchar;
     v_rec RECORD;
     v_key varchar;
BEGIN
     SELECT
           pg_class.oid  INTO v_oid
     FROM
           pg_class
           INNER JOIN pg_namespace ON (pg_class.relnamespace = pg_namespace.oid AND lower(pg_namespace.nspname) = a_schema_name)
     WHERE
           pg_class.relname=a_table_name;
     IF NOT FOUND THEN
         RETURN;
     END IF;

    v_sql='
     SELECT
           pg_attribute.attname AS fields_name,
           pg_attribute.attnum AS fields_index,
           pgsql_type(pg_type.typname::varchar) AS fields_type,
           pg_attribute.atttypmod-4 as fields_length,
           CASE WHEN pg_attribute.attnotnull  THEN ''not null''
           ELSE ''''
           END AS fields_not_null,
           pg_attrdef.adnum AS fields_default,
           pg_description.description AS fields_comment
     FROM
           pg_attribute
           INNER JOIN pg_class  ON pg_attribute.attrelid = pg_class.oid
           INNER JOIN pg_type   ON pg_attribute.atttypid = pg_type.oid
           LEFT OUTER JOIN pg_attrdef ON pg_attrdef.adrelid = pg_class.oid AND pg_attrdef.adnum = pg_attribute.attnum
           LEFT OUTER JOIN pg_description ON pg_description.objoid = pg_class.oid AND pg_description.objsubid = pg_attribute.attnum
     WHERE
           pg_attribute.attnum > 0
           AND attisdropped <> ''t''
           AND pg_class.oid = ' || v_oid || '
     ORDER BY pg_attribute.attnum' ;

     FOR v_rec IN EXECUTE v_sql LOOP
         v_ret.fields_name=v_rec.fields_name;
         v_ret.fields_type=v_rec.fields_type;
         IF v_rec.fields_length > 0 THEN
            v_ret.fields_length:=v_rec.fields_length;
         ELSE
            v_ret.fields_length:=NULL;
         END IF;
         v_ret.fields_not_null=v_rec.fields_not_null;
         v_ret.fields_default=v_rec.fields_default;
         v_ret.fields_comment=v_rec.fields_comment;
         SELECT constraint_name INTO v_key FROM information_schema.key_column_usage WHERE table_schema=a_schema_name AND table_name=a_table_name AND column_name=v_rec.fields_name;
         IF FOUND THEN
            v_ret.fields_key_name=v_key;
         ELSE
            v_ret.fields_key_name='';
         END IF;
         RETURN NEXT v_ret;
     END LOOP;
     RETURN ;
END;
$function$
;

CREATE OR REPLACE FUNCTION public.to_days(c1 timestamp with time zone)
 RETURNS integer
 LANGUAGE plpgsql
AS $function$  
DECLARE
     v_type integer;
BEGIN
      v_type:= date(c1) - date '0001-01-01' + 366;
     RETURN v_type;
END;
$function$
;