com.yqboots.initializer.core.builder.excel.DataDictSheetBuilder.java Source code

Introduction

Here is the source code for com.yqboots.initializer.core.builder.excel.DataDictSheetBuilder.java

Source

/*
 *
 *  * Copyright 2015-2016 the original author or authors.
 *  *
 *  * Licensed under the Apache License, Version 2.0 (the "License");
 *  * you may not use this file except in compliance with the License.
 *  * You may obtain a copy of the License at
 *  *
 *  *      http://www.apache.org/licenses/LICENSE-2.0
 *  *
 *  * Unless required by applicable law or agreed to in writing, software
 *  * distributed under the License is distributed on an "AS IS" BASIS,
 *  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 *  * See the License for the specific language governing permissions and
 *  * limitations under the License.
 *
 */
package com.yqboots.initializer.core.builder.excel;

import com.yqboots.dict.core.DataDict;
import com.yqboots.dict.core.DataDicts;
import com.yqboots.fss.core.support.FileType;
import com.yqboots.initializer.core.ProjectMetadata;
import org.apache.commons.lang3.StringUtils;
import org.apache.poi.ss.usermodel.Cell;
import org.apache.poi.ss.usermodel.Row;
import org.apache.poi.ss.usermodel.Sheet;
import org.springframework.oxm.Marshaller;
import org.springframework.util.Assert;

import javax.xml.transform.stream.StreamResult;
import java.io.File;
import java.io.FileWriter;
import java.io.IOException;
import java.nio.file.Files;
import java.nio.file.Path;
import java.nio.file.Paths;
import java.util.ArrayList;
import java.util.List;

/**
 * It builds the Excel sheet to retrieve DataDict.
 *
 * @author Eric H B Zhan
 * @since 1.0.0
 */
public class DataDictSheetBuilder extends AbstractSheetBuilder {
    private final Marshaller marshaller;

    private final DataDictProperties properties;

    /**
     * Constructs the DataDictSheetBuilder.
     *
     * @param marshaller marshal the generated MenuItem to XML
     * @param properties properties for MenuItem
     */
    public DataDictSheetBuilder(final Marshaller marshaller, DataDictProperties properties) {
        super(properties.getSheetName());
        this.marshaller = marshaller;
        this.properties = properties;
    }

    @Override
    protected void formatChecking(final Sheet sheet) {
        // get the title row
        final Row row = sheet.getRow(0);

        final Cell nameCell = row.getCell(0);
        final Cell textCell = row.getCell(1);
        final Cell valueCell = row.getCell(2);

        Assert.isTrue(StringUtils.equalsIgnoreCase(nameCell.getStringCellValue(), "name"),
                "Column 'name' is required");
        Assert.isTrue(StringUtils.equalsIgnoreCase(textCell.getStringCellValue(), "text"),
                "Column 'text' is required");
        Assert.isTrue(StringUtils.equalsIgnoreCase(valueCell.getStringCellValue(), "value"),
                "Column 'value' is required");
    }

    @Override
    protected void doBuild(final Path root, final ProjectMetadata metadata, final Sheet sheet) throws IOException {
        final List<DataDict> items = new ArrayList<>();

        for (Row row : sheet) {
            // ignore the header
            if (row.getRowNum() < 1) {
                continue;
            }

            items.add(getDataDicts(row));
        }

        // generate an XML for the application importing into Database
        final Path targetPath = Paths.get(root + File.separator + properties.getExportRelativePath());
        if (!Files.exists(targetPath)) {
            Files.createDirectories(targetPath);
        }

        final Path file = Paths.get(targetPath + File.separator + properties.getExportName() + FileType.DOT_XML);
        try (FileWriter writer = new FileWriter(file.toFile())) {
            marshaller.marshal(new DataDicts(items), new StreamResult(writer));
        }
    }

    private static DataDict getDataDicts(final Row row) {
        DataDict result = new DataDict();
        Assert.notNull(row.getCell(0));
        Assert.notNull(row.getCell(1));
        Assert.notNull(row.getCell(2));

        Cell cell = row.getCell(0);
        result.setName(cell.getStringCellValue());
        cell = row.getCell(1);
        result.setText(cell.getStringCellValue());
        cell = row.getCell(2);
        result.setValue(cell.getStringCellValue());
        cell = row.getCell(3);
        if (cell != null) {
            result.setDescription(cell.getStringCellValue());
        }

        return result;
    }
}