Spring Boot - Chargement des données initiales

Question

Je me demande quelle est la meilleure façon de charger les données initiales de la base de données avant le démarrage de l'application ? Ce que je cherche, c'est quelque chose qui va remplir ma base de données H2 avec des données.

Par exemple, j'ai un modèle de domaine "User". Je peux accéder aux utilisateurs en allant dans /users mais, au départ, il n'y aura pas d'utilisateurs dans la base de données et je devrai les créer. Existe-t-il un moyen de remplir automatiquement la base de données avec des données ?

Actuellement, j'ai un Bean qui est instancié par le conteneur et qui crée des utilisateurs pour moi.

Exemple :

@Component
public class DataLoader {

    private UserRepository userRepository;

    @Autowired
    public DataLoader(UserRepository userRepository) {
        this.userRepository = userRepository;
        LoadUsers();
    }

    private void LoadUsers() {
        userRepository.save(new User("lala", "lala", "lala"));
    }
}

Mais je doute fort que ce soit la meilleure façon de procéder. Ou bien l'est-elle ?

Answer 1

Vous pouvez créer un données.sql dans votre src/main/ressources et il sera exécuté automatiquement au démarrage. Dans ce fichier, vous pouvez ajouter quelques déclarations d'insertion, par exemple.. :

INSERT INTO users (username, firstname, lastname) VALUES
  ('lala', 'lala', 'lala'),
  ('lolo', 'lolo', 'lolo');

---

De même, vous pouvez créer un schéma.sql (ou schema-h2.sql) pour créer votre schéma :

CREATE TABLE task (
  id          INTEGER PRIMARY KEY,
  description VARCHAR(64) NOT NULL,
  completed   BIT NOT NULL);

Normalement, vous ne devriez pas avoir à le faire puisque Spring boot configure déjà Hibernate pour créer votre schéma basé sur vos entités pour une base de données en mémoire. Si vous voulez vraiment utiliser schema.sql, vous devez désactiver cette fonctionnalité en l'ajoutant à votre application.properties :

spring.jpa.hibernate.ddl-auto=none

Vous trouverez de plus amples informations dans la documentation sur Initialisation de la base de données .

---

Si vous utilisez Spring boot 2 L'initialisation de la base de données ne fonctionne que pour les bases de données embarquées (H2, HSQLDB, ...). Si vous souhaitez l'utiliser pour d'autres bases de données également, vous devez modifier l'option spring.datasource.initialization-mode propriété :

spring.datasource.initialization-mode=always

---

Si vous utilisez plusieurs fournisseurs de bases de données, vous pouvez nommer votre fichier données-h2.sql o données-mysql.sql en fonction de la plate-forme de base de données que vous souhaitez utiliser.

Pour que cela fonctionne, vous devez configurer l'interface de l'utilisateur. spring.datasource.platform propriété cependant :

spring.datasource.platform=h2

Answer 2

Si je veux simplement insérer des données de test simples, j'implémente souvent un fichier ApplicationRunner . Les implémentations de cette interface sont exécutées au démarrage de l'application et peuvent utiliser, par exemple, un référentiel auto-connecté pour insérer des données de test.

Je pense qu'une telle implémentation serait légèrement plus explicite que la vôtre, car l'interface implique que votre implémentation contient quelque chose que vous souhaitez faire directement après que votre application soit prête.

Votre mise en œuvre ressemblerait à quelque chose comme ceci :

@Component
public class DataLoader implements ApplicationRunner {

    private UserRepository userRepository;

    @Autowired
    public DataLoader(UserRepository userRepository) {
        this.userRepository = userRepository;
    }

    public void run(ApplicationArguments args) {
        userRepository.save(new User("lala", "lala", "lala"));
    }
}

Answer 3

Il existe de multiples façons d'y parvenir. Je préfère utiliser l'une des options suivantes :

Option 1 : Initialisation avec CommandLineRunner haricot :

@Bean
public CommandLineRunner loadData(CustomerRepository repository) {
    return (args) -> {
        // save a couple of customers
        repository.save(new Customer("Jack", "Bauer"));
        repository.save(new Customer("Chloe", "O'Brian"));
        repository.save(new Customer("Kim", "Bauer"));
        repository.save(new Customer("David", "Palmer"));
        repository.save(new Customer("Michelle", "Dessler"));

        // fetch all customers
        log.info("Customers found with findAll():");
        log.info("-------------------------------");
        for (Customer customer : repository.findAll()) {
            log.info(customer.toString());
        }
        log.info("");

        // fetch an individual customer by ID
        Customer customer = repository.findOne(1L);
        log.info("Customer found with findOne(1L):");
        log.info("--------------------------------");
        log.info(customer.toString());
        log.info("");

        // fetch customers by last name
        log.info("Customer found with findByLastNameStartsWithIgnoreCase('Bauer'):");
        log.info("--------------------------------------------");
        for (Customer bauer : repository
                .findByLastNameStartsWithIgnoreCase("Bauer")) {
            log.info(bauer.toString());
        }
        log.info("");
    }
}

Option 2 : Initialisation avec schéma et données SQL scripts

Conditions préalables :

application.properties

spring.jpa.hibernate.ddl-auto=none

Explication :

Sans ddl-auto Les scripts SQL seront ignorés par Hibernate et déclenchera le comportement par défaut - le balayage du projet pour les @Entity et/ou @Table les classes annotées.

Ensuite, dans votre MyApplication classe coller ceci :

@Bean(name = "dataSource")
public DriverManagerDataSource dataSource() {
    DriverManagerDataSource dataSource = new DriverManagerDataSource();
    dataSource.setDriverClassName("org.h2.Driver");
    dataSource.setUrl("jdbc:h2:~/myDB;MV_STORE=false");
    dataSource.setUsername("sa");
    dataSource.setPassword("");

    // schema init
    Resource initSchema = new ClassPathResource("scripts/schema-h2.sql");
    Resource initData = new ClassPathResource("scripts/data-h2.sql");
    DatabasePopulator databasePopulator = new ResourceDatabasePopulator(initSchema, initData);
    DatabasePopulatorUtils.execute(databasePopulator, dataSource);

    return dataSource;
}

Donde scripts est situé sous resources dossier (IntelliJ Idea)

J'espère que cela aidera quelqu'un

Mise à jour 04-2021 : Les deux options peuvent être combinées avec Profils de printemps car cela vous permettra d'éviter de créer des fichiers de configuration supplémentaires, ce qui vous facilitera la vie en tant que développeur.

Answer 4

Vous pouvez utiliser quelque chose comme ça :

@SpringBootApplication  
public class Application {

@Autowired
private UserRepository userRepository;

public static void main(String[] args) {
    SpringApplication.run(Application.class, args);
}

@Bean
InitializingBean sendDatabase() {
    return () -> {
        userRepository.save(new User("John"));
        userRepository.save(new User("Rambo"));
      };
   }
}

Answer 5

Spring Boot vous permet d'utiliser un simple script pour initialiser votre base de données, à l'aide des éléments suivants Lot de printemps .

Néanmoins, si vous souhaitez utiliser quelque chose d'un peu plus élaboré pour gérer les versions des bases de données, etc., Spring Boot s'intègre bien à Voie de migration .

Voir aussi :

• Initialisation de la base de données Spring Boot