com.ijuru.ijambo.dao.PlayerDAO.java Source code

Introduction

Here is the source code for com.ijuru.ijambo.dao.PlayerDAO.java

Source

/**
 * Copyright 2012 Rowan Seymour
 * 
 * This file is part of Ijambo.
 *
 * Ijambo 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.
 *
 * Ijambo 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 Ijambo. If not, see <http://www.gnu.org/licenses/>.
 */

package com.ijuru.ijambo.dao;

import com.ijuru.ijambo.Player;
import com.mongodb.BasicDBObject;
import com.mongodb.DB;
import com.mongodb.DBCollection;

/**
 * Data access for words
 */
public class PlayerDAO {

    private DB db;

    /**
     * Constructs new DAO
     * @param db the database
     */
    public PlayerDAO(DB db) {
        this.db = db;
    }

    /**
     * Gets the specified player
     * @param identifier the player identifier
     * @return the player
     */
    public Player getPlayer(String identifier) {
        DBCollection players = db.getCollection("players");

        BasicDBObject query = new BasicDBObject();
        query.put("identifier", identifier);

        BasicDBObject obj = (BasicDBObject) players.findOne(query);
        if (obj != null)
            return new Player(obj.getString("identifier"), obj.getString("prevAnswer"));
        else
            return null;
    }

    /**
     * Saves a player
     * @param player the player
     */
    public void save(Player player) {
        DBCollection players = db.getCollection("players");

        remove(player);

        BasicDBObject obj = new BasicDBObject();
        obj.put("identifier", player.getIdentifier());
        obj.put("prevAnswer", player.getPrevAnswer());

        players.insert(obj);
    }

    /**
     * Removes a player
     * @param player the player
     */
    public void remove(Player player) {
        DBCollection players = db.getCollection("players");

        BasicDBObject query = new BasicDBObject();
        query.put("identifier", player.getIdentifier());

        players.findAndRemove(query);
    }
}