parent
fead10291c
commit
dbd92b41bc
@ -0,0 +1,146 @@
|
||||
package space.kklochko.jdbc_hospital_example.db.repositories;
|
||||
|
||||
import space.kklochko.jdbc_hospital_example.db.factories.DataBaseConnection;
|
||||
import space.kklochko.jdbc_hospital_example.db.repositories.blocks.CreateTable;
|
||||
import space.kklochko.jdbc_hospital_example.db.repositories.blocks.TransactionalInsert;
|
||||
import space.kklochko.jdbc_hospital_example.db.repositories.blocks.TransactionalUpdate;
|
||||
import space.kklochko.jdbc_hospital_example.models.Patient;
|
||||
|
||||
import java.sql.*;
|
||||
import java.util.ArrayList;
|
||||
import java.util.UUID;
|
||||
|
||||
public class PatientRepository extends AbstractRepository<Patient> {
|
||||
private final String createTableStatement = "create table if not exists patients ("
|
||||
+ "id uuid default uuid_generate_v1(),"
|
||||
+ "name varchar(100) NOT NULL,"
|
||||
+ "age smallint NOT NULL,"
|
||||
+ "phone varchar(25) NOT NULL)";
|
||||
|
||||
public PatientRepository(DataBaseConnection db){
|
||||
super(db);
|
||||
}
|
||||
|
||||
public boolean createTable() {
|
||||
CreateTable createTable = new CreateTable();
|
||||
return createTable.create(db, createTableStatement);
|
||||
}
|
||||
|
||||
public boolean create(Patient object) {
|
||||
try (Connection connection = db.connect()) {
|
||||
connection.setAutoCommit(false);
|
||||
|
||||
String sql = "insert into patients (name, age, phone) values (?, ?, ?) returning id;";
|
||||
|
||||
if(object.getId() != null) {
|
||||
sql = "insert into patients (id, name, age, phone) values (?, ?, ?, ?) returning id;";
|
||||
}
|
||||
|
||||
PreparedStatement preparedStatement = connection.prepareStatement(sql);
|
||||
|
||||
if(object.getId() == null) {
|
||||
preparedStatement.setString(1, object.getName());
|
||||
preparedStatement.setShort(2, object.getAge());
|
||||
preparedStatement.setString(3, object.getPhone());
|
||||
} else {
|
||||
preparedStatement.setObject(1, object.getId());
|
||||
preparedStatement.setString(2, object.getName());
|
||||
preparedStatement.setShort(3, object.getAge());
|
||||
preparedStatement.setString(4, object.getPhone());
|
||||
}
|
||||
|
||||
ResultSet resultSet = (new TransactionalInsert()).run(connection, preparedStatement);
|
||||
|
||||
if (resultSet.next()) {
|
||||
UUID uuid = (UUID) resultSet.getObject("id");
|
||||
object.setId(uuid);
|
||||
return true;
|
||||
}
|
||||
|
||||
return false;
|
||||
} catch (SQLException e) {
|
||||
throw new RuntimeException(e);
|
||||
}
|
||||
}
|
||||
|
||||
public Patient read(UUID id) {
|
||||
try (Connection connection = db.connect()) {
|
||||
String sql = "select id, name, age, phone from patients where id = ?;";
|
||||
PreparedStatement preparedStatement = connection.prepareStatement(sql);
|
||||
preparedStatement.setObject(1, id);
|
||||
ResultSet resultSet = preparedStatement.executeQuery();
|
||||
|
||||
if (resultSet.next()) {
|
||||
UUID uuid = (UUID) resultSet.getObject("id");
|
||||
String name = resultSet.getString("name");
|
||||
short age = resultSet.getShort("age");
|
||||
String phone = resultSet.getString("phone");
|
||||
|
||||
return new Patient(uuid, name, age, phone);
|
||||
}
|
||||
|
||||
return null;
|
||||
} catch (SQLException e) {
|
||||
throw new RuntimeException(e);
|
||||
}
|
||||
}
|
||||
|
||||
public ArrayList<Patient> readAll() {
|
||||
try (Connection connection = db.connect()) {
|
||||
String sql = "select id, name, age, phone from patients;";
|
||||
Statement statement = connection.createStatement();
|
||||
ResultSet resultSet = statement.executeQuery(sql);
|
||||
|
||||
ArrayList<Patient> patients = new ArrayList<>();
|
||||
|
||||
while (resultSet.next()) {
|
||||
UUID uuid = (UUID) resultSet.getObject("id");
|
||||
String name = resultSet.getString("name");
|
||||
short age = resultSet.getShort("age");
|
||||
String phone = resultSet.getString("phone");
|
||||
|
||||
patients.add(new Patient(uuid, name, age, phone));
|
||||
}
|
||||
|
||||
return patients;
|
||||
} catch (SQLException e) {
|
||||
throw new RuntimeException(e);
|
||||
}
|
||||
}
|
||||
|
||||
public boolean update(Patient object) {
|
||||
try (Connection connection = db.connect()) {
|
||||
connection.setAutoCommit(false);
|
||||
|
||||
String sql = "update patients set name = ?, age = ?, phone = ? where id = ?;";
|
||||
PreparedStatement preparedStatement = connection.prepareStatement(sql);
|
||||
|
||||
preparedStatement.setString(1, object.getName());
|
||||
preparedStatement.setShort(2, object.getAge());
|
||||
preparedStatement.setString(3, object.getPhone());
|
||||
preparedStatement.setObject(4, object.getId());
|
||||
|
||||
int rowCount = (new TransactionalUpdate()).run(connection, preparedStatement);
|
||||
|
||||
return rowCount > 0;
|
||||
} catch (SQLException e) {
|
||||
throw new RuntimeException(e);
|
||||
}
|
||||
}
|
||||
|
||||
public boolean delete(UUID id) {
|
||||
try (Connection connection = db.connect()) {
|
||||
connection.setAutoCommit(false);
|
||||
|
||||
String sql = "delete from patients where id = ?;";
|
||||
PreparedStatement preparedStatement = connection.prepareStatement(sql);
|
||||
preparedStatement.setObject(1, id);
|
||||
|
||||
int rowCount = (new TransactionalUpdate()).run(connection, preparedStatement);
|
||||
|
||||
return rowCount > 0;
|
||||
} catch (SQLException e) {
|
||||
throw new RuntimeException(e);
|
||||
}
|
||||
}
|
||||
}
|
||||
Loading…
Reference in new issue