Examples with SQLExecuteResult com.varsql.core.sql.executor.SQLExecuteResult used on opensource projects

Example 1 with SQLExecuteResult

use of com.varsql.core.sql.executor.SQLExecuteResult in project varsql by varsqlinfo.

the class FileImportExportServiceImpl method jsonImport.

private SQLExecuteResult jsonImport(FileInfoEntity fileInfo, String conuid) throws SQLException {
    SqlStatementInfo ssi = new SqlStatementInfo();
    ssi.setConuid(conuid);
    ssi.setExportType(VarsqlFileType.JSON.name());
    ssi.addCustom(FileImportExecutor.IMPORT_FILE_PARAM_NAME, FileServiceUtils.getPath(fileInfo.getFilePath()).toUri());
    SQLExecuteResult ser = new FileImportExecutor().execute(ssi);
    return ser;
}

Example 2 with SQLExecuteResult

use of com.varsql.core.sql.executor.SQLExecuteResult in project varsql by varsqlinfo.

the class FileImportExportServiceImpl method importFile.

/**
 * @Method Name  : importFile
 * @Method 설명 : import file
 * @작성자   : ytkim
 * @작성일   : 2020. 11. 22.
 * @변경이력  :
 */
public ResponseResult importFile(FileImportInfo fii) {
    String conuid = fii.getConuid();
    String importType = fii.getImportType();
    List<FileInfoEntity> importFileList = fileInfoEntityRepository.findAllById(Arrays.asList(CommUtils.getFileIds(fii.getFileIds())));
    ResponseResult result = new ResponseResult();
    List<FileImportResult> resultInfo = new ArrayList<>();
    importFileList.forEach(fileInfo -> {
        FileImportResult fileImportResult = new FileImportResult();
        try {
            fileImportResult.setFileName(fileInfo.getFileName());
            SQLExecuteResult ser = null;
            if ("sql".equals(importType)) {
                ser = sqlImport(fileInfo, conuid);
            } else if ("xml".equals(importType)) {
                ser = xmlImport(fileInfo, conuid);
            } else if ("json".equals(importType)) {
                ser = jsonImport(fileInfo, conuid);
            } else if ("csv".equals(importType)) {
                ser = csvImport(fileInfo, conuid);
            }
            if (ser.getResultCode() != null) {
                fileImportResult.setResultCode(ser.getResultCode());
            }
            fileImportResult.setResultCount(ser.getExecuteCount());
            fileImportResult.setMessage(ser.getMessage());
        } catch (IOException | SQLException e) {
            fileImportResult.setResultCode(VarsqlAppCode.ERROR);
            fileImportResult.setMessage(e.getMessage());
            logger.error("file import error importType : {} , conuid : {} , fileInfo :{} ", importType, fileInfo);
            logger.error("error message : {}", e.getMessage(), e);
        }
        resultInfo.add(fileImportResult);
    });
    result.setItemList(resultInfo);
    return result;
}

Example 3 with SQLExecuteResult

use of com.varsql.core.sql.executor.SQLExecuteResult in project varsql by varsqlinfo.

the class FileImportExportServiceImpl method sqlImport.

/**
 * @method  : sqlImport
 * @desc : sql import
 * @author   : ytkim
 * @date   : 2021. 5. 23.
 * @param fileInfo
 * @param conuid
 * @return
 * @throws IOException
 * @throws SQLException
 */
private SQLExecuteResult sqlImport(FileInfoEntity fileInfo, String conuid) throws IOException, SQLException {
    SqlStatementInfo ssi = new SqlStatementInfo();
    ssi.setConuid(conuid);
    ssi.setSqlParam("{}");
    String insertQuery = FileUtils.readFileToString(FileServiceUtils.getFileInfoToFile(fileInfo));
    ssi.setSql(insertQuery);
    SQLExecuteResult ser = new SqlUpdateExecutor().execute(ssi, new AbstractSQLExecutorHandler() {

        @Override
        public boolean handle(SQLHandlerParameter handleParam) {
            return true;
        }
    });
    return ser;
}

Example 4 with SQLExecuteResult

use of com.varsql.core.sql.executor.SQLExecuteResult in project varsql by varsqlinfo.

the class FileImportExportServiceImpl method xmlImport.

private SQLExecuteResult xmlImport(FileInfoEntity fileInfo, String conuid) throws SQLException {
    SqlStatementInfo ssi = new SqlStatementInfo();
    ssi.setConuid(conuid);
    ssi.setExportType(VarsqlFileType.XML.name());
    ssi.addCustom(FileImportExecutor.IMPORT_FILE_PARAM_NAME, FileServiceUtils.getPath(fileInfo.getFilePath()).toUri());
    SQLExecuteResult ser = new FileImportExecutor().execute(ssi);
    return ser;
}

Example 5 with SQLExecuteResult

use of com.varsql.core.sql.executor.SQLExecuteResult in project varsql by varsqlinfo.

the class SQLServiceImpl method dataExport.

/**
 * 데이터 내보내기.
 * @param paramMap
 * @param req
 */
