private void getIgnoreList() {
Connection con = null;
try {
con = pool.getConnection(timeout);
Statement s = con.createStatement();
s.executeQuery("SELECT `name`, `type` FROM `ignores`");
ResultSet rs = s.getResultSet();
this.ignore_list.clear();
this.soft_ignore_list.clear();
while (rs.next()) {
if (rs.getString("type").equals("hard")) {
this.ignore_list.add(rs.getString("name").toLowerCase());
} else {
this.soft_ignore_list.add(rs.getString("name").toLowerCase());
}
}
rs.close();
s.close();
} catch (SQLException ex) {
Logger.getLogger(DBAccess.class.getName()).log(Level.SEVERE, null, ex);
} finally {
try {
if (con != null) {
con.close();
}
} catch (SQLException ex) {
Logger.getLogger(DBAccess.class.getName()).log(Level.SEVERE, null, ex);
}
}
}
String getSetting(String key) {
Connection con = null;
String value = "";
try {
con = pool.getConnection(timeout);
PreparedStatement s = con.prepareStatement("SELECT `value` FROM `settings` WHERE `key` = ?");
s.setString(1, key);
s.executeQuery();
ResultSet rs = s.getResultSet();
while (rs.next()) {
value = rs.getString("value");
}
rs.close();
s.close();
} catch (SQLException ex) {
Logger.getLogger(DBAccess.class.getName()).log(Level.SEVERE, null, ex);
} finally {
try {
if (con != null) {
con.close();
}
} catch (SQLException ex) {
Logger.getLogger(DBAccess.class.getName()).log(Level.SEVERE, null, ex);
}
}
return value;
}
void joinChannel(Channel channel) {
Connection con = null;
try {
con = pool.getConnection(timeout);
PreparedStatement s = con.prepareStatement("INSERT INTO `channels` (`channel`) VALUES (?)");
s.setString(1, channel.getName().toLowerCase());
s.executeUpdate();
s.close();
if (!this.channel_data.containsKey(channel.getName().toLowerCase())) {
ChannelInfo new_channel = new ChannelInfo(channel.getName().toLowerCase());
new_channel.setDefaultOptions();
this.channel_data.put(channel.getName().toLowerCase(), new_channel);
}
this.saveChannelSettings(this.channel_data.get(channel.getName().toLowerCase()));
} catch (SQLException ex) {
Logger.getLogger(DBAccess.class.getName()).log(Level.SEVERE, null, ex);
} finally {
try {
if (con != null) {
con.close();
}
} catch (SQLException ex) {
Logger.getLogger(DBAccess.class.getName()).log(Level.SEVERE, null, ex);
}
}
}
private void getPokemon() {
Connection con = null;
try {
con = pool.getConnection(timeout);
try (Statement s = con.createStatement()) {
ResultSet rs;
s.executeQuery("SELECT `name` FROM `pokemon`");
rs = s.getResultSet();
this.pokemon.clear();
while (rs.next()) {
this.pokemon.add(rs.getString("name"));
}
rs.close();
s.close();
}
} catch (SQLException ex) {
Logger.getLogger(DBAccess.class.getName()).log(Level.SEVERE, null, ex);
} finally {
try {
if (con != null) {
con.close();
}
} catch (SQLException ex) {
Logger.getLogger(DBAccess.class.getName()).log(Level.SEVERE, null, ex);
}
}
}
String getRandomChannelWithVelociraptors(String exclude) {
Connection con = null;
String value = "";
try {
con = pool.getConnection(timeout);
PreparedStatement s =
con.prepareStatement(
"SELECT `channel` FROM `channels` WHERE `channel` != ? AND `active_velociraptors` > 0 ORDER BY (RAND() * active_velociraptors) DESC LIMIT 1;");
s.setString(1, exclude);
s.executeQuery();
ResultSet rs = s.getResultSet();
while (rs.next()) {
value = rs.getString("channel");
}
rs.close();
s.close();
} catch (SQLException ex) {
Logger.getLogger(DBAccess.class.getName()).log(Level.SEVERE, null, ex);
} finally {
try {
if (con != null) {
con.close();
}
} catch (SQLException ex) {
Logger.getLogger(DBAccess.class.getName()).log(Level.SEVERE, null, ex);
}
}
return value;
}
void update_war(int db_id, long duration, long remaining, long time_to_start, int current_chain) {
Connection con = null;
try {
con = pool.getConnection(timeout);
PreparedStatement s =
con.prepareStatement(
"UPDATE `wars` SET `duration` = ? ,`remaining` = ?, `time_to_start` = ?, `current_chain` = ? WHERE id = ?");
s.setLong(1, duration);
s.setLong(2, remaining);
s.setLong(3, time_to_start);
s.setInt(4, current_chain);
s.setInt(5, db_id);
s.executeUpdate();
s.close();
} catch (SQLException ex) {
Logger.getLogger(DBAccess.class.getName()).log(Level.SEVERE, null, ex);
} finally {
try {
if (con != null) {
con.close();
}
} catch (SQLException ex) {
Logger.getLogger(DBAccess.class.getName()).log(Level.SEVERE, null, ex);
}
}
}
public void addToChannelGroup(String group, ChannelInfo channel) {
Connection con = null;
try {
con = pool.getConnection(timeout);
PreparedStatement s =
con.prepareStatement(
"REPLACE INTO `channel_groups` SET `name` = ?, `channel_id` = (SELECT `id` FROM `channels` WHERE `channel` = ?)");
s.setString(1, group.toLowerCase());
s.setString(2, channel.channel.toLowerCase());
s.executeUpdate();
s.close();
// Will do nothing if the channel is not in the list.
if (!this.channel_groups.containsKey(group.toLowerCase())) {
this.channel_groups.put(group.toLowerCase(), new HashSet<>());
}
this.channel_groups.get(group.toLowerCase()).add(channel);
} catch (SQLException ex) {
Logger.getLogger(DBAccess.class.getName()).log(Level.SEVERE, null, ex);
} finally {
try {
if (con != null) {
con.close();
}
} catch (SQLException ex) {
Logger.getLogger(DBAccess.class.getName()).log(Level.SEVERE, null, ex);
}
}
}
public void removeFromChannelGroup(String group, ChannelInfo channel) {
Connection con = null;
try {
con = pool.getConnection(timeout);
PreparedStatement s =
con.prepareStatement(
"DELETE `channel_groups`.* FROM `channel_groups` INNER JOIN `channels` ON (`channel_groups`.`channel_id` = `channels`.`id`) WHERE `channels`.`channel` = ? AND `channel_groups`.`name` = ?");
s.setString(1, channel.channel);
s.setString(2, group);
s.executeUpdate();
s.close();
// Will do nothing if the channel is not in the list.
this.channel_groups.get(group.toLowerCase()).remove(channel);
} catch (SQLException ex) {
Logger.getLogger(DBAccess.class.getName()).log(Level.SEVERE, null, ex);
} finally {
try {
if (con != null) {
con.close();
}
} catch (SQLException ex) {
Logger.getLogger(DBAccess.class.getName()).log(Level.SEVERE, null, ex);
}
}
}
int create_war(
Channel channel,
String starter,
String name,
long base_duration,
long duration,
long remaining,
long time_to_start,
int total_chains,
int current_chain,
int break_duration,
boolean do_randomness) {
int id = 0;
Connection con = null;
try {
con = pool.getConnection(timeout);
PreparedStatement s =
con.prepareStatement(
"INSERT INTO `wars` (`channel`, `starter`, `name`, `base_duration`, `randomness`, `delay`, `duration`, `remaining`, `time_to_start`, `total_chains`, `current_chain`) VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)",
Statement.RETURN_GENERATED_KEYS);
s.setString(1, channel.getName().toLowerCase());
s.setString(2, starter);
s.setString(3, name);
s.setLong(4, base_duration);
s.setBoolean(5, do_randomness);
s.setLong(6, break_duration);
s.setLong(7, duration);
s.setLong(8, remaining);
s.setLong(9, time_to_start);
s.setInt(10, total_chains);
s.setInt(11, current_chain);
s.executeUpdate();
ResultSet rs = s.getGeneratedKeys();
if (rs.next()) {
id = rs.getInt(1);
}
rs.close();
s.close();
} catch (SQLException ex) {
Logger.getLogger(DBAccess.class.getName()).log(Level.SEVERE, null, ex);
} finally {
try {
if (con != null) {
con.close();
}
} catch (SQLException ex) {
Logger.getLogger(DBAccess.class.getName()).log(Level.SEVERE, null, ex);
}
}
return id;
}
private DBAccess() {
// Initialize the connection pool, to prevent SQL timeout issues
String url =
"jdbc:mysql://"
+ Tim.config.getString("sql_server")
+ ":3306/"
+ Tim.config.getString("sql_database")
+ "?useUnicode=true&characterEncoding=UTF-8&serverTimezone=UTC";
pool =
new ConnectionPool(
"local",
5,
25,
150,
180000,
url,
Tim.config.getString("sql_user"),
Tim.config.getString("sql_password"));
pool.setValidator(new Select1Validator());
pool.setAsyncDestroy(true);
}
ConcurrentHashMap<String, WordWar> loadWars() {
Connection con = null;
Channel channel;
String user;
ConcurrentHashMap<String, WordWar> wars = new ConcurrentHashMap<>(32);
try {
con = pool.getConnection(timeout);
Statement s = con.createStatement();
ResultSet rs = s.executeQuery("SELECT * FROM `wars`");
while (rs.next()) {
channel = Tim.channelStorage.channelList.get(rs.getString("channel"));
user = rs.getString("starter");
WordWar war =
new WordWar(
rs.getLong("base_duration"),
rs.getLong("duration"),
rs.getLong("remaining"),
rs.getLong("time_to_start"),
rs.getInt("total_chains"),
rs.getInt("current_chain"),
rs.getInt("delay"),
rs.getBoolean("randomness"),
rs.getString("name"),
user,
channel,
rs.getInt("id"));
wars.put(war.getName(false).toLowerCase(), war);
}
rs.close();
s.close();
} catch (SQLException ex) {
Logger.getLogger(DBAccess.class.getName()).log(Level.SEVERE, null, ex);
} finally {
try {
if (con != null) {
con.close();
}
} catch (SQLException ex) {
Logger.getLogger(DBAccess.class.getName()).log(Level.SEVERE, null, ex);
}
}
return wars;
}
void deleteWar(int id) {
Connection con = null;
try {
con = pool.getConnection(timeout);
PreparedStatement s = con.prepareStatement("DELETE FROM `wars` WHERE `id` = ?");
s.setInt(1, id);
s.executeUpdate();
s.close();
} catch (SQLException ex) {
Logger.getLogger(DBAccess.class.getName()).log(Level.SEVERE, null, ex);
} finally {
try {
if (con != null) {
con.close();
}
} catch (SQLException ex) {
Logger.getLogger(DBAccess.class.getName()).log(Level.SEVERE, null, ex);
}
}
}
void deleteIgnore(String username) {
Connection con = null;
try {
con = pool.getConnection(timeout);
PreparedStatement s = con.prepareStatement("DELETE FROM `ignores` WHERE `name` = ?;");
s.setString(1, username.toLowerCase());
s.executeUpdate();
s.close();
} catch (SQLException ex) {
Logger.getLogger(DBAccess.class.getName()).log(Level.SEVERE, null, ex);
} finally {
try {
if (con != null) {
con.close();
}
} catch (SQLException ex) {
Logger.getLogger(DBAccess.class.getName()).log(Level.SEVERE, null, ex);
}
}
}
void saveIgnore(String username, String type) {
Connection con = null;
try {
con = pool.getConnection(timeout);
PreparedStatement s =
con.prepareStatement("INSERT INTO `ignores` (`name`, `type`) VALUES (?, ?);");
s.setString(1, username.toLowerCase());
s.setString(2, type.toLowerCase());
s.executeUpdate();
s.close();
} catch (SQLException ex) {
Logger.getLogger(DBAccess.class.getName()).log(Level.SEVERE, null, ex);
} finally {
try {
if (con != null) {
con.close();
}
} catch (SQLException ex) {
Logger.getLogger(DBAccess.class.getName()).log(Level.SEVERE, null, ex);
}
}
}
private void getChannelGroups() {
Connection con = null;
try {
con = pool.getConnection(timeout);
Statement s = con.createStatement();
s.executeQuery(
"SELECT `name`, `channels`.`channel` FROM `channel_groups` INNER JOIN `channels` ON (`channel_groups`.`channel_id` = `channels`.`id`)");
ResultSet rs = s.getResultSet();
this.channel_groups.clear();
while (rs.next()) {
if (!this.channel_groups.containsKey(rs.getString("name").toLowerCase())) {
this.channel_groups.put(rs.getString("name").toLowerCase(), new HashSet<>());
}
this.channel_groups
.get(rs.getString("name").toLowerCase())
.add(this.channel_data.get(rs.getString("channel")));
}
rs.close();
s.close();
} catch (SQLException ex) {
Logger.getLogger(DBAccess.class.getName()).log(Level.SEVERE, null, ex);
} finally {
try {
if (con != null) {
con.close();
}
} catch (SQLException ex) {
Logger.getLogger(DBAccess.class.getName()).log(Level.SEVERE, null, ex);
}
}
}
void deleteChannel(Channel channel) {
Connection con = null;
try {
con = pool.getConnection(timeout);
PreparedStatement s = con.prepareStatement("DELETE FROM `channels` WHERE `channel` = ?");
s.setString(1, channel.getName().toLowerCase());
s.executeUpdate();
s.close();
// Will do nothing if the channel is not in the list.
this.channel_data.remove(channel.getName().toLowerCase());
} catch (SQLException ex) {
Logger.getLogger(DBAccess.class.getName()).log(Level.SEVERE, null, ex);
} finally {
try {
if (con != null) {
con.close();
}
} catch (SQLException ex) {
Logger.getLogger(DBAccess.class.getName()).log(Level.SEVERE, null, ex);
}
}
}
public void saveChannelSettings(ChannelInfo channel) {
Connection con = null;
try {
con = pool.getConnection(timeout);
SimpleDateFormat sdf = new SimpleDateFormat("yyyy-MM-dd HH:mm:ss");
PreparedStatement s =
con.prepareStatement(
"UPDATE `channels` SET reactive_chatter_level = ?, chatter_name_multiplier = ?, random_chatter_level = ?, tweet_bucket_max = ?, "
+ "tweet_bucket_charge_rate = ?, auto_muzzle_wars = ?, velociraptor_sightings = ?, active_velociraptors = ?, dead_velociraptors = ?, killed_velociraptors = ?, last_sighting_date = ? WHERE channel = ?;");
s.setInt(1, channel.reactiveChatterLevel);
s.setInt(2, channel.chatterNameMultiplier);
s.setInt(3, channel.randomChatterLevel);
s.setFloat(4, channel.tweetBucketMax);
s.setFloat(5, channel.tweetBucketChargeRate);
s.setBoolean(6, channel.auto_muzzle_wars);
s.setInt(7, channel.velociraptorSightings);
s.setInt(8, channel.activeVelociraptors);
s.setInt(9, channel.deadVelociraptors);
s.setInt(10, channel.killedVelociraptors);
if (channel.lastSighting != null) {
s.setString(11, sdf.format(channel.lastSighting));
} else {
s.setString(11, "1999-01-01");
}
s.setString(12, channel.channel);
s.executeUpdate();
s = con.prepareStatement("DELETE FROM `channel_chatter_settings` WHERE `channel` = ?;");
s.setString(1, channel.channel);
s.executeUpdate();
s = con.prepareStatement("DELETE FROM `channel_command_settings` WHERE `channel` = ?;");
s.setString(1, channel.channel);
s.executeUpdate();
s = con.prepareStatement("DELETE FROM `channel_twitter_feeds` WHERE `channel` = ?;");
s.setString(1, channel.channel);
s.executeUpdate();
s =
con.prepareStatement(
"INSERT INTO `channel_chatter_settings` SET `channel` = ?, `setting` = ?, `value` = ?;");
s.setString(1, channel.channel);
for (Map.Entry<String, Boolean> setting : channel.chatter_enabled.entrySet()) {
s.setString(2, setting.getKey());
s.setBoolean(3, setting.getValue());
s.executeUpdate();
}
s =
con.prepareStatement(
"INSERT INTO `channel_command_settings` SET `channel` = ?, `setting` = ?, `value` = ?;");
s.setString(1, channel.channel);
for (Map.Entry<String, Boolean> setting : channel.commands_enabled.entrySet()) {
s.setString(2, setting.getKey());
s.setBoolean(3, setting.getValue());
s.executeUpdate();
}
s =
con.prepareStatement(
"INSERT INTO `channel_twitter_feeds` SET `channel` = ?, `account` = ?;");
s.setString(1, channel.channel);
for (String account : channel.twitter_accounts) {
s.setString(2, account);
s.executeUpdate();
}
s.close();
} catch (SQLException ex) {
Logger.getLogger(DBAccess.class.getName()).log(Level.SEVERE, null, ex);
} finally {
try {
if (con != null) {
con.close();
}
} catch (SQLException ex) {
Logger.getLogger(DBAccess.class.getName()).log(Level.SEVERE, null, ex);
}
}
}
private void getChannelList() {
Connection con = null;
ChannelInfo ci;
this.channel_data.clear();
try {
con = pool.getConnection(timeout);
ResultSet rs;
try (Statement s = con.createStatement()) {
rs = s.executeQuery("SELECT * FROM `channels`");
PreparedStatement s2;
ResultSet rs2;
while (rs.next()) {
ci = new ChannelInfo(rs.getString("channel"));
ci.setDefaultOptions();
ci.setReactiveChatter(
rs.getInt("reactive_chatter_level"), rs.getInt("chatter_name_multiplier"));
ci.setRandomChatter(rs.getInt("random_chatter_level"));
ci.setTwitterTimers(
rs.getFloat("tweet_bucket_max"), rs.getFloat("tweet_bucket_charge_rate"));
ci.setWarAutoMuzzle(rs.getBoolean("auto_muzzle_wars"));
ci.velociraptorSightings = rs.getInt("velociraptor_sightings");
ci.activeVelociraptors = rs.getInt("active_velociraptors");
ci.deadVelociraptors = rs.getInt("dead_velociraptors");
ci.killedVelociraptors = rs.getInt("killed_velociraptors");
if (rs.getDate("last_sighting_date") != null) {
long time = rs.getDate("last_sighting_date").getTime();
ci.lastSighting = new java.util.Date(time);
}
s2 =
con.prepareStatement(
"SELECT `setting`, `value` FROM `channel_chatter_settings` WHERE `channel` = ?");
s2.setString(1, rs.getString("channel"));
s2.executeQuery();
rs2 = s2.getResultSet();
while (rs2.next()) {
ci.addChatterSetting(rs2.getString("setting"), rs2.getBoolean("value"));
}
s2.close();
rs2.close();
s2 =
con.prepareStatement(
"SELECT `setting`, `value` FROM `channel_command_settings` WHERE `channel` = ?");
s2.setString(1, rs.getString("channel"));
s2.executeQuery();
rs2 = s2.getResultSet();
while (rs2.next()) {
ci.addCommandSetting(rs2.getString("setting"), rs2.getBoolean("value"));
}
s2.close();
rs2.close();
s2 =
con.prepareStatement(
"SELECT `account` FROM `channel_twitter_feeds` WHERE `channel` = ?");
s2.setString(1, rs.getString("channel"));
s2.executeQuery();
rs2 = s2.getResultSet();
while (rs2.next()) {
ci.addTwitterAccount(rs2.getString("account"));
}
s2.close();
rs2.close();
this.channel_data.put(ci.channel, ci);
this.saveChannelSettings(ci);
}
}
rs.close();
} catch (SQLException ex) {
Logger.getLogger(DBAccess.class.getName()).log(Level.SEVERE, null, ex);
} finally {
try {
if (con != null) {
con.close();
}
} catch (SQLException ex) {
Logger.getLogger(DBAccess.class.getName()).log(Level.SEVERE, null, ex);
}
}
}