Voomra/trader
Archived
0
Fork 0
Code Activity

Create repository

Browse Source
This commit is contained in:
iMoHax
2014-07-16 16:38:48 +04:00
committed by iMoHax
commit 15bfe2e0f3
79 changed files with 5741 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

50
utils/pom.xml Normal file
View File

@@ -0,0 +1,50 @@
<?xml version="1.0" encoding="UTF-8"?>
<project xmlns="http://maven.apache.org/POM/4.0.0"
xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance"
xsi:schemaLocation="http://maven.apache.org/POM/4.0.0 http://maven.apache.org/xsd/maven-4.0.0.xsd">
<modelVersion>4.0.0</modelVersion>
<parent>
<groupId>trader</groupId>
<artifactId>Trader</artifactId>
<version>1.0</version>
<relativePath>../pom.xml</relativePath>
</parent>
<artifactId>utils</artifactId>
<packaging>jar</packaging>
<dependencies>
<dependency>
<groupId>${project.groupId}</groupId>
<artifactId>core</artifactId>
<version>${project.version}</version>
</dependency>
<dependency>
<groupId>org.slf4j</groupId>
<artifactId>slf4j-api</artifactId>
</dependency>
<dependency>
<groupId>org.apache.poi</groupId>
<artifactId>poi</artifactId>
</dependency>
<dependency>
<groupId>org.apache.poi</groupId>
<artifactId>poi-ooxml</artifactId>
</dependency>
<dependency>
<groupId>org.apache.poi</groupId>
<artifactId>poi-ooxml-schemas</artifactId>
</dependency>
</dependencies>
<build>
<sourceDirectory>src/main/java</sourceDirectory>
<testSourceDirectory>src/test/java</testSourceDirectory>
<resources><resource><directory>src/main/resources</directory><filtering>false</filtering></resource></resources>
<testResources><testResource><directory>src/test/resources</directory><filtering>false</filtering></testResource></testResources>
</build>
</project>

4
utils/src/main/java/META-INF/MANIFEST.MF Normal file
View File

@@ -0,0 +1,4 @@
Manifest-Version: 1.0
Class-Path: slf4j-api-1.7.5.jar poi-3.10-FINAL-20140208.jar poi-ooxml-
3.10-FINAL-20140208.jar poi-ooxml-schemas-3.10-FINAL-20140208.jar

149
utils/src/main/java/ru/trader/store/XSSFImporter.java Normal file
View File

