Mercury-IM/data/src/main/java/org/mercury_im/messenger/data/mapping/DirectChatMapping.java

45 lines
1.3 KiB
Java

package org.mercury_im.messenger.data.mapping;
import org.mercury_im.messenger.data.model.DirectChatModel;
import org.mercury_im.messenger.entity.chat.DirectChat;
import org.mercury_im.messenger.entity.chat.IDirectChat;
import org.mercury_im.messenger.entity.contact.Peer;
import javax.inject.Inject;
public class DirectChatMapping extends AbstractMapping<DirectChat, DirectChatModel> {
private final PeerMapping peerMapping;
@Inject
public DirectChatMapping(PeerMapping peerMapping) {
this.peerMapping = peerMapping;
}
@Override
public DirectChat newEntity(DirectChatModel model) {
return new IDirectChat();
}
@Override
public DirectChatModel newModel(DirectChat entity) {
return new DirectChatModel();
}
@Override
public DirectChatModel mapToModel(DirectChat entity, DirectChatModel model) {
model.setId(entity.getId());
model.setPeer(peerMapping.toModel(entity.getPeer(), model.getPeer()));
return model;
}
@Override
public DirectChat mapToEntity(DirectChatModel model, DirectChat entity) {
entity.setId(model.getId());
Peer peer = peerMapping.toEntity(model.getPeer(), entity.getPeer());
entity.setPeer(peer);
entity.setAccount(peer.getAccount());
return entity;
}
}