221 lines
		
	
	
		
			5.1 KiB
		
	
	
	
		
			Java
		
	
	
	
	
	
			
		
		
	
	
			221 lines
		
	
	
		
			5.1 KiB
		
	
	
	
		
			Java
		
	
	
	
	
	
| /*
 | |
|  * Copyright (C) 2004-2015 L2J Server
 | |
|  * 
 | |
|  * This file is part of L2J Server.
 | |
|  * 
 | |
|  * L2J Server is free software: you can redistribute it and/or modify
 | |
|  * it under the terms of the GNU General Public License as published by
 | |
|  * the Free Software Foundation, either version 3 of the License, or
 | |
|  * (at your option) any later version.
 | |
|  * 
 | |
|  * L2J Server is distributed in the hope that it will be useful,
 | |
|  * but WITHOUT ANY WARRANTY; without even the implied warranty of
 | |
|  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
 | |
|  * General Public License for more details.
 | |
|  * 
 | |
|  * You should have received a copy of the GNU General Public License
 | |
|  * along with this program. If not, see <http://www.gnu.org/licenses/>.
 | |
|  */
 | |
| package com.l2jserver.gameserver.cache;
 | |
| 
 | |
| import java.io.BufferedInputStream;
 | |
| import java.io.File;
 | |
| import java.io.FileInputStream;
 | |
| import java.util.HashMap;
 | |
| import java.util.Map;
 | |
| import java.util.concurrent.ConcurrentHashMap;
 | |
| import java.util.logging.Level;
 | |
| import java.util.logging.Logger;
 | |
| 
 | |
| import com.l2jserver.Config;
 | |
| import com.l2jserver.gameserver.util.Util;
 | |
| import com.l2jserver.util.file.filter.HTMLFilter;
 | |
| 
 | |
| /**
 | |
|  * @author Layane
 | |
|  */
 | |
| public class HtmCache
 | |
