Search in sources :

Example 1 with NotLoginException

use of com.qiwenshare.common.exception.NotLoginException in project qiwen-file by qiwenshare.

the class OfficeController method IndexServlet.

@RequestMapping(value = "/IndexServlet", method = RequestMethod.POST)
@ResponseBody
public void IndexServlet(HttpServletResponse response, HttpServletRequest request) throws IOException {
    String token = request.getParameter("token");
    Long userId = userService.getUserIdByToken(token);
    if (userId == null) {
        throw new NotLoginException();
    }
    PrintWriter writer = response.getWriter();
    Scanner scanner = new Scanner(request.getInputStream()).useDelimiter("\\A");
    String body = scanner.hasNext() ? scanner.next() : "";
    JSONObject jsonObj = JSON.parseObject(body);
    log.info("===saveeditedfile:" + jsonObj.get("status"));
    ;
    String status = jsonObj != null ? jsonObj.get("status").toString() : "";
    if ("2".equals(status) || "6".equals(status)) {
        String type = request.getParameter("type");
        String downloadUri = (String) jsonObj.get("url");
        if ("edit".equals(type)) {
            // 修改报告
            String userFileId = request.getParameter("userFileId");
            UserFile userFile = userFileService.getById(userFileId);
            FileBean fileBean = fileService.getById(userFile.getFileId());
            Long pointCount = fileService.getFilePointCount(userFile.getFileId());
            if (pointCount > 1) {
                // 该场景,暂不支持编辑修改
                writer.write("{\"error\":1}");
                return;
            }
            URL url = new URL(downloadUri);
            HttpURLConnection connection = (HttpURLConnection) url.openConnection();
            int fileLength = 0;
            try {
                InputStream stream = connection.getInputStream();
                Writer writer1 = ufopFactory.getWriter(fileBean.getStorageType());
                WriteFile writeFile = new WriteFile();
                writeFile.setFileUrl(fileBean.getFileUrl());
                writeFile.setFileSize(connection.getContentLength());
                writer1.write(stream, writeFile);
            } catch (Exception e) {
                log.error(e.getMessage());
            } finally {
                if ("2".equals(status)) {
                    LambdaUpdateWrapper<UserFile> userFileUpdateWrapper = new LambdaUpdateWrapper<>();
                    userFileUpdateWrapper.set(UserFile::getUploadTime, DateUtil.getCurrentTime()).eq(UserFile::getUserFileId, userFileId);
                    userFileService.update(userFileUpdateWrapper);
                }
                LambdaUpdateWrapper<FileBean> lambdaUpdateWrapper = new LambdaUpdateWrapper<>();
                fileLength = connection.getContentLength();
                log.info("当前修改文件大小为:" + Long.valueOf(fileLength));
                DownloadFile downloadFile = new DownloadFile();
                downloadFile.setFileUrl(fileBean.getFileUrl());
                InputStream inputStream = ufopFactory.getDownloader(fileBean.getStorageType()).getInputStream(downloadFile);
                String md5Str = DigestUtils.md5Hex(inputStream);
                lambdaUpdateWrapper.set(FileBean::getIdentifier, md5Str).set(FileBean::getFileSize, Long.valueOf(fileLength)).set(FileBean::getModifyTime, DateUtil.getCurrentTime()).set(FileBean::getModifyUserId, userId).eq(FileBean::getFileId, fileBean.getFileId());
                fileService.update(lambdaUpdateWrapper);
                connection.disconnect();
            }
        }
    }
    if ("3".equals(status) || "7".equals(status)) {
        // 不强制手动保存时为6,"6".equals(status)
        log.debug("====保存失败:");
        writer.write("{\"error\":1}");
    } else {
        log.debug("状态为:0");
        writer.write("{\"error\":" + "0" + "}");
    }
}
Also used : Scanner(java.util.Scanner) WriteFile(com.qiwenshare.ufop.operation.write.domain.WriteFile) DownloadFile(com.qiwenshare.ufop.operation.download.domain.DownloadFile) FileInputStream(java.io.FileInputStream) InputStream(java.io.InputStream) URL(java.net.URL) IOException(java.io.IOException) NotLoginException(com.qiwenshare.common.exception.NotLoginException) LambdaUpdateWrapper(com.baomidou.mybatisplus.core.conditions.update.LambdaUpdateWrapper) HttpURLConnection(java.net.HttpURLConnection) JSONObject(com.alibaba.fastjson.JSONObject) NotLoginException(com.qiwenshare.common.exception.NotLoginException) UserFile(com.qiwenshare.file.domain.UserFile) FileBean(com.qiwenshare.file.domain.FileBean) Writer(com.qiwenshare.ufop.operation.write.Writer) PrintWriter(java.io.PrintWriter) PrintWriter(java.io.PrintWriter)

Aggregations

JSONObject (com.alibaba.fastjson.JSONObject)1 LambdaUpdateWrapper (com.baomidou.mybatisplus.core.conditions.update.LambdaUpdateWrapper)1 NotLoginException (com.qiwenshare.common.exception.NotLoginException)1 FileBean (com.qiwenshare.file.domain.FileBean)1 UserFile (com.qiwenshare.file.domain.UserFile)1 DownloadFile (com.qiwenshare.ufop.operation.download.domain.DownloadFile)1 Writer (com.qiwenshare.ufop.operation.write.Writer)1 WriteFile (com.qiwenshare.ufop.operation.write.domain.WriteFile)1 FileInputStream (java.io.FileInputStream)1 IOException (java.io.IOException)1 InputStream (java.io.InputStream)1 PrintWriter (java.io.PrintWriter)1 HttpURLConnection (java.net.HttpURLConnection)1 URL (java.net.URL)1 Scanner (java.util.Scanner)1