@@ -0,0 +1,149 @@
package ru.trader.store;
import org.apache.poi.openxml4j.exceptions.InvalidFormatException;
import org.apache.poi.openxml4j.exceptions.OpenXML4JException;
import org.apache.poi.openxml4j.opc.OPCPackage;
import org.apache.poi.xssf.eventusermodel.ReadOnlySharedStringsTable;
import org.apache.poi.xssf.eventusermodel.XSSFReader;
import org.apache.poi.xssf.eventusermodel.XSSFSheetXMLHandler;
import org.apache.poi.xssf.model.StylesTable;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.xml.sax.ContentHandler;
import org.xml.sax.InputSource;
import org.xml.sax.SAXException;
import org.xml.sax.XMLReader;
import org.xml.sax.helpers.XMLReaderFactory;
import ru.trader.core.*;
import java.io.File;
import java.io.IOException;
import java.io.InputStream;
import java.util.ArrayList;
import java.util.Iterator;
import java.util.List;
import java.util.regex.Matcher;
import java.util.regex.Pattern;
public class XSSFImporter {
private final static Logger LOG = LoggerFactory.getLogger(XSSFImporter.class);
private OPCPackage pkg;
private final XSSFReader reader;
protected final Iterator<InputStream> sheetItr;
protected final Market market;
protected final List<Vendor> vendors;
public XSSFImporter(File file) throws IOException {
pkg = null;
try {
pkg = OPCPackage.open(file.getPath());
reader = new XSSFReader(pkg);
sheetItr = reader.getSheetsData();
market = new SimpleMarket();
vendors = new ArrayList<>();
} catch (OpenXML4JException e) {
throw new IOException(e);
}
}
public Market doImport() throws IOException, SAXException {
if (pkg==null) throw new IllegalStateException("No init package");
try {
XMLReader saxParser = XMLReaderFactory.createXMLReader();
saxParser.setContentHandler(getHandler());
saxParser.parse(getInputSource());
market.addVendors(vendors);
} finally {
if (pkg!=null) {
try { pkg.close();} catch (IOException e) {LOG.warn("Error on close pkg",e);}
pkg = null;
}
}
return market;
}
protected InputSource getInputSource() {
return new InputSource(sheetItr.next());
}
protected ContentHandler getHandler() throws IOException, SAXException {
StylesTable styles;
try {
styles = reader.getStylesTable();
} catch (InvalidFormatException e) {
throw new SAXException(e);
}
ReadOnlySharedStringsTable sharedStrings = new ReadOnlySharedStringsTable(pkg);
return new XSSFSheetXMLHandler(styles, sharedStrings, new LoaderSheetContentsHandler(), false);
}
protected class LoaderSheetContentsHandler implements XSSFSheetXMLHandler.SheetContentsHandler {
private Item item;
@Override
public void startRow(int rowNum) {
}
@Override
public void endRow() {
}
@Override
public void cell(String cellReference, String formattedValue) {
LOG.trace("Parce cell {}, value = {}", cellReference, formattedValue);
if (formattedValue==null||formattedValue.isEmpty()) return;
RC cell = new RC(cellReference);
if (cell.row == 1) {
if (cell.column > 1 &&cell.column % 2 == 0){
LOG.trace("add vendor");
vendors.add(new SimpleVendor(formattedValue));
}
} else if (cell.row>2){
if (cell.column == 1){
LOG.trace("create item");
item = new Item(formattedValue);
market.add(item);
} else {
LOG.trace("add offer");
Offer offer = new Offer(cell.column % 2 == 0? OFFER_TYPE.BUY : OFFER_TYPE.SELL, item, Double.valueOf(formattedValue));
vendors.get(cell.column/2 -1).add(offer);
}
}
}
@Override
public void headerFooter(String text, boolean isHeader, String tagName) {
}
}
private static final Pattern CELL_REF = Pattern.compile("(\\D+)(\\d+)");
private class RC {
int row;
int column;
public RC(String cellReference){
Matcher matcher = CELL_REF.matcher(cellReference);
if (matcher.find()){
column = wordToColumn(matcher.group(1));
row = Integer.valueOf(matcher.group(2));
}
}
}
public static int wordToColumn(String word){
word = word.toUpperCase();
int column = -1;
for (int i = 0; i < word.length(); ++i) {
int c = word.charAt(i);
column = (column + 1) * 26 + c - 'A';
}
return column+1;
}
}

27
utils/utils.iml Normal file
View File

@@ -0,0 +1,27 @@
<?xml version="1.0" encoding="UTF-8"?>
<module org.jetbrains.idea.maven.project.MavenProjectsManager.isMavenModule="true" type="JAVA_MODULE" version="4">
<component name="NewModuleRootManager" LANGUAGE_LEVEL="JDK_1_8" inherit-compiler-output="false">
<output url="file://$MODULE_DIR$/target/classes" />
<output-test url="file://$MODULE_DIR$/target/test-classes" />
<content url="file://$MODULE_DIR$">
<sourceFolder url="file://$MODULE_DIR$/src/main/java" isTestSource="false" />
<sourceFolder url="file://$MODULE_DIR$/src/test/java" isTestSource="true" />
<sourceFolder url="file://$MODULE_DIR$/src/main/resources" type="java-resource" />
<excludeFolder url="file://$MODULE_DIR$/target" />
</content>
<orderEntry type="inheritedJdk" />
<orderEntry type="sourceFolder" forTests="false" />
<orderEntry type="module" module-name="core" />
<orderEntry type="library" name="Maven: org.slf4j:slf4j-api:1.7.7" level="project" />
<orderEntry type="library" name="Maven: org.jetbrains:annotations:13.0" level="project" />
<orderEntry type="library" name="Maven: org.apache.poi:poi:3.10-FINAL" level="project" />
<orderEntry type="library" name="Maven: commons-codec:commons-codec:1.5" level="project" />
<orderEntry type="library" name="Maven: org.apache.poi:poi-ooxml:3.10-FINAL" level="project" />
<orderEntry type="library" name="Maven: org.apache.poi:poi-ooxml-schemas:3.10-FINAL" level="project" />
<orderEntry type="library" name="Maven: org.apache.xmlbeans:xmlbeans:2.3.0" level="project" />
<orderEntry type="library" name="Maven: stax:stax-api:1.0.1" level="project" />
<orderEntry type="library" name="Maven: dom4j:dom4j:1.6.1" level="project" />
<orderEntry type="library" name="Maven: xml-apis:xml-apis:1.0.b2" level="project" />
</component>
</module>