44 lines
1.7 KiB
Java
44 lines
1.7 KiB
Java
package edu.amu.pl.demo;
|
|
|
|
import edu.amu.pl.demo.risk.entity.Client;
|
|
import edu.amu.pl.demo.risk.entity.Illness;
|
|
import edu.amu.pl.demo.risk.repository.ClientRepository;
|
|
import edu.amu.pl.demo.risk.repository.IllnessRepository;
|
|
import org.springframework.beans.factory.annotation.Autowired;
|
|
import org.springframework.boot.CommandLineRunner;
|
|
import org.springframework.boot.SpringApplication;
|
|
import org.springframework.boot.autoconfigure.EnableAutoConfiguration;
|
|
import org.springframework.boot.autoconfigure.SpringBootApplication;
|
|
import org.springframework.context.annotation.Bean;
|
|
import org.springframework.context.annotation.ComponentScan;
|
|
import org.springframework.context.annotation.Configuration;
|
|
|
|
import java.util.Set;
|
|
|
|
@SpringBootApplication
|
|
public class DemoJakartaApplication {
|
|
|
|
public static void main(String[] args) {
|
|
SpringApplication.run(DemoJakartaApplication.class, args);
|
|
}
|
|
|
|
@Bean
|
|
public CommandLineRunner populate(
|
|
@Autowired IllnessRepository illnessRepository, @Autowired ClientRepository repository) {
|
|
return (args) -> {
|
|
// Hypothetical conditions with corresponding risk factors
|
|
var hypertension = illnessRepository.save(new Illness(null, "Nadciśnienie", 0.7d));
|
|
var liverCancer = illnessRepository.save(new Illness(null, "Rak wątroby", 0.95d));
|
|
var obesity = illnessRepository.save(new Illness(null, "Otyłość", 0.6d));
|
|
var stupidity = illnessRepository.save(new Illness(null, "Głupota", 0.8d));
|
|
|
|
// Clients with their conditions
|
|
repository.save(new Client(null, Set.of(hypertension, obesity, stupidity)));
|
|
repository.save(new Client(null, Set.of(liverCancer, obesity)));
|
|
repository.save(new Client(null, Set.of()));
|
|
|
|
};
|
|
}
|
|
}
|
|
|