rmscmd.java

来自「Hecl编程语言是一个高层次的脚本语言的Java实现。其用意是要小」· Java 代码 · 共 525 行

/* Copyright 2005-2006 David N. Welton <[e-mail]>
 * Wolfgang S. Kechel - [e-mail]

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 org.hecl.rms;

import java.io.ByteArrayInputStream;
import java.io.ByteArrayOutputStream;
import java.io.DataInputStream;
import java.io.DataOutputStream;
import java.io.IOException;
import java.util.Hashtable;
import java.util.Vector;

import org.hecl.HeclException;
import org.hecl.Interp;
import org.hecl.IntThing;
import org.hecl.ListThing;
import org.hecl.Operator;
import org.hecl.RealThing;
import org.hecl.StringThing;
import org.hecl.Thing;

import javax.microedition.rms.RecordStore;
import javax.microedition.rms.RecordEnumeration;


/**
 * The <code>RMSCmd</code> class implements the record store related commands.
 *
 * @author <a href="mailto:[e-mail]">David N. Welton</a>
 * @author <a href="mailto:[e-mail]">Wolfgang S. Kechel</a>
 * @version 1.1
 */
public class RMSCmd extends Operator {
    public static final int RMS_LIST = 1;
    public static final int RMS_CREATE = 2;
    public static final int RMS_GET = 3;
    public static final int RMS_SET = 4;
    public static final int RMS_SIZE = 5;
    public static final int RMS_SIZEAVAIL = 6;
    public static final int RMS_DELETE = 7;
    public static final int RMS_ADD = 8;
    public static final int RMS_HSET = 9;
    public static final int RMS_HGET = 10;
    public static final int RMS_HEXISTS = 11;
    public static final int RMS_HKEYS = 12;
    public static final int RMS_HDEL = 13;
//#if midp >= 2.0
    public static final int RMS_SETMODE = 14;
//#endif

    protected RMSCmd(int cmdcode,int minargs,int maxargs) {
	super(cmdcode,minargs,maxargs);
    }

    protected static void closeRS(RecordStore rs) throws HeclException {
	if(rs != null) {
	    try {
		rs.closeRecordStore();
	    }
	    catch(Exception e) {
		throw new HeclException("Can't close recordstore.");
	    }
	}
    }
    
