Update code structure for wrap JNI API.