Changeset deea3c4 for src/main/java/com/example/rezevirajmasa/demo/config/ModelMapperConfig.java

Timestamp:

04/28/25 14:21:17 (3 weeks ago)

Author:

Aleksandar Panovski <apano77@…>

Branches:

main

Children:

e15e8d9

Parents:

f5b256e

Message:

Big change done fully handle_reservation_update() trigger works

File:

• src/main/java/com/example/rezevirajmasa/demo/config/ModelMapperConfig.java (modified) (1 diff)

src/main/java/com/example/rezevirajmasa/demo/config/ModelMapperConfig.java

@@ -1 @@
-package com.example.rezevirajmasa.demo.config;public class ModelMapperConfig {
+package com.example.rezevirajmasa.demo.config;
+
+import com.example.rezevirajmasa.demo.dto.ReservationDTO;
+import com.example.rezevirajmasa.demo.dto.RestaurantDTO;
+import com.example.rezevirajmasa.demo.dto.TableDTO;
+import com.example.rezevirajmasa.demo.model.Reservation;
+import com.example.rezevirajmasa.demo.model.Restaurant;
+import com.example.rezevirajmasa.demo.model.TableEntity;
+import org.modelmapper.ModelMapper;
+import org.springframework.context.annotation.Bean;
+import org.springframework.context.annotation.Configuration;
+
+@Configuration
+public class ModelMapperConfig {
+    @Bean
+    public ModelMapper modelMapper() {
+        ModelMapper modelMapper = new ModelMapper();
+
+        // Map Restaurant to RestaurantDTO
+        modelMapper.typeMap(Restaurant.class, RestaurantDTO.class).addMappings(mapper -> {
+            mapper.map(Restaurant::getTablesList, RestaurantDTO::setTablesList);
+        });
+
+        // Map TableEntity to TableDTO
+        modelMapper.typeMap(TableEntity.class, TableDTO.class).addMappings(mapper -> {
+            mapper.map(TableEntity::getReservations, TableDTO::setReservations);
+        });
+
+        // Map Reservation to ReservationDTO
+        modelMapper.typeMap(Reservation.class, ReservationDTO.class);
+
+        return modelMapper;
+    }
 }