Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Reintroduce ComThread and move base functions into ObjectFactory #670

Merged
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Original file line number Diff line number Diff line change
Expand Up @@ -55,7 +55,7 @@ public class CallbackProxy implements IDispatchCallback {
private static float DEFAULT_FLOAT;
private static double DEFAULT_DOUBLE;

public CallbackProxy(Factory factory, Class<?> comEventCallbackInterface,
public CallbackProxy(ObjectFactory factory, Class<?> comEventCallbackInterface,
IComEventCallbackListener comEventCallbackListener) {
this.factory = factory;
this.comEventCallbackInterface = comEventCallbackInterface;
Expand All @@ -65,7 +65,7 @@ public CallbackProxy(Factory factory, Class<?> comEventCallbackInterface,
this.dispatchListener = new DispatchListener(this);
}

Factory factory;
ObjectFactory factory;
Class<?> comEventCallbackInterface;
IComEventCallbackListener comEventCallbackListener;
REFIID listenedToRiid;
Expand Down
Original file line number Diff line number Diff line change
@@ -0,0 +1,150 @@
/* Copyright (c) 2014 Dr David H. Akehurst (itemis), All Rights Reserved
*
* This library is free software; you can redistribute it and/or
* modify it under the terms of the GNU Lesser General Public
* License as published by the Free Software Foundation; either
* version 2.1 of the License, or (at your option) any later version.
*
* This library 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
* Lesser General Public License for more details.
*/
package com.sun.jna.platform.win32.COM.util;

import java.lang.Thread.UncaughtExceptionHandler;
import java.util.concurrent.Callable;
import java.util.concurrent.ExecutionException;
import java.util.concurrent.ExecutorService;
import java.util.concurrent.Executors;
import java.util.concurrent.ThreadFactory;
import java.util.concurrent.TimeUnit;
import java.util.concurrent.TimeoutException;

import com.sun.jna.platform.win32.Ole32;
import com.sun.jna.platform.win32.WinNT;
import com.sun.jna.platform.win32.COM.COMUtils;

public class ComThread {
private static ThreadLocal<Boolean> isCOMThread = new ThreadLocal<Boolean>();

ExecutorService executor;
Runnable firstTask;
boolean requiresInitialisation;
long timeoutMilliseconds;
UncaughtExceptionHandler uncaughtExceptionHandler;

public ComThread(final String threadName, long timeoutMilliseconds, UncaughtExceptionHandler uncaughtExceptionHandler) {
this(threadName, timeoutMilliseconds, uncaughtExceptionHandler, Ole32.COINIT_MULTITHREADED);
}

public ComThread(final String threadName, long timeoutMilliseconds, UncaughtExceptionHandler uncaughtExceptionHandler, final int coinitialiseExFlag) {
this.requiresInitialisation = true;
this.timeoutMilliseconds = timeoutMilliseconds;
this.uncaughtExceptionHandler = uncaughtExceptionHandler;
this.firstTask = new Runnable() {
@Override
public void run() {
try {
//If we do not use COINIT_MULTITHREADED, it is necessary to have
// a message loop see -
// [http://www.codeguru.com/cpp/com-tech/activex/apts/article.php/c5529/Understanding-COM-Apartments-Part-I.htm]
// [http://www.codeguru.com/cpp/com-tech/activex/apts/article.php/c5533/Understanding-COM-Apartments-Part-II.htm]
WinNT.HRESULT hr = Ole32.INSTANCE.CoInitializeEx(null, coinitialiseExFlag);
isCOMThread.set(true);
COMUtils.checkRC(hr);
ComThread.this.requiresInitialisation = false;
} catch (Throwable t) {
ComThread.this.uncaughtExceptionHandler.uncaughtException(Thread.currentThread(), t);
}
}
};
executor = Executors.newSingleThreadExecutor(new ThreadFactory() {

@Override
public Thread newThread(Runnable r) {
if (!ComThread.this.requiresInitialisation) {
// something has gone wrong!
throw new RuntimeException("ComThread executor has a problem.");
}
Thread thread = new Thread(r, threadName);
//make sure this is a daemon thread, or it will stop JVM existing
// if program does not call terminate();
thread.setDaemon(true);

thread.setUncaughtExceptionHandler(new UncaughtExceptionHandler() {
@Override
public void uncaughtException(Thread t, Throwable e) {
ComThread.this.requiresInitialisation = true;
ComThread.this.uncaughtExceptionHandler.uncaughtException(t, e);
}
});

return thread;
}
});

}

/**
* Stop the COM Thread.
*
* @param timeoutMilliseconds
* number of milliseconds to wait for a clean shutdown before a
* forced shutdown is attempted
*/
public void terminate(long timeoutMilliseconds) {
try {

executor.submit(new Runnable() {
@Override
public void run() {
Ole32.INSTANCE.CoUninitialize();
}
}).get(timeoutMilliseconds, TimeUnit.MILLISECONDS);

executor.shutdown();

} catch (InterruptedException e) {
e.printStackTrace();
} catch (ExecutionException e) {
e.printStackTrace();
} catch (TimeoutException e) {
executor.shutdownNow();
}
}

@Override
protected void finalize() throws Throwable {
if (!executor.isShutdown()) {
this.terminate(100);
}
}

static void setComThread(boolean value) {
isCOMThread.set(value);
}

public <T> T execute(Callable<T> task) throws TimeoutException, InterruptedException, ExecutionException {
// If the call is done on a COM thread, invoke directly
// if the call comes from outside the invokation is dispatched
// into the Dispatch Thread.
Boolean comThread = isCOMThread.get();
if(comThread == null) {
comThread = false;
}
if(comThread) {
try {
return task.call();
} catch (Exception ex) {
throw new ExecutionException(ex);
}
} else {
if (this.requiresInitialisation) {
executor.execute(firstTask);
}
return executor.submit(task).get(this.timeoutMilliseconds, TimeUnit.MILLISECONDS);
}
}

}
Original file line number Diff line number Diff line change
Expand Up @@ -133,7 +133,7 @@ public static VARIANT toVariant(Object value) {
}
}

public static Object toJavaObject(VARIANT value, Class<?> targetClass, Factory factory, boolean addReference, boolean freeValue) {
public static Object toJavaObject(VARIANT value, Class<?> targetClass, ObjectFactory factory, boolean addReference, boolean freeValue) {
if (null==value
|| value.getVarType().intValue() == VT_EMPTY
|| value.getVarType().intValue() == VT_NULL) {
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -13,9 +13,6 @@
package com.sun.jna.platform.win32.COM.util;

import java.util.Iterator;
import java.util.concurrent.Callable;
import java.util.concurrent.ExecutionException;
import java.util.concurrent.TimeoutException;

import com.sun.jna.platform.win32.WinDef;
import com.sun.jna.platform.win32.WinNT;
Expand All @@ -36,7 +33,7 @@
public class EnumMoniker implements Iterable<IDispatch> {

protected EnumMoniker(IEnumMoniker raw, com.sun.jna.platform.win32.COM.IRunningObjectTable rawRot,
Factory factory) {
ObjectFactory factory) {

assert COMUtils.comIsInitialized() : "COM not initialized";

Expand All @@ -50,7 +47,7 @@ protected EnumMoniker(IEnumMoniker raw, com.sun.jna.platform.win32.COM.IRunningO
this.cacheNext();
}

Factory factory;
ObjectFactory factory;
com.sun.jna.platform.win32.COM.IRunningObjectTable rawRot;
IEnumMoniker raw;
Moniker rawNext;
Expand Down
Loading