Spring Boot - Đang tải dữ liệu ban đầu

Tôi đang tự hỏi cách tốt nhất để tải dữ liệu cơ sở dữ liệu ban đầu trước khi ứng dụng bắt đầu là gì? Những gì tôi đang tìm kiếm là thứ gì đó sẽ lấp đầy cơ sở dữ liệu H2 của tôi với dữ liệu.

Ví dụ: tôi có một mô hình miền "Người dùng" Tôi có thể truy cập người dùng bằng cách truy cập / người dùng nhưng ban đầu sẽ không có bất kỳ người dùng nào trong cơ sở dữ liệu nên tôi phải tạo chúng. Có cách nào để điền vào cơ sở dữ liệu với dữ liệu tự động?

Hiện tại tôi có một Bean được khởi tạo bởi container và tạo người dùng cho tôi.

Thí dụ:

@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"));
    }
}

Nhưng tôi rất nghi ngờ đó là cách tốt nhất để làm điều đó. Hoặc là nó?

Bạn có thể chỉ cần tạo một tệp data.sql trong thư mục src / main / resource của mình và nó sẽ được thực hiện tự động khi khởi động. Trong tệp này, bạn chỉ cần thêm một số câu lệnh chèn, ví dụ:

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

Tương tự như vậy, bạn có thể tạo một schema.sql tệp (hoặc schema-h2.sql) cũng như để tạo ra sơ đồ của bạn:

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

Mặc dù thông thường, bạn không cần phải làm điều này vì Spring boot đã cấu hình Hibernate để tạo lược đồ của bạn dựa trên các thực thể của bạn cho một cơ sở dữ liệu trong bộ nhớ. Nếu bạn thực sự muốn sử dụng lược đồ. Bạn sẽ phải tắt tính năng này bằng cách thêm tính năng này vào ứng dụng của bạn.

spring.jpa.hibernate.ddl-auto=none

Thông tin thêm có thể được tìm thấy tại tài liệu về khởi tạo cơ sở dữ liệu .

Nếu bạn đang sử dụng Spring boot 2 , khởi tạo cơ sở dữ liệu chỉ hoạt động đối với cơ sở dữ liệu nhúng (H2, HSQLDB, ...). Nếu bạn muốn sử dụng nó cho các cơ sở dữ liệu khác, bạn cần thay đổi thuộc spring.datasource.initialization-modetính:

spring.datasource.initialization-mode=always

Nếu bạn đang sử dụng nhiều nhà cung cấp cơ sở dữ liệu, bạn có thể đặt tên tệp dữ liệu của mình-h2.sql hoặc data-mysql.sql tùy thuộc vào nền tảng cơ sở dữ liệu nào bạn muốn sử dụng.

Để thực hiện công việc đó, bạn sẽ phải định cấu hình thuộc spring.datasource.platformtính:

spring.datasource.platform=h2

Nếu tôi chỉ muốn chèn dữ liệu thử nghiệm đơn giản, tôi thường thực hiện a ApplicationRunner. Việc triển khai giao diện này được chạy khi khởi động ứng dụng và có thể sử dụng ví dụ như kho lưu trữ tự động để chèn một số dữ liệu thử nghiệm.

Tôi nghĩ việc triển khai như vậy sẽ rõ ràng hơn một chút so với giao diện của bạn bởi vì giao diện ngụ ý rằng việc triển khai của bạn có chứa thứ gì đó bạn muốn làm trực tiếp sau khi ứng dụng của bạn sẵn sàng.

Việc thực hiện của bạn sẽ trông sth. như thế này:

@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"));
    }
}

Theo gợi ý, hãy thử điều này:

@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("");
    }
}

Tùy chọn 2: Khởi tạo với lược đồ và tập lệnh dữ liệu

Điều kiện tiên quyết: trong application.propertiesbạn phải đề cập đến điều này:

spring.jpa.hibernate.ddl-auto=none(nếu không các tập lệnh sẽ bị bỏ qua bởi chế độ ngủ đông và nó sẽ quét dự án cho @Entityvà / hoặc @Tablecác lớp chú thích)

Sau đó, trong MyApplicationlớp của bạn dán này:

@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;
}

Nơi scriptsthư mục được đặt trong resourcesthư mục (IntelliJ Idea)

Hy vọng nó sẽ giúp được ai đó

Bạn có thể thêm một thuộc spring.datasource.datatính để application.propertiesliệt kê các tệp sql bạn muốn chạy. Như thế này:

spring.datasource.data=classpath:accounts.sql, classpath:books.sql, classpath:reviews.sql

Các câu lệnh chèn sql trong mỗi tệp này sau đó sẽ được chạy, cho phép bạn giữ mọi thứ gọn gàng.

Nếu bạn đặt các tệp trong đường dẫn lớp, ví dụ như src/main/resourceschúng sẽ được áp dụng. Hoặc thay thế classpath:bằng file:và sử dụng một đường dẫn tuyệt đối đến tệp

Bạn có thể sử dụng một cái gì đó như thế này:

@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"));
      };
   }
}

Spring Boot cho phép bạn sử dụng một tập lệnh đơn giản để khởi tạo cơ sở dữ liệu của mình, sử dụng Spring Batch .

