| /* |
| * Copyright (c) 1998, 2021 Oracle and/or its affiliates. All rights reserved. |
| * |
| * This program and the accompanying materials are made available under the |
| * terms of the Eclipse Public License v. 2.0 which is available at |
| * http://www.eclipse.org/legal/epl-2.0, |
| * or the Eclipse Distribution License v. 1.0 which is available at |
| * http://www.eclipse.org/org/documents/edl-v10.php. |
| * |
| * SPDX-License-Identifier: EPL-2.0 OR BSD-3-Clause |
| */ |
| |
| // Contributors: |
| // Oracle - initial API and implementation from Oracle TopLink |
| package org.eclipse.persistence.testing.tests.queries; |
| |
| import org.eclipse.persistence.testing.models.employee.domain.*; |
| import org.eclipse.persistence.expressions.*; |
| import org.eclipse.persistence.queries.*; |
| import org.eclipse.persistence.sessions.Session; |
| import org.eclipse.persistence.testing.framework.*; |
| import org.eclipse.persistence.sessions.server.ClientSession; |
| import org.eclipse.persistence.sessions.server.Server; |
| |
| /** |
| * Test the cursored stream feature by performing a cursor read on the database |
| * and comparing the contents to a normal query read. |
| */ |
| public class CursoredStreamReleaseConnectionsTest extends TestCase { |
| protected int size; |
| protected boolean useUOW; |
| protected Server serverSession; |
| protected ClientSession clientSession; |
| |
| public CursoredStreamReleaseConnectionsTest(boolean useUOW) { |
| super(); |
| this.useUOW = useUOW; |
| setDescription("This test verifies that the connections are released after executing a query using cursed stream"); |
| if (useUOW) { |
| this.setName(getName() + " -UnitOfWork"); |
| } else { |
| this.setName(getName() + " -ClientSession"); |
| } |
| } |
| |
| @Override |
| public void setup() { |
| org.eclipse.persistence.sessions.Project proj = new org.eclipse.persistence.testing.models.employee.relational.EmployeeProject(); |
| proj.setDatasourceLogin(getSession().getDatasourceLogin().clone()); |
| serverSession = proj.createServerSession(); |
| serverSession.setSessionLog(getSession().getSessionLog()); |
| serverSession.login(); |
| } |
| |
| @Override |
| public void reset() { |
| if (clientSession != null) { |
| this.clientSession.release(); |
| } |
| this.serverSession.logout(); |
| } |
| |
| @Override |
| public void test() { |
| CursoredStream stream = null; |
| |
| try { |
| ExpressionBuilder builder = new ExpressionBuilder(); |
| ReportQuery query = new ReportQuery(Employee.class, builder); |
| query.addAttribute("lastName"); |
| query.useCursoredStream(); |
| query.retrievePrimaryKeys(); |
| clientSession = serverSession.acquireClientSession(); |
| if (useUOW) { |
| Session uow = clientSession.acquireUnitOfWork(); |
| stream = (CursoredStream)uow.executeQuery(query); |
| } else { |
| stream = (CursoredStream)clientSession.executeQuery(query); |
| } |
| } finally { |
| if (stream != null) { |
| stream.close(); |
| } |
| } |
| } |
| |
| /** |
| * Verify if number of query objects matches number of cursor objects |
| */ |
| @Override |
| public void verify() { |
| int total = serverSession.getReadConnectionPool().getTotalNumberOfConnections(); |
| int totalAvailable = serverSession.getReadConnectionPool().getConnectionsAvailable().size(); |
| int totalUsedConnections = total - totalAvailable; |
| |
| if (totalUsedConnections != 0) { |
| throw new TestErrorException("Not all of the connections were released."); |
| } |
| } |
| } |