/*
* Juick
* Copyright (C) 2008-2011, Ugnich Anton
*
* This program is free software: you can redistribute it and/or modify
* it under the terms of the GNU Affero General Public License as
* published by the Free Software Foundation, either version 3 of the
* License, or (at your option) any later version.
*
* This program 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 Affero General Public License for more details.
*
* You should have received a copy of the GNU Affero General Public License
* along with this program. If not, see .
*/
package com.juick.server;
import com.juick.Tag;
import org.springframework.dao.EmptyResultDataAccessException;
import org.springframework.jdbc.core.JdbcTemplate;
import java.sql.Connection;
import java.sql.PreparedStatement;
import java.sql.ResultSet;
import java.sql.SQLException;
import java.sql.Statement;
import java.util.ArrayList;
import java.util.List;
import java.util.logging.Level;
import java.util.logging.Logger;
import org.springframework.jdbc.support.GeneratedKeyHolder;
import org.springframework.jdbc.support.KeyHolder;
/**
*
* @author Ugnich Anton
*/
public class TagQueries {
private static final Logger LOGGER = Logger.getLogger(TagQueries.class.getName());
public static com.juick.Tag getTag(JdbcTemplate sql, int tid) {
return sql.queryForObject("SELECT synonym_id,name FROM tags WHERE tag_id=?",
new Object[]{tid}, (rs, num) -> {
Tag ret = new Tag();
ret.TID = tid;
ret.SynonymID = rs.getInt(1);
ret.Name = rs.getString(2);
return ret;
});
}
public static com.juick.Tag getTag(JdbcTemplate sql, String tag, boolean autoCreate) {
Tag ret = null;
try {
ret = sql.queryForObject("SELECT tag_id,synonym_id,name FROM tags WHERE name=?",
new Object[]{tag}, (rs, rowNum) -> {
Tag ret1 = new Tag();
ret1.TID = rs.getInt(1);
ret1.SynonymID = rs.getInt(2);
ret1.Name = rs.getString(3);
return ret1;
});
} catch (EmptyResultDataAccessException e) {
// tag not found
}
if (ret == null && autoCreate) {
ret = new com.juick.Tag();
ret.TID = createTag(sql, tag);
ret.Name = tag;
}
return ret;
}
public static List getTags(JdbcTemplate sql, String[] tags, boolean autoCreate) {
List ret = new ArrayList<>();
for (String tag : tags) {
if (!tag.isEmpty()) {
Tag t = getTag(sql, tag, autoCreate);
if (t != null) {
ret.add(t);
}
}
}
return ret;
}
public static boolean getTagNoIndex(Connection sql, int tag_id) {
boolean ret = false;
PreparedStatement stmt = null;
ResultSet rs = null;
try {
stmt = sql.prepareStatement("SELECT noindex FROM tags WHERE tag_id=?");
stmt.setInt(1, tag_id);
rs = stmt.executeQuery();
if (rs.first()) {
ret = rs.getInt(1) == 1;
}
} catch (SQLException e) {
LOGGER.log(Level.SEVERE, "sql exception", e);
} finally {
Utils.finishSQL(rs, stmt);
}
return ret;
}
public static int createTag(JdbcTemplate sql, String name) {
KeyHolder holder = new GeneratedKeyHolder();
sql.update(con -> {
PreparedStatement stmt = con.prepareStatement("INSERT INTO tags(name) VALUES (?)",
Statement.RETURN_GENERATED_KEYS);
stmt.setString(1, name);
return stmt;
}, holder);
return holder.getKey().intValue();
}
public static List getUserTagsAll(Connection sql, int uid) {
List tags = new ArrayList<>();
PreparedStatement stmt = null;
ResultSet rs = null;
try {
stmt = sql.prepareStatement("SELECT tags.name,COUNT(messages.message_id) FROM (messages INNER JOIN messages_tags ON (messages.user_id=? AND messages.message_id=messages_tags.message_id)) INNER JOIN tags ON messages_tags.tag_id=tags.tag_id GROUP BY tags.tag_id ORDER BY tags.name ASC");
stmt.setInt(1, uid);
rs = stmt.executeQuery();
rs.beforeFirst();
while (rs.next()) {
com.juick.Tag t = new com.juick.Tag();
t.Name = rs.getString(1);
t.UsageCnt = rs.getInt(2);
tags.add(t);
}
} catch (SQLException e) {
LOGGER.log(Level.SEVERE, "sql exception", e);
} finally {
Utils.finishSQL(rs, stmt);
}
return tags;
}
public static List getUserBLTags(Connection sql, int uid) {
List tags = new ArrayList<>();
PreparedStatement stmt = null;
ResultSet rs = null;
try {
stmt = sql.prepareStatement("SELECT tags.name FROM tags INNER JOIN bl_tags ON (bl_tags.user_id=? AND bl_tags.tag_id=tags.tag_id) ORDER BY tags.name");
stmt.setInt(1, uid);
rs = stmt.executeQuery();
rs.beforeFirst();
while (rs.next()) {
tags.add(rs.getString(1));
}
} catch (SQLException e) {
LOGGER.log(Level.SEVERE, "sql exception", e);
} finally {
Utils.finishSQL(rs, stmt);
}
return tags;
}
}