class InMemoryAtomikosDataSourceRepository implements JtaDataSourceRepository {

    @Getter
    @Setter
    private Map<String, AtomikosDataSourceConfig> jta = new HashMap<>();


    @PostConstruct
    public void init() {<FILL_FUNCTION_BODY>}

    @Override
    public List<AtomikosDataSourceConfig> findAll() {
        return new ArrayList<>(jta.values());
    }

    @Override
    public AtomikosDataSourceConfig findById(String dataSourceId) {
        return jta.get(dataSourceId);
    }

    @Override
    public AtomikosDataSourceConfig add(AtomikosDataSourceConfig config) {
        return jta.put(config.getId(), config);
    }

    @Override
    public AtomikosDataSourceConfig remove(String dataSourceId) {
        return jta.remove(dataSourceId);
    }
}

        jta.forEach((id, config) -> {
            if (config.getId() == null) {
                config.setId(id);
            } else if (!config.getId().equals(id)) {
                jta.put(config.getId(), config);
            }
        });
    
<no_super_class>