Sqlserver
 sql >> Base de données >  >> RDS >> Sqlserver

Comment récupérer * tout * d'une procédure stockée à l'aide de JDBC

Lorsque nous exécutons une procédure stockée dans JDBC, nous obtenons une série de zéro ou plusieurs "résultats". Nous pouvons ensuite traiter ces "résultats" de manière séquentielle en appelant CallableStatement#getMoreResults() . Chaque "résultat" peut contenir

  • zéro ou plusieurs lignes de données que nous pouvons récupérer avec un ResultSet objet,
  • un nombre de mises à jour pour une instruction DML (INSERT, UPDATE, DELETE) que nous pouvons récupérer avec CallableStatement#getUpdateCount() , ou
  • une erreur qui lève une exception SQLServerException.

Pour "Issue 1" le problème est souvent que la procédure stockée ne commence pas par SET NOCOUNT ON; et exécute une instruction DML avant de faire un SELECT pour produire un ensemble de résultats. Le nombre de mises à jour pour le DML est renvoyé comme premier "résultat", et les lignes de données sont "bloquées derrière" jusqu'à ce que nous appelions getMoreResults .

"Issue 2" est essentiellement le même problème. La procédure stockée produit un "résultat" (généralement un SELECT ou éventuellement un nombre de mises à jour) avant que l'erreur ne se produise. L'erreur est renvoyée dans un "résultat" ultérieur et ne provoque pas d'exception tant que nous ne l'avons pas "récupérée" à l'aide de getMoreResults .

Dans de nombreux cas, le problème peut être évité en ajoutant simplement SET NOCOUNT ON; comme première instruction exécutable dans la procédure stockée. Cependant, un changement de procédure stockée n'est pas toujours possible et il n'en reste pas moins que pour obtenir tout de retour de la procédure stockée, nous devons continuer à appeler getMoreResults jusqu'à ce que, comme le dit le Javadoc :

There are no more results when the following is true: 

     // stmt is a Statement object
     ((stmt.getMoreResults() == false) && (stmt.getUpdateCount() == -1))

Cela semble assez simple, mais comme d'habitude, "le diable est dans les détails", comme l'illustre l'exemple suivant. Pour une procédure stockée SQL Server...

ALTER PROCEDURE dbo.TroublesomeSP AS
BEGIN
    -- note: no `SET NOCOUNT ON;`
    DECLARE @tbl TABLE (id VARCHAR(3) PRIMARY KEY);

    DROP TABLE NonExistent;
    INSERT INTO @tbl (id) VALUES ('001');
    SELECT id FROM @tbl;
    INSERT INTO @tbl (id) VALUES ('001');  -- duplicate key error
    SELECT 1/0;  -- error _inside_ ResultSet
    INSERT INTO @tbl (id) VALUES ('101');
    INSERT INTO @tbl (id) VALUES ('201'),('202');
    SELECT id FROM @tbl;
END

... le code Java suivant renverra tout ...

try (CallableStatement cs = conn.prepareCall("{call dbo.TroublesomeSP}")) {
    boolean resultSetAvailable = false;
    int numberOfResultsProcessed = 0;
    try {
        resultSetAvailable = cs.execute();
    } catch (SQLServerException sse) {
        System.out.printf("Exception thrown on execute: %s%n%n", sse.getMessage());
        numberOfResultsProcessed++;
    }
    int updateCount = -2;  // initialize to impossible(?) value
    while (true) {
        boolean exceptionOccurred = true; 
        do {
            try {
                if (numberOfResultsProcessed > 0) {
                    resultSetAvailable = cs.getMoreResults();
                }
                exceptionOccurred = false;
                updateCount = cs.getUpdateCount();
            } catch (SQLServerException sse) {
                System.out.printf("Current result is an exception: %s%n%n", sse.getMessage());
            }
            numberOfResultsProcessed++;
        } while (exceptionOccurred);

        if ((!resultSetAvailable) && (updateCount == -1)) {
            break;  // we're done
        }

        if (resultSetAvailable) {
            System.out.println("Current result is a ResultSet:");
            try (ResultSet rs = cs.getResultSet()) {
                try {
                    while (rs.next()) {
                        System.out.println(rs.getString(1));
                    }
                } catch (SQLServerException sse) {
                    System.out.printf("Exception while processing ResultSet: %s%n", sse.getMessage());
                }
            }
        } else {
            System.out.printf("Current result is an update count: %d %s affected%n",
                    updateCount,
                    updateCount == 1 ? "row was" : "rows were");
        }
        System.out.println();
    }
    System.out.println("[end of results]");
}

... produisant la sortie de console suivante :

Exception thrown on execute: Cannot drop the table 'NonExistent', because it does not exist or you do not have permission.

Current result is an update count: 1 row was affected

Current result is a ResultSet:
001

Current result is an exception: Violation of PRIMARY KEY constraint 'PK__#314D4EA__3213E83F3335971A'. Cannot insert duplicate key in object '[email protected]'. The duplicate key value is (001).

Current result is a ResultSet:
Exception while processing ResultSet: Divide by zero error encountered.

Current result is an update count: 1 row was affected

Current result is an update count: 2 rows were affected

Current result is a ResultSet:
001
101
201
202

[end of results]