    public Thing operate(int cmd, Interp interp, Thing[] argv) throws HeclException {
	RecordStore rs = null;
	byte[] data = null;
	String name = argv.length>1?argv[1].toString() : null;
	int recordid = 1;
	
	switch (cmd) {
	  case RMS_LIST:
	    ;
	    {
		Vector v = new Vector();
		if(name == null) {
		    String[] names = RecordStore.listRecordStores();
		    if (names != null) {
			for (int i = 0; i < names.length; i++) {
			    v.addElement(new Thing(names[i]));
			}
		    }
		} else {
		    int[] res = getRecordIds(name);
		    for(int i=0; i<res.length; ++i)
			v.addElement(IntThing.create(res[i]));
		}
		return ListThing.create(v);
	    }
		
	  case RMS_CREATE:
	    ;
	    {
//#if midp < 2.0

		// name already extracted

//#else
		// [options] name
		name = null;
		int authmode = RecordStore.AUTHMODE_PRIVATE;
		boolean writeflag = false;
		boolean argend = false;
		
		for(int i=1; i<argv.length; ++i) {
		    String s = argv[1].toString();
		    if(!argend && s.equals("--")) {
			argend = true;
			continue;
		    }
		    if(!argend && s.startsWith("-")) {
			// option
			if(s.equals("-any"))
			    authmode = RecordStore.AUTHMODE_ANY;
			if(s.equals("-private"))
			    authmode = RecordStore.AUTHMODE_PRIVATE;
			else if(s.equals("-writable"))
			    writeflag = true;
			else
			    throw new HeclException("unknown option '"+s+"'");
		    } else {
			name = s;
		    }
		}
		if(name == null)
		    throw new HeclException("name required");
//#endif
		try {
		    rs = RecordStore.openRecordStore(name, true
						     //#ifdef MIDP20
						     , authmode, writeflag
						     //#endif
			);
		}
		catch(Exception e) {
		    /* FIXME - we ought to do something a little bit more clever here. */
		    throw new HeclException(e.toString());
		}
		finally {
		    closeRS(rs);
		}
		return new Thing(name);
	    }
	    
	  case RMS_GET:
	    // rs_get name [recordid]
	    if(argv.length>2) {
		recordid = IntThing.get(argv[2]);
	    }
	    try {
		rs =  RecordStore.openRecordStore(name, false);
		data = rs.getRecord(recordid);
	    } catch (Exception e) {
		/* FIXME - we ought to do something a little bit more clever here. */
		throw new HeclException(e.toString());
	    }
	    finally {
		closeRS(rs);
	    }
	    return new Thing(new String(data));
	    
	  case RMS_SET:
	    // rs_put name [recordid=1] data
	    //System.err.println("argv.length="+argv.length);
	    
	    if(argv.length>3) {
		recordid = IntThing.get(argv[2]);
		data = argv[3].toString().getBytes();
	    } else {
		data = argv[2].toString().getBytes();
	    }
	    
	    try {
		rs = RecordStore.openRecordStore(name, true);
		//rs.addRecord(data, 0, data.length);
		rs.setRecord(recordid,data, 0, data.length);
	    } catch (Exception e) {
		throw new HeclException(e.toString());
	    }
	    finally {
		closeRS(rs);
	    }
	    return IntThing.create(data.length);
	    
	  case RMS_ADD:
	    // add data
	    try {
		rs = RecordStore.openRecordStore(name, false);
		data = argv[2].toString().getBytes();
		recordid = rs.addRecord(data,0,data.length);
	    }
	    catch(Exception e) {
		throw new HeclException("Can't add record: " + e.getMessage());
	    }
	    finally {
		closeRS(rs);
	    }
	    return IntThing.create(recordid);
	    
	  case RMS_SIZE:
	  case RMS_SIZEAVAIL:
	    // size recordid
	    // we reuse the recordid var!
	    try {
		rs =  RecordStore.openRecordStore(name, true);
		recordid = cmd == RMS_SIZE ? rs.getSize() : rs.getSizeAvailable();
	    } catch (Exception e) {
		throw new HeclException(e.toString());
	    }
	    finally {
		closeRS(rs);
	    }
	    return IntThing.create(recordid);

	  case RMS_DELETE:
	    recordid = -1;
	    if(argv.length > 2) {
		recordid = IntThing.get(argv[2]);
	    }
	    if(recordid < 0) {
		try {
		    RecordStore.deleteRecordStore(name);
		}
		catch(Exception e) {
		    throw new HeclException("Can't delete recordstore '"
					    +name+"': " +e.getMessage());
		}
		break;
	    }
	    try {
		rs = RecordStore.openRecordStore(name, false);
		rs.deleteRecord(recordid);
	    }
	    catch(Exception e) {
		throw new HeclException("Can't delete recordstore '"
					+name+"["+recordid+"]': " +e.getMessage());
	    }
	    finally {
		closeRS(rs);
	    }
	    break;

	  case RMS_HSET:
	    // rs_hset name key value
	    data = argv[3].toString().getBytes();
	    try {
		rs = RecordStore.openRecordStore(name, true);
		//rs.addRecord(data, 0, data.length);
		String key = argv[2].toString();
		Object[] v = recordOf(rs,key);
		recordid = ((Integer)v[0]).intValue();
		data = toData(key,argv[3].toString());
		if(recordid < 0)
		    rs.addRecord(data, 0, data.length);
		else
		    rs.setRecord(recordid,data, 0, data.length);
		recordid = data.length;
		data = null;
	    } catch (Exception e) {
		throw new HeclException(e.toString());
	    }
	    finally {
		closeRS(rs);
	    }
	    return IntThing.create(recordid);
	    
	  case RMS_HGET:
	    // hget name key
	    ;				    // trick emacs indentation
	    {
		Object[] v = null;
		
		try {
		    rs = RecordStore.openRecordStore(name, true);
		    v = recordOf(rs,argv[2].toString());
		    recordid = ((Integer)v[0]).intValue();
		} catch (Exception e) {
		    throw new HeclException(e.toString());
		}
		finally {
		    closeRS(rs);
		}
		return new Thing(recordid >= 0 && v[1] != null ? (String)v[1] : "");
	    }
	    
	  case RMS_HEXISTS:
	    // hget name key
	    try {	
		rs = RecordStore.openRecordStore(name, true);
		Object[] v = recordOf(rs,argv[2].toString());
		recordid = ((Integer)v[0]).intValue();
	    } catch (Exception e) {
		throw new HeclException(e.toString());
	    }
	    finally {
		closeRS(rs);
	    }
	    return IntThing.create(recordid >= 0 ? 1 : 0);

	  case RMS_HKEYS:
	    ;
	    {
		Vector v = null;
		
		try {
		    rs = RecordStore.openRecordStore(name, true);
		    v = hkeys(rs);
		} catch (Exception e) {
		    throw new HeclException(e.toString());
		}
		finally {
		    closeRS(rs);
		}
		if(v == null)
		    v = new Vector();
		
		int n = v.size();
		for(int i=0; i<n; ++i)
		    v.setElementAt(new Thing((String)v.elementAt(i)),i);
		return ListThing.create(v);
	    }

	  case RMS_HDEL:
	    try {
		rs = RecordStore.openRecordStore(name, true);
		Object[] v = recordOf(rs,argv[2].toString());
		recordid = ((Integer)v[0]).intValue();
		if(recordid >= 0)
		    rs.deleteRecord(recordid);
	    } catch (Exception e) {
		throw new HeclException(e.toString());
	    }
	    finally {
		closeRS(rs);
	    }
	    break;

//#if midp >= 2.0
	  case RMS_SETMODE:
	    try {
		int authmode = "any".equals(argv[2].toString()) ? 
		    RecordStore.AUTHMODE_ANY : RecordStore.AUTHMODE_PRIVATE;
		boolean writeflag = Thing.isTrue(argv[3]);
		rs = RecordStore.openRecordStore(name, true);
		rs.setMode(authmode,writeflag);
	    } catch (Exception e) {
		throw new HeclException(e.toString());
	    }
	    finally {
		closeRS(rs);
	    }
	    break;
//#endif

	  default:
	    throw new HeclException("Unknown rms command '"
				    + argv[0].toString() + "' with code '"
				    + cmd + "'.");
	}
	return null;
    }


