|
|
@ -21,11 +21,11 @@ package io.entgra.device.mgt.core.certificate.mgt.core.dao.impl;
|
|
|
|
import io.entgra.device.mgt.core.device.mgt.common.CertificatePaginationRequest;
|
|
|
|
import io.entgra.device.mgt.core.device.mgt.common.CertificatePaginationRequest;
|
|
|
|
import io.entgra.device.mgt.core.certificate.mgt.core.dto.CertificateResponse;
|
|
|
|
import io.entgra.device.mgt.core.certificate.mgt.core.dto.CertificateResponse;
|
|
|
|
import io.entgra.device.mgt.core.certificate.mgt.core.impl.CertificateGenerator;
|
|
|
|
import io.entgra.device.mgt.core.certificate.mgt.core.impl.CertificateGenerator;
|
|
|
|
|
|
|
|
import org.apache.commons.lang.StringUtils;
|
|
|
|
import org.apache.commons.logging.Log;
|
|
|
|
import org.apache.commons.logging.Log;
|
|
|
|
import org.apache.commons.logging.LogFactory;
|
|
|
|
import org.apache.commons.logging.LogFactory;
|
|
|
|
import io.entgra.device.mgt.core.certificate.mgt.core.dao.CertificateManagementDAOException;
|
|
|
|
import io.entgra.device.mgt.core.certificate.mgt.core.dao.CertificateManagementDAOException;
|
|
|
|
import io.entgra.device.mgt.core.certificate.mgt.core.dao.CertificateManagementDAOFactory;
|
|
|
|
import io.entgra.device.mgt.core.certificate.mgt.core.dao.CertificateManagementDAOFactory;
|
|
|
|
import io.entgra.device.mgt.core.certificate.mgt.core.dao.CertificateManagementDAOUtil;
|
|
|
|
|
|
|
|
import io.entgra.device.mgt.core.certificate.mgt.core.service.PaginationResult;
|
|
|
|
import io.entgra.device.mgt.core.certificate.mgt.core.service.PaginationResult;
|
|
|
|
import org.wso2.carbon.context.PrivilegedCarbonContext;
|
|
|
|
import org.wso2.carbon.context.PrivilegedCarbonContext;
|
|
|
|
|
|
|
|
|
|
|
@ -47,8 +47,6 @@ public class SQLServerCertificateDAOImpl extends AbstractCertificateDAOImpl {
|
|
|
|
@Override
|
|
|
|
@Override
|
|
|
|
public PaginationResult getAllCertificates(CertificatePaginationRequest request) throws CertificateManagementDAOException {
|
|
|
|
public PaginationResult getAllCertificates(CertificatePaginationRequest request) throws CertificateManagementDAOException {
|
|
|
|
int tenantId = PrivilegedCarbonContext.getThreadLocalCarbonContext().getTenantId();
|
|
|
|
int tenantId = PrivilegedCarbonContext.getThreadLocalCarbonContext().getTenantId();
|
|
|
|
PreparedStatement stmt = null;
|
|
|
|
|
|
|
|
ResultSet resultSet = null;
|
|
|
|
|
|
|
|
CertificateResponse certificateResponse;
|
|
|
|
CertificateResponse certificateResponse;
|
|
|
|
List<CertificateResponse> certificates = new ArrayList<>();
|
|
|
|
List<CertificateResponse> certificates = new ArrayList<>();
|
|
|
|
PaginationResult paginationResult;
|
|
|
|
PaginationResult paginationResult;
|
|
|
@ -64,24 +62,24 @@ public class SQLServerCertificateDAOImpl extends AbstractCertificateDAOImpl {
|
|
|
|
String query = "SELECT * " +
|
|
|
|
String query = "SELECT * " +
|
|
|
|
"FROM DM_DEVICE_CERTIFICATE " +
|
|
|
|
"FROM DM_DEVICE_CERTIFICATE " +
|
|
|
|
"WHERE TENANT_ID = ? ";
|
|
|
|
"WHERE TENANT_ID = ? ";
|
|
|
|
if (serialNumber != null && !serialNumber.isEmpty()) {
|
|
|
|
if (StringUtils.isNotEmpty(serialNumber)) {
|
|
|
|
query += "AND SERIAL_NUMBER = ? ";
|
|
|
|
query += "AND SERIAL_NUMBER = ? ";
|
|
|
|
isCertificateSerialNumberProvided = true;
|
|
|
|
isCertificateSerialNumberProvided = true;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
if (deviceIdentifier != null && !deviceIdentifier.isEmpty()) {
|
|
|
|
if (StringUtils.isNotEmpty(deviceIdentifier)) {
|
|
|
|
query += "AND DEVICE_IDENTIFIER = ? ";
|
|
|
|
query += "AND DEVICE_IDENTIFIER = ? ";
|
|
|
|
isCertificateDeviceIdentifierProvided = true;
|
|
|
|
isCertificateDeviceIdentifierProvided = true;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
if (username != null && !username.isEmpty()) {
|
|
|
|
if (StringUtils.isNotEmpty(username)) {
|
|
|
|
query += "AND USERNAME LIKE ? ";
|
|
|
|
query += "AND USERNAME LIKE ? ";
|
|
|
|
isCertificateUsernameProvided = true;
|
|
|
|
isCertificateUsernameProvided = true;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
query += "ORDER BY ID OFFSET ? ROWS FETCH NEXT ? ROWS ONLY";
|
|
|
|
query += "ORDER BY ID OFFSET ? ROWS FETCH NEXT ? ROWS ONLY";
|
|
|
|
|
|
|
|
|
|
|
|
stmt = conn.prepareStatement(query);
|
|
|
|
try (PreparedStatement stmt = conn.prepareStatement(query)) {
|
|
|
|
int paramIdx = 1;
|
|
|
|
int paramIdx = 1;
|
|
|
|
stmt.setInt(paramIdx++, tenantId);
|
|
|
|
stmt.setInt(paramIdx++, tenantId);
|
|
|
|
if (isCertificateSerialNumberProvided) {
|
|
|
|
if (isCertificateSerialNumberProvided) {
|
|
|
@ -95,7 +93,7 @@ public class SQLServerCertificateDAOImpl extends AbstractCertificateDAOImpl {
|
|
|
|
}
|
|
|
|
}
|
|
|
|
stmt.setInt(paramIdx++, request.getStartIndex());
|
|
|
|
stmt.setInt(paramIdx++, request.getStartIndex());
|
|
|
|
stmt.setInt(paramIdx++, request.getRowCount());
|
|
|
|
stmt.setInt(paramIdx++, request.getRowCount());
|
|
|
|
resultSet = stmt.executeQuery();
|
|
|
|
try (ResultSet resultSet = stmt.executeQuery()) {
|
|
|
|
while (resultSet.next()) {
|
|
|
|
while (resultSet.next()) {
|
|
|
|
certificateResponse = new CertificateResponse();
|
|
|
|
certificateResponse = new CertificateResponse();
|
|
|
|
byte[] certificateBytes = resultSet.getBytes("CERTIFICATE");
|
|
|
|
byte[] certificateBytes = resultSet.getBytes("CERTIFICATE");
|
|
|
@ -110,13 +108,12 @@ public class SQLServerCertificateDAOImpl extends AbstractCertificateDAOImpl {
|
|
|
|
paginationResult = new PaginationResult();
|
|
|
|
paginationResult = new PaginationResult();
|
|
|
|
paginationResult.setData(certificates);
|
|
|
|
paginationResult.setData(certificates);
|
|
|
|
paginationResult.setRecordsTotal(this.getCertificateCount(request));
|
|
|
|
paginationResult.setRecordsTotal(this.getCertificateCount(request));
|
|
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
}
|
|
|
|
} catch (SQLException e) {
|
|
|
|
} catch (SQLException e) {
|
|
|
|
String errorMsg = "SQL error occurred while retrieving the certificates.";
|
|
|
|
String errorMsg = "SQL error occurred while retrieving the certificates.";
|
|
|
|
log.error(errorMsg, e);
|
|
|
|
log.error(errorMsg, e);
|
|
|
|
throw new CertificateManagementDAOException(errorMsg, e);
|
|
|
|
throw new CertificateManagementDAOException(errorMsg, e);
|
|
|
|
} finally {
|
|
|
|
|
|
|
|
CertificateManagementDAOUtil.cleanupResources(stmt, resultSet);
|
|
|
|
|
|
|
|
}
|
|
|
|
}
|
|
|
|
return paginationResult;
|
|
|
|
return paginationResult;
|
|
|
|
}
|
|
|
|
}
|
|
|
@ -131,56 +128,52 @@ public class SQLServerCertificateDAOImpl extends AbstractCertificateDAOImpl {
|
|
|
|
String serialNumber = request.getSerialNumber();
|
|
|
|
String serialNumber = request.getSerialNumber();
|
|
|
|
String deviceIdentifier = request.getDeviceIdentifier();
|
|
|
|
String deviceIdentifier = request.getDeviceIdentifier();
|
|
|
|
String username = request.getUsername();
|
|
|
|
String username = request.getUsername();
|
|
|
|
PreparedStatement stmt = null;
|
|
|
|
|
|
|
|
ResultSet rs = null;
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
try {
|
|
|
|
try {
|
|
|
|
Connection conn = this.getConnection();
|
|
|
|
Connection conn = this.getConnection();
|
|
|
|
StringBuilder queryBuilder = new StringBuilder("SELECT COUNT(*) AS DEVICE_CERTIFICATE_COUNT FROM DM_DEVICE_CERTIFICATE WHERE TENANT_ID = ?");
|
|
|
|
String sql = "SELECT COUNT(*) AS DEVICE_CERTIFICATE_COUNT " +
|
|
|
|
|
|
|
|
"FROM DM_DEVICE_CERTIFICATE " +
|
|
|
|
|
|
|
|
"WHERE TENANT_ID = ?";
|
|
|
|
|
|
|
|
|
|
|
|
if (serialNumber != null && !serialNumber.isEmpty()) {
|
|
|
|
if (StringUtils.isNotEmpty(serialNumber)) {
|
|
|
|
queryBuilder.append(" AND SERIAL_NUMBER = ?");
|
|
|
|
sql += " AND SERIAL_NUMBER = ?";
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
if (deviceIdentifier != null && !deviceIdentifier.isEmpty()) {
|
|
|
|
if (StringUtils.isNotEmpty(deviceIdentifier)) {
|
|
|
|
queryBuilder.append(" AND DEVICE_IDENTIFIER = ?");
|
|
|
|
sql += " AND DEVICE_IDENTIFIER = ?";
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
if (username != null && !username.isEmpty()) {
|
|
|
|
if (StringUtils.isNotEmpty(username)) {
|
|
|
|
queryBuilder.append(" AND USERNAME LIKE ?");
|
|
|
|
sql += " AND USERNAME LIKE ?";
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
String sql = queryBuilder.toString();
|
|
|
|
try (PreparedStatement stmt = conn.prepareStatement(sql)) {
|
|
|
|
|
|
|
|
|
|
|
|
stmt = conn.prepareStatement(sql);
|
|
|
|
|
|
|
|
stmt.setInt(1, tenantId);
|
|
|
|
stmt.setInt(1, tenantId);
|
|
|
|
|
|
|
|
|
|
|
|
int paramIdx = 2;
|
|
|
|
int paramIdx = 2;
|
|
|
|
if (serialNumber != null && !serialNumber.isEmpty()) {
|
|
|
|
if (StringUtils.isNotEmpty(serialNumber)) {
|
|
|
|
stmt.setString(paramIdx++, serialNumber);
|
|
|
|
stmt.setString(paramIdx++, serialNumber);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
if (deviceIdentifier != null && !deviceIdentifier.isEmpty()) {
|
|
|
|
if (StringUtils.isNotEmpty(deviceIdentifier)) {
|
|
|
|
stmt.setString(paramIdx++, deviceIdentifier);
|
|
|
|
stmt.setString(paramIdx++, deviceIdentifier);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
if (username != null && !username.isEmpty()) {
|
|
|
|
if (StringUtils.isNotEmpty(username)) {
|
|
|
|
stmt.setString(paramIdx, "%" + username + "%");
|
|
|
|
stmt.setString(paramIdx, "%" + username + "%");
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
rs = stmt.executeQuery();
|
|
|
|
try (ResultSet rs = stmt.executeQuery()) {
|
|
|
|
|
|
|
|
|
|
|
|
if (rs.next()) {
|
|
|
|
if (rs.next()) {
|
|
|
|
certificateCount = rs.getInt("DEVICE_CERTIFICATE_COUNT");
|
|
|
|
certificateCount = rs.getInt("DEVICE_CERTIFICATE_COUNT");
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
}
|
|
|
|
} catch (SQLException e) {
|
|
|
|
} catch (SQLException e) {
|
|
|
|
String errorMsg = "SQL error occurred while retrieving the certificate count.";
|
|
|
|
String errorMsg = "SQL error occurred while retrieving the certificate count.";
|
|
|
|
log.error(errorMsg, e);
|
|
|
|
log.error(errorMsg, e);
|
|
|
|
throw new CertificateManagementDAOException(errorMsg, e);
|
|
|
|
throw new CertificateManagementDAOException(errorMsg, e);
|
|
|
|
} finally {
|
|
|
|
|
|
|
|
CertificateManagementDAOUtil.cleanupResources(stmt, rs);
|
|
|
|
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
return certificateCount;
|
|
|
|
return certificateCount;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|