Created and updated methods to pull data from db.

This commit is contained in:
Grant Duchars 2021-01-28 22:50:07 -05:00 committed by Gnarwhal
parent e7e5168073
commit c160703048
Signed by: Gnarwhal
GPG key ID: 0989A73D8C421174
4 changed files with 179 additions and 65 deletions

View file

@ -1,12 +1,12 @@
package achievements.services;
import achievements.data.Achievements;
import achievements.data.Games;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;
import javax.annotation.PostConstruct;
import java.sql.Connection;
import java.sql.SQLException;
import java.sql.*;
@Service
public class DbService {
@ -16,32 +16,72 @@ public class DbService {
private Connection db;
@PostConstruct
private void init() {
db = dbs.getConnection();
}
public Achievements getAchievements() {
final String QUERY = "SELECT * FROM [dbo].[Achievement]";
private void init() { db = dbs.getConnection(); }
public Achievements getAchievements(String gameName) {
try {
var statement = db.createStatement();
// Create Query
CallableStatement stmt = db.prepareCall("{? = call GetAchievements(?)}");
stmt.registerOutParameter(1, Types.INTEGER);
stmt.setString(2, gameName);
// Read Result(s)
ResultSet results = stmt.executeQuery();
var achievements = new Achievements();
var queryResults = statement.executeQuery(QUERY);
while (queryResults.next()) {
achievements.getAchievements().add(new Achievements.Achievement(
queryResults.getInt("GameID"),
queryResults.getString("Name"),
queryResults.getString("Description"),
queryResults.getInt("Stages")
));
while (results.next()) {
// Add Result(s) to data class
int achievementGameID = results.getInt("GameID");
String achievementGameName = results.getString("GameName");
String achievementName = results.getString("Name");
String achievementDescription = results.getString("Description");
int achievementStages = results.getInt("Stages");
// Checks if getting from specific game or all achievements
if (!gameName.equals("%")) {
achievements.setGameID(achievementGameID);
achievements.setGameName(achievementGameName);
}
achievements.addAchievement(new Achievements.Achievement(achievementName, achievementDescription, achievementStages));
}
statement.close();
stmt.close();
return achievements;
} catch (SQLException e) {
e.printStackTrace();
return null;
}
}
public Games getGames(String name) {
try {
// Create Query
CallableStatement stmt = db.prepareCall("{? = call GetGame(?)}");
stmt.registerOutParameter(1, Types.INTEGER);
stmt.setString(2, name);
// Read Result(s)
ResultSet results = stmt.executeQuery();
var games = new Games();
while (results.next()) {
// Add Result(s) to data class
int gameID = results.getInt("ID");
String gameName = results.getString("Name");
String gamePlatform = results.getString("PlatformName");
if (!games.getGames().isEmpty()) {
var lastGame = games.getGames().get(games.getGames().size()-1);
if (lastGame.getId() == gameID) {
lastGame.addToPlatforms(gamePlatform);
} else {
games.addGame(new Games.Game(gameID,gameName,gamePlatform));
}
} else {
games.addGame(new Games.Game(gameID,gameName,gamePlatform));
}
}
stmt.close();
return games;
} catch (SQLException e) {
e.printStackTrace();
return null;
}
}
}