com.jivesoftware.os.routing.bird.endpoints.base.LocateStringResource.java Source code

Introduction

Here is the source code for com.jivesoftware.os.routing.bird.endpoints.base.LocateStringResource.java

Source

/*
 * Copyright 2013 Jive Software, Inc
 *
 * 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.jivesoftware.os.routing.bird.endpoints.base;

import com.jivesoftware.os.mlogger.core.MetricLogger;
import com.jivesoftware.os.mlogger.core.MetricLoggerFactory;
import java.io.BufferedInputStream;
import java.io.ByteArrayInputStream;
import java.io.FileInputStream;
import java.io.InputStream;
import java.io.StringWriter;
import java.util.Enumeration;
import java.util.HashMap;
import java.util.Map;
import java.util.zip.ZipEntry;
import java.util.zip.ZipFile;
import java.util.zip.ZipInputStream;
import org.apache.commons.io.IOUtils;

class LocateStringResource {

    private static final MetricLogger LOG = MetricLoggerFactory.getLogger();
    private final String jarFileName;
    private final String resourceName;

    LocateStringResource(String jarFileName, String resourceName) {
        this.jarFileName = jarFileName;
        this.resourceName = resourceName;
    }

    String getStringResource() {
        Map<String, Long> htSizes = new HashMap<>();
        try {
            try (final ZipFile zf = new ZipFile(jarFileName)) {
                Enumeration<? extends ZipEntry> e = zf.entries();
                while (e.hasMoreElements()) {
                    ZipEntry ze = e.nextElement();
                    if (!ze.getName().equals(resourceName)) {
                        continue;
                    }
                    htSizes.put(ze.getName(), ze.getSize());
                }
            }
            // extract resources and put them into the hashtable.
            FileInputStream fis = new FileInputStream(jarFileName);
            BufferedInputStream bis = new BufferedInputStream(fis);
            try (final ZipInputStream zis = new ZipInputStream(bis)) {
                ZipEntry ze;
                while ((ze = zis.getNextEntry()) != null) {
                    if (!ze.getName().equals(resourceName)) {
                        continue;
                    }
                    if (ze.isDirectory()) {
                        continue;
                    }
                    int size = (int) ze.getSize();
                    // -1 means unknown size.
                    if (size == -1) {
                        size = htSizes.get(ze.getName()).intValue();
                    }
                    byte[] b = new byte[size];
                    int rb = 0;
                    int chunk = 0;
                    while ((size - rb) > 0) {
                        chunk = zis.read(b, rb, size - rb);
                        if (chunk == -1) {
                            break;
                        }
                        rb += chunk;
                    }
                    InputStream inputStream = new ByteArrayInputStream(b);
                    StringWriter writer = new StringWriter();
                    IOUtils.copy(inputStream, writer, "UTF-8");
                    return writer.toString();
                }
            }
        } catch (Exception e) {
            LOG.warn("Failed to locate " + resourceName, e);
        }
        return null;
    }

}