Class SqlTableTag

zap/src/main/java/org/zaproxy/zap/db/sql/SqlTableTag.java:32–172

import org.parosproxy.paros.db.TableTag;

public class SqlTableTag extends SqlAbstractTable implements TableTag {

    private static final String TABLE_NAME = DbSQL.getSQL("tag.table_name");

    private static final String TAGID = DbSQL.getSQL("tag.field.tagid");
    private static final String HISTORYID = DbSQL.getSQL("tag.field.historyid");
    private static final String TAG = DbSQL.getSQL("tag.field.tag");

    public SqlTableTag() {}

    @Override
    protected void reconnect(Connection conn) throws DatabaseException {
        try {
            if (!DbUtils.hasTable(conn, TABLE_NAME)) {
                // Need to create the table
                DbUtils.execute(conn, DbSQL.getSQL("tag.ps.createtable"));
            }
        } catch (SQLException e) {
            throw new DatabaseException(e);
        }
    }

    @Override
    public synchronized RecordTag read(long tagId) throws DatabaseException {
        SqlPreparedStatementWrapper psRead = null;
        try {
            psRead = DbSQL.getSingleton().getPreparedStatement("tag.ps.read");
            psRead.getPs().setLong(1, tagId);

            try (ResultSet rs = psRead.getPs().executeQuery()) {
                return build(rs);
            }
        } catch (SQLException e) {
            throw new DatabaseException(e);
        } finally {
            DbSQL.getSingleton().releasePreparedStatement(psRead);
        }
    }

    @Override
    public synchronized RecordTag insert(long historyId, String tag) throws DatabaseException {
        SqlPreparedStatementWrapper psInsertTag = null;
        try {
            psInsertTag = DbSQL.getSingleton().getPreparedStatement("tag.ps.insert");
            psInsertTag.getPs().setLong(1, historyId);
            psInsertTag.getPs().setString(2, tag);
            psInsertTag.getPs().executeUpdate();

            try (ResultSet rs = psInsertTag.getLastInsertedId()) {
                rs.next();
                long id = rs.getLong(1);
                return read(id);
            }
        } catch (SQLException e) {
            throw new DatabaseException(e);
        } finally {
            DbSQL.getSingleton().releasePreparedStatement(psInsertTag);
        }