| @@ -1,44 +1,22 @@ | |||||
| package cn.afterturn.easypoi.cache; | package cn.afterturn.easypoi.cache; | ||||
| import java.util.concurrent.TimeUnit; | |||||
| import com.google.common.cache.CacheBuilder; | |||||
| import com.google.common.cache.CacheLoader; | |||||
| import com.google.common.cache.LoadingCache; | |||||
| import cn.afterturn.easypoi.excel.entity.ExcelToHtmlParams; | import cn.afterturn.easypoi.excel.entity.ExcelToHtmlParams; | ||||
| import cn.afterturn.easypoi.excel.html.ExcelToHtmlService; | import cn.afterturn.easypoi.excel.html.ExcelToHtmlService; | ||||
| /** | /** | ||||
| * Excel 转变成为Html 的缓存 | * Excel 转变成为Html 的缓存 | ||||
| * | |||||
| * @author JueYue | * @author JueYue | ||||
| * 2015年8月7日 下午1:29:47 | |||||
| * 2015年8月7日 下午1:29:47 | |||||
| */ | */ | ||||
| public class HtmlCache { | public class HtmlCache { | ||||
| private static LoadingCache<ExcelToHtmlParams, String> loadingCache; | |||||
| static { | |||||
| loadingCache = CacheBuilder.newBuilder().expireAfterWrite(7, TimeUnit.DAYS).maximumSize(200) | |||||
| .build(new CacheLoader<ExcelToHtmlParams, String>() { | |||||
| @Override | |||||
| public String load(ExcelToHtmlParams params) throws Exception { | |||||
| return new ExcelToHtmlService(params).printPage(); | |||||
| } | |||||
| }); | |||||
| } | |||||
| public static String getHtml(ExcelToHtmlParams params) { | public static String getHtml(ExcelToHtmlParams params) { | ||||
| try { | try { | ||||
| return loadingCache.get(params); | |||||
| return new ExcelToHtmlService(params).printPage(); | |||||
| } catch (Exception e) { | } catch (Exception e) { | ||||
| throw new RuntimeException(e); | throw new RuntimeException(e); | ||||
| } | } | ||||
| } | } | ||||
| public static void setLoadingCache(LoadingCache<ExcelToHtmlParams, String> loadingCache) { | |||||
| HtmlCache.loadingCache = loadingCache; | |||||
| } | |||||
| } | } | ||||
| @@ -1,13 +1,13 @@ | |||||
| /** | /** | ||||
| * Copyright 2013-2015 JueYue (qrb.jueyue@gmail.com) | * Copyright 2013-2015 JueYue (qrb.jueyue@gmail.com) | ||||
| * | |||||
| * Licensed under the Apache License, Version 2.0 (the "License"); | |||||
| * you may not use this file except in compliance with the License. | |||||
| * You may obtain a copy of the License at | |||||
| * | |||||
| * http://www.apache.org/licenses/LICENSE-2.0 | |||||
| * | |||||
| * Unless required by applicable law or agreed to in writing, software | |||||
| * <p> | |||||
| * Licensed under the Apache License, Version 2.0 (the "License"); | |||||
| * you may not use this file except in compliance with the License. | |||||
| * You may obtain a copy of the License at | |||||
| * <p> | |||||
| * http://www.apache.org/licenses/LICENSE-2.0 | |||||
| * <p> | |||||
| * Unless required by applicable law or agreed to in writing, software | |||||
| * distributed under the License is distributed on an "AS IS" BASIS, | * distributed under the License is distributed on an "AS IS" BASIS, | ||||
| * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. | * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. | ||||
| * See the License for the specific language governing permissions and | * See the License for the specific language governing permissions and | ||||
| @@ -18,58 +18,43 @@ package cn.afterturn.easypoi.cache; | |||||
| import java.awt.image.BufferedImage; | import java.awt.image.BufferedImage; | ||||
| import java.io.ByteArrayOutputStream; | import java.io.ByteArrayOutputStream; | ||||
| import java.io.InputStream; | import java.io.InputStream; | ||||
| import java.util.concurrent.TimeUnit; | |||||
| import javax.imageio.ImageIO; | import javax.imageio.ImageIO; | ||||
| import org.apache.poi.util.IOUtils; | import org.apache.poi.util.IOUtils; | ||||
| import com.google.common.cache.CacheBuilder; | |||||
| import com.google.common.cache.CacheLoader; | |||||
| import com.google.common.cache.LoadingCache; | |||||
| import cn.afterturn.easypoi.cache.manager.POICacheManager; | import cn.afterturn.easypoi.cache.manager.POICacheManager; | ||||
| import org.slf4j.Logger; | |||||
| import org.slf4j.LoggerFactory; | |||||
| /** | /** | ||||
| * 图片缓存处理 | * 图片缓存处理 | ||||
| * | |||||
| * @author JueYue | * @author JueYue | ||||
| * 2016年1月8日 下午4:16:32 | |||||
| * 2016年1月8日 下午4:16:32 | |||||
| */ | */ | ||||
| public class ImageCache { | public class ImageCache { | ||||
| private static LoadingCache<String, byte[]> loadingCache; | |||||
| static { | |||||
| loadingCache = CacheBuilder.newBuilder().expireAfterWrite(1, TimeUnit.DAYS) | |||||
| .maximumSize(2000).build(new CacheLoader<String, byte[]>() { | |||||
| @Override | |||||
| public byte[] load(String imagePath) throws Exception { | |||||
| InputStream is = POICacheManager.getFile(imagePath); | |||||
| BufferedImage bufferImg = ImageIO.read(is); | |||||
| ByteArrayOutputStream byteArrayOut = new ByteArrayOutputStream(); | |||||
| try { | |||||
| ImageIO.write(bufferImg, | |||||
| imagePath.substring(imagePath.indexOf(".") + 1, imagePath.length()), | |||||
| byteArrayOut); | |||||
| return byteArrayOut.toByteArray(); | |||||
| } finally { | |||||
| IOUtils.closeQuietly(is); | |||||
| IOUtils.closeQuietly(byteArrayOut); | |||||
| } | |||||
| } | |||||
| }); | |||||
| } | |||||
| private static final Logger LOGGER = LoggerFactory | |||||
| .getLogger(ImageCache.class); | |||||
| public static byte[] getImage(String imagePath) { | public static byte[] getImage(String imagePath) { | ||||
| InputStream is = POICacheManager.getFile(imagePath); | |||||
| ByteArrayOutputStream byteArrayOut = new ByteArrayOutputStream(); | |||||
| try { | try { | ||||
| return loadingCache.get(imagePath); | |||||
| BufferedImage bufferImg = ImageIO.read(is); | |||||
| ImageIO.write(bufferImg, | |||||
| imagePath.substring(imagePath.indexOf(".") + 1, imagePath.length()), | |||||
| byteArrayOut); | |||||
| return byteArrayOut.toByteArray(); | |||||
| } catch (Exception e) { | } catch (Exception e) { | ||||
| LOGGER.error(e.getMessage(), e); | |||||
| return null; | return null; | ||||
| } finally { | |||||
| IOUtils.closeQuietly(is); | |||||
| IOUtils.closeQuietly(byteArrayOut); | |||||
| } | } | ||||
| } | } | ||||
| public static void setLoadingCache(LoadingCache<String, byte[]> loadingCache) { | |||||
| ImageCache.loadingCache = loadingCache; | |||||
| } | |||||
| } | } | ||||
| @@ -26,8 +26,9 @@ import java.net.URLConnection; | |||||
| /** | /** | ||||
| * 文件加载类,根据路径加载指定文件 | * 文件加载类,根据路径加载指定文件 | ||||
| * | |||||
| * @author JueYue | * @author JueYue | ||||
| * 2014年2月10日 | |||||
| * 2014年2月10日 | |||||
| * @version 1.0 | * @version 1.0 | ||||
| */ | */ | ||||
| public class FileLoaderImpl implements IFileLoader { | public class FileLoaderImpl implements IFileLoader { | ||||
| @@ -66,7 +67,7 @@ public class FileLoaderImpl implements IFileLoader { | |||||
| } | } | ||||
| baos.flush(); | baos.flush(); | ||||
| return baos.toByteArray(); | return baos.toByteArray(); | ||||
| } catch (IOException e) { | |||||
| } catch (Exception e) { | |||||
| LOGGER.error(e.getMessage(), e); | LOGGER.error(e.getMessage(), e); | ||||
| } finally { | } finally { | ||||
| IOUtils.closeQuietly(fileis); | IOUtils.closeQuietly(fileis); | ||||
| @@ -1,13 +1,13 @@ | |||||
| /** | /** | ||||
| * Copyright 2013-2015 JueYue (qrb.jueyue@gmail.com) | * Copyright 2013-2015 JueYue (qrb.jueyue@gmail.com) | ||||
| * | |||||
| * Licensed under the Apache License, Version 2.0 (the "License"); | |||||
| * you may not use this file except in compliance with the License. | |||||
| * You may obtain a copy of the License at | |||||
| * | |||||
| * http://www.apache.org/licenses/LICENSE-2.0 | |||||
| * | |||||
| * Unless required by applicable law or agreed to in writing, software | |||||
| * <p> | |||||
| * Licensed under the Apache License, Version 2.0 (the "License"); | |||||
| * you may not use this file except in compliance with the License. | |||||
| * You may obtain a copy of the License at | |||||
| * <p> | |||||
| * http://www.apache.org/licenses/LICENSE-2.0 | |||||
| * <p> | |||||
| * Unless required by applicable law or agreed to in writing, software | |||||
| * distributed under the License is distributed on an "AS IS" BASIS, | * distributed under the License is distributed on an "AS IS" BASIS, | ||||
| * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. | * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. | ||||
| * See the License for the specific language governing permissions and | * See the License for the specific language governing permissions and | ||||
| @@ -18,19 +18,13 @@ package cn.afterturn.easypoi.cache.manager; | |||||
| import java.io.ByteArrayInputStream; | import java.io.ByteArrayInputStream; | ||||
| import java.io.InputStream; | import java.io.InputStream; | ||||
| import java.util.Arrays; | import java.util.Arrays; | ||||
| import java.util.concurrent.ExecutionException; | |||||
| import java.util.concurrent.TimeUnit; | |||||
| import org.slf4j.Logger; | import org.slf4j.Logger; | ||||
| import org.slf4j.LoggerFactory; | import org.slf4j.LoggerFactory; | ||||
| import com.google.common.cache.CacheBuilder; | |||||
| import com.google.common.cache.CacheLoader; | |||||
| import com.google.common.cache.LoadingCache; | |||||
| /** | /** | ||||
| * 缓存管理 | * 缓存管理 | ||||
| * | |||||
| * | |||||
| * @author JueYue | * @author JueYue | ||||
| * 2014年2月10日 | * 2014年2月10日 | ||||
| * 2015年10月17日 | * 2015年10月17日 | ||||
| @@ -38,57 +32,41 @@ import com.google.common.cache.LoadingCache; | |||||
| */ | */ | ||||
| public final class POICacheManager { | public final class POICacheManager { | ||||
| private static final Logger LOGGER = LoggerFactory | |||||
| .getLogger(POICacheManager.class); | |||||
| private static LoadingCache<String, byte[]> loadingCache; | |||||
| private static final Logger LOGGER = LoggerFactory | |||||
| .getLogger(POICacheManager.class); | |||||
| private static IFileLoader fileLoder; | |||||
| private static IFileLoader fileLoader = new FileLoaderImpl(); | |||||
| private static ThreadLocal<IFileLoader> LOCAL_FILELOADER = new ThreadLocal<IFileLoader>(); | |||||
| static { | |||||
| loadingCache = CacheBuilder.newBuilder().expireAfterWrite(1, TimeUnit.HOURS).maximumSize(50) | |||||
| .build(new CacheLoader<String, byte[]>() { | |||||
| @Override | |||||
| public byte[] load(String url) throws Exception { | |||||
| if (LOCAL_FILELOADER.get() != null) { | |||||
| return LOCAL_FILELOADER.get().getFile(url); | |||||
| } | |||||
| return fileLoder.getFile(url); | |||||
| } | |||||
| }); | |||||
| //设置默认实现 | |||||
| fileLoder = new FileLoaderImpl(); | |||||
| } | |||||
| private static ThreadLocal<IFileLoader> LOCAL_FILELOADER = new ThreadLocal<IFileLoader>(); | |||||
| public static InputStream getFile(String id) { | public static InputStream getFile(String id) { | ||||
| try { | try { | ||||
| byte[] result; | |||||
| //复杂数据,防止操作原数据 | //复杂数据,防止操作原数据 | ||||
| byte[] result = Arrays.copyOf(loadingCache.get(id), loadingCache.get(id).length); | |||||
| if (LOCAL_FILELOADER.get() != null) { | |||||
| result = LOCAL_FILELOADER.get().getFile(id); | |||||
| } | |||||
| result = fileLoader.getFile(id); | |||||
| result = Arrays.copyOf(result, result.length); | |||||
| return new ByteArrayInputStream(result); | return new ByteArrayInputStream(result); | ||||
| } catch (ExecutionException e) { | |||||
| } catch (Exception e) { | |||||
| LOGGER.error(e.getMessage(), e); | LOGGER.error(e.getMessage(), e); | ||||
| } | } | ||||
| return null; | return null; | ||||
| } | } | ||||
| public static void setFileLoder(IFileLoader fileLoder) { | |||||
| POICacheManager.fileLoder = fileLoder; | |||||
| public static void setFileLoader(IFileLoader fileLoader) { | |||||
| POICacheManager.fileLoader = fileLoader; | |||||
| } | } | ||||
| /** | /** | ||||
| * 一次线程有效 | * 一次线程有效 | ||||
| * @param fileLoder | |||||
| * @param fileLoader | |||||
| */ | */ | ||||
| public static void setFileLoderOnce(IFileLoader fileLoder) { | |||||
| if (fileLoder != null) { | |||||
| LOCAL_FILELOADER.set(fileLoder); | |||||
| public static void setFileLoaderOnce(IFileLoader fileLoader) { | |||||
| if (fileLoader != null) { | |||||
| LOCAL_FILELOADER.set(fileLoader); | |||||
| } | } | ||||
| } | } | ||||
| public static void setLoadingCache(LoadingCache<String, byte[]> loadingCache) { | |||||
| POICacheManager.loadingCache = loadingCache; | |||||
| } | |||||
| } | } | ||||