| {
 | |
| 	private static final Logger _log = Logger.getLogger(HtmCache.class.getName());
 | |
| 	
 | |
| 	private static final HTMLFilter HTML_FILTER = new HTMLFilter();
 | |
| 	
 | |
| 	private static final Map<String, String> _cache = Config.LAZY_CACHE ? new ConcurrentHashMap<>() : new HashMap<>();
 | |
| 	
 | |
| 	private int _loadedFiles;
 | |
| 	private long _bytesBuffLen;
 | |
| 	
 | |
| 	protected HtmCache()
 | |
| 	{
 | |
| 		reload();
 | |
| 	}
 | |
| 	
 | |
| 	public void reload()
 | |
| 	{
 | |
| 		reload(Config.DATAPACK_ROOT);
 | |
| 	}
 | |
| 	
 | |
| 	public void reload(File f)
 | |
| 	{
 | |
| 		if (!Config.LAZY_CACHE)
 | |
| 		{
 | |
| 			_log.info("Html cache start...");
 | |
| 			parseDir(f);
 | |
| 			_log.info("Cache[HTML]: " + String.format("%.3f", getMemoryUsage()) + " megabytes on " + getLoadedFiles() + " files loaded");
 | |
| 		}
 | |
| 		else
 | |
| 		{
 | |
| 			_cache.clear();
 | |
| 			_loadedFiles = 0;
 | |
| 			_bytesBuffLen = 0;
 | |
| 			_log.info("Cache[HTML]: Running lazy cache");
 | |
| 		}
 | |
| 	}
 | |
| 	
 | |
| 	public void reloadPath(File f)
 | |
| 	{
 | |
| 		parseDir(f);
 | |
| 		_log.info("Cache[HTML]: Reloaded specified path.");
 | |
| 	}
 | |
| 	
 | |
| 	public double getMemoryUsage()
 | |
| 	{
 | |
| 		return ((float) _bytesBuffLen / 1048576);
 | |
| 	}
 | |
| 	
 | |
| 	public int getLoadedFiles()
 | |
| 	{
 | |
| 		return _loadedFiles;
 | |
| 	}
 | |
| 	
 | |
| 	private void parseDir(File dir)
 | |
| 	{
 | |
| 		final File[] files = dir.listFiles();
 | |
| 		if (files != null)
 | |
| 		{
 | |
| 			for (File file : files)
 | |
| 			{
 | |
| 				if (!file.isDirectory())
 | |
| 				{
 | |
| 					loadFile(file);
 | |
| 				}
 | |
| 				else
 | |
| 				{
 | |
| 					parseDir(file);
 | |
| 				}
 | |
| 			}
 | |
| 		}
 | |
| 	}
 | |
| 	
 | |
| 	public String loadFile(File file)
 | |
| 	{
 | |
| 		if (!HTML_FILTER.accept(file))
 | |
| 		{
 | |
| 			return null;
 | |
| 		}
 | |
| 		
 | |
| 		final String relpath = Util.getRelativePath(Config.DATAPACK_ROOT, file);
 | |
| 		String content = null;
 | |
| 		try (FileInputStream fis = new FileInputStream(file);
 | |
| 			BufferedInputStream bis = new BufferedInputStream(fis))
 | |
| 		{
 | |
| 			final int bytes = bis.available();
 | |
| 			byte[] raw = new byte[bytes];
 | |
| 			
 | |
| 			bis.read(raw);
 | |
| 			content = new String(raw, "UTF-8");
 | |
| 			content = content.replaceAll("(?s)<!--.*?-->", ""); // Remove html comments
 | |
| 			
 | |
| 			String oldContent = _cache.put(relpath, content);
 | |
| 			if (oldContent == null)
 | |
| 			{
 | |
| 				_bytesBuffLen += bytes;
 | |
| 				_loadedFiles++;
 | |
| 			}
 | |
| 			else
 | |
| 			{
 | |
| 				_bytesBuffLen = (_bytesBuffLen - oldContent.length()) + bytes;
 | |
| 			}
 | |
| 		}
 | |
| 		catch (Exception e)
 | |
| 		{
 | |
| 			_log.log(Level.WARNING, "Problem with htm file " + e.getMessage(), e);
 | |
| 		}
 | |
| 		return content;
 | |
| 	}
 | |
| 	
 | |
| 	public String getHtmForce(String prefix, String path)
 | |
| 	{
 | |
| 		String content = getHtm(prefix, path);
 | |
| 		if (content == null)
 | |
| 		{
 | |
| 			content = "<html><body>My text is missing:<br>" + path + "</body></html>";
 | |
| 			_log.warning("Cache[HTML]: Missing HTML page: " + path);
 | |
| 		}
 | |
| 		return content;
 | |
| 	}
 | |
| 	
 | |
| 	public String getHtm(String prefix, String path)
 | |
| 	{
 | |
| 		String newPath = null;
 | |
| 		String content;
 | |
| 		if ((prefix != null) && !prefix.isEmpty())
 | |
| 		{
 | |
| 			newPath = prefix + path;
 | |
| 			content = getHtm(newPath);
 | |
| 			if (content != null)
 | |
| 			{
 | |
| 				return content;
 | |
| 			}
 | |
| 		}
 | |
| 		
 | |
| 		content = getHtm(path);
 | |
| 		if ((content != null) && (newPath != null))
 | |
| 		{
 | |
| 			_cache.put(newPath, content);
 | |
| 		}
 | |
| 		
 | |
| 		return content;
 | |
| 	}
 | |
| 	
 | |
| 	private String getHtm(String path)
 | |
| 	{
 | |
| 		if ((path == null) || path.isEmpty())
 | |
| 		{
 | |
| 			return ""; // avoid possible NPE
 | |
| 		}
 | |
| 		
 | |
| 		String content = _cache.get(path);
 | |
| 		// TODO: Check why some files do not get in cache on server startup.
 | |
| 		if (content == null)
 | |
| 		{
 | |
| 			content = loadFile(new File(Config.DATAPACK_ROOT, path));
 | |
| 		}
 | |
| 		return content;
 | |
| 	}
 | |
| 	
 | |
| 	public boolean contains(String path)
 | |
| 	{
 | |
| 		return _cache.containsKey(path);
 | |
| 	}
 | |
| 	
 | |
| 	/**
 | |
| 	 * @param path The path to the HTM
 | |
| 	 * @return {@code true} if the path targets a HTM or HTML file, {@code false} otherwise.
 | |
| 	 */
 | |
| 	public boolean isLoadable(String path)
 | |
| 	{
 | |
| 		return HTML_FILTER.accept(new File(Config.DATAPACK_ROOT, path));
 | |
| 	}
 | |
| 	
 | |
| 	public static HtmCache getInstance()
 | |
| 	{
 | |
| 		return SingletonHolder._instance;
 | |
| 	}
 | |
| 	
 | |
| 	private static class SingletonHolder
 | |
| 	{
 | |
| 		protected static final HtmCache _instance = new HtmCache();
 | |
| 	}
 | |
| }
 | 