    public static int[] getRecordIds(String name) throws HeclException {
	int count = 0;
	int[] res = new int[0];
	RecordStore rs = null;
	try {
	    rs = RecordStore.openRecordStore(name,false);
	    synchronized(rs) {
		RecordEnumeration records = rs.enumerateRecords(null,null,false);
		while(records.hasNextElement()) {
		    records.nextRecordId();
		    ++count;
		}
		records.reset();
		res = new int[count];
		int i = 0;
		while(records.hasNextElement()) {
		    res[i++] = records.nextRecordId();
		}
		records.destroy();
	    }
	} catch (Exception e) {
	    /* FIXME - we ought to do something a little bit more clever here. */
	    throw new HeclException(e.toString());
	}
	finally {
	    closeRS(rs);
	}
	return res;
    }
    

    private static byte[] toData(String key,String value) throws Exception {
	ByteArrayOutputStream bos = null;
	DataOutputStream dos = null;
	byte[] data = null;
	
	try {
	    bos = new ByteArrayOutputStream();
	    dos = new DataOutputStream(bos);
	    dos.writeUTF(key);
	    dos.writeUTF(value);
	    data = bos.toByteArray();
	    dos.close();
	    bos.close();
	}
	catch(Exception e) {
	    throw new HeclException("Can't create data: "+e.getMessage());
	}
	return data;
    }
    
	    
    /*
      Convert all rms records into a hashtable
    private static Hashtable asTable(RecordStore rs) throws Exception {
	Hashtable t = new Hashtable();
	
	RecordEnumeration enumeration = rs.enumerateRecords(null, null, false);
	for (int i = 0; i < enumeration.numRecords(); i++) {
	    int tmpid = enumeration.nextRecordId();
	    byte[] data = rs.getRecord(tmpid);
	    
	    ByteArrayInputStream bis = new ByteArrayInputStream(data);
	    DataInputStream dis = new DataInputStream(bis);
	    
	    t.put(dis.readUTF(),dis.readUTF());
	    dis.close();
	    bis.close();
	}
	enumeration.destroy();
	return t;
    }
    */

