All Downloads are FREE. Search and download functionalities are using the official Maven repository.

org.drools.decisiontable.DecisionTableProviderImpl Maven / Gradle / Ivy

There is a newer version: 9.44.0.Final
Show newest version
/*
 * Copyright 2015 Red Hat, Inc. and/or its affiliates.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * 
 *      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 org.drools.decisiontable;

import java.io.File;
import java.io.IOException;
import java.io.InputStream;
import java.io.Reader;
import java.io.UncheckedIOException;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

import org.drools.decisiontable.parser.xls.PropertiesSheetListener;
import org.drools.drl.extensions.DecisionTableProvider;
import org.drools.template.parser.DecisionTableParseException;
import org.drools.util.StringUtils;
import org.kie.api.io.Resource;
import org.kie.internal.builder.DecisionTableConfiguration;
import org.kie.internal.builder.RuleTemplateConfiguration;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

public class DecisionTableProviderImpl
    implements
    DecisionTableProvider {

    private static final transient Logger logger = LoggerFactory.getLogger( DecisionTableProviderImpl.class );

    @Override
    public String loadFromResource(Resource resource,
                                   DecisionTableConfiguration configuration) {

        try {
            return compileResource( resource, configuration );
        } catch (IOException e) {
            throw new UncheckedIOException( e );
        } catch (Exception e) {
            throw new DecisionTableParseException(resource, e);
        }
    }

    @Override
    public List loadFromInputStreamWithTemplates(Resource resource,
                                                         DecisionTableConfiguration configuration) {
        List drls = new ArrayList<>( configuration.getRuleTemplateConfigurations().size() );
        ExternalSpreadsheetCompiler converter = new ExternalSpreadsheetCompiler();
        for ( RuleTemplateConfiguration template : configuration.getRuleTemplateConfigurations() ) {
            try {
                drls.add(converter.compile(resource.getInputStream(),
                                           template.getTemplate().getInputStream(),
                                           InputType.getInputTypeFromDecisionTableInputType(configuration.getInputType()),
                                           template.getRow(),
                                           template.getCol()));
            } catch (IOException e) {
                logger.error( "Cannot open " + template.getTemplate(), e );
            } catch (Exception e) {
                throw new DecisionTableParseException(resource, e);
            }
        }
        return drls;
    }

    private String compileResource(Resource resource,
                                   DecisionTableConfiguration configuration) throws IOException {
        SpreadsheetCompiler compiler = new SpreadsheetCompiler(configuration.isTrimCell());

        switch ( configuration.getInputType() ) {
            case XLS :
            case XLSX :
                if ( StringUtils.isEmpty( configuration.getWorksheetName() ) ) {
                    return compiler.compile( resource,
                                             InputType.XLS );
                } else {
                    return compiler.compile( resource.getInputStream(),
                                             configuration.getWorksheetName() );
                }
            case CSV : {
                return compiler.compile( resource.getInputStream(),
                                         InputType.CSV );
            }
        }

        return null;
    }

    @Override
    public Map> loadPropertiesFromFile(File file, DecisionTableConfiguration configuration) {
        switch (configuration.getInputType()) {
            case XLS :
            case XLSX :
                PropertiesSheetListener propertiesSheetListener = new PropertiesSheetListener();
                InputType.XLS.createParser(propertiesSheetListener).parseFile(file);
                return propertiesSheetListener.getProperties();
            default :
                return new HashMap<>();
        }
    }

    @Override
    public Map> loadPropertiesFromInputStream(InputStream inputStream, DecisionTableConfiguration configuration) {
        switch (configuration.getInputType()) {
            case XLS :
            case XLSX :
                PropertiesSheetListener propertiesSheetListener = new PropertiesSheetListener();
                InputType.XLS.createParser(propertiesSheetListener).parseFile(inputStream);
                return propertiesSheetListener.getProperties();
            default :
                return new HashMap<>();
        }
    }

    /**
     * Adapts a Reader as an InputStream. Adapted from
     * StringInputStream.
     */
    public static class ReaderInputStream extends InputStream {

        /** Source Reader */
        private Reader in;

        private String encoding = System.getProperty( "file.encoding" );

        private byte[] slack;

        private int    begin;

        /**
         * Construct a ReaderInputStream for the specified
         * Reader.
         * 
         * @param reader
         *            Reader. Must not be null.
         */
        public ReaderInputStream(Reader reader) {
            in = reader;
        }

        /**
         * Construct a ReaderInputStream for the specified
         * Reader, with the specified encoding.
         * 
         * @param reader
         *            non-null Reader.
         * @param encoding
         *            non-null String encoding.
         */
        public ReaderInputStream(Reader reader,
                                 String encoding) {
            this( reader );
            if ( encoding == null ) {
                throw new IllegalArgumentException( "encoding must not be null" );
            } else {
                this.encoding = encoding;
            }
        }

        /**
         * Reads from the Reader, returning the same value.
         * 
         * @return the value of the next character in the Reader.
         * 
         * @exception IOException
         *                if the original Reader fails to be read
         */
        public synchronized int read() throws IOException {
            if ( in == null ) {
                throw new IOException( "Stream Closed" );
            }

            byte result;
            if ( slack != null && begin < slack.length ) {
                result = slack[begin];
                if ( ++begin == slack.length ) {
                    slack = null;
                }
            } else {
                byte[] buf = new byte[1];
                if ( read( buf,
                           0,
                           1 ) <= 0 ) {
                    result = -1;
                }
                result = buf[0];
            }

            if ( result < -1 ) {
                result += 256;
            }

            return result;
        }

        /**
         * Reads from the Reader into a byte array
         * 
         * @param b
         *            the byte array to read into
         * @param off
         *            the offset in the byte array
         * @param len
         *            the length in the byte array to fill
         * @return the actual number read into the byte array, -1 at the end of
         *         the stream
         * @exception IOException
         *                if an error occurs
         */
        public synchronized int read(byte[] b,
                                     int off,
                                     int len) throws IOException {
            if ( in == null ) {
                throw new IOException( "Stream Closed" );
            }

            while ( slack == null ) {
                char[] buf = new char[len]; // might read too much
                int n = in.read( buf );
                if ( n == -1 ) {
                    return -1;
                }
                if ( n > 0 ) {
                    slack = new String( buf,
                                        0,
                                        n ).getBytes( encoding );
                    begin = 0;
                }
            }

            if ( len > slack.length - begin ) {
                len = slack.length - begin;
            }

            System.arraycopy( slack,
                              begin,
                              b,
                              off,
                              len );

            if ( (begin += len) >= slack.length ) {
                slack = null;
            }

            return len;
        }

        /**
         * Marks the read limit of the StringReader.
         * 
         * @param limit
         *            the maximum limit of bytes that can be read before the
         *            mark position becomes invalid
         */
        public synchronized void mark(final int limit) {
            try {
                in.mark( limit );
            } catch ( IOException ioe ) {
                throw new RuntimeException( ioe.getMessage() );
            }
        }

        /**
         * @return the current number of bytes ready for reading
         * @exception IOException
         *                if an error occurs
         */
        public synchronized int available() throws IOException {
            if ( in == null ) {
                throw new IOException( "Stream Closed" );
            }
            if ( slack != null ) {
                return slack.length - begin;
            }
            if ( in.ready() ) {
                return 1;
            } else {
                return 0;
            }
        }

        /**
         * @return false - mark is not supported
         */
        public boolean markSupported() {
            return false; // would be imprecise
        }

        /**
         * Resets the StringReader.
         * 
         * @exception IOException
         *                if the StringReader fails to be reset
         */
        public synchronized void reset() throws IOException {
            if ( in == null ) {
                throw new IOException( "Stream Closed" );
            }
            slack = null;
            in.reset();
        }

        /**
         * Closes the Stringreader.
         * 
         * @exception IOException
         *                if the original StringReader fails to be closed
         */
        public synchronized void close() throws IOException {
            if ( in != null ) {
                in.close();
                slack = null;
                in = null;
            }
        }
    }

}




© 2015 - 2024 Weber Informatics LLC | Privacy Policy