mongodb/db/javajs.h

// javajs.h

/**
*    Copyright (C) 2008 10gen Inc.
*
*    This program is free software: you can redistribute it and/or  modify
*    it under the terms of the GNU Affero General Public License, version 3,
*    as published by the Free Software Foundation.
*
*    This program 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 Affero General Public License for more details.
*
*    You should have received a copy of the GNU Affero General Public License
*    along with this program.  If not, see <http://www.gnu.org/licenses/>.
*/

/* this file contains code to call into java (into the 10gen sandbox) from inside the database */

#pragma once

#include "../stdafx.h"

#define J_USE_OBJ

#include <jni.h>
#include <boost/thread/tss.hpp>
#include <errno.h>
#include <sys/types.h>

#if !defined(_WIN32)
#include <dirent.h>
#endif

#include "jsobj.h"

namespace mongo {

    void jasserted(const char *msg, const char *file, unsigned line);
#define jassert(_Expression) if ( ! ( _Expression ) ){ jasserted(#_Expression, __FILE__, __LINE__); }

    int javajstest();

    const char * findEd();
    const char * findEd(const char *);
    const string findJars();

    class BSONObj;

    class JavaJSImpl {
    public:
        JavaJSImpl(const char * = 0);
        ~JavaJSImpl();

        jlong scopeCreate();
        int scopeInit( jlong id , BSONObj * obj );
        int scopeSetThis( jlong id , BSONObj * obj );
        jboolean scopeReset( jlong id );
        void scopeFree( jlong id );

        double scopeGetNumber( jlong id , const char * field );
        string scopeGetString( jlong id , const char * field );
        jboolean scopeGetBoolean( jlong id , const char * field );
        BSONObj scopeGetObject( jlong id , const char * field );
        char scopeGetType( jlong id , const char * field );

        int scopeSetNumber( jlong id , const char * field , double val );
        int scopeSetString( jlong id , const char * field , const char * val );
        int scopeSetObject( jlong id , const char * field , BSONObj * obj );
        int scopeSetBoolean( jlong id , const char * field , jboolean val );

        jlong functionCreate( const char * code );

        /* return values:
           public static final int NO_SCOPE = -1;
           public static final int NO_FUNCTION = -2;
           public static final int INVOKE_ERROR = -3;
           public static final int INVOKE_SUCCESS = 0;
          */
        int invoke( jlong scope , jlong function );

        void printException();

        void run( const char * js );

        void detach( JNIEnv * env ) {
            _jvm->DetachCurrentThread();
        }

    private:

        jobject create( const char * name ) {
            jclass c = findClass( name );
            if ( ! c )
                return 0;

            jmethodID cons = _getEnv()->GetMethodID( c , "<init>" , "()V" );
            if ( ! cons )
                return 0;

            return _getEnv()->NewObject( c , cons );
        }

        jclass findClass( const char * name ) {
            return _getEnv()->FindClass( name );
        }


    private:

        JNIEnv * _getEnv();

        JavaVM * _jvm;
        JNIEnv * _mainEnv;
        JavaVMInitArgs * _vmArgs;

        boost::thread_specific_ptr<JNIEnv> * _envs;

        jclass _dbhook;
        jclass _dbjni;

        jmethodID _scopeCreate;
        jmethodID _scopeInit;
        jmethodID _scopeSetThis;
        jmethodID _scopeReset;
        jmethodID _scopeFree;

        jmethodID _scopeGetNumber;
        jmethodID _scopeGetString;
        jmethodID _scopeGetObject;
        jmethodID _scopeGetBoolean;
        jmethodID _scopeGuessObjectSize;
        jmethodID _scopeGetType;

        jmethodID _scopeSetNumber;
        jmethodID _scopeSetString;
        jmethodID _scopeSetObject;
        jmethodID _scopeSetBoolean;

        jmethodID _functionCreate;

        jmethodID _invoke;

    };

    extern JavaJSImpl *JavaJS;

// a javascript "scope"
    class Scope {
    public:
        Scope() {
            s = JavaJS->scopeCreate();
        }
        ~Scope() {
            JavaJS->scopeFree(s);
            s = 0;
        }
        void reset() {
            JavaJS->scopeReset(s);
        }

        void init( const char * data ) {
            BSONObj o( data , 0 );
            JavaJS->scopeInit( s , & o );
        }

        double getNumber(const char *field) {
            return JavaJS->scopeGetNumber(s,field);
        }
        string getString(const char *field) {
            return JavaJS->scopeGetString(s,field);
        }
        jboolean getBoolean(const char *field) {
            return JavaJS->scopeGetBoolean(s,field);
        }
        BSONObj getObject(const char *field ) {
            return JavaJS->scopeGetObject(s,field);
        }
        int type(const char *field ) {
            return JavaJS->scopeGetType(s,field);
        }

        void setNumber(const char *field, double val ) {
            JavaJS->scopeSetNumber(s,field,val);
        }
        void setString(const char *field, const char * val ) {
            JavaJS->scopeSetString(s,field,val);
        }
        void setObject(const char *field, BSONObj& obj ) {
            JavaJS->scopeSetObject(s,field,&obj);
        }
        void setBoolean(const char *field, jboolean val ) {
            JavaJS->scopeSetBoolean(s,field,val);
        }

        int invoke(jlong function) {
            return JavaJS->invoke(s,function);
        }

        jlong s;
    };

    JNIEXPORT void JNICALL java_native_say(JNIEnv *, jclass, jobject outBuffer );
    JNIEXPORT jint JNICALL java_native_call(JNIEnv *, jclass, jobject outBuffer , jobject inBuffer );

} // namespace mongo