Как получить динамический набор результатов при вызове функции postgresql - PullRequest
Новая
       36

Как получить динамический набор результатов при вызове функции postgresql

1 голос
/ 10 июня 2019

Я написал хранимую процедуру в PostgreSQL с условием if и else if. В каждом условии есть разные запросы, которые возвращают разные результаты. Как я могу получить доступ к результату в Java, используя метод resultset.getString()?

Хранимая процедура: 

CREATE FUNCTION public."testDynamicReturn"(_tagId character varying)
    RETURNS ??? -- what should be written in the place of ???
    LANGUAGE 'plpgsql'

AS $BODY$
BEGIN

        IF _tagId == 'stu' then 
            RETURN query
            SELECT "StudentRollNo.","StudentName","GuardianName" FROM public."DetailsOfStudents";
        ELSE if _tagId == 'teach' then
            RETURN query
            SELECT "TeacherName","TeacherAddress" FROM public."DetailsOfTeachers";

END;
$BODY$;

Java-код: 

    public static void main(String[] args) {
        Connection connection = DBUtility.getDBConnection();
        String sp = "{ call \"testDynamicReturn\"(?)}";

        try {
            CallableStatement statement = connection.prepareCall(sp);
            statement.setString(1, "stu");
            statement.execute();

            ResultSet resultSet = statement.getResultSet();
            while (resultSet.next()) {
                String a = resultSet.getString(???);
                System.out.println(a);
            }
        } catch (SQLException e) {
            // TODO Auto-generated catch block
            e.printStackTrace();
        }
    }

Если я отправлю "stu" в качестве параметра хранимой процедуры, результат будет: 

--------------------------------------------
StudentRollNo.| StudentName | GuardianName |
--------------------------------------------
15600223596   | John        |   Doe        |
--------------------------------------------

И если я отправлю "teach" в качестве параметра хранимой процедуры, то результат будет: 

------------------------------
TeacherName | TeacherAddress |
------------------------------
Adam        | New york       |
------------------------------

Пожалуйста, ответьте, как этого добиться.

1 Ответ

0 голосов
/ 10 июня 2019

Вот функция: 

CREATE OR REPLACE FUNCTION  public.testDynamicReturn(_tagId character varying)
    RETURNS TABLE (out_rollNum character varying, out_name character varying, out_gaurdian character varying) 
    AS 
    $func$
    BEGIN
     IF _tagId = 'stu' then 
               --validate your select, ensure it works independently
               RETURN QUERY  SELECT "StudentRollNo.","StudentName","GuardianName" FROM public."DetailsOfStudents";  
               END IF;
         IF _tagId = 'teach' then 
               --validate your select, ensure it works independently
               RETURN QUERY  SELECT "StudentRollNo.","StudentName", CAST (null AS character varying) "GuardianName" FROM public."DetailsOfStudents";
               END IF;

    END
    $func$  LANGUAGE plpgsql;

Теперь код для вызова этой динамической функции: 

 PreparedStatement statement = conn.prepareStatement("select * from public.testDynamicReturn('stu')");

ResultSet resultSet = statement.executeQuery();

while (resultSet.next()) {
 //can give column name
 String rollNum= resultSet.getString("out_rollNum");   
  String name= resultSet.getString("out_name");   
  //will be null for 'stu'but you already know
  String gaurdian= resultSet.getString("out_gaurdian");    
}
...