เก็บผลลัพธ์การสืบค้นในตัวแปรโดยใช้ใน PL / pgSQL

วิธีกำหนดผลลัพธ์ของแบบสอบถามให้กับตัวแปรใน PL / pgSQL ซึ่งเป็นภาษาขั้นตอนของ PostgreSQL

ฉันมีฟังก์ชั่น:

CREATE OR REPLACE FUNCTION test(x numeric)
RETURNS character varying AS
$BODY$
DECLARE
name   character varying(255);
begin
 name ='SELECT name FROM test_table where id='||x;

 if(name='test')then
  --do somthing
 else
  --do the else part
 end if;
end;
return -- return my process result here
$BODY$
LANGUAGE plpgsql VOLATILE

ในฟังก์ชันข้างต้นฉันต้องการเก็บผลลัพธ์ของแบบสอบถามนี้:

'SELECT name FROM test_table where id='||x;

nameให้กับตัวแปร

จะดำเนินการอย่างไร?

ฉันคิดว่าคุณกำลังมองหาSELECT INTO:

select test_table.name into name from test_table where id = x;

นั่นจะดึงอาร์กิวเมนต์ของฟังก์ชันของคุณnameมาจากtest_tableที่idใดและปล่อยไว้ในnameตัวแปร อย่าปล่อยคำนำหน้าชื่อตารางไว้มิtest_table.nameฉะนั้นคุณจะได้รับการร้องเรียนเกี่ยวกับการอ้างอิงที่ไม่ชัดเจน

ตราบเท่าที่คุณกำหนดตัวแปรเดียวคุณยังสามารถใช้การกำหนดธรรมดาในฟังก์ชัน plpgsql:

name := (SELECT t.name from test_table t where t.id = x);

หรือใช้SELECT INTOเหมือนที่@mu จัดให้แล้ว .

ใช้งานได้เช่นกัน:

name := t.name from test_table t where t.id = x;

แต่ควรใช้หนึ่งในสองวิธีแรกที่ชัดเจนกว่าดังที่ @Pavel แสดงความคิดเห็น

ฉันย่อรูปแบบด้วยนามแฝงตารางเพิ่มเติม
ปรับปรุง: ฉันออกตัวอย่างเช่นรหัสของฉันและแนะนำให้ใช้IF EXISTS()แทนเช่นการให้บริการโดย @Pavel

รูปแบบปกติคือEXISTS(subselect):

BEGIN
  IF EXISTS(SELECT name
              FROM test_table t
             WHERE t.id = x
               AND t.name = 'test')
  THEN
     ---
  ELSE
     ---
  END IF;

รูปแบบนี้ใช้ใน PL / SQL, PL / pgSQL, SQL / PSM, ...

สร้างตารางการเรียนรู้:

CREATE TABLE "public"."learning" (
    "api_id" int4 DEFAULT nextval('share_api_api_id_seq'::regclass) NOT NULL,
    "title" varchar(255) COLLATE "default"
);

แทรกตารางการเรียนรู้ข้อมูล:

INSERT INTO "public"."learning" VALUES ('1', 'Google AI-01');
INSERT INTO "public"."learning" VALUES ('2', 'Google AI-02');
INSERT INTO "public"."learning" VALUES ('3', 'Google AI-01');

ขั้นตอน: 01

CREATE OR REPLACE FUNCTION get_all (pattern VARCHAR) RETURNS TABLE (
        learn_id INT,
        learn_title VARCHAR
) AS $$
BEGIN
    RETURN QUERY SELECT
        api_id,
        title
    FROM
        learning
    WHERE
        title = pattern ;
END ; $$ LANGUAGE 'plpgsql';

ขั้นตอน: 02

SELECT * FROM get_all('Google AI-01');

ขั้นตอน: 03

DROP FUNCTION get_all();

การสาธิต:

คุณสามารถใช้ตัวอย่างต่อไปนี้เพื่อจัดเก็บผลลัพธ์คิวรีในตัวแปรโดยใช้ PL / pgSQL:

 select * into demo from maintenanceactivitytrack ; 
    raise notice'p_maintenanceid:%',demo;