Tuy nhiên, nếu bạn muốn sử dụng một cái gì đó chi tiết hơn một chút để quản lý các phiên bản DB, v.v., Spring Boot tích hợp tốt với Flyway .

Xem thêm:

• Khởi tạo cơ sở dữ liệu khởi động mùa xuân

Trong Spring Boot 2 data.sql không hoạt động với tôi như trong spring boot 1.5

nhập khẩu

Ngoài ra, một tệp có tên import.sqltrong thư mục gốc của classpath được thực thi khi khởi động nếu Hibernate tạo lược đồ từ đầu (nghĩa là, nếu thuộc tính ddl-auto được đặt thành tạo hoặc tạo-thả).

Lưu ý rất quan trọng nếu bạn chèn Khóa không thể trùng lặp, không sử dụng thuộc tính ddl-auto được đặt thành cập nhật vì với mỗi lần khởi động lại sẽ chèn lại cùng một dữ liệu

Để biết thêm thông tin bạn tham gia websit mùa xuân

https://docs.spring.io/spring-boot/docs/civerse/reference/html/howto-database-initialization.html

Đây là cách tôi có được rằng:

@Component
public class ApplicationStartup implements ApplicationListener<ApplicationReadyEvent> {

    /**
     * This event is executed as late as conceivably possible to indicate that
     * the application is ready to service requests.
     */

    @Autowired
    private MovieRepositoryImpl movieRepository;

    @Override
    public void onApplicationEvent(final ApplicationReadyEvent event) {
        seedData();
    }

    private void seedData() {
        movieRepository.save(new Movie("Example"));

        // ... add more code
    }

}

Cảm ơn tác giả của bài viết này:

http://blog.netgloo.com/2014/11/13/run-code-at-spring-boot-startup/

Bạn có thể chỉ cần tạo một import.sqltệp trong src/main/resourcesvà Hibernate sẽ thực thi nó khi lược đồ được tạo.

Tôi đã giải quyết vấn đề tương tự theo cách này:

@Component
public class DataLoader {

    @Autowired
    private UserRepository userRepository;

    //method invoked during the startup
    @PostConstruct
    public void loadData() {
        userRepository.save(new User("user"));
    }

    //method invoked during the shutdown
    @PreDestroy
    public void removeData() {
        userRepository.deleteAll();
    }
}

Nếu ai đó đang vật lộn để làm cho nó hoạt động ngay cả khi làm theo câu trả lời được chấp nhận , đối với tôi chỉ làm việc thêm vào chi tiết src/test/resources/application.ymlH2 của tôi datasource:

spring:
  datasource:
    platform: h2
    url: jdbc:h2:mem:test;DB_CLOSE_DELAY=-1
    driver-class-name: org.h2.Driver
    username: sa
    password:

bạn có thể đăng ký và nghe sự kiện để đạt được điều đó như dưới đây:

@EventListener
public void seed(ContextRefreshedEvent event) {
    userRepository.save(new User("lala", "lala", "lala"));
}

Khi ContextRefreshEvent được kích hoạt, chúng tôi có quyền truy cập vào tất cả các loại đậu tự động trong ứng dụng - bao gồm các mô hình và kho lưu trữ.

Nếu bạn muốn chèn chỉ vài hàng và bạn có Cài đặt JPA. Bạn có thể sử dụng dưới đây

    @SpringBootApplication
        @Slf4j
        public class HospitalManagementApplication {

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

            @Bean
            ApplicationRunner init(PatientRepository repository) {
                return (ApplicationArguments args) ->  dataSetup(repository);
            } 

            public void dataSetup(PatientRepository repository){
            //inserts

     }

Điều này cũng sẽ làm việc.

    @Bean
    CommandLineRunner init (StudentRepo studentRepo){
        return args -> {
            // Adding two students objects
            List<String> names = Arrays.asList("udara", "sampath");
            names.forEach(name -> studentRepo.save(new Student(name)));
        };
    }

Giải pháp nhỏ gọn nhất (cho dữ liệu động) đưa giải pháp @ mathias-dpunkt vào MainApp (với Lombok @AllArgsConstructor):

@SpringBootApplication
@AllArgsConstructor
public class RestaurantVotingApplication implements ApplicationRunner {
  private final VoteRepository voteRepository;
  private final UserRepository userRepository;

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

  @Override
  public void run(ApplicationArguments args) {
    voteRepository.save(new Vote(userRepository.getOne(1), LocalDate.now(), LocalTime.now()));
  }
}

Bạn đã gần tới!

@Component
public class DataLoader implements CommandLineRunner {

    private UserRepository userRepository;

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

    @Override
    public void run(String... args) throws Exception {
         LoadUsers()
    }

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

Bạn có thể sử dụng mã dưới đây. Trong đoạn mã sau đây, việc chèn cơ sở dữ liệu xảy ra trong quá trình khởi động ứng dụng khởi động mùa xuân.

@SpringBootApplication
public class Application implements CommandLineRunner {
    
    @Autowired
    private IService<Car> service;

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

    @Override
    public void run(String... args) throws Exception {
        for(int i=1; i<=1000; i++) {
            Car car = new Car();
            car.setName("Car Name "+i);
            book.setPrice(50 + i);
            service.saveOrUpdate(car);
        }
    }

}