현성필
73 lines
2.5 KiB
Java
73 lines
2.5 KiB
Java
package com.eactive.httpmockserver.api.mapper;
|
|
|
|
import com.eactive.httpmockserver.api.dto.MockHeaderDTO;
|
|
import com.eactive.httpmockserver.api.dto.MockResponseDTO;
|
|
import com.eactive.httpmockserver.api.dto.MockRouteDTO;
|
|
import com.eactive.httpmockserver.api.entity.MockResponse;
|
|
import com.eactive.httpmockserver.api.entity.MockRoute;
|
|
import com.eactive.httpmockserver.common.mapper.CommonMapper;
|
|
import org.mapstruct.Mapper;
|
|
import org.mapstruct.Mapping;
|
|
import org.mapstruct.Named;
|
|
import org.mapstruct.ReportingPolicy;
|
|
import org.springframework.stereotype.Component;
|
|
|
|
import java.util.ArrayList;
|
|
import java.util.HashMap;
|
|
import java.util.List;
|
|
import java.util.Map;
|
|
|
|
@Mapper(componentModel = "spring",
|
|
uses = {CommonMapper.class},
|
|
unmappedTargetPolicy = ReportingPolicy.IGNORE)
|
|
@Component
|
|
public abstract class MockRouteMapper {
|
|
public abstract MockRoute map(MockRouteDTO dto);
|
|
|
|
public abstract MockRouteDTO map(MockRoute entity);
|
|
|
|
@Mapping(target = "headers", source = "headers", qualifiedByName = "mapHeadersToMap")
|
|
@Mapping(target ="defaultResponse", source = "defaultResponse", qualifiedByName = "mapEntityDefaultResponse")
|
|
public abstract MockResponse map(MockResponseDTO dto);
|
|
|
|
@Mapping(target = "headers", source = "headers", qualifiedByName = "mapHeadersToList")
|
|
@Mapping(target ="defaultResponse", source = "defaultResponse", qualifiedByName = "mapDTODefaultResponse")
|
|
public abstract MockResponseDTO map(MockResponse entity);
|
|
|
|
@Named("mapDTODefaultResponse")
|
|
public String map(boolean value) {
|
|
return value ? "true" : "false";
|
|
}
|
|
|
|
@Named("mapEntityDefaultResponse")
|
|
public boolean map(String value) {
|
|
if (value == null) return false;
|
|
return value.equals("true");
|
|
}
|
|
|
|
@Named("mapHeadersToList")
|
|
public List<MockHeaderDTO> mapHeadersToList(Map<String, String> headers) {
|
|
List<MockHeaderDTO> list = new ArrayList<>();
|
|
if (headers != null) {
|
|
headers.forEach((key, value) -> {
|
|
MockHeaderDTO header = new MockHeaderDTO();
|
|
header.setKey(key);
|
|
header.setValue(value);
|
|
list.add(header);
|
|
});
|
|
}
|
|
return list;
|
|
}
|
|
|
|
@Named("mapHeadersToMap")
|
|
public Map<String, String> mapHeadersToMap(List<MockHeaderDTO> headers) {
|
|
Map<String, String> map = new HashMap<>();
|
|
if (headers != null) {
|
|
headers.forEach(header -> {
|
|
map.put(header.getKey(), header.getValue());
|
|
});
|
|
}
|
|
return map;
|
|
}
|
|
}
|