    private static Vector hkeys(RecordStore rs) throws Exception {
	Vector v = new Vector();
	
	RecordEnumeration enumeration = rs.enumerateRecords(null, null, false);
	for (int i = 0; i < enumeration.numRecords(); i++) {
	    int tmpid = enumeration.nextRecordId();
	    byte[] data = rs.getRecord(tmpid);
	    
	    ByteArrayInputStream bis = new ByteArrayInputStream(data);
	    DataInputStream dis = new DataInputStream(bis);
	    
	    v.addElement(dis.readUTF());
	    dis.close();
	    bis.close();
	}
	return v;
    }

    public static Object[] recordOf(RecordStore rs,String key)
	throws Exception {
	String value = null;
	int id = -1;
	RecordEnumeration enumeration = rs.enumerateRecords(null, null, false);
	for (int i = 0; i < enumeration.numRecords(); i++) {
	    int tmpid = enumeration.nextRecordId();
	    byte[] data = rs.getRecord(tmpid);
	    
	    ByteArrayInputStream bis = new ByteArrayInputStream(data);
	    DataInputStream dis = new DataInputStream(bis);
	    
	    String tmpkey = dis.readUTF();
	    bis.close();
	    dis.close();
	    if (tmpkey.equals(key)) {
		id = tmpid;
		value = dis.readUTF();
		break;
	    }
	}
	enumeration.destroy();
	return new Object[]{new Integer(id),value};
    }
    
    public static void load(Interp ip) throws HeclException {
	Operator.load(ip,cmdtable);
    }

    public static void unload(Interp ip) throws HeclException {
	Operator.unload(ip,cmdtable);
    }

    private static Hashtable cmdtable = new Hashtable();

    static {
        cmdtable.put("rms.list", new RMSCmd(RMS_LIST,0,1));
        cmdtable.put("rms.create", new RMSCmd(RMS_CREATE,1,0));
        cmdtable.put("rms.get", new RMSCmd(RMS_GET,1,2));
        cmdtable.put("rms.set", new RMSCmd(RMS_SET,2,3));
        cmdtable.put("rms.size", new RMSCmd(RMS_SIZE,1,1));
        cmdtable.put("rms.sizeavail", new RMSCmd(RMS_SIZEAVAIL,1,1));
        cmdtable.put("rms.delete", new RMSCmd(RMS_DELETE,1,2));
        cmdtable.put("rms.add", new RMSCmd(RMS_ADD,2,2));
        cmdtable.put("rms.hset", new RMSCmd(RMS_HSET,3,3));
        cmdtable.put("rms.hget", new RMSCmd(RMS_HGET,2,2));
        cmdtable.put("rms.hexists", new RMSCmd(RMS_HEXISTS,2,2));
        cmdtable.put("rms.hkeys", new RMSCmd(RMS_HKEYS,1,1));
        cmdtable.put("rms.hdel", new RMSCmd(RMS_HDEL,2,2));
//#if midp >= 2.0
        cmdtable.put("rms.setmode", new RMSCmd(RMS_SETMODE,3,3));
//#endif
    }
}