com.xinferin.dao.DAOCustomerRegistrationImpl.java Source code

Introduction

Here is the source code for com.xinferin.dao.DAOCustomerRegistrationImpl.java

Source

/*
    
 *   XLicenser Source Code
 *   Copyright (C) 2015 XINFERIN Technologies 
 *  
 *    All Rights Reserved
 * 
 */

package com.xinferin.dao;

import java.text.DateFormat;
import java.text.SimpleDateFormat;
import java.util.UUID;

import javax.sql.DataSource;

import org.apache.commons.codec.binary.Base64;
import org.springframework.jdbc.core.namedparam.MapSqlParameterSource;
import org.springframework.jdbc.core.namedparam.NamedParameterJdbcTemplate;
import org.springframework.jdbc.support.GeneratedKeyHolder;
import org.springframework.jdbc.support.KeyHolder;
import org.springframework.transaction.annotation.Propagation;
import org.springframework.transaction.annotation.Transactional;

import com.xinferin.licensing.LicenceGenerator;
import com.xinferin.model.CustomerRegistration;
import com.xinferin.model.Licence;
import com.xinferin.model.LicenceType;
import com.xinferin.model.Product;
import com.xinferin.model.Provider;

public class DAOCustomerRegistrationImpl implements DAOCustomerRegistration {

    private NamedParameterJdbcTemplate jdbcTemplate;
    private DataSource dataSource;

    public DAOCustomerRegistrationImpl(DataSource dataSource) {
        this.dataSource = dataSource;
        jdbcTemplate = new NamedParameterJdbcTemplate(dataSource);
    }

    @Override
    @Transactional(propagation = Propagation.REQUIRED, rollbackFor = Exception.class)
    public void register(CustomerRegistration creg) throws Exception {

        DAOProductImpl daoProductImpl = new DAOProductImpl(dataSource);
        Product product = daoProductImpl.get(creg.getProduct());
        if (product == null)
            throw new ProductNotFoundException(creg.getProduct());

        DAOProviderImpl daoProviderImpl = new DAOProviderImpl(dataSource);
        Provider provider = daoProviderImpl.get(creg.getProvider());
        if (provider == null)
            throw new ProviderNotFoundException(creg.getProvider());

        // add a new customer
        DAOCustomerImpl daoCustomerImpl = new DAOCustomerImpl(dataSource);
        int customerId = daoCustomerImpl.add(creg.getCustomer());
        if (customerId < 0)
            throw new CustomerNotCreatedException();

        // register the new customer
        int registrationId = addRegistration(customerId, product.getId(), provider.getId(), creg.getLicenceType(),
                creg.getPrice());
        if (registrationId < 0)
            throw new RegistrationNotCreatedException();

        // first add the licence before updating with the newly generated licence id the key_info field
        Licence licence = new Licence();
        licence.setProduct_id(product.getId());

        DAOLicenceImpl daoLicenceImpl = new DAOLicenceImpl(dataSource);
        int licenceId = daoLicenceImpl.add(licence);
        licence.setId(licenceId);

        // use the LicenceGenerator to create a new licence key
        // LICENCE_ID | PRODUCT | CUSTOMER FULLNAME | CUSTOMER EMAIL | LICENCE TYPE | EXPIRY DATE

        DateFormat df = new SimpleDateFormat("yyyy-MM-dd");
        String key_info = licenceId + "|" + creg.getProduct() + "|" + creg.getCustomer().getFname() + ", "
                + creg.getCustomer().getLname() + "|" + creg.getCustomer().getEmail() + "|" + creg.getLicenceType()
                + "|" + df.format(creg.getExpiryDate());

        LicenceGenerator lg = new LicenceGenerator(product.getLicence_params());
        byte[] generatedKey = lg.signData(key_info.getBytes());
        String signedB64 = Base64.encodeBase64String(generatedKey);

        // update the new licence with the generated key.
        licence.setKey_info(Base64.encodeBase64String(key_info.getBytes()));
        licence.setGenerated_key(signedB64);
        licence.setCustomer_key(UUID.randomUUID().toString());
        licence.setExpiry_date(creg.getExpiryDate());
        daoLicenceImpl.update(licence);

        // bind the registration with the licence.
        bindRegistrationAndLicence(registrationId, licenceId);
    }

    private int addRegistration(int customerId, int productId, int providerId, LicenceType licenceType,
            double price) {
        String sql = "INSERT INTO registration (date, customer_id, product_id, provider_id, licence_type, price)"
                + " VALUES (NOW(), :customer_id, :product_id, :provider_id, :licence_type, :price)";

        MapSqlParameterSource args = new MapSqlParameterSource();
        args.addValue("customer_id", customerId);
        args.addValue("product_id", productId);
        args.addValue("provider_id", providerId);
        args.addValue("licence_type", licenceType.getLicenceType());
        args.addValue("price", price);
        KeyHolder keyHolder = new GeneratedKeyHolder();

        jdbcTemplate.update(sql, args, keyHolder);

        return keyHolder.getKey().intValue();
    }

    private void bindRegistrationAndLicence(int registration_id, int licence_id) {
        String sql = "INSERT INTO licence_registration (registration_id, licence_id)"
                + " VALUES (:registration_id, :licence_id)";

        MapSqlParameterSource args = new MapSqlParameterSource();
        args.addValue("registration_id", registration_id);
        args.addValue("licence_id", licence_id);

        jdbcTemplate.update(sql, args);
    }
}