Many resources are needed to download a project. Please understand that we have to compensate our server costs. Thank you in advance. Project price only 1 $
You can buy this project and download/modify it how often you want.
package com.unbxd.client.feed;
import com.unbxd.client.ConnectionManager;
import com.unbxd.client.feed.exceptions.FeedInputException;
import com.unbxd.client.feed.exceptions.FeedUploadException;
import com.unbxd.client.feed.response.FeedResponse;
import org.apache.http.HttpResponse;
import org.apache.http.client.methods.HttpPost;
import org.apache.http.entity.mime.MultipartEntityBuilder;
import org.apache.http.entity.mime.content.FileBody;
import org.apache.http.impl.client.CloseableHttpClient;
import org.apache.http.impl.client.HttpClients;
import org.apache.log4j.Logger;
import org.codehaus.jackson.map.ObjectMapper;
import org.w3c.dom.Document;
import javax.xml.transform.Transformer;
import javax.xml.transform.TransformerFactory;
import javax.xml.transform.dom.DOMSource;
import javax.xml.transform.stream.StreamResult;
import java.io.*;
import java.util.*;
import java.util.zip.ZipEntry;
import java.util.zip.ZipOutputStream;
/**
* Created with IntelliJ IDEA.
* User: sourabh
* Date: 07/07/14
* Time: 3:30 PM
*
* http://unbxd.com/docs/
*
* Client class for calling Feed APIs
*/
public class FeedClient {
private static final Logger LOG = Logger.getLogger(FeedClient.class);
private String siteKey;
private String secretKey;
private boolean secure;
private List _fields;
private Map _addedDocs;
private Map _updatedDocs;
private Set _deletedDocs;
private List _taxonomyNodes;
private Map> _taxonomyMappings;
protected FeedClient(String siteKey, String secretKey, boolean secure){
this.siteKey = siteKey;
this.secretKey = secretKey;
this.secure = secure;
_fields = new ArrayList();
_addedDocs = new HashMap();
_updatedDocs = new HashMap();
_deletedDocs = new HashSet();
_taxonomyNodes = new ArrayList();
_taxonomyMappings = new HashMap>();
}
private String getFeedUrl(){
return (secure ? "https://" : "http://") + "feed.unbxdapi.com/upload/v2/" + secretKey + "/" + siteKey;
}
/**
* Adds schema for a field. Schema needs to be added only once.
*
* @param fieldName Name of the field. Following rules apply for field names.
*
*
Should be alphnumeric
*
Can contain hyphens and underscores
*
Can not start and end with -- or __
*
Can not start with numbers
*
* @param datatype Datatype of the field. Refer {@link DataType}
* @param multivalued True for allowing multiple values for each document
* @param autosuggest True to include field in autosuggest response
* @return this
*/
public FeedClient addSchema(String fieldName, DataType datatype, boolean multivalued, boolean autosuggest){
_fields.add(new FeedField(fieldName, datatype, multivalued, autosuggest));
return this;
}
/**
* Adds schema for a field. Schema needs to be added only once.
*
* @param fieldName Name of the field. Following rules apply for field names.
*
*
Should be alphnumeric
*
Can contain hyphens and underscores
*
Can not start and end with -- or __
*
Can not start with numbers
*
* @param datatype Datatype of the field. Refer {@link DataType}
* @return this
*/
public FeedClient addSchema(String fieldName, DataType datatype){
_fields.add(new FeedField(fieldName, datatype, false, false));
return this;
}
/**
* Adds a product to the field. If a product with the same uniqueId is found to be already present the product will be overwritten
* @param product
* @return this
*/
public FeedClient addProduct(FeedProduct product) {
_addedDocs.put(product.getUniqueId(), product);
return this;
}
/**
* Adds a list of products to the field. If a product with the same uniqueId is found to be already present the product will be overwritten
* @param products
* @return this
*/
public FeedClient addProducts(List products) {
for(FeedProduct product : products){
this.addProduct(product);
}
return this;
}
/**
* Add a variant to a product.
* @param parentUniqueId Unique Id of the parent product
* @param variantAttributes Attributes of the variant
* @return this
* @throws FeedInputException
*/
public FeedClient addVariant(String parentUniqueId, Map variantAttributes) throws FeedInputException {
if(!_addedDocs.containsKey(parentUniqueId)){
throw new FeedInputException("Parent product needs to be added");
}
_addedDocs.get(parentUniqueId).addAssociatedProduct(variantAttributes);
return this;
}
/**
* Add variants to a product.
* @param parentUniqueId Unique Id of the parent product
* @param variants List of attributes of the variants
* @return this
* @throws FeedInputException
*/
public FeedClient addVariants(String parentUniqueId, List> variants) throws FeedInputException {
for(Map variant : variants){
this.addVariant(parentUniqueId, variant);
}
return this;
}
/**
* Upserts a product.
* @param productDelta Delta of product attributes. uniqueId is mandatory
* @return this
*/
public FeedClient updateProduct(FeedProduct productDelta) {
_updatedDocs.put(productDelta.getUniqueId(), productDelta);
return this;
}
/**
* Upserts products.
* @param productsDeltas Deltas of products attributes. uniqueId is mandatory
* @return this
*/
public FeedClient updateProducts(List productsDeltas) {
for(FeedProduct product : productsDeltas){
this.updateProduct(product);
}
return this;
}
/**
* Deletes a product with given uniqueId
* @param uniqueId
* @return this
*/
public FeedClient deleteProduct(String uniqueId){
_deletedDocs.add(uniqueId);
return this;
}
/**
* Deletes products with given uniqueIds
* @param uniqueIds
* @return this
*/
public FeedClient deleteProducts(List uniqueIds){
_deletedDocs.addAll(uniqueIds);
return this;
}
/**
* Adds a taxonomy node
* @param node
* @return this
*/
public FeedClient addTaxonomyNode(TaxonomyNode node){
_taxonomyNodes.add(node);
return this;
}
/**
* Add taxonomy nodes
* @param nodes
* @return
*/
public FeedClient addTaxonomyNodes(List nodes){
_taxonomyNodes.addAll(nodes);
return this;
}
/**
* Maps a uniqueId with taxonomy nodes
* @param uniqueId
* @param nodeIds List of taxnonomy node Id
* @return this
*/
public FeedClient addTaxonomyMapping(String uniqueId, List nodeIds){
_taxonomyMappings.put(uniqueId, nodeIds);
return this;
}
/**
* Maps uniqueIds with taxonomy nodes
* @param mappings Map of Unique Id - List of taxonomy nodes Ids
* @return this
*/
public FeedClient addTaxonomyMappings(Map> mappings){
_taxonomyMappings.putAll(mappings);
return this;
}
private File zipIt(File inFile) throws IOException {
byte[] buffer = new byte[1024];
ZipOutputStream zos = null;
FileInputStream in = null;
try{
in = new FileInputStream(inFile);
File file = File.createTempFile(siteKey, ".zip");
FileOutputStream fos = new FileOutputStream(file);
zos = new ZipOutputStream(fos);
ZipEntry ze= new ZipEntry(inFile.getName());
zos.putNextEntry(ze);
int len;
while ((len = in.read(buffer)) > 0) {
zos.write(buffer, 0, len);
}
in.close();
zos.closeEntry();
return file;
}finally {
//remember close it
if(zos != null)
zos.close();
if(in != null)
in.close();
}
}
/**
* Uploads the feed to Unbxd platform. The feed is zipped before being pushed.
* Please be realistic and don't push millions of products in a single call. Try to confine it to a few thousands. 10K would be a good number
*
* @param isFullImport If true it will clear the old feed entirely and upload new products. Please use with care.
* @return {@link FeedResponse}
* @throws FeedUploadException
*/
public FeedResponse push(boolean isFullImport) throws FeedUploadException {
CloseableHttpClient httpClient = HttpClients.custom().setConnectionManager(ConnectionManager.getConnectionManager()).build();
Document doc = new FeedFile(_fields, _addedDocs.values(), _updatedDocs.values(), _deletedDocs, _taxonomyNodes, _taxonomyMappings).getDoc();
File file = null;
File zipFile = null;
try {
long t = new Date().getTime();
file = File.createTempFile(siteKey, ".xml");
FileOutputStream fos = null;
try{
fos = new FileOutputStream(file);
TransformerFactory tFactory = TransformerFactory.newInstance();
Transformer transformer = tFactory.newTransformer();
DOMSource source = new DOMSource(doc);
StreamResult result = new StreamResult(fos);
transformer.transform(source, result);
}finally {
if(fos != null)
fos.close();
}
zipFile = this.zipIt(file);
LOG.debug("Stored at : " + zipFile.getAbsolutePath());
String url = getFeedUrl();
if(isFullImport){
url += "?fullimport=true";
}
HttpPost post = new HttpPost(url);
MultipartEntityBuilder entity = MultipartEntityBuilder.create();
entity.addPart("file", new FileBody(zipFile));
post.setEntity(entity.build());
HttpResponse response = httpClient.execute(post);
t = new Date().getTime() - t;
LOG.debug("Took : " + t + " millisecs");
if(response.getStatusLine().getStatusCode() == 200){
try{
ObjectMapper mapper = new ObjectMapper();
Map map = mapper.readValue(new InputStreamReader(response.getEntity().getContent()), Map.class);
return new FeedResponse(map);
}catch (Exception e){
LOG.error("Failed to parse response", e);
throw new FeedUploadException(e);
}
}else {
BufferedReader rd = new BufferedReader(new InputStreamReader(response.getEntity().getContent()));
StringBuffer sb = new StringBuffer();
String line = "";
while ((line = rd.readLine()) != null) {
sb.append(line);
}
String responseText = sb.toString();
LOG.error(responseText);
throw new FeedUploadException(responseText);
}
} catch (Exception e) {
LOG.error(e.getMessage(), e);
throw new FeedUploadException(e);
} finally {
if(null != file) {
file.delete();
}
if(null != zipFile) {
zipFile.delete();
}
}
}
}
