net.sf.esfinge.gamification.mechanics.database.sql.RankingStorage Maven / Gradle / Ivy

Go to download
package net.sf.esfinge.gamification.mechanics.database.sql;

import java.sql.Connection;
import java.sql.PreparedStatement;
import java.sql.ResultSet;
import java.sql.SQLException;
import java.util.HashMap;
import java.util.Map;

import net.sf.esfinge.gamification.achievement.Achievement;
import net.sf.esfinge.gamification.achievement.Ranking;
import net.sf.esfinge.gamification.mechanics.database.Storage;

public class RankingStorage implements Storage {

	private Connection connection;

	public RankingStorage(Connection c) {
		connection = c;
	}

	public void insert(Object user, Achievement a) throws SQLException {
		Ranking r = (Ranking) a;
		PreparedStatement stmt;

		stmt = connection.prepareStatement("insert into gamification.ranking"
				+ " (userid, name, level) values (?,?,?)");
		stmt.setString(1, user.toString());
		stmt.setString(2, r.getName());
		stmt.setString(3, r.getLevel());
		stmt.execute();
	}

	public Ranking select(Object user, String name) throws SQLException {
		PreparedStatement stmt;
		stmt = connection.prepareStatement("select * from gamification.ranking"
				+ " where userid=? and name = ?");
		stmt.setString(1, user.toString());
		stmt.setString(2, name);
		ResultSet rs = stmt.executeQuery();
		if (rs.next()) {
			String level = rs.getString("level");
			Ranking r = new Ranking(name, level);
			return r;
		}
		return null;
	}

	public Map select(Object user) throws SQLException {
		Map map = new HashMap();
		PreparedStatement stmt;
		stmt = connection
				.prepareStatement("select * from gamification.ranking "
						+ "where userid=?");
		stmt.setString(1, user.toString());
		ResultSet rs = stmt.executeQuery();
		while (rs.next()) {
			String name = rs.getString("name");
			String level = rs.getString("level");
			Ranking r = new Ranking(name, level);
			map.put(r.getName(), r);
		}

		return map;
	}

	public void update(Object user, Achievement a) throws SQLException {
		Ranking r = (Ranking) a;
		PreparedStatement stmt;
		stmt = connection.prepareStatement("update gamification.ranking "
				+ "set level = ? where userid=? and name=?");
		stmt.setString(2, user.toString());
		stmt.setString(3, r.getName());
		stmt.setString(1, r.getLevel());
		stmt.execute();
	}

	@Override
	public void delete(Object user, Achievement a) throws SQLException {
		PreparedStatement stmt;
		stmt = connection.prepareStatement("delete from gamification.ranking"
				+ " where userid=? and name = ?");
		stmt.setString(1, user.toString());
		stmt.setString(2, a.getName());
		stmt.execute();

	}

	@Override
	public Map selectAll() throws SQLException {
		Map map = null;
		PreparedStatement stmt;
		stmt = connection.prepareStatement("select userid, name, level from gamification.ranking");
		ResultSet rs = stmt.executeQuery();
		if(rs != null) {
			map = new HashMap<>();
			while(rs.next()) {
				String name = rs.getString("name");
				String level = rs.getString("level");
				Ranking ranking = new Ranking(name, level);
				map.put(rs.getString("userid"), ranking);
			}
		}
		return map;
	}
}