com.brihaspathee.zeus.mapper.impl.AccountMapperImpl Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of member-mgmt-service Show documentation
Member Management Service
The newest version!
package com.brihaspathee.zeus.mapper.impl;

import com.brihaspathee.zeus.domain.entity.Account;
import com.brihaspathee.zeus.domain.entity.Broker;
import com.brihaspathee.zeus.dto.account.AccountDto;
import com.brihaspathee.zeus.dto.account.BrokerDto;
import com.brihaspathee.zeus.mapper.interfaces.*;
import lombok.RequiredArgsConstructor;
import lombok.extern.slf4j.Slf4j;
import org.springframework.stereotype.Component;

import java.util.ArrayList;
import java.util.HashSet;
import java.util.Set;
import java.util.stream.Collectors;

/**
 * Created in Intellij IDEA
 * User: Balaji Varadharajan
 * Date: 20, April 2022
 * Time: 4:25 PM
 * Project: Zeus
 * Package Name: com.brihaspathee.zeus.mapper.impl
 * To change this template use File | Settings | File and Code Template
 */
@Slf4j
@Component
@RequiredArgsConstructor
public class AccountMapperImpl implements AccountMapper {

    /**
     * Member Mapper to map the member entity
     */
    private final MemberMapper memberMapper;

    /**
     * Enrollment span mapper to map the enrollment span entity
     */
    private final EnrollmentSpanMapper enrollmentSpanMapper;

    /**
     * Account attribute mapper to map the account attribute entity
     */
    private final AccountAttributeMapper attributeMapper;

    /**
     * Broker mapper to map the broker entity
     */
    private final BrokerMapper brokerMapper;

    /**
     * Sponsor mapper to map the sponsor entity
     */
    private final SponsorMapper sponsorMapper;

    /**
     * Payer mapper to map the payer entity
     */
    private final PayerMapper payerMapper;

    /**
     * Convert account dto to account entity
     * @param accountDto
     * @return
     */
    @Override
    public Account accountDtoToAccount(AccountDto accountDto) {
        if(accountDto == null){
            return null;
        }
        Account account = Account.builder()
                .accountSK(accountDto.getAccountSK())
                .accountNumber(accountDto.getAccountNumber())
                .lineOfBusinessTypeCode(accountDto.getLineOfBusinessTypeCode())
                .ztcn(accountDto.getZtcn())
                .source(accountDto.getSource())
                .members(memberMapper.memberDtosToMembers(accountDto.getMembers()))
                .enrollmentSpans(enrollmentSpanMapper.enrollmentSpanDtosToEnrollmentSpans(accountDto.getEnrollmentSpans()))
                .accountAttributes(attributeMapper.accountAttributeDtosToAccountAttributes(accountDto.getAccountAttributes()))
                .createdDate(accountDto.getCreatedDate())
                .updatedDate(accountDto.getUpdatedDate())
                .build();
        if(accountDto.getBrokers() != null && !accountDto.getBrokers().isEmpty()){
            account.setBrokers(new HashSet<>(brokerMapper.brokerDtosToBrokers(new ArrayList<>(accountDto.getBrokers()))));
        }
        if(accountDto.getSponsors() != null && !accountDto.getSponsors().isEmpty()){
            account.setSponsors(new HashSet<>(sponsorMapper.sponsorDtosToSponsors(new ArrayList<>(accountDto.getSponsors()))));
        }
        if(accountDto.getPayers() != null && !accountDto.getPayers().isEmpty()){
            account.setPayers(new HashSet<>(payerMapper.payerDtosToPayers(new ArrayList<>(accountDto.getPayers()))));
        }
        return account;
    }

    /**
     * Convert account entity to account dto
     * @param account
     * @return
     */
    @Override
    public AccountDto accountToAccountDto(Account account) {
        if(account == null){
            return null;
        }
        AccountDto accountDto = AccountDto.builder()
                .accountSK(account.getAccountSK())
                .accountNumber(account.getAccountNumber())
                .lineOfBusinessTypeCode(account.getLineOfBusinessTypeCode())
                .ztcn(account.getZtcn())
                .source(account.getSource())
                .members(memberMapper.membersToMemberDtos(account.getMembers()))
                .enrollmentSpans(enrollmentSpanMapper.enrollmentSpansToEnrollmentSpanDtos(account.getEnrollmentSpans()))
                .accountAttributes(attributeMapper.accountAttributesToAccountAttributeDtos(account.getAccountAttributes()))
                .createdDate(account.getCreatedDate())
                .updatedDate(account.getUpdatedDate())
                .build();
        if(account.getBrokers() != null && !account.getBrokers().isEmpty()){
            accountDto.setBrokers(new HashSet<>(brokerMapper.brokersToBrokerDtos(new ArrayList<>(account.getBrokers()))));
        }
        if(account.getSponsors() != null && !account.getSponsors().isEmpty()){
            accountDto.setSponsors(new HashSet<>(sponsorMapper.sponsorsToSponsorDtos(new ArrayList<>(account.getSponsors()))));
        }
        if(account.getPayers() != null && !account.getPayers().isEmpty()){
            accountDto.setPayers(new HashSet<>(payerMapper.payersToPayerDtos(new ArrayList<>(account.getPayers()))));
        }
        return accountDto;
    }

    /**
     * Convert account dtos to account entities
     * @param accountDtos
     * @return
     */
    @Override
    public Set accountDtosToAccount(Set accountDtos) {
        if(accountDtos !=null && !accountDtos.isEmpty()){
            return accountDtos.stream().map(this::accountDtoToAccount).collect(Collectors.toSet());
        }else{
            return null;
        }
    }

    /**
     * Convert account entities to account dtos
     * @param accounts
     * @return
     */
    @Override
    public Set accountToAccountDtos(Set accounts) {
        if(accounts !=null && !accounts.isEmpty()){
            return accounts.stream().map(this::accountToAccountDto).collect(Collectors.toSet());
        }else{
            return null;
        }
    }
}