com.talent.nio.utils.ByteUtils.java Source code

Introduction

Here is the source code for com.talent.nio.utils.ByteUtils.java

Source

/*
 * Copyright (c) 2013, OpenCloudDB/MyCAT and/or its affiliates. All rights reserved.
 * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
 *
 * This code is free software;Designed and Developed mainly by many Chinese 
 * opensource volunteers. you can redistribute it and/or modify it under the 
 * terms of the GNU General Public License version 2 only, as published by the
 * Free Software Foundation.
 *
 * This code is distributed in the hope that it will be useful, but WITHOUT
 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
 * FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License
 * version 2 for more details (a copy is included in the LICENSE file that
 * accompanied this code).
 *
 * You should have received a copy of the GNU General Public License version
 * 2 along with this work; if not, write to the Free Software Foundation,
 * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.
 * 
 * Any questions about this component can be directed to it's project Web address 
 * https://code.google.com/p/opencloudb/.
 *
 */
package com.talent.nio.utils;

import io.netty.buffer.ByteBuf;
import io.netty.buffer.Unpooled;

import java.io.IOException;
import java.util.ArrayList;
import java.util.List;

import org.apache.commons.lang.ArrayUtils;

/**
 * 
 * 
 * @filename: com.talent.nio.utils.EncodeUtils
 * @copyright: Copyright (c)2010
 * @company: talent
 * @author: 
 * @version: 1.0
 * @create time: 2013-5-17 ?9:50:23
 * @record <table cellPadding="3" cellSpacing="0" style="width:600px">
 *         <thead style="font-weight:bold;background-color:#e3e197">
 *         <tr>
 *         <td>date</td>
 *         <td>author</td>
 *         <td>version</td>
 *         <td>description</td>
 *         </tr>
 *         </thead> <tbody style="background-color:#ffffeb">
 *         <tr>
 *         <td>2012-5-17</td>
 *         <td></td>
 *         <td>1.0</td>
 *         <td>create</td>
 *         </tr>
 *         </tbody>
 *         </table>
 */
public class ByteUtils {
    /**
     * 
     * @author tanyaowu
     * @param bytes1
     * @param bytes2
     * @return
     */
    public static byte[] joinBytes(byte[] bytes1, byte[] bytes2) {
        if (bytes1 == null && bytes2 == null) {
            return null;
        }

        if (bytes1 == null || bytes1.length == 0) {
            return bytes2 == null ? bytes1 : bytes2;
        }
        if (bytes2 == null || bytes2.length == 0) {
            return bytes1 == null ? bytes2 : bytes1;
        }

        byte[] ret = new byte[bytes1.length + bytes2.length];
        System.arraycopy(bytes1, 0, ret, 0, bytes1.length);
        System.arraycopy(bytes2, 0, ret, bytes1.length, bytes2.length);
        return ret;
    }

    /**
     * 
     * @param buffer
     * @return
     * @throws IOException
     */
    public static List<String> toLinesList(ByteBuf buffer) throws IOException {
        List<String> retList = new ArrayList<String>(20);

        int lastPosition = 0;
        int byteCountInOneLine = 0; // 

        byte lastByte = 0; // 
        int length = buffer.capacity();
        for (int i = 0; i < length; i++) {
            byte b = buffer.getByte(i);// .get();
            boolean isLastByte = (length - 1 == i); // ??
            byteCountInOneLine++;

            if (b == '\n') {
                if ((i > 0 && lastByte == '\r')) {
                    if (byteCountInOneLine == 2) // ?????
                    {
                        retList.add("\r\n"); // 
                    } else {
                        byte[] bs1 = new byte[byteCountInOneLine];
                        buffer.getBytes(lastPosition, bs1);
                        String line1 = new String(bs1, "utf-8");
                        retList.add(line1);
                    }
                } else {
                    if (byteCountInOneLine == 1) // ?????
                    {
                        retList.add("\n"); // 
                    } else {
                        byte[] bs1 = new byte[byteCountInOneLine];
                        buffer.getBytes(lastPosition, bs1);
                        String line1 = new String(bs1, "utf-8");
                        retList.add(line1);
                    }
                }

                byteCountInOneLine = 0;
                lastPosition = i + 1;
            } else if (isLastByte) {
                byte[] bs1 = new byte[byteCountInOneLine];
                buffer.getBytes(lastPosition, bs1);
                String line1 = new String(bs1, "utf-8");
                retList.add(line1);
            }

            lastByte = b;
        }
        return retList;
    }

    /**
     * 
     * @param source
     * @param srcBegin
     * @return
     */
    public static byte[] subbytes(byte[] source, int srcBegin) {
        return subbytes(source, srcBegin, source.length);
    }

    /**
     * 
     * @param source
     * @param srcBegin
     * @param length
     * @return
     */
    public static byte[] subbytes(byte[] source, int srcBegin, int length) {
        byte[] destination = new byte[length];
        getBytes(source, srcBegin, length, destination, 0);

        return destination;
    }

    /**
     * 
     * @param source
     * @param srcBegin
     * @param length
     * @param destination
     * @param dstBegin
     */
    public static void getBytes(byte[] source, int srcBegin, int length, byte[] destination, int dstBegin) {
        System.arraycopy(source, srcBegin, destination, dstBegin, length);
    }

    public static void main(String[] args) throws IOException {
        byte[] bs = "1hello world\r\nhehe".getBytes();
        System.out.println(ArrayUtils.toString(bs));
        ByteBuf buf = Unpooled.copiedBuffer(bs);
        String xString = buf.toString();
        ByteUtils.toLinesList(buf);

        bs = "2hello world hehe\r\n".getBytes();
        buf = Unpooled.copiedBuffer(bs);
        ByteUtils.toLinesList(buf);

        bs = "3hello world\r\nhehe\r\n\r\n".getBytes();
        buf = Unpooled.copiedBuffer(bs);
        ByteUtils.toLinesList(buf);

        bs = "4\rhe\nllo world\r\nhehe".getBytes();
        buf = Unpooled.copiedBuffer(bs);
        ByteUtils.toLinesList(buf);
    }

}