com.xxdb.streaming.client.MessageParser Maven / Gradle / Ivy

Show more of this group Show more artifacts with this name
Show all versions of api-java Show documentation
The messaging and data conversion protocol between Java and DolphinDB server
There is a newer version: 1.0.27
package com.xxdb.streaming.client;

import java.io.BufferedInputStream;
import java.io.IOException;
import java.net.Socket;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;

import com.xxdb.data.*;
import com.xxdb.io.BigEndianDataInputStream;
import com.xxdb.io.ExtendedDataInput;
import com.xxdb.io.LittleEndianDataInputStream;

class MessageParser implements Runnable{
	private final int MAX_FORM_VALUE = Entity.DATA_FORM.values().length -1;
	private final int MAX_TYPE_VALUE = Entity.DATA_TYPE.values().length -1;

	BufferedInputStream bis = null;
	Socket socket = null;
	MessageDispatcher dispatcher;
	String topic;
	HashMap nameToIndex = null;

	public MessageParser(Socket socket, MessageDispatcher dispatcher){
		this.socket = socket;
		this.dispatcher = dispatcher;
	}
	private final static char[] hexArray = "0123456789ABCDEF".toCharArray();
	public static String bytesToHex(byte[] bytes) {
		char[] hexChars = new char[bytes.length * 2];
		for ( int j = 0; j < bytes.length; j++ ) {
			int v = bytes[j] & 0xFF;
			hexChars[j * 2] = hexArray[v >>> 4];
			hexChars[j * 2 + 1] = hexArray[v & 0x0F];
		}
		return new String(hexChars);
	}
	public void run(){
		Socket socket = this.socket;
		try {
		if(bis == null) bis= new BufferedInputStream(socket.getInputStream());
		long offset = -1;
		ExtendedDataInput in = null;

		while(true){
			if (in == null) {
				Boolean isLittle = bis.read() != 0;
				if (isLittle == true)
					in = new LittleEndianDataInputStream(bis);
				else
					in = new BigEndianDataInputStream(bis);
			} 
			else {
				in.readBoolean();
			}
			
			in.readLong();
			long msgid = in.readLong();
			if (offset == -1) {
				offset = msgid;
			} else {
				assert(offset == msgid);
			}
			topic = in.readString();
			short flag = in.readShort();
			EntityFactory factory = new BasicEntityFactory();
			int form = flag>>8;

			int type = flag & 0xff;

			if(form < 0 || form > MAX_FORM_VALUE)
				throw new IOException("Invalid form value: " + form);
			if(type <0 || type > MAX_TYPE_VALUE){
				throw new IOException("Invalid type value: " + type);
				
			}
			Entity.DATA_FORM df = Entity.DATA_FORM.values()[form];
			Entity.DATA_TYPE dt = Entity.DATA_TYPE.values()[type];
			Entity body;
			try
			{
				body =  factory.createEntity(df, dt, in);
			}
			catch(Exception exception) {
				throw exception;
			}
			if (body.isTable()) {
				assert(body.rows() == 0);
				nameToIndex = new HashMap<>();
				BasicTable schema = (BasicTable) body;
				int columns = schema.columns();
				for (int i = 0; i < columns; i++) {
					String name = schema.getColumnName(i);
					nameToIndex.put(name.toLowerCase(), i);
				}
			}
			else if (body.isVector()){
				BasicAnyVector dTable = (BasicAnyVector)body;
				
				int colSize = dTable.rows();
				int rowSize = dTable.getEntity(0).rows();
				if(rowSize>=1){
					if(rowSize==1){
						BasicMessage rec = new BasicMessage(msgid,topic,dTable,nameToIndex);
						dispatcher.dispatch(rec);
					} else {
						List messages = new ArrayList<>(rowSize);
						for(int i=0;i