src/main/java/com/l2jserver/gameserver/data/sql/impl/NpcBufferTable.java - develop - tofu/l2interlude

src/main/java/com/l2jserver/gameserver/data/sql/impl/NpcBufferTable.java
160 строк5 KB
Zoey76
01 янв 2025, 07:57
01 янв 2025, 07:5716130a5
/*
 * Copyright © 2004-2025 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.data.sql.impl;

import static com.l2jserver.gameserver.config.Configuration.general;

import java.util.HashMap;
import java.util.Map;

import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import com.l2jserver.commons.database.ConnectionFactory;
import com.l2jserver.gameserver.model.holders.ItemHolder;
import com.l2jserver.gameserver.model.holders.SkillHolder;

public class NpcBufferTable {
	
	private static final Logger LOG = LoggerFactory.getLogger(NpcBufferTable.class);
	
	private final Map<Integer, NpcBufferSkills> _buffers = new HashMap<>();
	
	public static class NpcBufferData {
		private final SkillHolder _skill;
		private final ItemHolder _fee;
		
		protected NpcBufferData(int skillId, int skillLevel, int feeId, int feeAmount) {
			_skill = new SkillHolder(skillId, skillLevel);
			_fee = new ItemHolder(feeId, feeAmount);
		}
		
		public SkillHolder getSkill() {
			return _skill;
		}
		
		public ItemHolder getFee() {
			return _fee;
		}
	}
	
	private static class NpcBufferSkills {
		private final Map<Integer, NpcBufferData> _skills = new HashMap<>();
		
		public void addSkill(int skillId, int skillLevel, int skillFeeId, int skillFeeAmount, int buffGroup) {
			_skills.put(buffGroup, new NpcBufferData(skillId, skillLevel, skillFeeId, skillFeeAmount));
		}
		
		public NpcBufferData getSkillGroupInfo(int buffGroup) {
			return _skills.get(buffGroup);
		}
	}
	
	protected NpcBufferTable() {
		int skillCount = 0;
		try (var con = ConnectionFactory.getInstance().getConnection();
			var st = con.createStatement();
			var rs = st.executeQuery("SELECT `npc_id`,`skill_id`,`skill_level`,`skill_fee_id`,`skill_fee_amount`,`buff_group` FROM `npc_buffer` ORDER BY `npc_id`")) {
			int lastNpcId = 0;
			NpcBufferSkills skills = null;
			
			while (rs.next()) {
				int npcId = rs.getInt("npc_id");
				int skillId = rs.getInt("skill_id");
				int skillLevel = rs.getInt("skill_level");
				int skillFeeId = rs.getInt("skill_fee_id");
				int skillFeeAmount = rs.getInt("skill_fee_amount");
				int buffGroup = rs.getInt("buff_group");
				
				if (npcId != lastNpcId) {
					if (lastNpcId != 0) {
						_buffers.put(lastNpcId, skills);
					}
					
					skills = new NpcBufferSkills();
					skills.addSkill(skillId, skillLevel, skillFeeId, skillFeeAmount, buffGroup);
				} else if (skills != null) {
					skills.addSkill(skillId, skillLevel, skillFeeId, skillFeeAmount, buffGroup);
				}
				
				lastNpcId = npcId;
				skillCount++;
			}
			
			if (lastNpcId != 0) {
				_buffers.put(lastNpcId, skills);
			}
		} catch (Exception ex) {
			LOG.warn("There has been an error reading npc_buffer table!", ex);
		}
		
		if (general().customNpcBufferTables()) {
			try (var con = ConnectionFactory.getInstance().getConnection();
				var st = con.createStatement();
				var rs = st.executeQuery("SELECT `npc_id`,`skill_id`,`skill_level`,`skill_fee_id`,`skill_fee_amount`,`buff_group` FROM `custom_npc_buffer` ORDER BY `npc_id`")) {
				int lastNpcId = 0;
				NpcBufferSkills skills = null;
				while (rs.next()) {
					int npcId = rs.getInt("npc_id");
					int skillId = rs.getInt("skill_id");
					int skillLevel = rs.getInt("skill_level");
					int skillFeeId = rs.getInt("skill_fee_id");
					int skillFeeAmount = rs.getInt("skill_fee_amount");
					int buffGroup = rs.getInt("buff_group");
					
					if (npcId != lastNpcId) {
						if (lastNpcId != 0) {
							_buffers.put(lastNpcId, skills);
						}
						
						skills = new NpcBufferSkills();
						skills.addSkill(skillId, skillLevel, skillFeeId, skillFeeAmount, buffGroup);
					} else if (skills != null) {
						skills.addSkill(skillId, skillLevel, skillFeeId, skillFeeAmount, buffGroup);
					}
					lastNpcId = npcId;
					skillCount++;
				}
				
				if (lastNpcId != 0) {
					_buffers.put(lastNpcId, skills);
				}
			} catch (Exception ex) {
				LOG.warn("There has been an error reading custom_npc_buffer table!", ex);
			}
		}
		LOG.info("Loaded {} buffers and {} skills.", _buffers.size(), skillCount);
	}
	
	public NpcBufferData getSkillInfo(int npcId, int buffGroup) {
		final NpcBufferSkills skills = _buffers.get(npcId);
		if (skills != null) {
			return skills.getSkillGroupInfo(buffGroup);
		}
		return null;
	}
	
	public static NpcBufferTable getInstance() {
		return SingletonHolder._instance;
	}
	
	private static class SingletonHolder {
		protected static final NpcBufferTable _instance = new NpcBufferTable();
	}
}