iframe-proxy | Sunbelt Computer Software

110 lines (80 loc) · 3.02 KB

package com.github.hackathon;
import java.sql.Connection;
import java.sql.DriverManager;
import java.sql.PreparedStatement;
import java.sql.SQLException;
import java.sql.Statement;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;
import org.springframework.boot.CommandLineRunner;
import org.springframework.boot.SpringApplication;
import org.springframework.boot.autoconfigure.SpringBootApplication;
import org.springframework.context.ApplicationContext;
import org.springframework.context.annotation.Bean;
import com.github.hackathon.advancedsecurityjava.Models.Book;
import org.apache.logging.log4j.LogManager;
import org.apache.logging.log4j.Logger;
@SpringBootApplication
public class Application {
	public static String connectionString = "jdbc:sqlite:database.sqlite";
	public static final Logger logger = LogManager.getLogger();
	public static void main(String[] args) {
			Class.forName("org.sqlite.JDBC");
		} catch (ClassNotFoundException error) {
		error.printStackTrace();
		System.exit(1);
		createDatabase();
		List<Book> books = new ArrayList<Book>();
		books.add(new Book("The Hobbit", "JRR Tolkien", true));
		books.add(new Book("The Fellowship of the Ring", "JRR Tolkien", true));
		books.add(new Book("The Eye of the World", "Robert Jordan"));
		books.add(new Book("A Game of Thrones", "George R. R. Martin", true));
		books.add(new Book("The Way of Kings", "Brandon Sanderson"));
		// Create database entries
		createDatabaseEntries(books);
		SpringApplication.run(Application.class, args);
	public CommandLineRunner commandLineRunner(ApplicationContext ctx) {
		return args -> {
			System.out.println("Let's inspect the beans provided by Spring Boot:");
			String[] beanNames = ctx.getBeanDefinitionNames();
			Arrays.sort(beanNames);
			for (String beanName : beanNames) {
				System.out.println(beanName);
	public static void createDatabase() {
		try (Connection connection = DriverManager.getConnection(connectionString);
			Statement stmt = connection.createStatement()) {
		  // Create tables if they don't exist
		  stmt.execute(
			  "CREATE TABLE IF NOT EXISTS Books (id INTEGER PRIMARY KEY, name TEXT NOT NULL, author TEXT NOT NULL, read INTEGER, UNIQUE(name))");
		} catch (SQLException error) {
		  error.printStackTrace();
		  System.exit(1);
	  public static void createDatabaseEntries(List<Book> books) {
		try (Connection connection = DriverManager.getConnection(connectionString)) {
		  String query = "INSERT INTO Books (name, author, read) VALUES(?, ?, ?)";
		  for (Book book : books) {
			try (PreparedStatement prepStmt = connection.prepareStatement(query);) {
			  prepStmt.setString(1, book.name);
			  prepStmt.setString(2, book.author);
			  prepStmt.setInt(3, book.read? 1 : 0);
			  prepStmt.executeUpdate();
			} catch (SQLException error) {
			  logger.warn("Failed to create book (already exists?) :: " + book.name);
		} catch (SQLException error) {
		  error.printStackTrace();
		  System.exit(2);

Sunbelt Computer Software

PL/B Language Development and Support

Provide feedback

Saved searches

Use saved searches to filter your results more quickly

Uh oh!

Sunbelt Computer Software

PL/B Language Development and Support

Uh oh!

FilesExpand file tree

Application.java

Latest commit

History

Application.java

File metadata and controls