@SuppressWarnings("rawtypes")
public void dataExport(ParamMap paramMap, SqlExecuteDTO sqlExecuteInfo, HttpServletRequest req, HttpServletResponse res) throws Exception {
    String objectName = sqlExecuteInfo.getObjectName();
    if (!sqlExecuteInfo.getBaseSchema().equals(sqlExecuteInfo.getSchema())) {
        objectName = sqlExecuteInfo.getSchema() + "." + objectName;
    }
    StringBuilder reqQuerySb = new StringBuilder().append("select * from ").append(objectName).append(" where 1=1 ");
    String conditionQuery = paramMap.getString("conditionQuery");
    if (!StringUtils.isBlank(conditionQuery)) {
        conditionQuery = StringUtils.lTrim(conditionQuery);
        if (conditionQuery.startsWith("where")) {
            conditionQuery = conditionQuery.replaceFirst("where", "");
        }
        if (conditionQuery.startsWith("and")) {
            conditionQuery = conditionQuery.replaceFirst("and", "");
        }
        conditionQuery = " and " + conditionQuery;
        reqQuerySb.append(conditionQuery);
    }
    sqlExecuteInfo.setSqlParam("{}");
    sqlExecuteInfo.setSql(reqQuerySb.toString());
    String exportCharset = paramMap.getString("charset", VarsqlConstants.CHAR_SET);
    OutputStream outstream = null;
    AbstractWriter writer = null;
    try {
        VarsqlFileType exportType = sqlExecuteInfo.getExportType();
        Path fileExportPath = FileServiceUtils.getSavePath(UploadFileType.EXPORT);
        String downloadFilePath = FileUtils.pathConcat(fileExportPath.toAbsolutePath().toString(), exportType.concatExtension(VartechUtils.generateUUID()));
        outstream = new FileOutputStream(downloadFilePath);
        if (VarsqlFileType.CSV.equals(exportType)) {
            writer = new CSVWriter(outstream, ',', exportCharset);
        } else if (VarsqlFileType.JSON.equals(exportType)) {
            writer = new JSONWriter(outstream, "row", exportCharset);
        } else if (VarsqlFileType.XML.equals(exportType)) {
            writer = new XMLWriter(outstream, "row", exportCharset);
        } else if (VarsqlFileType.EXCEL.equals(exportType)) {
            writer = new ExcelWriter(outstream);
        } else {
            writer = new SQLWriter(outstream, DBType.getDBType(sqlExecuteInfo.getDbType()), objectName, exportCharset);
        }
        logger.debug("data export downloadFilePath :{} , query : {}", downloadFilePath, sqlExecuteInfo.getSql());
        final String tableName = objectName;
        SQLExecuteResult ser = new SelectExecutor().execute(sqlExecuteInfo, new AbstractSQLExecutorHandler(writer) {

            private boolean firstFlag = true;

            @Override
            public boolean handle(SQLHandlerParameter handleParam) {
                if (firstFlag) {
                    WriteMetadataInfo whi = new WriteMetadataInfo("exportInfo");
                    List<ExportColumnInfo> columns = new LinkedList<ExportColumnInfo>();
                    handleParam.getColumnInfoList().forEach(item -> {
                        ExportColumnInfo gci = new ExportColumnInfo();
                        gci.setName(item.getLabel());
                        gci.setAlias(item.getKey());
                        gci.setType(item.getDbType());
                        gci.setNumber(item.isNumber());
                        gci.setLob(item.isLob());
                        columns.add(gci);
                    });
                    whi.addMetedata("tableName", tableName);
                    whi.addMetedata("columns", columns);
                    getWriter().setMetadata(whi);
                    if (VarsqlFileType.SQL.equals(exportType)) {
                        ((SQLWriter) getWriter()).setColumnInfos(handleParam.getColumnInfoList());
                    }
                    firstFlag = false;
                }
                try {
                    getWriter().addRow(handleParam.getRowObject());
                } catch (IOException e) {
                    logger.error(e.getMessage(), e);
                    return false;
                }
                return true;
            }
        });
        writer.writeAndClose();
        if (ser.getResultCode() != null) {
            throw new DataDownloadException(ser.getResultCode(), ser.getMessage(), new VarsqlAppException(ser.getMessage()));
        }
        String exportFileName = ValidateUtils.getValidFileName(paramMap.getString("fileName", objectName));
        VarsqlUtils.setResponseDownAttr(res, req, exportType.concatExtension(exportFileName));
        try (FileInputStream fileInputStream = new FileInputStream(new File(downloadFilePath));
            OutputStream downloadStream = res.getOutputStream()) {
            byte[] buf = new byte[8192];
            int read = 0;
            while ((read = fileInputStream.read(buf)) != -1) {
                downloadStream.write(buf, 0, read);
            }
            downloadStream.flush();
            downloadStream.close();
            fileInputStream.close();
        }
    } catch (Exception e) {
        throw e;
    } finally {
        IOUtils.close(writer);
        IOUtils.close(outstream);
    }
}