core/java/android/webkit/WebStorageClassic.java - platform/frameworks/base - Gitiles

 /*
  * Copyright (C) 2012 The Android Open Source Project
  *
  * 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 android.webkit;

 import android.os.Handler;
 import android.os.Message;

 import java.util.Collection;
 import java.util.HashMap;
 import java.util.Map;
 import java.util.Set;

 /** @hide */
 public class WebStorageClassic extends WebStorage {
     // Global instance of a WebStorage
     private static WebStorageClassic sWebStorage;

     // Message ids
     static final int UPDATE = 0;
     static final int SET_QUOTA_ORIGIN = 1;
     static final int DELETE_ORIGIN = 2;
     static final int DELETE_ALL = 3;
     static final int GET_ORIGINS = 4;
     static final int GET_USAGE_ORIGIN = 5;
     static final int GET_QUOTA_ORIGIN = 6;

     // Message ids on the UI thread
     static final int RETURN_ORIGINS = 0;
     static final int RETURN_USAGE_ORIGIN = 1;
     static final int RETURN_QUOTA_ORIGIN = 2;

     private static final String ORIGINS = "origins";
     private static final String ORIGIN = "origin";
     private static final String CALLBACK = "callback";
     private static final String USAGE = "usage";
     private static final String QUOTA = "quota";

     private Map <String, Origin> mOrigins;

     private Handler mHandler = null;
     private Handler mUIHandler = null;

     /**
      * @hide
      * Message handler, UI side
      * @hide
      */
     public void createUIHandler() {
         if (mUIHandler == null) {
             mUIHandler = new Handler() {
                 @Override
                 public void handleMessage(Message msg) {
                     switch (msg.what) {
                         case RETURN_ORIGINS: {
                             Map values = (Map) msg.obj;
                             Map origins = (Map) values.get(ORIGINS);
                             ValueCallback<Map> callback = (ValueCallback<Map>) values.get(CALLBACK);
                             callback.onReceiveValue(origins);
                             } break;

                         case RETURN_USAGE_ORIGIN: {
                             Map values = (Map) msg.obj;
                             ValueCallback<Long> callback = (ValueCallback<Long>) values.get(CALLBACK);
                             callback.onReceiveValue((Long)values.get(USAGE));
                             } break;

                         case RETURN_QUOTA_ORIGIN: {
                             Map values = (Map) msg.obj;
                             ValueCallback<Long> callback = (ValueCallback<Long>) values.get(CALLBACK);
                             callback.onReceiveValue((Long)values.get(QUOTA));
                             } break;
                     }
                 }
             };
         }
     }

     /**
      * Message handler, WebCore side
      * @hide
      */
     public synchronized void createHandler() {
         if (mHandler == null) {
             mHandler = new Handler() {
                 @Override
                 public void handleMessage(Message msg) {
                     switch (msg.what) {
                         case SET_QUOTA_ORIGIN: {
                             Origin website = (Origin) msg.obj;
                             nativeSetQuotaForOrigin(website.getOrigin(),
                                                     website.getQuota());
                             } break;

                         case DELETE_ORIGIN: {
                             Origin website = (Origin) msg.obj;
                             nativeDeleteOrigin(website.getOrigin());
                             } break;

                         case DELETE_ALL:
                             nativeDeleteAllData();
                             break;

                         case GET_ORIGINS: {
                             syncValues();
                             ValueCallback callback = (ValueCallback) msg.obj;
                             Map origins = new HashMap(mOrigins);
                             Map values = new HashMap<String, Object>();
                             values.put(CALLBACK, callback);
                             values.put(ORIGINS, origins);
                             postUIMessage(Message.obtain(null, RETURN_ORIGINS, values));
                             } break;

                         case GET_USAGE_ORIGIN: {
                             syncValues();
                             Map values = (Map) msg.obj;
                             String origin = (String) values.get(ORIGIN);
                             ValueCallback callback = (ValueCallback) values.get(CALLBACK);
                             Origin website = mOrigins.get(origin);
                             Map retValues = new HashMap<String, Object>();
                             retValues.put(CALLBACK, callback);
                             if (website != null) {
                                 long usage = website.getUsage();
                                 retValues.put(USAGE, new Long(usage));
                             }
                             postUIMessage(Message.obtain(null, RETURN_USAGE_ORIGIN, retValues));
                             } break;

                         case GET_QUOTA_ORIGIN: {
                             syncValues();
                             Map values = (Map) msg.obj;
                             String origin = (String) values.get(ORIGIN);
                             ValueCallback callback = (ValueCallback) values.get(CALLBACK);
                             Origin website = mOrigins.get(origin);
                             Map retValues = new HashMap<String, Object>();
                             retValues.put(CALLBACK, callback);
                             if (website != null) {
                                 long quota = website.getQuota();
                                 retValues.put(QUOTA, new Long(quota));
                             }
                             postUIMessage(Message.obtain(null, RETURN_QUOTA_ORIGIN, retValues));
                             } break;

                         case UPDATE:
                             syncValues();
                             break;
                     }
                 }
             };
         }
     }

     /*
      * When calling getOrigins(), getUsageForOrigin() and getQuotaForOrigin(),
      * we need to get the values from WebCore, but we cannot block while doing so
      * as we used to do, as this could result in a full deadlock (other WebCore
      * messages received while we are still blocked here, see http://b/2127737).
      *
      * We have to do everything asynchronously, by providing a callback function.
      * We post a message on the WebCore thread (mHandler) that will get the result
      * from WebCore, and we post it back on the UI thread (using mUIHandler).
      * We can then use the callback function to return the value.
      */

     @Override
     public void getOrigins(ValueCallback<Map> callback) {
         if (callback != null) {
             if (WebViewCore.THREAD_NAME.equals(Thread.currentThread().getName())) {
                 syncValues();
                 callback.onReceiveValue(mOrigins);
             } else {
                 postMessage(Message.obtain(null, GET_ORIGINS, callback));
             }
         }
     }

     /**
      * Returns a list of origins having a database
      * should only be called from WebViewCore.
      */
     Collection<Origin> getOriginsSync() {
         if (WebViewCore.THREAD_NAME.equals(Thread.currentThread().getName())) {
             update();
             return mOrigins.values();
         }
         return null;
     }

     @Override
     public void getUsageForOrigin(String origin, ValueCallback<Long> callback) {
         if (callback == null) {
             return;
         }
         if (origin == null) {
             callback.onReceiveValue(null);
             return;
         }
         if (WebViewCore.THREAD_NAME.equals(Thread.currentThread().getName())) {
             syncValues();
             Origin website = mOrigins.get(origin);
             callback.onReceiveValue(new Long(website.getUsage()));
         } else {
             HashMap values = new HashMap<String, Object>();
             values.put(ORIGIN, origin);
             values.put(CALLBACK, callback);
             postMessage(Message.obtain(null, GET_USAGE_ORIGIN, values));
         }
     }

     @Override
     public void getQuotaForOrigin(String origin, ValueCallback<Long> callback) {
         if (callback == null) {
             return;
         }
         if (origin == null) {
             callback.onReceiveValue(null);
             return;
         }
         if (WebViewCore.THREAD_NAME.equals(Thread.currentThread().getName())) {
             syncValues();
             Origin website = mOrigins.get(origin);
             callback.onReceiveValue(new Long(website.getUsage()));
         } else {
             HashMap values = new HashMap<String, Object>();
             values.put(ORIGIN, origin);
             values.put(CALLBACK, callback);
             postMessage(Message.obtain(null, GET_QUOTA_ORIGIN, values));
         }
     }

     @Override
     public void setQuotaForOrigin(String origin, long quota) {
         if (origin != null) {
             if (WebViewCore.THREAD_NAME.equals(Thread.currentThread().getName())) {
                 nativeSetQuotaForOrigin(origin, quota);
             } else {
                 postMessage(Message.obtain(null, SET_QUOTA_ORIGIN,
                     new Origin(origin, quota)));
             }
         }
     }

     @Override
     public void deleteOrigin(String origin) {
         if (origin != null) {
             if (WebViewCore.THREAD_NAME.equals(Thread.currentThread().getName())) {
                 nativeDeleteOrigin(origin);
             } else {
                 postMessage(Message.obtain(null, DELETE_ORIGIN,
                     new Origin(origin)));
             }
         }
     }

     @Override
     public void deleteAllData() {
         if (WebViewCore.THREAD_NAME.equals(Thread.currentThread().getName())) {
             nativeDeleteAllData();
         } else {
             postMessage(Message.obtain(null, DELETE_ALL));
         }
     }

     /**
      * Sets the maximum size of the ApplicationCache.
      * This should only ever be called on the WebKit thread.
      * Not part of the base-class API: this is only used by dump render tree.
      */
     public void setAppCacheMaximumSize(long size) {
         nativeSetAppCacheMaximumSize(size);
     }

     /**
      * Utility function to send a message to our handler
      */
     private synchronized void postMessage(Message msg) {
         if (mHandler != null) {
             mHandler.sendMessage(msg);
         }
     }

     /**
      * Utility function to send a message to the handler on the UI thread
      */
     private void postUIMessage(Message msg) {
         if (mUIHandler != null) {
             mUIHandler.sendMessage(msg);
         }
     }

     /**
      * Get the singleton instance of this class.
      * @return The singleton {@link WebStorage} instance.
      */
     public static WebStorageClassic getInstance() {
       if (sWebStorage == null) {
           sWebStorage = new WebStorageClassic();
       }
       return sWebStorage;
     }

     /**
      * @hide
      * Post a Sync request
      */
     public void update() {
         if (WebViewCore.THREAD_NAME.equals(Thread.currentThread().getName())) {
             syncValues();
         } else {
             postMessage(Message.obtain(null, UPDATE));
         }
     }

     /**
      * Run on the WebCore thread
      * set the local values with the current ones
      */
     private void syncValues() {
         Set<String> tmp = nativeGetOrigins();
         mOrigins = new HashMap<String, Origin>();
         for (String origin : tmp) {
             Origin website = new Origin(origin,
                                  nativeGetQuotaForOrigin(origin),
                                  nativeGetUsageForOrigin(origin));
             mOrigins.put(origin, website);
         }
     }

     WebStorageClassic() {}

     // Native functions
     private static native Set nativeGetOrigins();
     private static native long nativeGetUsageForOrigin(String origin);
     private static native long nativeGetQuotaForOrigin(String origin);
     private static native void nativeSetQuotaForOrigin(String origin, long quota);
     private static native void nativeDeleteOrigin(String origin);
     private static native void nativeDeleteAllData();
     private static native void nativeSetAppCacheMaximumSize(long size);
 }
	/*
	* Copyright (C) 2012 The Android Open Source Project
	*
	* 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 android.webkit;

	import android.os.Handler;
	import android.os.Message;

	import java.util.Collection;
	import java.util.HashMap;
	import java.util.Map;
	import java.util.Set;

	/** @hide */
	public class WebStorageClassic extends WebStorage {
	// Global instance of a WebStorage
	private static WebStorageClassic sWebStorage;

	// Message ids
	static final int UPDATE = 0;
	static final int SET_QUOTA_ORIGIN = 1;
	static final int DELETE_ORIGIN = 2;
	static final int DELETE_ALL = 3;
	static final int GET_ORIGINS = 4;
	static final int GET_USAGE_ORIGIN = 5;
	static final int GET_QUOTA_ORIGIN = 6;

	// Message ids on the UI thread
	static final int RETURN_ORIGINS = 0;
	static final int RETURN_USAGE_ORIGIN = 1;
	static final int RETURN_QUOTA_ORIGIN = 2;

	private static final String ORIGINS = "origins";
	private static final String ORIGIN = "origin";
	private static final String CALLBACK = "callback";
	private static final String USAGE = "usage";
	private static final String QUOTA = "quota";

	private Map <String, Origin> mOrigins;

	private Handler mHandler = null;
	private Handler mUIHandler = null;

	/**
	* @hide
	* Message handler, UI side
	* @hide
	*/
	public void createUIHandler() {
	if (mUIHandler == null) {
	mUIHandler = new Handler() {
	@Override
	public void handleMessage(Message msg) {
	switch (msg.what) {
	case RETURN_ORIGINS: {
	Map values = (Map) msg.obj;
	Map origins = (Map) values.get(ORIGINS);
	ValueCallback<Map> callback = (ValueCallback<Map>) values.get(CALLBACK);
	callback.onReceiveValue(origins);
	} break;

	case RETURN_USAGE_ORIGIN: {
	Map values = (Map) msg.obj;
	ValueCallback<Long> callback = (ValueCallback<Long>) values.get(CALLBACK);
	callback.onReceiveValue((Long)values.get(USAGE));
	} break;

	case RETURN_QUOTA_ORIGIN: {
	Map values = (Map) msg.obj;
	ValueCallback<Long> callback = (ValueCallback<Long>) values.get(CALLBACK);
	callback.onReceiveValue((Long)values.get(QUOTA));
	} break;
	}
	}
	};
	}
	}

	/**
	* Message handler, WebCore side
	* @hide
	*/
	public synchronized void createHandler() {
	if (mHandler == null) {
	mHandler = new Handler() {
	@Override
	public void handleMessage(Message msg) {
	switch (msg.what) {
	case SET_QUOTA_ORIGIN: {
	Origin website = (Origin) msg.obj;
	nativeSetQuotaForOrigin(website.getOrigin(),
	website.getQuota());
	} break;

	case DELETE_ORIGIN: {
	Origin website = (Origin) msg.obj;
	nativeDeleteOrigin(website.getOrigin());
	} break;

	case DELETE_ALL:
	nativeDeleteAllData();
	break;

	case GET_ORIGINS: {
	syncValues();
	ValueCallback callback = (ValueCallback) msg.obj;
	Map origins = new HashMap(mOrigins);
	Map values = new HashMap<String, Object>();
	values.put(CALLBACK, callback);
	values.put(ORIGINS, origins);
	postUIMessage(Message.obtain(null, RETURN_ORIGINS, values));
	} break;

	case GET_USAGE_ORIGIN: {
	syncValues();
	Map values = (Map) msg.obj;
	String origin = (String) values.get(ORIGIN);
	ValueCallback callback = (ValueCallback) values.get(CALLBACK);
	Origin website = mOrigins.get(origin);
	Map retValues = new HashMap<String, Object>();
	retValues.put(CALLBACK, callback);
	if (website != null) {
	long usage = website.getUsage();
	retValues.put(USAGE, new Long(usage));
	}
	postUIMessage(Message.obtain(null, RETURN_USAGE_ORIGIN, retValues));
	} break;

	case GET_QUOTA_ORIGIN: {
	syncValues();
	Map values = (Map) msg.obj;
	String origin = (String) values.get(ORIGIN);
	ValueCallback callback = (ValueCallback) values.get(CALLBACK);
	Origin website = mOrigins.get(origin);
	Map retValues = new HashMap<String, Object>();
	retValues.put(CALLBACK, callback);
	if (website != null) {
	long quota = website.getQuota();
	retValues.put(QUOTA, new Long(quota));
	}
	postUIMessage(Message.obtain(null, RETURN_QUOTA_ORIGIN, retValues));
	} break;

	case UPDATE:
	syncValues();
	break;
	}
	}
	};
	}
	}

	/*
	* When calling getOrigins(), getUsageForOrigin() and getQuotaForOrigin(),
	* we need to get the values from WebCore, but we cannot block while doing so
	* as we used to do, as this could result in a full deadlock (other WebCore
	* messages received while we are still blocked here, see http://b/2127737).
	*
	* We have to do everything asynchronously, by providing a callback function.
	* We post a message on the WebCore thread (mHandler) that will get the result
	* from WebCore, and we post it back on the UI thread (using mUIHandler).
	* We can then use the callback function to return the value.
	*/

	@Override
	public void getOrigins(ValueCallback<Map> callback) {
	if (callback != null) {
	if (WebViewCore.THREAD_NAME.equals(Thread.currentThread().getName())) {
	syncValues();
	callback.onReceiveValue(mOrigins);
	} else {
	postMessage(Message.obtain(null, GET_ORIGINS, callback));
	}
	}
	}

	/**
	* Returns a list of origins having a database
	* should only be called from WebViewCore.
	*/
	Collection<Origin> getOriginsSync() {
	if (WebViewCore.THREAD_NAME.equals(Thread.currentThread().getName())) {
	update();
	return mOrigins.values();
	}
	return null;
	}

	@Override
	public void getUsageForOrigin(String origin, ValueCallback<Long> callback) {
	if (callback == null) {
	return;
	}
	if (origin == null) {
	callback.onReceiveValue(null);
	return;
	}
	if (WebViewCore.THREAD_NAME.equals(Thread.currentThread().getName())) {
	syncValues();
	Origin website = mOrigins.get(origin);
	callback.onReceiveValue(new Long(website.getUsage()));
	} else {
	HashMap values = new HashMap<String, Object>();
	values.put(ORIGIN, origin);
	values.put(CALLBACK, callback);
	postMessage(Message.obtain(null, GET_USAGE_ORIGIN, values));
	}
	}

	@Override
	public void getQuotaForOrigin(String origin, ValueCallback<Long> callback) {
	if (callback == null) {
	return;
	}
	if (origin == null) {
	callback.onReceiveValue(null);
	return;
	}
	if (WebViewCore.THREAD_NAME.equals(Thread.currentThread().getName())) {
	syncValues();
	Origin website = mOrigins.get(origin);
	callback.onReceiveValue(new Long(website.getUsage()));
	} else {
	HashMap values = new HashMap<String, Object>();
	values.put(ORIGIN, origin);
	values.put(CALLBACK, callback);
	postMessage(Message.obtain(null, GET_QUOTA_ORIGIN, values));
	}
	}

	@Override
	public void setQuotaForOrigin(String origin, long quota) {
	if (origin != null) {
	if (WebViewCore.THREAD_NAME.equals(Thread.currentThread().getName())) {
	nativeSetQuotaForOrigin(origin, quota);
	} else {
	postMessage(Message.obtain(null, SET_QUOTA_ORIGIN,
	new Origin(origin, quota)));
	}
	}
	}

	@Override
	public void deleteOrigin(String origin) {
	if (origin != null) {
	if (WebViewCore.THREAD_NAME.equals(Thread.currentThread().getName())) {
	nativeDeleteOrigin(origin);
	} else {
	postMessage(Message.obtain(null, DELETE_ORIGIN,
	new Origin(origin)));
	}
	}
	}

	@Override
	public void deleteAllData() {
	if (WebViewCore.THREAD_NAME.equals(Thread.currentThread().getName())) {
	nativeDeleteAllData();
	} else {
	postMessage(Message.obtain(null, DELETE_ALL));
	}
	}

	/**
	* Sets the maximum size of the ApplicationCache.
	* This should only ever be called on the WebKit thread.
	* Not part of the base-class API: this is only used by dump render tree.
	*/
	public void setAppCacheMaximumSize(long size) {
	nativeSetAppCacheMaximumSize(size);
	}

	/**
	* Utility function to send a message to our handler
	*/
	private synchronized void postMessage(Message msg) {
	if (mHandler != null) {
	mHandler.sendMessage(msg);
	}
	}

	/**
	* Utility function to send a message to the handler on the UI thread
	*/
	private void postUIMessage(Message msg) {
	if (mUIHandler != null) {
	mUIHandler.sendMessage(msg);
	}
	}

	/**
	* Get the singleton instance of this class.
	* @return The singleton {@link WebStorage} instance.
	*/
	public static WebStorageClassic getInstance() {
	if (sWebStorage == null) {
	sWebStorage = new WebStorageClassic();
	}
	return sWebStorage;
	}

	/**
	* @hide
	* Post a Sync request
	*/
	public void update() {
	if (WebViewCore.THREAD_NAME.equals(Thread.currentThread().getName())) {
	syncValues();
	} else {
	postMessage(Message.obtain(null, UPDATE));
	}
	}

	/**
	* Run on the WebCore thread
	* set the local values with the current ones
	*/
	private void syncValues() {
	Set<String> tmp = nativeGetOrigins();
	mOrigins = new HashMap<String, Origin>();
	for (String origin : tmp) {
	Origin website = new Origin(origin,
	nativeGetQuotaForOrigin(origin),
	nativeGetUsageForOrigin(origin));
	mOrigins.put(origin, website);
	}
	}

	WebStorageClassic() {}

	// Native functions
	private static native Set nativeGetOrigins();
	private static native long nativeGetUsageForOrigin(String origin);
	private static native long nativeGetQuotaForOrigin(String origin);
	private static native void nativeSetQuotaForOrigin(String origin, long quota);
	private static native void nativeDeleteOrigin(String origin);
	private static native void nativeDeleteAllData();
	private static native void nativeSetAppCacheMaximumSize(long size);
	}