3 Commits

Author SHA1 Message Date
l-jonas f336a2932f Update whatsnew 2018-12-02 10:49:49 +01:00
l-jonas 0b3e2bf914 Release 0.3.6 2018-12-02 10:49:08 +01:00
l-jonas 6d9009daff Refactor index handler (#104)
* Add base for new database API

* Implement new database API for the android db implementation

* Implement new API for the default database implementation

* Fix compilation errors

* Move IndexHandler to own package

* Move some code out of the IndexHandler

* Fix compilation errors

* Make IndexInfo a data class

* Make FolderStats a data class

* Make FolderInfo a data class

* Move code out of IndexHandler

* Use one transaction per index message

* Start replacing callbacks by BroadcastChannels

* Fix compilation errors

* Replace callbacks by BroadcastChannels

* Remove IndexHandler.folderInfoByFolder

* Move code out of IndexHandler

* Use index update events to notify IndexBrowsers

* Remove preloading from the IndexBrowser

* Use channels to handle index update messages

* Remove the old ExecutorUtils

* Remove functions from the IndexHandler

* Refactor FolderBrowser

* Remove writeAccessLock

* Remove the indexWaitLock

* Send index change events after the transaction

* Remove markActive from the IndexHandler

* Refactor the IndexBrowser

* Fix showing folder content

* Fix document provider integration

* Fix index sequence handling

* Use a LinearLayout as base for folder entries

* Add theoretically showing of the missing index updates

* Move folder stats update events out of the database

* Send index update events when receiving a cluster config

* Fix counting missing index updates

* Send events after the db transaction at handle cluster config

* Handle index updates in batches

* Add logging of time for index processing

* Deduplicate index updates

* Read old records in bulk

* Update folder stats in bulk

* Fix typo

* Modularize IndexElementProcessor

* Prepare bulk FileInfo updates

* Update FileInfo in bulk

* Make logger private

* Use IO dispatcher

* Reconnect better

* Fix detecting new folders

* Dispatch crashes from background threads to the main thread

* Fix random crash on library shutdown

* Add option for more detailed crash reports

* Fix transaction running check and remove old code at DB abstraction

* Sort directory listings
2018-12-02 10:47:09 +01:00
57 changed files with 2319 additions and 2011 deletions
+2 -2
View File
@@ -19,8 +19,8 @@ android {
applicationId "net.syncthing.lite"
minSdkVersion 21
targetSdkVersion 26
versionCode 15
versionName "0.3.5"
versionCode 16
versionName "0.3.6"
multiDexEnabled true
playAccountConfig = playAccountConfigs.defaultAccountConfig
}
@@ -4,16 +4,15 @@ import android.app.Activity
import android.content.Intent
import android.databinding.DataBindingUtil
import android.os.Bundle
import android.util.Log
import kotlinx.coroutines.CompletableDeferred
import kotlinx.coroutines.Dispatchers
import kotlinx.coroutines.GlobalScope
import kotlinx.coroutines.launch
import net.syncthing.java.bep.IndexBrowser
import kotlinx.coroutines.*
import kotlinx.coroutines.channels.ConflatedBroadcastChannel
import kotlinx.coroutines.channels.consumeEach
import net.syncthing.java.bep.index.browser.DirectoryContentListing
import net.syncthing.java.bep.index.browser.DirectoryListing
import net.syncthing.java.bep.index.browser.DirectoryNotFoundListing
import net.syncthing.java.bep.index.browser.IndexBrowser
import net.syncthing.java.core.beans.FileInfo
import net.syncthing.java.core.beans.FolderInfo
import net.syncthing.java.core.utils.PathUtils
import net.syncthing.lite.BuildConfig
import net.syncthing.lite.R
import net.syncthing.lite.adapters.FolderContentsAdapter
import net.syncthing.lite.adapters.FolderContentsListener
@@ -22,30 +21,43 @@ import net.syncthing.lite.dialogs.FileMenuDialogFragment
import net.syncthing.lite.dialogs.FileUploadDialog
import net.syncthing.lite.dialogs.ReconnectIssueDialogFragment
import net.syncthing.lite.dialogs.downloadfile.DownloadFileDialogFragment
import org.jetbrains.anko.doAsync
class FolderBrowserActivity : SyncthingActivity() {
companion object {
private const val TAG = "FolderBrowserActivity"
private const val REQUEST_SELECT_UPLOAD_FILE = 171
private const val STATUS_PATH = "path"
const val EXTRA_FOLDER_NAME = "folder_name"
}
private lateinit var binding: ActivityFolderBrowserBinding
private lateinit var indexBrowser: IndexBrowser
private val adapter = FolderContentsAdapter()
private lateinit var folder: String
private val path = ConflatedBroadcastChannel<String>()
private val listing = ConflatedBroadcastChannel<DirectoryListing?>(null)
override fun onCreate(savedInstanceState: Bundle?) {
super.onCreate(savedInstanceState)
binding = DataBindingUtil.setContentView(this, R.layout.activity_folder_browser)
binding.mainListViewUploadHereButton.setOnClickListener { showUploadHereDialog() }
val binding: ActivityFolderBrowserBinding = DataBindingUtil.setContentView(this, R.layout.activity_folder_browser)
val adapter = FolderContentsAdapter()
binding.listView.adapter = adapter
binding.mainListViewUploadHereButton.setOnClickListener {
startActivityForResult(
Intent(Intent.ACTION_OPEN_DOCUMENT).apply {
addCategory(Intent.CATEGORY_OPENABLE)
type = "*/*"
},
REQUEST_SELECT_UPLOAD_FILE
)
}
adapter.listener = object: FolderContentsListener {
override fun onItemClicked(fileInfo: FileInfo) {
navigateToFolder(fileInfo)
if (fileInfo.isDirectory()) {
path.offer(fileInfo.path)
} else {
DownloadFileDialogFragment.newInstance(fileInfo).show(supportFragmentManager)
}
}
override fun onItemLongClicked(fileInfo: FileInfo): Boolean {
@@ -58,26 +70,75 @@ class FolderBrowserActivity : SyncthingActivity() {
}
}
}
val folder = intent.getStringExtra(EXTRA_FOLDER_NAME)
libraryHandler.syncthingClient {
indexBrowser = it.indexHandler.newIndexBrowser(folder, true, true)
indexBrowser.setOnFolderChangedListener(this::onFolderChanged)
}
ReconnectIssueDialogFragment.showIfNeeded(this)
folder = intent.getStringExtra(EXTRA_FOLDER_NAME)
path.offer(if (savedInstanceState == null) IndexBrowser.ROOT_PATH else savedInstanceState.getString(STATUS_PATH))
launch {
var job = Job()
path.consumeEach { path ->
job.cancel()
job = Job()
binding.listView.scrollToPosition(0)
listing.send(null)
async(job) {
libraryHandler.libraryManager.streamDirectoryListing(folder, path).consumeEach {
listing.send(it)
}
}
}
}
launch {
listing.openSubscription().consumeEach { listing ->
if (listing == null) {
binding.isLoading = true
} else {
supportActionBar?.title = if (PathUtils.isRoot(listing.path)) folder else PathUtils.getFileName(listing.path)
binding.isLoading = false
adapter.data = if (listing is DirectoryContentListing)
listing.entries.sortedWith(IndexBrowser.sortAlphabeticallyDirectoriesFirst)
else
emptyList()
}
}
}
}
override fun onDestroy() {
super.onDestroy()
Thread {
indexBrowser.setOnFolderChangedListener(null)
indexBrowser.close()
}.start()
override fun onSaveInstanceState(outState: Bundle) {
super.onSaveInstanceState(outState)
outState.putString(STATUS_PATH, path.value)
}
private fun goUp(): Boolean {
val currentListing = listing.value
val parentPath = when (currentListing) {
is DirectoryContentListing -> currentListing.parentEntry?.path
is DirectoryNotFoundListing -> currentListing.theoreticalParentPath
else -> null
}
return if (parentPath == null) {
false
} else {
path.offer(parentPath)
true
}
}
override fun onBackPressed() {
//click item '0', ie '..' (go to parent)
navigateToFolder(adapter.data[0])
if (!goUp()) {
super.onBackPressed()
}
}
override fun onActivityResult(requestCode: Int, resultCode: Int, intent: Intent?) {
@@ -85,85 +146,18 @@ class FolderBrowserActivity : SyncthingActivity() {
libraryHandler.syncthingClient { syncthingClient ->
GlobalScope.launch (Dispatchers.Main) {
// FIXME: it would be better if the dialog would use the library handler
FileUploadDialog(this@FolderBrowserActivity, syncthingClient, intent!!.data,
indexBrowser.folder, indexBrowser.currentPath,
{ showFolderListView(indexBrowser.currentPath) }).show()
FileUploadDialog(
this@FolderBrowserActivity,
syncthingClient,
intent!!.data,
folder,
path.value,
{ /* nothing to do on success */ }
).show()
}
}
} else {
super.onActivityResult(requestCode, resultCode, intent)
}
}
private fun showFolderListView(path: String) {
indexBrowser.navigateToNearestPath(path)
navigateToFolder(indexBrowser.currentPathInfo())
}
private fun navigateToFolder(fileInfo: FileInfo) {
Log.d(TAG, "navigate to path = '" + fileInfo.path + "' from path = '" + indexBrowser.currentPath + "'")
if (indexBrowser.isRoot() && PathUtils.isParent(fileInfo.path)) {
finish()
} else {
if (fileInfo.isDirectory()) {
doAsync {
indexBrowser.navigateTo(fileInfo)
}
Log.d(TAG, "load folder cache bg")
binding.isLoading = true
} else {
if (BuildConfig.DEBUG) {
Log.i(TAG, "pulling file = " + fileInfo)
}
DownloadFileDialogFragment.newInstance(fileInfo).show(supportFragmentManager)
}
}
}
private fun onFolderChanged() {
GlobalScope.launch {
val list = indexBrowser.listFiles()
Log.i("navigateToFolder", "list for path = '" + indexBrowser.currentPath + "' list = " + list.size + " records")
Log.d("navigateToFolder", "list for path = '" + indexBrowser.currentPath + "' list = " + list)
assert(!list.isEmpty())//list must contain at least the 'parent' path
val title = if (indexBrowser.isRoot()) {
val result = CompletableDeferred<String?>()
libraryHandler.folderBrowser {
result.complete(it.getFolderInfo(indexBrowser.folder)?.label)
}
result.await()
} else {
indexBrowser.currentPathInfo().fileName
}
runOnUiThread {
binding.isLoading = false
adapter.data = list
binding.listView.scrollToPosition(0)
supportActionBar?.title = title
}
}
}
private fun updateFolderListView() {
showFolderListView(indexBrowser.currentPath)
}
private fun showUploadHereDialog() {
val intent = Intent(Intent.ACTION_OPEN_DOCUMENT)
intent.addCategory(Intent.CATEGORY_OPENABLE)
intent.type = "*/*"
startActivityForResult(intent, REQUEST_SELECT_UPLOAD_FILE)
}
override fun onIndexUpdateComplete(folderInfo: FolderInfo) {
super.onIndexUpdateComplete(folderInfo)
updateFolderListView()
}
}
@@ -4,22 +4,18 @@ import android.app.AlertDialog
import android.databinding.DataBindingUtil
import android.os.Bundle
import android.support.design.widget.Snackbar
import android.support.v7.app.AppCompatActivity
import android.view.LayoutInflater
import net.syncthing.java.core.beans.FolderInfo
import net.syncthing.lite.BuildConfig
import net.syncthing.lite.R
import net.syncthing.lite.async.CoroutineActivity
import net.syncthing.lite.databinding.DialogLoadingBinding
import net.syncthing.lite.library.LibraryHandler
import org.jetbrains.anko.contentView
import org.slf4j.impl.HandroidLoggerAdapter
abstract class SyncthingActivity : AppCompatActivity() {
abstract class SyncthingActivity : CoroutineActivity() {
val libraryHandler: LibraryHandler by lazy {
LibraryHandler(
context = this@SyncthingActivity,
onIndexUpdateProgressListener = this::onIndexUpdateProgress,
onIndexUpdateCompleteListener = this::onIndexUpdateComplete
context = this@SyncthingActivity
)
}
private var loadingDialog: AlertDialog? = null
@@ -58,19 +54,6 @@ abstract class SyncthingActivity : AppCompatActivity() {
loadingDialog?.dismiss()
}
open fun onIndexUpdateProgress(folderInfo: FolderInfo, percentage: Int) {
val message = getString(R.string.index_update_progress_label, folderInfo.label, percentage)
snackBar?.setText(message) ?: run {
snackBar = Snackbar.make(contentView!!, message, Snackbar.LENGTH_INDEFINITE)
snackBar?.show()
}
}
open fun onIndexUpdateComplete(folderInfo: FolderInfo) {
snackBar?.dismiss()
snackBar = null
}
open fun onLibraryLoaded() {
// nothing to do
}
@@ -2,8 +2,10 @@ package net.syncthing.lite.adapters
import android.support.v7.widget.RecyclerView
import android.text.format.DateUtils
import android.util.Log
import android.view.LayoutInflater
import android.view.ViewGroup
import net.syncthing.java.bep.folder.FolderStatus
import net.syncthing.java.core.beans.FolderInfo
import net.syncthing.java.core.beans.FolderStats
import net.syncthing.lite.R
@@ -11,7 +13,7 @@ import net.syncthing.lite.databinding.ListviewFolderBinding
import kotlin.properties.Delegates
class FoldersListAdapter: RecyclerView.Adapter<FolderListViewHolder>() {
var data: List<Pair<FolderInfo, FolderStats>> by Delegates.observable(listOf()) {
var data: List<FolderStatus> by Delegates.observable(listOf()) {
_, _, _ -> notifyDataSetChanged()
}
@@ -22,7 +24,7 @@ class FoldersListAdapter: RecyclerView.Adapter<FolderListViewHolder>() {
}
override fun getItemCount() = data.size
override fun getItemId(position: Int) = data[position].first.folderId.hashCode().toLong()
override fun getItemId(position: Int) = data[position].info.folderId.hashCode().toLong()
override fun onCreateViewHolder(parent: ViewGroup, viewType: Int) = FolderListViewHolder (
ListviewFolderBinding.inflate(
@@ -32,15 +34,23 @@ class FoldersListAdapter: RecyclerView.Adapter<FolderListViewHolder>() {
override fun onBindViewHolder(holder: FolderListViewHolder, position: Int) {
val binding = holder.binding
val (folderInfo, folderStats) = data[position]
val item = data[position]
val (folderInfo, folderStats) = item
val context = holder.itemView.context
Log.d("FolderListAdapter", "$item")
binding.folderName = context.getString(R.string.folder_label_format, folderInfo.label, folderInfo.folderId)
binding.lastModification = context.getString(R.string.last_modified_time,
DateUtils.getRelativeDateTimeString(context, folderStats.lastUpdate.time, DateUtils.MINUTE_IN_MILLIS, DateUtils.WEEK_IN_MILLIS, 0))
binding.info = context.getString(R.string.folder_content_info, folderStats.describeSize(), folderStats.fileCount, folderStats.dirCount)
binding.info = context.getString(R.string.folder_content_info, folderStats.sizeDescription, folderStats.fileCount, folderStats.dirCount)
binding.info2 = if (item.missingIndexUpdates == 0L)
null
else
context.getString(R.string.pending_index_updates, item.missingIndexUpdates)
binding.root.setOnClickListener {
listener?.onFolderClicked(folderInfo, folderStats)
@@ -52,4 +62,4 @@ class FolderListViewHolder(val binding: ListviewFolderBinding): RecyclerView.Vie
interface FolderListAdapterListener {
fun onFolderClicked(folderInfo: FolderInfo, folderStats: FolderStats)
}
}
@@ -4,6 +4,8 @@ import android.app.Application
import android.content.ClipData
import android.content.ClipboardManager
import android.content.Context
import android.os.Handler
import android.os.Looper
import android.util.Log
import net.syncthing.lite.BuildConfig
import org.jetbrains.anko.defaultSharedPreferences
@@ -14,6 +16,7 @@ class Application: Application() {
companion object {
private const val LOG_TAG = "Application"
private const val PREF_ENABLE_CRASH_HANDLER = "crash_handler"
private val handler = Handler(Looper.getMainLooper())
}
override fun onCreate() {
@@ -21,12 +24,13 @@ class Application: Application() {
val clipboard = getSystemService(Context.CLIPBOARD_SERVICE) as ClipboardManager
val defaultHandler = Thread.getDefaultUncaughtExceptionHandler()
val mainThread = Thread.currentThread()
if (defaultHandler == null) {
Log.w(LOG_TAG, "could not get default crash handler")
}
Thread.setDefaultUncaughtExceptionHandler { thread, ex ->
fun handleCrash(ex: Throwable) {
Log.w(LOG_TAG, "app crashed", ex)
val enableCustomCrashHandling = defaultSharedPreferences.getBoolean(PREF_ENABLE_CRASH_HANDLER, false)
@@ -43,10 +47,18 @@ class Application: Application() {
}
if (defaultHandler != null) {
defaultHandler.uncaughtException(thread, ex)
defaultHandler.uncaughtException(mainThread, ex)
} else {
System.exit(1)
}
}
Thread.setDefaultUncaughtExceptionHandler { thread, ex ->
if (Looper.getMainLooper() === Looper.myLooper()) {
handleCrash(ex)
} else {
handler.post { handleCrash(ex) }
}
}
}
}
@@ -0,0 +1,19 @@
package net.syncthing.lite.async
import android.support.v7.app.AppCompatActivity
import kotlinx.coroutines.CoroutineScope
import kotlinx.coroutines.Dispatchers
import kotlinx.coroutines.Job
import kotlin.coroutines.CoroutineContext
abstract class CoroutineActivity: AppCompatActivity(), CoroutineScope {
val job = Job()
override val coroutineContext: CoroutineContext
get() = job + Dispatchers.Main
override fun onDestroy() {
super.onDestroy()
job.cancel()
}
}
@@ -0,0 +1,19 @@
package net.syncthing.lite.async
import android.support.v4.app.Fragment
import kotlinx.coroutines.CoroutineScope
import kotlinx.coroutines.Dispatchers
import kotlinx.coroutines.Job
import kotlin.coroutines.CoroutineContext
abstract class CoroutineFragment: Fragment(), CoroutineScope {
val job = Job()
override val coroutineContext: CoroutineContext
get() = job + Dispatchers.Main
override fun onDestroy() {
super.onDestroy()
job.cancel()
}
}
@@ -2,12 +2,10 @@ package net.syncthing.lite.fragments
import android.arch.lifecycle.Observer
import android.os.Bundle
import android.util.Log
import android.view.LayoutInflater
import android.view.View
import android.view.ViewGroup
import kotlinx.coroutines.Dispatchers
import kotlinx.coroutines.GlobalScope
import kotlinx.coroutines.channels.consumeEach
import kotlinx.coroutines.launch
import net.syncthing.java.core.beans.FolderInfo
import net.syncthing.java.core.beans.FolderStats
@@ -18,49 +16,30 @@ import net.syncthing.lite.databinding.FragmentFoldersBinding
import org.jetbrains.anko.intentFor
class FoldersFragment : SyncthingFragment() {
override fun onCreateView(inflater: LayoutInflater, container: ViewGroup?, savedInstanceState: Bundle?): View? {
val adapter = FoldersListAdapter()
private val TAG = "FoldersFragment"
private lateinit var binding: FragmentFoldersBinding
override fun onCreateView(inflater: LayoutInflater, container: ViewGroup?,
savedInstanceState: Bundle?): View? {
binding = FragmentFoldersBinding.inflate(layoutInflater, container, false)
adapter.listener = object : FolderListAdapterListener {
override fun onFolderClicked(folderInfo: FolderInfo, folderStats: FolderStats) {
startActivity(
activity!!.intentFor<FolderBrowserActivity>(
FolderBrowserActivity.EXTRA_FOLDER_NAME to folderInfo.folderId
)
)
}
}
val binding = FragmentFoldersBinding.inflate(layoutInflater, container, false)
binding.list.adapter = adapter
libraryHandler.isListeningPortTaken.observe(this, Observer { binding.listeningPortTaken = it })
launch {
libraryHandler.subscribeToFolderStatusList().consumeEach {
adapter.data = it
binding.isEmpty = it.isEmpty()
}
}
return binding.root
}
override fun onLibraryLoaded() {
showAllFoldersListView()
}
private fun showAllFoldersListView() {
libraryHandler.folderBrowser { folderBrowser ->
val list = folderBrowser.folderInfoAndStatsList()
GlobalScope.launch (Dispatchers.Main) {
Log.i(TAG, "list folders = " + list + " (" + list.size + " records)")
val adapter = FoldersListAdapter().apply { data = list }
binding.list.adapter = adapter
adapter.listener = object : FolderListAdapterListener {
override fun onFolderClicked(folderInfo: FolderInfo, folderStats: FolderStats) {
startActivity(
activity!!.intentFor<FolderBrowserActivity>(
FolderBrowserActivity.EXTRA_FOLDER_NAME to folderInfo.folderId
)
)
}
}
binding.isEmpty = list.isEmpty()
}
}
}
override fun onIndexUpdateComplete(folderInfo: FolderInfo) {
super.onIndexUpdateComplete(folderInfo)
showAllFoldersListView()
}
}
@@ -13,6 +13,7 @@ class SettingsFragment : PreferenceFragmentCompat() {
val localDeviceName = findPreference("local_device_name") as EditTextPreference
val appVersion = findPreference("app_version")
val forceStop = findPreference("force_stop")
(activity as SyncthingActivity?)?.let { activity ->
val versionName = activity.packageManager.getPackageInfo(activity.packageName, 0)?.versionName
@@ -27,5 +28,11 @@ class SettingsFragment : PreferenceFragmentCompat() {
true
}
}
forceStop.setOnPreferenceClickListener {
System.exit(0)
true
}
}
}
@@ -1,15 +1,10 @@
package net.syncthing.lite.fragments
import android.support.v4.app.Fragment
import net.syncthing.java.core.beans.FolderInfo
import net.syncthing.lite.async.CoroutineFragment
import net.syncthing.lite.library.LibraryHandler
abstract class SyncthingFragment : Fragment() {
val libraryHandler: LibraryHandler by lazy { LibraryHandler(
context = context!!,
onIndexUpdateProgressListener = this::onIndexUpdateProgress,
onIndexUpdateCompleteListener = this::onIndexUpdateComplete
)}
abstract class SyncthingFragment : CoroutineFragment() {
val libraryHandler: LibraryHandler by lazy { LibraryHandler(context = context!!)}
override fun onStart() {
super.onStart()
@@ -27,8 +22,4 @@ abstract class SyncthingFragment : Fragment() {
}
open fun onLibraryLoaded() {}
open fun onIndexUpdateProgress(folderInfo: FolderInfo, percentage: Int) {}
open fun onIndexUpdateComplete(folderInfo: FolderInfo) {}
}
}
@@ -5,16 +5,16 @@ import android.arch.lifecycle.MutableLiveData
import android.content.Context
import android.os.Handler
import android.os.Looper
import android.util.Log
import kotlinx.coroutines.Dispatchers
import kotlinx.coroutines.GlobalScope
import kotlinx.coroutines.Job
import kotlinx.coroutines.channels.BroadcastChannel
import kotlinx.coroutines.channels.Channel
import kotlinx.coroutines.channels.consumeEach
import kotlinx.coroutines.launch
import net.syncthing.java.bep.FolderBrowser
import net.syncthing.java.bep.folder.FolderBrowser
import net.syncthing.java.bep.folder.FolderStatus
import net.syncthing.java.client.SyncthingClient
import net.syncthing.java.core.beans.DeviceId
import net.syncthing.java.core.beans.FileInfo
import net.syncthing.java.core.beans.FolderInfo
import net.syncthing.java.core.beans.IndexInfo
import net.syncthing.java.core.configuration.Configuration
import org.jetbrains.anko.doAsync
import java.util.concurrent.atomic.AtomicBoolean
@@ -25,18 +25,19 @@ import java.util.concurrent.atomic.AtomicBoolean
*
* It's possible to do multiple start and stop cycles with one instance of this class.
*/
class LibraryHandler(context: Context,
private val onIndexUpdateProgressListener: (FolderInfo, Int) -> Unit = {_, _ -> },
private val onIndexUpdateCompleteListener: (FolderInfo) -> Unit = {}) {
class LibraryHandler(context: Context) {
companion object {
private const val TAG = "LibraryHandler"
private val handler = Handler(Looper.getMainLooper())
}
private val libraryManager = DefaultLibraryManager.with(context)
val libraryManager = DefaultLibraryManager.with(context)
private val isStarted = AtomicBoolean(false)
private val isListeningPortTakenInternal = MutableLiveData<Boolean>().apply { postValue(false) }
private val indexUpdateCompleteMessages = BroadcastChannel<String>(capacity = 16)
private val folderStatusList = BroadcastChannel<List<FolderStatus>>(capacity = Channel.CONFLATED)
private var job: Job = Job()
val isListeningPortTaken: LiveData<Boolean> = isListeningPortTakenInternal
@@ -62,9 +63,21 @@ class LibraryHandler(context: Context,
val client = libraryInstance.syncthingClient
client.indexHandler.registerOnIndexRecordAcquiredListener(this::onIndexRecordAcquired)
client.indexHandler.registerOnFullIndexAcquiredListenersListener(this::onRemoteIndexAcquired)
client.discoveryHandler.registerMessageFromUnknownDeviceListener(internalMessageFromUnknownDeviceListener)
job = Job()
GlobalScope.launch (job) {
libraryInstance.syncthingClient.indexHandler.subscribeToOnFullIndexAcquiredEvents().consumeEach {
indexUpdateCompleteMessages.send(it)
}
}
GlobalScope.launch (job) {
libraryInstance.folderBrowser.folderInfoAndStatusStream().consumeEach {
folderStatusList.send(it)
}
}
}
}
@@ -73,10 +86,10 @@ class LibraryHandler(context: Context,
throw IllegalStateException("already stopped")
}
job!!.cancel()
syncthingClient {
try {
it.indexHandler.unregisterOnIndexRecordAcquiredListener(this::onIndexRecordAcquired)
it.indexHandler.unregisterOnFullIndexAcquiredListenersListener(this::onRemoteIndexAcquired)
it.discoveryHandler.unregisterMessageFromUnknownDeviceListener(internalMessageFromUnknownDeviceListener)
} catch (e: IllegalArgumentException) {
// ignored, no idea why this is thrown
@@ -86,22 +99,6 @@ class LibraryHandler(context: Context,
libraryManager.stopLibraryUsage()
}
private fun onIndexRecordAcquired(folderInfo: FolderInfo, newRecords: List<FileInfo>, indexInfo: IndexInfo) {
Log.i(TAG, "handleIndexRecordEvent trigger folder list update from index record acquired")
GlobalScope.launch (Dispatchers.Main) {
onIndexUpdateProgressListener(folderInfo, (indexInfo.getCompleted() * 100).toInt())
}
}
private fun onRemoteIndexAcquired(folderInfo: FolderInfo) {
Log.i(TAG, "handleIndexAcquiredEvent trigger folder list update from index acquired")
GlobalScope.launch (Dispatchers.Main) {
onIndexUpdateCompleteListener(folderInfo)
}
}
/*
* The callback is executed asynchronously.
* As soon as it returns, there is no guarantee about the availability of the library
@@ -139,4 +136,7 @@ class LibraryHandler(context: Context,
fun unregisterMessageFromUnknownDeviceListener(listener: (DeviceId) -> Unit) {
messageFromUnknownDeviceListeners.remove(listener)
}
fun subscribeToOnFullIndexAcquiredEvents() = indexUpdateCompleteMessages.openSubscription()
fun subscribeToFolderStatusList() = folderStatusList.openSubscription()
}
@@ -7,6 +7,7 @@ import net.syncthing.java.core.configuration.Configuration
import net.syncthing.repository.android.SqliteIndexRepository
import net.syncthing.repository.android.TempDirectoryLocalRepository
import net.syncthing.repository.android.database.RepositoryDatabase
import org.jetbrains.anko.defaultSharedPreferences
import java.io.File
import java.net.DatagramSocket
import java.net.InetAddress
@@ -53,12 +54,13 @@ class LibraryInstance (context: Context) {
closeDatabaseOnClose = false,
clearTempStorageHook = { tempRepository.deleteAllData() }
),
tempRepository = tempRepository
tempRepository = tempRepository,
enableDetailedException = context.defaultSharedPreferences.getBoolean("detailed_exception", false)
)
val folderBrowser = syncthingClient.indexHandler.newFolderBrowser()
val folderBrowser = syncthingClient.indexHandler.folderBrowser
val indexBrowser = syncthingClient.indexHandler.indexBrowser
fun shutdown() {
folderBrowser.close()
syncthingClient.close()
configuration.persistNow()
}
@@ -2,6 +2,12 @@ package net.syncthing.lite.library
import android.os.Handler
import android.os.Looper
import kotlinx.coroutines.GlobalScope
import kotlinx.coroutines.Job
import kotlinx.coroutines.async
import kotlinx.coroutines.channels.ConflatedBroadcastChannel
import kotlinx.coroutines.channels.consumeEach
import kotlinx.coroutines.channels.produce
import java.util.concurrent.Executors
import kotlin.coroutines.resume
import kotlin.coroutines.suspendCoroutine
@@ -35,7 +41,7 @@ class LibraryManager (
// only this Thread should access instance and userCounter
private val startStopExecutor = Executors.newSingleThreadExecutor()
private var instance: LibraryInstance? = null
private val instanceStream = ConflatedBroadcastChannel<LibraryInstance?>(null)
private var userCounter = 0
fun startLibraryUsage(callback: (LibraryInstance) -> Unit) {
@@ -43,12 +49,12 @@ class LibraryManager (
val newUserCounter = ++userCounter
handler.post { userCounterListener(newUserCounter) }
if (instance == null) {
instance = synchronousInstanceCreator()
if (instanceStream.value == null) {
instanceStream.offer(synchronousInstanceCreator())
handler.post { isRunningListener(true) }
}
handler.post { callback(instance!!) }
handler.post { callback(instanceStream.value!!) }
}
}
@@ -60,6 +66,16 @@ class LibraryManager (
}
}
suspend fun <T> withLibrary(action: suspend (LibraryInstance) -> T): T {
val instance = startLibraryUsageCoroutine()
return try {
action(instance)
} finally {
stopLibraryUsage()
}
}
fun stopLibraryUsage() {
startStopExecutor.submit {
val newUserCounter = --userCounter
@@ -77,8 +93,8 @@ class LibraryManager (
fun shutdownIfThereAreZeroUsers(listener: (wasShutdownPerformed: Boolean) -> Unit = {}) {
startStopExecutor.submit {
if (userCounter == 0) {
instance?.shutdown()
instance = null
instanceStream.value?.shutdown()
instanceStream.offer(null)
handler.post { isRunningListener(false) }
handler.post { listener(true) }
@@ -87,4 +103,21 @@ class LibraryManager (
}
}
}
fun streamDirectoryListing(folder: String, path: String) = GlobalScope.produce {
var job = Job()
instanceStream.openSubscription().consumeEach { instance ->
job.cancel()
job = Job()
if (instance != null) {
async (job) {
instance.indexBrowser.streamDirectoryListing(folder, path).consumeEach {
send(it)
}
}
}
}
}
}
@@ -10,33 +10,36 @@ import android.provider.DocumentsProvider
import android.util.Log
import kotlinx.coroutines.cancel
import kotlinx.coroutines.runBlocking
import net.syncthing.java.bep.IndexBrowser
import net.syncthing.java.bep.index.browser.DirectoryContentListing
import net.syncthing.java.bep.index.browser.DirectoryNotFoundListing
import net.syncthing.java.core.beans.FileInfo
import net.syncthing.java.core.beans.FolderInfo
import net.syncthing.java.core.beans.FolderStats
import net.syncthing.lite.R
import java.io.FileNotFoundException
import java.net.URLConnection
import java.util.concurrent.CountDownLatch
class SyncthingProvider : DocumentsProvider() {
companion object {
private const val Tag = "SyncthingProvider"
private val DefaultRootProjection = arrayOf(
Root.COLUMN_ROOT_ID,
Root.COLUMN_FLAGS,
Root.COLUMN_TITLE,
Root.COLUMN_SUMMARY,
Root.COLUMN_DOCUMENT_ID,
Root.COLUMN_ICON)
Root.COLUMN_ICON
)
private val DefaultDocumentProjection = arrayOf(
Document.COLUMN_DOCUMENT_ID,
Document.COLUMN_DISPLAY_NAME,
Document.COLUMN_SIZE,
Document.COLUMN_MIME_TYPE,
Document.COLUMN_LAST_MODIFIED,
Document.COLUMN_FLAGS)
Document.COLUMN_FLAGS
)
}
override fun onCreate(): Boolean {
@@ -45,96 +48,122 @@ class SyncthingProvider : DocumentsProvider() {
}
// this instance is not started -> it connects and disconnects on demand
private val libraryHandler: LibraryHandler by lazy { LibraryHandler(context) }
private val libraryManager: LibraryManager by lazy { DefaultLibraryManager.with(context) }
override fun queryRoots(projection: Array<String>?): Cursor {
Log.d(Tag, "queryRoots($projection)")
val latch = CountDownLatch(1)
var folders: List<Pair<FolderInfo, FolderStats>>? = null
libraryHandler.folderBrowser { folderBrowser ->
folders = folderBrowser.folderInfoAndStatsList()
latch.countDown()
}
latch.await()
val result = MatrixCursor(projection ?: DefaultRootProjection)
folders!!.forEach { folder ->
val row = result.newRow()
row.add(Root.COLUMN_ROOT_ID, folder.first.folderId)
row.add(Root.COLUMN_SUMMARY, folder.first.label)
row.add(Root.COLUMN_FLAGS, 0)
row.add(Root.COLUMN_TITLE, context.getString(R.string.app_name))
row.add(Root.COLUMN_DOCUMENT_ID, getDocIdForFile(folder.first))
row.add(Root.COLUMN_ICON, R.mipmap.ic_launcher)
return runBlocking {
libraryManager.withLibrary { instance ->
MatrixCursor(projection ?: DefaultRootProjection).apply {
instance.folderBrowser.folderInfoAndStatusList().forEach { folder ->
newRow().apply {
add(Root.COLUMN_ROOT_ID, folder.info.folderId)
add(Root.COLUMN_SUMMARY, folder.info.label)
add(Root.COLUMN_FLAGS, 0)
add(Root.COLUMN_TITLE, context.getString(R.string.app_name))
add(Root.COLUMN_DOCUMENT_ID, getDocIdForFile(folder.info))
add(Root.COLUMN_ICON, R.mipmap.ic_launcher)
}
}
}
}
}
return result
}
override fun queryChildDocuments(parentDocumentId: String, projection: Array<String>?,
sortOrder: String?): Cursor {
Log.d(Tag, "queryChildDocuments($parentDocumentId, $projection, $sortOrder)")
val result = MatrixCursor(projection ?: DefaultDocumentProjection)
getIndexBrowser(getFolderIdForDocId(parentDocumentId))
.listFiles(getPathForDocId(parentDocumentId))
.forEach { fileInfo ->
includeFile(result, fileInfo)
return runBlocking {
libraryManager.withLibrary { instance ->
val listing = instance.indexBrowser.getDirectoryListing(
folder = getFolderIdForDocId(parentDocumentId),
path = getPathForDocId(parentDocumentId)
)
when (listing) {
is DirectoryNotFoundListing -> throw FileNotFoundException()
is DirectoryContentListing -> {
val result = MatrixCursor(projection ?: DefaultDocumentProjection)
listing.entries.forEach { entry ->
includeFile(result, entry)
}
result
}
}
return result
}
}
}
override fun queryDocument(documentId: String, projection: Array<String>?): Cursor {
Log.d(Tag, "queryDocument($documentId, $projection)")
val result = MatrixCursor(projection ?: DefaultDocumentProjection)
val fileInfo = getIndexBrowser(getFolderIdForDocId(documentId))
.getFileInfoByAbsolutePath(getPathForDocId(documentId))
includeFile(result, fileInfo)
return result
return runBlocking {
libraryManager.withLibrary { instance ->
val fileInfo = instance.indexBrowser.getFileInfoByAbsolutePathAllowNull(
folder = getFolderIdForDocId(documentId),
path = getPathForDocId(documentId)
) ?: throw FileNotFoundException()
MatrixCursor(projection ?: DefaultDocumentProjection).apply {
includeFile(this, fileInfo)
}
}
}
}
@Throws(FileNotFoundException::class)
override fun openDocument(documentId: String, mode: String, signal: CancellationSignal?):
ParcelFileDescriptor {
Log.d(Tag, "openDocument($documentId, $mode, $signal)")
val fileInfo = getIndexBrowser(getFolderIdForDocId(documentId))
.getFileInfoByAbsolutePath(getPathForDocId(documentId))
val accessMode = ParcelFileDescriptor.parseMode(mode)
if (accessMode != ParcelFileDescriptor.MODE_READ_ONLY) {
throw NotImplementedError()
}
val outputFile = runBlocking {
signal?.setOnCancelListener {
this.coroutineContext.cancel()
}
return runBlocking {
libraryManager.withLibrary { instance ->
val fileInfo = instance.indexBrowser.getFileInfoByAbsolutePathAllowNull(
folder = getFolderIdForDocId(documentId),
path = getPathForDocId(documentId)
) ?: throw FileNotFoundException()
val libraryInstance = libraryManager.startLibraryUsageCoroutine()
signal?.setOnCancelListener {
this.coroutineContext.cancel()
}
try {
DownloadFileTask.downloadFileCoroutine(
val outputFile = DownloadFileTask.downloadFileCoroutine(
externalCacheDir = context.externalCacheDir,
syncthingClient = libraryInstance.syncthingClient,
syncthingClient = instance.syncthingClient,
fileInfo = fileInfo,
onProgress = { /* ignore the progress */ }
)
} finally {
libraryManager.stopLibraryUsage()
ParcelFileDescriptor.open(outputFile, ParcelFileDescriptor.MODE_READ_ONLY)
}
}
return ParcelFileDescriptor.open(outputFile, ParcelFileDescriptor.MODE_READ_ONLY)
}
private fun includeFile(result: MatrixCursor, fileInfo: FileInfo) {
val row = result.newRow()
row.add(Document.COLUMN_DOCUMENT_ID, getDocIdForFile(fileInfo))
row.add(Document.COLUMN_DISPLAY_NAME, fileInfo.fileName)
row.add(Document.COLUMN_SIZE, fileInfo.size)
val mime = if (fileInfo.isDirectory()) Document.MIME_TYPE_DIR
else URLConnection.guessContentTypeFromName(fileInfo.fileName)
row.add(Document.COLUMN_MIME_TYPE, mime)
row.add(Document.COLUMN_LAST_MODIFIED, fileInfo.lastModified)
row.add(Document.COLUMN_FLAGS, 0)
result.newRow().apply {
add(Document.COLUMN_DOCUMENT_ID, getDocIdForFile(fileInfo))
add(Document.COLUMN_DISPLAY_NAME, fileInfo.fileName)
add(Document.COLUMN_SIZE, fileInfo.size)
add(
Document.COLUMN_MIME_TYPE,
if (fileInfo.isDirectory())
Document.MIME_TYPE_DIR
else
URLConnection.guessContentTypeFromName(fileInfo.fileName)
)
add(Document.COLUMN_LAST_MODIFIED, fileInfo.lastModified)
add(Document.COLUMN_FLAGS, 0)
}
}
private fun getFolderIdForDocId(docId: String) = docId.split(":")[0]
@@ -144,15 +173,4 @@ class SyncthingProvider : DocumentsProvider() {
private fun getDocIdForFile(folderInfo: FolderInfo) = folderInfo.folderId + ":"
private fun getDocIdForFile(fileInfo: FileInfo) = fileInfo.folder + ":" + fileInfo.path
private fun getIndexBrowser(folderId: String): IndexBrowser {
val latch = CountDownLatch(1)
var indexBrowser: IndexBrowser? = null
libraryHandler.syncthingClient {
indexBrowser = it.indexHandler.newIndexBrowser(folderId)
latch.countDown()
}
latch.await()
return indexBrowser!!
}
}
+1 -3
View File
@@ -1,3 +1 @@
- new connection handling
- option for users to get detailed crash reports
- bugfixes
- new (faster) index handling
+22 -13
View File
@@ -13,9 +13,17 @@
<variable
name="info"
type="String" />
<variable
name="info2"
type="String" />
<import type="android.view.View" />
<import type="android.text.TextUtils" />
</data>
<RelativeLayout
<LinearLayout
android:orientation="vertical"
android:background="?selectableItemBackground"
android:layout_width="match_parent"
android:layout_height="wrap_content"
@@ -29,13 +37,9 @@
android:text="@{folderName}"
android:id="@+id/folder_name_view"
android:layout_width="match_parent"
android:layout_alignParentEnd="true"
android:layout_height="wrap_content"
android:ellipsize="end"
android:maxLines="1"
android:layout_alignParentTop="true"
android:gravity="top"
android:textAlignment="gravity"
android:textSize="20sp"
android:textStyle="bold"/>
@@ -47,9 +51,7 @@
android:layout_height="wrap_content"
android:ellipsize="end"
android:maxLines="1"
android:layout_below="@id/folder_name_view"
android:textSize="14sp"
android:layout_alignParentStart="true" />
android:textSize="14sp" />
<TextView
tools:text="Additional information"
@@ -59,11 +61,18 @@
android:layout_height="wrap_content"
android:ellipsize="end"
android:maxLines="1"
android:layout_below="@id/folder_lastmod_info"
android:textSize="14sp"
android:layout_alignParentStart="true" />
android:textSize="14sp" />
<TextView
android:visibility="@{TextUtils.isEmpty(info2) ? View.GONE : View.VISIBLE}"
tools:text="Index Update Progress"
android:text="@{info2}"
android:layout_width="match_parent"
android:layout_height="wrap_content"
android:ellipsize="end"
android:maxLines="1"
android:textSize="14sp" />
</RelativeLayout>
</LinearLayout>
</layout>
</layout>
+8
View File
@@ -44,6 +44,13 @@
<string name="settings_shutdown_delay_summary">Time before shuting down the Syncthing client after its last usage</string>
<string name="settings_crash_handler_title">Custom Crash-Handler</string>
<string name="settings_crash_handler_summary">Copy the error message to the clipboard when the App crashes</string>
<string name="settings_detailed_exception_title">Enable more detailed crash reports</string>
<string name="settings_detailed_exception_summary">
This could leak private data.
You should only use it with the custom crash handler and should not send a crash report without review.
Changes of this need an App restart (use force stop to be safe).
</string>
<string name="settings_force_stop">Force stop this App</string>
<string name="device_id_dialog_title">Enter Device ID</string>
<string name="settings_shutdown_delay_10_seconds">10 seconds</string>
<string name="settings_shutdown_delay_30_seconds">30 seconds</string>
@@ -56,4 +63,5 @@
This does not apply to local discovery connections.
</string>
<string name="dialog_file_save_as">Save as</string>
<string name="pending_index_updates">%d index updates pending</string>
</resources>
+9
View File
@@ -29,10 +29,19 @@
android:title="@string/settings_crash_handler_title"
android:summary="@string/settings_crash_handler_summary" />
<CheckBoxPreference
android:key="detailed_exception"
android:title="@string/settings_detailed_exception_title"
android:summary="@string/settings_detailed_exception_summary" />
<Preference
android:key="app_version"
android:title="@string/settings_app_version_title"/>
<Preference
android:key="force_stop"
android:title="@string/settings_force_stop" />
</PreferenceCategory>
</PreferenceScreen>
@@ -18,6 +18,7 @@ import com.google.protobuf.ByteString
import kotlinx.coroutines.*
import kotlinx.coroutines.channels.Channel
import net.syncthing.java.bep.connectionactor.ConnectionActorWrapper
import net.syncthing.java.bep.index.IndexHandler
import net.syncthing.java.bep.utils.longSumBy
import net.syncthing.java.core.beans.BlockInfo
import net.syncthing.java.core.beans.FileBlocks
@@ -17,10 +17,19 @@ package net.syncthing.java.bep
import com.google.protobuf.ByteString
import kotlinx.coroutines.GlobalScope
import kotlinx.coroutines.async
import kotlinx.coroutines.channels.consumeEach
import kotlinx.coroutines.launch
import kotlinx.coroutines.runBlocking
import net.syncthing.java.bep.BlockExchangeProtos.Vector
import net.syncthing.java.bep.connectionactor.ConnectionActorWrapper
import net.syncthing.java.core.beans.*
import net.syncthing.java.bep.index.FolderStatsUpdateCollector
import net.syncthing.java.bep.index.IndexElementProcessor
import net.syncthing.java.bep.index.IndexHandler
import net.syncthing.java.bep.index.IndexMessageProcessor
import net.syncthing.java.core.beans.BlockInfo
import net.syncthing.java.core.beans.DeviceId
import net.syncthing.java.core.beans.FileInfo.Version
import net.syncthing.java.core.beans.FolderStats
import net.syncthing.java.core.utils.BlockUtils
import net.syncthing.java.core.utils.NetworkUtils
import org.apache.commons.io.IOUtils
@@ -46,7 +55,7 @@ class BlockPusher(private val localDeviceId: DeviceId,
private val logger = LoggerFactory.getLogger(javaClass)
suspend fun pushDelete(folderId: String, targetPath: String): BlockExchangeProtos.IndexUpdate {
val fileInfo = indexHandler.waitForRemoteIndexAcquired(connectionHandler).getFileInfoByPath(folderId, targetPath)!!
val fileInfo = indexHandler.waitForRemoteIndexAcquiredWithTimeout(connectionHandler).getFileInfoByPath(folderId, targetPath)!!
NetworkUtils.assertProtocol(connectionHandler.hasFolder(fileInfo.folder), {"supplied connection handler $connectionHandler will not share folder ${fileInfo.folder}"})
return sendIndexUpdate(folderId, BlockExchangeProtos.FileInfo.newBuilder()
.setName(targetPath)
@@ -62,7 +71,7 @@ class BlockPusher(private val localDeviceId: DeviceId,
}
suspend fun pushFile(inputStream: InputStream, folderId: String, targetPath: String): FileUploadObserver {
val fileInfo = indexHandler.waitForRemoteIndexAcquired(connectionHandler).getFileInfoByPath(folderId, targetPath)
val fileInfo = indexHandler.waitForRemoteIndexAcquiredWithTimeout(connectionHandler).getFileInfoByPath(folderId, targetPath)
NetworkUtils.assertProtocol(connectionHandler.hasFolder(folderId), {"supplied connection handler $connectionHandler will not share folder $folderId"})
assert(fileInfo == null || fileInfo.folder == folderId)
assert(fileInfo == null || fileInfo.path == targetPath)
@@ -99,20 +108,22 @@ class BlockPusher(private val localDeviceId: DeviceId,
}
logger.debug("send index update for file = {}", targetPath)
val indexListener = { folderInfo: FolderInfo, newRecords: List<FileInfo>, _: IndexInfo ->
if (folderInfo.folderId == folderId) {
for (fileInfo2 in newRecords) {
if (fileInfo2.path == targetPath && fileInfo2.hash == dataSource.getHash()) { //TODO check not invalid
// sentBlocks.addAll(dataSource.getHashes());
isCompleted.set(true)
synchronized(updateLock) {
updateLock.notifyAll()
val indexListenerStream = indexHandler.subscribeToOnIndexRecordAcquiredEvents()
GlobalScope.launch {
indexListenerStream.consumeEach { (indexFolderId, newRecords, _) ->
if (indexFolderId == folderId) {
for (fileInfo2 in newRecords) {
if (fileInfo2.path == targetPath && fileInfo2.hash == dataSource.getHash()) { //TODO check not invalid
// sentBlocks.addAll(dataSource.getHashes());
isCompleted.set(true)
synchronized(updateLock) {
updateLock.notifyAll()
}
}
}
}
}
}
indexHandler.registerOnIndexRecordAcquiredListener(indexListener)
val indexUpdate = sendIndexUpdate(folderId, BlockExchangeProtos.FileInfo.newBuilder()
.setName(targetPath)
.setSize(fileSize)
@@ -128,9 +139,27 @@ class BlockPusher(private val localDeviceId: DeviceId,
override fun close() {
logger.debug("closing upload process")
monitoringProcessExecutorService.shutdown()
indexHandler.unregisterOnIndexRecordAcquiredListener(indexListener)
indexListenerStream.cancel()
requestHandlerRegistry.unregisterListener(requestFilter)
val fileInfo1 = indexHandler.pushRecord(indexUpdate.folder, indexUpdate.filesList.single())
val (fileInfo1, folderStatsUpdate) = indexHandler.indexRepository.runInTransaction {
val folderStatsUpdateCollector = FolderStatsUpdateCollector(folderId)
// TODO: notify the IndexBrowsers again (as it was earlier)
val fileInfo = IndexElementProcessor.pushRecord(
it,
indexUpdate.folder,
indexUpdate.filesList.single(),
folderStatsUpdateCollector,
it.findFileInfo(folderId, indexUpdate.filesList.single().name)
)
IndexMessageProcessor.handleFolderStatsUpdate(it, folderStatsUpdateCollector)
val folderStatsUpdate = it.findFolderStats(folderId) ?: FolderStats.createDummy(folderId)
fileInfo to folderStatsUpdate
}
runBlocking { indexHandler.sendFolderStatsUpdate(folderStatsUpdate) }
logger.info("sent file info record = {}", fileInfo1)
}
@@ -151,7 +180,7 @@ class BlockPusher(private val localDeviceId: DeviceId,
private suspend fun sendIndexUpdate(folderId: String, fileInfoBuilder: BlockExchangeProtos.FileInfo.Builder,
oldVersions: Iterable<Version>?): BlockExchangeProtos.IndexUpdate {
run {
val nextSequence = indexHandler.sequencer().nextSequence()
val nextSequence = indexHandler.getNextSequenceNumber()
val list = oldVersions ?: emptyList()
logger.debug("version list = {}", list)
val id = ByteBuffer.wrap(localDeviceId.toHashData()).long
@@ -1,58 +0,0 @@
/*
* Copyright (C) 2016 Davide Imbriaco
*
* This Java file is subject to the terms of the Mozilla Public
* License, v. 2.0. If a copy of the MPL was not distributed with this
* file, You can obtain one at http://mozilla.org/MPL/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 net.syncthing.java.bep
import net.syncthing.java.core.beans.FolderInfo
import net.syncthing.java.core.beans.FolderStats
import net.syncthing.java.core.interfaces.IndexRepository
import java.io.Closeable
class FolderBrowser internal constructor(private val indexHandler: IndexHandler) : Closeable {
private val folderStatsCache = mutableMapOf<String, FolderStats>()
private val indexRepositoryEventListener = { event: IndexRepository.FolderStatsUpdatedEvent ->
addFolderStats(event.getFolderStats())
}
fun folderInfoAndStatsList(): List<Pair<FolderInfo, FolderStats>> =
indexHandler.folderInfoList()
.map { folderInfo -> Pair(folderInfo, getFolderStats(folderInfo.folderId)) }
.sortedBy { it.first.label }
init {
indexHandler.indexRepository.setOnFolderStatsUpdatedListener(indexRepositoryEventListener)
addFolderStats(indexHandler.indexRepository.findAllFolderStats())
}
private fun addFolderStats(folderStatsList: List<FolderStats>) {
for (folderStats in folderStatsList) {
folderStatsCache.put(folderStats.folderId, folderStats)
}
}
fun getFolderStats(folder: String): FolderStats {
return folderStatsCache[folder] ?: let {
FolderStats.Builder()
.setFolder(folder)
.build()
}
}
fun getFolderInfo(folder: String): FolderInfo? {
return indexHandler.getFolderInfo(folder)
}
override fun close() {
indexHandler.indexRepository.setOnFolderStatsUpdatedListener(null)
}
}
@@ -1,184 +0,0 @@
/*
* Copyright (C) 2016 Davide Imbriaco
*
* This Java file is subject to the terms of the Mozilla Public
* License, v. 2.0. If a copy of the MPL was not distributed with this
* file, You can obtain one at http://mozilla.org/MPL/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 net.syncthing.java.bep
import net.syncthing.java.core.beans.FileInfo
import net.syncthing.java.core.interfaces.IndexRepository
import net.syncthing.java.core.utils.PathUtils
import net.syncthing.java.core.utils.awaitTerminationSafe
import net.syncthing.java.core.utils.submitLogging
import org.apache.commons.lang3.StringUtils
import org.slf4j.LoggerFactory
import java.io.Closeable
import java.util.*
import java.util.concurrent.Executors
class IndexBrowser internal constructor(private val indexRepository: IndexRepository, private val indexHandler: IndexHandler,
val folder: String, private val includeParentInList: Boolean = false,
private val allowParentInRoot: Boolean = false, ordering: Comparator<FileInfo>?) : Closeable {
private fun isParent(fileInfo: FileInfo) = PathUtils.isParent(fileInfo.path)
val ALPHA_ASC_DIR_FIRST: Comparator<FileInfo> =
compareBy<FileInfo>({!isParent(it)}, {!it.isDirectory()})
.thenBy { it.fileName.toLowerCase() }
val LAST_MOD_DESC: Comparator<FileInfo> =
compareBy<FileInfo>({!isParent(it)}, {it.lastModified})
.thenBy { it.fileName.toLowerCase() }
private val ordering = ordering ?: ALPHA_ASC_DIR_FIRST
private val logger = LoggerFactory.getLogger(javaClass)
var currentPath: String = PathUtils.ROOT_PATH
private set
private val PARENT_FILE_INFO: FileInfo
private val ROOT_FILE_INFO: FileInfo
private val executorService = Executors.newSingleThreadScheduledExecutor()
private val preloadJobs = mutableSetOf<String>()
private val preloadJobsLock = Any()
private var mOnPathChangedListener: (() -> Unit)? = null
private fun isCacheReady(): Boolean {
synchronized(preloadJobsLock) {
return preloadJobs.isEmpty()
}
}
internal fun onIndexChangedevent(folder: String) {
if (folder == this.folder) {
preloadFileInfoForCurrentPath()
}
}
fun currentPathInfo(): FileInfo = getFileInfoByAbsolutePath(currentPath)
fun currentPathFileName(): String? = PathUtils.getFileName(currentPath)
fun isRoot(): Boolean = PathUtils.isRoot(currentPath)
init {
assert(folder.isNotEmpty())
PARENT_FILE_INFO = FileInfo(folder = folder, type = FileInfo.FileType.DIRECTORY, path = PathUtils.PARENT_PATH)
ROOT_FILE_INFO = FileInfo(folder = folder, type = FileInfo.FileType.DIRECTORY, path = PathUtils.ROOT_PATH)
navigateToAbsolutePath(PathUtils.ROOT_PATH)
}
fun setOnFolderChangedListener(onPathChangedListener: (() -> Unit)?) {
mOnPathChangedListener = onPathChangedListener
}
private fun preloadFileInfoForCurrentPath() {
logger.debug("trigger preload for folder = '{}'", folder)
synchronized(preloadJobsLock) {
currentPath.let<String, Any> { currentPath ->
if (preloadJobs.contains(currentPath)) {
preloadJobs.remove(currentPath)
preloadJobs.add(currentPath) ///add last
} else {
preloadJobs.add(currentPath)
executorService.submitLogging(object : Runnable {
override fun run() {
val preloadPath =
synchronized(preloadJobsLock) {
assert(!preloadJobs.isEmpty())
preloadJobs.last() //pop last job
}
logger.info("folder preload BEGIN for folder = '{}' path = '{}'", folder, preloadPath)
getFileInfoByAbsolutePath(preloadPath)
if (!PathUtils.isRoot(preloadPath)) {
val parent = PathUtils.getParentPath(preloadPath)
getFileInfoByAbsolutePath(parent)
listFiles(parent)
}
for (record in listFiles(preloadPath)) {
if (record.path == PARENT_FILE_INFO.path && record.isDirectory()) {
listFiles(record.path)
}
}
logger.info("folder preload END for folder = '{}' path = '{}'", folder, preloadPath)
synchronized(preloadJobsLock) {
preloadJobs.remove(preloadPath)
if (isCacheReady()) {
logger.info("cache ready, notify listeners")
mOnPathChangedListener?.invoke()
} else {
logger.info("still {} job[s] left in cache loader", preloadJobs.size)
executorService.submitLogging(this)
}
}
}
})
}
}
}
}
fun listFiles(path: String = currentPath): List<FileInfo> {
logger.debug("doListFiles for path = '{}' BEGIN", path)
val list = ArrayList(indexRepository.findNotDeletedFilesByFolderAndParent(folder, path))
logger.debug("doListFiles for path = '{}' : {} records loaded)", path, list.size)
if (includeParentInList && (!PathUtils.isRoot(path) || allowParentInRoot)) {
list.add(0, PARENT_FILE_INFO)
}
return list.sortedWith(ordering)
}
fun getFileInfoByAbsolutePath(path: String): FileInfo {
return if (PathUtils.isRoot(path)) {
ROOT_FILE_INFO
} else {
logger.debug("doGetFileInfoByAbsolutePath for path = '{}' BEGIN", path)
val fileInfo = indexRepository.findNotDeletedFileInfo(folder, path) ?: error("file not found for path = $path")
logger.debug("doGetFileInfoByAbsolutePath for path = '{}' END", path)
fileInfo
}
}
fun navigateTo(fileInfo: FileInfo) {
assert(fileInfo.isDirectory())
assert(fileInfo.folder == folder)
return if (fileInfo.path == PARENT_FILE_INFO.path)
navigateToAbsolutePath(PathUtils.getParentPath(currentPath))
else
navigateToAbsolutePath(fileInfo.path)
}
fun navigateToNearestPath(oldPath: String) {
if (!StringUtils.isBlank(oldPath)) {
navigateToAbsolutePath(oldPath)
}
}
private fun navigateToAbsolutePath(newPath: String) {
if (PathUtils.isRoot(newPath)) {
currentPath = PathUtils.ROOT_PATH
} else {
val fileInfo = getFileInfoByAbsolutePath(newPath)
assert(fileInfo.isDirectory(), {"cannot navigate to path ${fileInfo.path}: not a directory"})
currentPath = fileInfo.path
}
logger.info("navigate to path = '{}'", currentPath)
preloadFileInfoForCurrentPath()
}
override fun close() {
logger.info("closing")
indexHandler.unregisterIndexBrowser(this)
executorService.shutdown()
executorService.awaitTerminationSafe()
}
}
@@ -1,451 +0,0 @@
/*
* Copyright (C) 2016 Davide Imbriaco
*
* This Java file is subject to the terms of the Mozilla Public
* License, v. 2.0. If a copy of the MPL was not distributed with this
* file, You can obtain one at http://mozilla.org/MPL/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 net.syncthing.java.bep
import net.syncthing.java.bep.connectionactor.ClusterConfigInfo
import net.syncthing.java.bep.connectionactor.ConnectionActorWrapper
import net.syncthing.java.core.beans.*
import net.syncthing.java.core.beans.FileInfo.Version
import net.syncthing.java.core.configuration.Configuration
import net.syncthing.java.core.interfaces.IndexRepository
import net.syncthing.java.core.interfaces.Sequencer
import net.syncthing.java.core.interfaces.TempRepository
import net.syncthing.java.core.utils.NetworkUtils
import net.syncthing.java.core.utils.awaitTerminationSafe
import net.syncthing.java.core.utils.trySubmitLogging
import org.apache.commons.lang3.tuple.Pair
import org.bouncycastle.util.encoders.Hex
import org.slf4j.LoggerFactory
import java.io.Closeable
import java.io.IOException
import java.util.*
import java.util.concurrent.Executors
class IndexHandler(private val configuration: Configuration, val indexRepository: IndexRepository,
private val tempRepository: TempRepository) : Closeable {
private val logger = LoggerFactory.getLogger(javaClass)
private val folderInfoByFolder = mutableMapOf<String, FolderInfo>()
private val indexMessageProcessor = IndexMessageProcessor()
private var lastIndexActivity: Long = 0
private val writeAccessLock = Object()
private val indexWaitLock = Object()
private val indexBrowsers = mutableSetOf<IndexBrowser>()
private val onIndexRecordAcquiredListeners = mutableSetOf<(FolderInfo, List<FileInfo>, IndexInfo) -> Unit>()
private val onFullIndexAcquiredListeners = mutableSetOf<(FolderInfo) -> Unit>()
private fun lastActive(): Long = System.currentTimeMillis() - lastIndexActivity
fun sequencer(): Sequencer = indexRepository.getSequencer()
fun folderList(): List<String> = folderInfoByFolder.keys.toList()
fun folderInfoList(): List<FolderInfo> = folderInfoByFolder.values.toList()
private fun markActive() {
lastIndexActivity = System.currentTimeMillis()
}
fun registerOnIndexRecordAcquiredListener(listener: (FolderInfo, List<FileInfo>, IndexInfo) -> Unit) {
onIndexRecordAcquiredListeners.add(listener)
}
fun unregisterOnIndexRecordAcquiredListener(listener: (FolderInfo, List<FileInfo>, IndexInfo) -> Unit) {
assert(onIndexRecordAcquiredListeners.contains(listener))
onIndexRecordAcquiredListeners.remove(listener)
}
fun registerOnFullIndexAcquiredListenersListener(listener: (FolderInfo) -> Unit) {
onFullIndexAcquiredListeners.add(listener)
}
fun unregisterOnFullIndexAcquiredListenersListener(listener: (FolderInfo) -> Unit) {
assert(onFullIndexAcquiredListeners.contains(listener))
onFullIndexAcquiredListeners.remove(listener)
}
init {
loadFolderInfoFromConfig()
}
private fun loadFolderInfoFromConfig() {
synchronized(writeAccessLock) {
for (folderInfo in configuration.folders) {
folderInfoByFolder.put(folderInfo.folderId, folderInfo) //TODO reference 'folder info' repository
}
}
}
@Synchronized
fun clearIndex() {
synchronized(writeAccessLock) {
indexRepository.clearIndex()
folderInfoByFolder.clear()
loadFolderInfoFromConfig()
}
}
internal fun isRemoteIndexAcquired(clusterConfigInfo: ClusterConfigInfo, peerDeviceId: DeviceId): Boolean {
var ready = true
for (folder in clusterConfigInfo.sharedFolderIds) {
val indexSequenceInfo = indexRepository.findIndexInfoByDeviceAndFolder(peerDeviceId, folder)
if (indexSequenceInfo == null || indexSequenceInfo.localSequence < indexSequenceInfo.maxSequence) {
logger.debug("waiting for index on folder = {} sequenceInfo = {}", folder, indexSequenceInfo)
ready = false
}
}
return ready
}
@Throws(InterruptedException::class)
fun waitForRemoteIndexAcquired(connectionHandler: ConnectionActorWrapper, timeoutSecs: Long? = null): IndexHandler {
val timeoutMillis = (timeoutSecs ?: DEFAULT_INDEX_TIMEOUT) * 1000
synchronized(indexWaitLock) {
while (!isRemoteIndexAcquired(connectionHandler.getClusterConfig(), connectionHandler.deviceId)) {
indexWaitLock.wait(timeoutMillis)
NetworkUtils.assertProtocol(/* TODO connectionHandler.getLastActive() < timeoutMillis || */ lastActive() < timeoutMillis,
{"unable to acquire index from connection $connectionHandler, timeout reached!"})
}
}
logger.debug("acquired all indexes on connection {}", connectionHandler)
return this
}
fun handleClusterConfigMessageProcessedEvent(clusterConfig: BlockExchangeProtos.ClusterConfig) {
synchronized(writeAccessLock) {
for (folderRecord in clusterConfig.foldersList) {
val folder = folderRecord.id
val folderInfo = updateFolderInfo(folder, folderRecord.label)
logger.debug("acquired folder info from cluster config = {}", folderInfo)
for (deviceRecord in folderRecord.devicesList) {
val deviceId = DeviceId.fromHashData(deviceRecord.id.toByteArray())
if (deviceRecord.hasIndexId() && deviceRecord.hasMaxSequence()) {
val folderIndexInfo = updateIndexInfo(folder, deviceId, deviceRecord.indexId, deviceRecord.maxSequence, null)
logger.debug("acquired folder index info from cluster config = {}", folderIndexInfo)
}
}
}
}
}
internal fun handleIndexMessageReceivedEvent(folderId: String, filesList: List<BlockExchangeProtos.FileInfo>, clusterConfigInfo: ClusterConfigInfo, peerDeviceId: DeviceId) {
indexMessageProcessor.handleIndexMessageReceivedEvent(folderId, filesList, clusterConfigInfo, peerDeviceId)
}
fun pushRecord(folder: String, bepFileInfo: BlockExchangeProtos.FileInfo): FileInfo? {
var fileBlocks: FileBlocks? = null
val builder = FileInfo.Builder()
.setFolder(folder)
.setPath(bepFileInfo.name)
.setLastModified(Date(bepFileInfo.modifiedS * 1000 + bepFileInfo.modifiedNs / 1000000))
.setVersionList((if (bepFileInfo.hasVersion()) bepFileInfo.version.countersList else null ?: emptyList()).map { record -> Version(record.id, record.value) })
.setDeleted(bepFileInfo.deleted)
when (bepFileInfo.type) {
BlockExchangeProtos.FileInfoType.FILE -> {
fileBlocks = FileBlocks(folder, builder.getPath()!!, ((bepFileInfo.blocksList ?: emptyList())).map { record ->
BlockInfo(record.offset, record.size, Hex.toHexString(record.hash.toByteArray()))
})
builder
.setTypeFile()
.setHash(fileBlocks.hash)
.setSize(bepFileInfo.size)
}
BlockExchangeProtos.FileInfoType.DIRECTORY -> builder.setTypeDir()
else -> {
logger.warn("unsupported file type = {}, discarding file info", bepFileInfo.type)
return null
}
}
return addRecord(builder.build(), fileBlocks)
}
private fun updateIndexInfo(folder: String, deviceId: DeviceId, indexId: Long?, maxSequence: Long?, localSequence: Long?): IndexInfo {
synchronized(writeAccessLock) {
var indexSequenceInfo = indexRepository.findIndexInfoByDeviceAndFolder(deviceId, folder)
var shouldUpdate = false
val builder: IndexInfo.Builder
if (indexSequenceInfo == null) {
shouldUpdate = true
assert(indexId != null, {"index sequence info not found, and supplied null index id (folder = $folder, device = $deviceId)"})
builder = IndexInfo.newBuilder()
.setFolder(folder)
.setDeviceId(deviceId.deviceId)
.setIndexId(indexId!!)
.setLocalSequence(0)
.setMaxSequence(-1)
} else {
builder = indexSequenceInfo.copyBuilder()
}
if (indexId != null && indexId != builder.getIndexId()) {
shouldUpdate = true
builder.setIndexId(indexId)
}
if (maxSequence != null && maxSequence > builder.getMaxSequence()) {
shouldUpdate = true
builder.setMaxSequence(maxSequence)
}
if (localSequence != null && localSequence > builder.getLocalSequence()) {
shouldUpdate = true
builder.setLocalSequence(localSequence)
}
if (shouldUpdate) {
indexSequenceInfo = builder.build()
indexRepository.updateIndexInfo(indexSequenceInfo)
}
return indexSequenceInfo!!
}
}
private fun addRecord(record: FileInfo, fileBlocks: FileBlocks?): FileInfo? {
synchronized(writeAccessLock) {
val lastModified = indexRepository.findFileInfoLastModified(record.folder, record.path)
return if (lastModified != null && record.lastModified < lastModified) {
logger.trace("discarding record = {}, modified before local record", record)
null
} else {
indexRepository.updateFileInfo(record, fileBlocks)
logger.trace("loaded new record = {}", record)
indexBrowsers.forEach {
it.onIndexChangedevent(record.folder)
}
record
}
}
}
fun getFileInfoByPath(folder: String, path: String): FileInfo? {
return indexRepository.findFileInfo(folder, path)
}
fun getFileInfoAndBlocksByPath(folder: String, path: String): Pair<FileInfo, FileBlocks>? {
val fileInfo = getFileInfoByPath(folder, path)
return if (fileInfo == null) {
null
} else {
assert(fileInfo.isFile())
val fileBlocks = indexRepository.findFileBlocks(folder, path)
checkNotNull(fileBlocks) {"file blocks not found for file info = $fileInfo"}
FileInfo.checkBlocks(fileInfo, fileBlocks)
Pair.of(fileInfo, fileBlocks)
}
}
private fun updateFolderInfo(folder: String, label: String?): FolderInfo {
var folderInfo: FolderInfo? = folderInfoByFolder[folder]
if (folderInfo == null || label.isNullOrEmpty()) {
folderInfo = FolderInfo(folder, label)
folderInfoByFolder.put(folderInfo.folderId, folderInfo)
}
return folderInfo
}
fun getFolderInfo(folder: String): FolderInfo? {
return folderInfoByFolder[folder]
}
fun getIndexInfo(device: DeviceId, folder: String): IndexInfo? {
return indexRepository.findIndexInfoByDeviceAndFolder(device, folder)
}
fun newFolderBrowser(): FolderBrowser {
return FolderBrowser(this)
}
fun newIndexBrowser(folder: String, includeParentInList: Boolean = false, allowParentInRoot: Boolean = false,
ordering: Comparator<FileInfo>? = null): IndexBrowser {
val indexBrowser = IndexBrowser(indexRepository, this, folder, includeParentInList, allowParentInRoot, ordering)
indexBrowsers.add(indexBrowser)
return indexBrowser
}
internal fun unregisterIndexBrowser(indexBrowser: IndexBrowser) {
assert(indexBrowsers.contains(indexBrowser))
indexBrowsers.remove(indexBrowser)
}
override fun close() {
assert(indexBrowsers.isEmpty())
assert(onIndexRecordAcquiredListeners.isEmpty())
assert(onFullIndexAcquiredListeners.isEmpty())
indexMessageProcessor.stop()
}
private inner class IndexMessageProcessor {
private val executorService = Executors.newSingleThreadExecutor()
private var queuedMessages = 0
private var queuedRecords: Long = 0
// private long lastRecordProcessingTime = 0;
// , delay = 0;
// private boolean addProcessingDelayForInterface = true;
// private final int MIN_DELAY = 0, MAX_DELAY = 5000, MAX_RECORD_PER_PROCESS = 16, DELAY_FACTOR = 1;
private var startTime: Long? = null
fun handleIndexMessageReceivedEvent(folderId: String, filesList: List<BlockExchangeProtos.FileInfo>, clusterConfigInfo: ClusterConfigInfo, peerDeviceId: DeviceId) {
logger.info("received index message event, preparing (queued records = {} event record count = {})", queuedRecords, filesList.size)
markActive()
// List<BlockExchangeProtos.FileInfo> fileList = event.getFilesList();
// for (int index = 0; index < fileList.size(); index += MAX_RECORD_PER_PROCESS) {
// BlockExchangeProtos.IndexUpdate data = BlockExchangeProtos.IndexUpdate.newBuilder()
// .addAllFiles(Iterables.limit(Iterables.skip(fileList, index), MAX_RECORD_PER_PROCESS))
// .setFolder(event.getFolder())
// .build();
// if (queuedMessages > 0) {
// storeAndProcessBg(data, clusterConfigInfo, peerDeviceId);
// } else {
// processBg(data, clusterConfigInfo, peerDeviceId);
// }
// }
val data = BlockExchangeProtos.IndexUpdate.newBuilder()
.addAllFiles(filesList)
.setFolder(folderId)
.build()
if (queuedMessages > 0) {
storeAndProcessBg(data, clusterConfigInfo, peerDeviceId)
} else {
processBg(data, clusterConfigInfo, peerDeviceId)
}
}
private fun processBg(data: BlockExchangeProtos.IndexUpdate, clusterConfigInfo: ClusterConfigInfo?, peerDeviceId: DeviceId) {
logger.debug("received index message event, queuing for processing")
queuedMessages++
queuedRecords += data.filesCount.toLong()
executorService.trySubmitLogging(object : ProcessingRunnable() {
override fun runProcess() {
doHandleIndexMessageReceivedEvent(data, clusterConfigInfo, peerDeviceId)
}
})
}
private fun storeAndProcessBg(data: BlockExchangeProtos.IndexUpdate, clusterConfigInfo: ClusterConfigInfo?, peerDeviceId: DeviceId) {
val key = tempRepository.pushTempData(data.toByteArray())
logger.debug("received index message event, stored to temp record {}, queuing for processing", key)
queuedMessages++
queuedRecords += data.filesCount.toLong()
executorService.trySubmitLogging(object : ProcessingRunnable() {
override fun runProcess() {
try {
doHandleIndexMessageReceivedEvent(key, clusterConfigInfo, peerDeviceId)
} catch (ex: IOException) {
logger.error("error processing index message", ex)
}
}
})
}
private abstract inner class ProcessingRunnable : Runnable {
override fun run() {
startTime = System.currentTimeMillis()
runProcess()
queuedMessages--
// lastRecordProcessingTime = stopwatch.elapsed(TimeUnit.MILLISECONDS) - delay;
// logger.info("processed a bunch of records, {}*{} remaining", queuedMessages, MAX_RECORD_PER_PROCESS);
// logger.debug("processed index message in {} secs", lastRecordProcessingTime / 1000d);
startTime = null
}
protected abstract fun runProcess()
// private boolean isVersionOlderThanSequence(BlockExchangeProtos.FileInfo fileInfo, long localSequence) {
// long fileSequence = fileInfo.getSequence();
// //TODO should we check last version instead of sequence? verify
// return fileSequence < localSequence;
// }
@Throws(IOException::class)
protected fun doHandleIndexMessageReceivedEvent(key: String, clusterConfigInfo: ClusterConfigInfo?, peerDeviceId: DeviceId) {
logger.debug("processing index message event from temp record {}", key)
markActive()
val data = tempRepository.popTempData(key)
val message = BlockExchangeProtos.IndexUpdate.parseFrom(data)
doHandleIndexMessageReceivedEvent(message, clusterConfigInfo, peerDeviceId)
}
protected fun doHandleIndexMessageReceivedEvent(message: BlockExchangeProtos.IndexUpdate, clusterConfigInfo: ClusterConfigInfo?, peerDeviceId: DeviceId) {
// synchronized (writeAccessLock) {
// if (addProcessingDelayForInterface) {
// delay = Math.min(MAX_DELAY, Math.max(MIN_DELAY, lastRecordProcessingTime * DELAY_FACTOR));
// logger.info("add delay of {} secs before processing index message (to allow UI to process)", delay / 1000d);
// try {
// Thread.sleep(delay);
// } catch (InterruptedException ex) {
// logger.warn("interrupted", ex);
// }
// } else {
// delay = 0;
// }
logger.info("processing index message with {} records (queue size: messages = {} records = {})", message.filesCount, queuedMessages, queuedRecords)
// String deviceId = connectionHandler.getDeviceId();
val folderId = message.folder
var sequence: Long = -1
val newRecords = mutableListOf<FileInfo>()
// IndexInfo oldIndexInfo = indexRepository.findIndexInfoByDeviceAndFolder(deviceId, folder);
// Stopwatch stopwatch = Stopwatch.createStarted();
logger.debug("processing {} index records for folder {}", message.filesList.size, folderId)
for (fileInfo in message.filesList) {
markActive()
// if (oldIndexInfo != null && isVersionOlderThanSequence(fileInfo, oldIndexInfo.getLocalSequence())) {
// logger.trace("skipping file {}, version older than sequence {}", fileInfo, oldIndexInfo.getLocalSequence());
// } else {
val newRecord = pushRecord(folderId, fileInfo)
if (newRecord != null) {
newRecords.add(newRecord)
}
sequence = Math.max(fileInfo.sequence, sequence)
markActive()
// }
}
val newIndexInfo = updateIndexInfo(folderId, peerDeviceId, null, null, sequence)
val elap = System.currentTimeMillis() - startTime!!
queuedRecords -= message.filesCount.toLong()
logger.info("processed {} index records, acquired {} ({} secs, {} record/sec)", message.filesCount, newRecords.size, elap / 1000.0, Math.round(message.filesCount / (elap / 1000.0) * 100) / 100.0)
if (logger.isInfoEnabled && newRecords.size <= 10) {
for (fileInfo in newRecords) {
logger.info("acquired record = {}", fileInfo)
}
}
val folderInfo = folderInfoByFolder[folderId]
if (!newRecords.isEmpty()) {
onIndexRecordAcquiredListeners.forEach { it(folderInfo!!, newRecords, newIndexInfo) }
}
logger.debug("index info = {}", newIndexInfo)
if (isRemoteIndexAcquired(clusterConfigInfo!!, peerDeviceId)) {
logger.debug("index acquired")
onFullIndexAcquiredListeners.forEach { it(folderInfo!!)}
}
// IndexHandler.this.notifyAll();
markActive()
synchronized(indexWaitLock) {
indexWaitLock.notifyAll()
}
}
}
fun stop() {
logger.info("stopping index record processor")
executorService.shutdown()
executorService.awaitTerminationSafe()
}
}
companion object {
private const val DEFAULT_INDEX_TIMEOUT: Long = 30
}
}
@@ -16,7 +16,7 @@ package net.syncthing.java.bep.connectionactor
import com.google.protobuf.ByteString
import net.syncthing.java.bep.BlockExchangeProtos
import net.syncthing.java.bep.IndexHandler
import net.syncthing.java.bep.index.IndexHandler
import net.syncthing.java.core.beans.DeviceId
import net.syncthing.java.core.beans.FolderInfo
import net.syncthing.java.core.configuration.Configuration
@@ -32,48 +32,50 @@ object ClusterConfigHandler {
): BlockExchangeProtos.ClusterConfig {
val builder = BlockExchangeProtos.ClusterConfig.newBuilder()
for (folder in configuration.folders) {
val folderBuilder = BlockExchangeProtos.Folder.newBuilder()
.setId(folder.folderId)
.setLabel(folder.label)
indexHandler.indexRepository.runInTransaction { indexTransaction ->
for (folder in configuration.folders) {
val folderBuilder = BlockExchangeProtos.Folder.newBuilder()
.setId(folder.folderId)
.setLabel(folder.label)
// add this device
folderBuilder.addDevices(
BlockExchangeProtos.Device.newBuilder()
.setId(ByteString.copyFrom(configuration.localDeviceId.toHashData()))
.setIndexId(indexHandler.sequencer().indexId())
.setMaxSequence(indexHandler.sequencer().currentSequence())
)
// add this device
folderBuilder.addDevices(
BlockExchangeProtos.Device.newBuilder()
.setId(ByteString.copyFrom(configuration.localDeviceId.toHashData()))
.setIndexId(indexTransaction.getSequencer().indexId())
.setMaxSequence(indexTransaction.getSequencer().currentSequence())
)
// add other device
val indexSequenceInfo = indexHandler.indexRepository.findIndexInfoByDeviceAndFolder(deviceId, folder.folderId)
// add other device
val indexSequenceInfo = indexTransaction.findIndexInfoByDeviceAndFolder(deviceId, folder.folderId)
folderBuilder.addDevices(
BlockExchangeProtos.Device.newBuilder()
.setId(ByteString.copyFrom(deviceId.toHashData()))
.apply {
indexSequenceInfo?.let {
setIndexId(indexSequenceInfo.indexId)
setMaxSequence(indexSequenceInfo.localSequence)
folderBuilder.addDevices(
BlockExchangeProtos.Device.newBuilder()
.setId(ByteString.copyFrom(deviceId.toHashData()))
.apply {
indexSequenceInfo?.let {
setIndexId(indexSequenceInfo.indexId)
setMaxSequence(indexSequenceInfo.localSequence)
logger.info("send delta index info device = {} index = {} max (local) sequence = {}",
indexSequenceInfo.deviceId,
indexSequenceInfo.indexId,
indexSequenceInfo.localSequence)
logger.info("send delta index info device = {} index = {} max (local) sequence = {}",
indexSequenceInfo.deviceId,
indexSequenceInfo.indexId,
indexSequenceInfo.localSequence)
}
}
}
)
)
builder.addFolders(folderBuilder)
builder.addFolders(folderBuilder)
// TODO: add the other devices to the cluster config
// TODO: add the other devices to the cluster config
}
}
return builder.build()
}
// TODO: understand this
internal fun handleReceivedClusterConfig(
internal suspend fun handleReceivedClusterConfig(
clusterConfig: BlockExchangeProtos.ClusterConfig,
configuration: Configuration,
otherDeviceId: DeviceId,
@@ -96,12 +98,19 @@ object ClusterConfigHandler {
if (ourDevice != null) {
folderInfo = folderInfo.copy(isShared = true)
logger.info("folder shared from device = {} folder = {}", otherDeviceId, folderInfo)
val folderIds = configuration.folders.map { it.folderId }
if (!folderIds.contains(folderInfo.folderId)) {
val fi = FolderInfo(folderInfo.folderId, folderInfo.label)
configuration.folders = configuration.folders + fi
newSharedFolders.add(fi)
val newFolderInfo = FolderInfo(folderInfo.folderId, folderInfo.label)
val oldFolderEntry = configuration.folders.find { it.folderId == folderInfo.folderId }
if (oldFolderEntry == null) {
configuration.folders = configuration.folders + newFolderInfo
newSharedFolders.add(newFolderInfo)
logger.info("new folder shared = {}", folderInfo)
} else {
if (oldFolderEntry != newFolderInfo) {
configuration.folders = configuration.folders.filter { it != oldFolderEntry }.toSet() + setOf(newFolderInfo)
}
}
} else {
logger.info("folder not shared from device = {} folder = {}", otherDeviceId, folderInfo)
@@ -17,7 +17,7 @@ package net.syncthing.java.bep.connectionactor
import kotlinx.coroutines.*
import kotlinx.coroutines.channels.*
import net.syncthing.java.bep.BlockExchangeProtos
import net.syncthing.java.bep.IndexHandler
import net.syncthing.java.bep.index.IndexHandler
import net.syncthing.java.core.beans.DeviceAddress
import net.syncthing.java.core.configuration.Configuration
import org.slf4j.LoggerFactory
@@ -42,4 +42,8 @@ object ConnectionActorUtil {
deferred.await()
}
suspend fun disconnect(actor: SendChannel<ConnectionAction>) {
actor.send(CloseConnectionAction)
}
}
@@ -81,6 +81,12 @@ class ConnectionActorWrapper (
// this triggers a disconnection
// the ConnectionActorGenerator will reconnect soon
fun reconnect() {
currentConnectionActor?.close()
val actor = currentConnectionActor
GlobalScope.launch {
if (actor != null) {
ConnectionActorUtil.disconnect(actor)
}
}
}
}
@@ -22,7 +22,7 @@ import kotlinx.coroutines.channels.consumeEach
import kotlinx.coroutines.sync.Mutex
import kotlinx.coroutines.sync.withLock
import net.syncthing.java.bep.BlockExchangeProtos
import net.syncthing.java.bep.IndexHandler
import net.syncthing.java.bep.index.IndexHandler
import net.syncthing.java.core.beans.DeviceAddress
import net.syncthing.java.core.configuration.Configuration
import net.syncthing.java.core.security.KeystoreHandler
@@ -0,0 +1,113 @@
/*
* Copyright (C) 2016 Davide Imbriaco
* Copyright (C) 2018 Jonas Lochmann
*
* This Java file is subject to the terms of the Mozilla Public
* License, v. 2.0. If a copy of the MPL was not distributed with this
* file, You can obtain one at http://mozilla.org/MPL/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 net.syncthing.java.bep.folder
import kotlinx.coroutines.*
import kotlinx.coroutines.channels.ConflatedBroadcastChannel
import kotlinx.coroutines.channels.consumeEach
import kotlinx.coroutines.channels.first
import kotlinx.coroutines.channels.produce
import kotlinx.coroutines.sync.Mutex
import kotlinx.coroutines.sync.withLock
import net.syncthing.java.bep.index.IndexHandler
import net.syncthing.java.core.beans.FolderStats
import net.syncthing.java.core.configuration.Configuration
import java.io.Closeable
class FolderBrowser internal constructor(private val indexHandler: IndexHandler, private val configuration: Configuration) : Closeable {
private val job = Job()
private val foldersStatus = ConflatedBroadcastChannel<Map<String, FolderStatus>>()
init {
GlobalScope.launch (job) {
// get initial status
val currentFolderStats = mutableMapOf<String, FolderStats>()
var currentIndexInfo = withContext(Dispatchers.IO) {
indexHandler.indexRepository.runInTransaction { indexTransaction ->
configuration.folders.map { it.folderId }.forEach { folderId ->
currentFolderStats[folderId] = indexTransaction.findFolderStats(folderId) ?: FolderStats.createDummy(folderId)
}
indexTransaction.findAllIndexInfos().groupBy { it.folderId }.toMutableMap()
}
}
// send status
suspend fun dispatch() {
foldersStatus.send(
configuration.folders.map { info ->
FolderStatus(
info = info,
stats = currentFolderStats[info.folderId] ?: FolderStats.createDummy(info.folderId),
indexInfo = currentIndexInfo[info.folderId] ?: emptyList()
)
}.associateBy { it.info.folderId }
)
}
dispatch()
// handle changes
val updateLock = Mutex()
async {
indexHandler.subscribeFolderStatsUpdatedEvents().consumeEach { folderStats ->
updateLock.withLock {
currentFolderStats[folderStats.folderId] = folderStats
dispatch()
}
}
}
async {
indexHandler.subscribeToOnIndexRecordAcquiredEvents().consumeEach { event ->
updateLock.withLock {
val oldList = currentIndexInfo[event.folderId] ?: emptyList()
val newList = oldList.filter { it.deviceId != event.indexInfo.deviceId } + event.indexInfo
currentIndexInfo[event.folderId] = newList
dispatch()
}
}
}
}
}
fun folderInfoAndStatusStream() = GlobalScope.produce {
foldersStatus.openSubscription().consumeEach { folderStats ->
send(
folderStats
.values
.sortedBy { it.info.label }
)
}
}
suspend fun folderInfoAndStatusList(): List<FolderStatus> = folderInfoAndStatusStream().first()
suspend fun getFolderStatus(folder: String): FolderStatus {
return getFolderStatus(folder, foldersStatus.openSubscription().first())
}
fun getFolderStatusSync(folder: String) = runBlocking { getFolderStatus(folder) }
private fun getFolderStatus(folder: String, folderStatus: Map<String, FolderStatus>) = folderStatus[folder] ?: FolderStatus.createDummy(folder)
override fun close() {
job.cancel()
}
}
@@ -0,0 +1,29 @@
package net.syncthing.java.bep.folder
import net.syncthing.java.bep.utils.longMaxBy
import net.syncthing.java.core.beans.FolderInfo
import net.syncthing.java.core.beans.FolderStats
import net.syncthing.java.core.beans.IndexInfo
data class FolderStatus(
val info: FolderInfo,
val stats: FolderStats,
val indexInfo: List<IndexInfo>
) {
companion object {
fun createDummy(folder: String) = FolderStatus(
info = FolderInfo(folder, folder),
stats = FolderStats.createDummy(folder),
indexInfo = emptyList()
)
}
val missingIndexUpdates: Long by lazy {
Math.max(
0,
indexInfo.longMaxBy ({ it.maxSequence }, 0) -
indexInfo.longMaxBy ({ it.localSequence }, 0)
)
}
}
@@ -0,0 +1,17 @@
package net.syncthing.java.bep.index
import java.util.*
class FolderStatsUpdateCollector (val folderId: String) {
var deltaFileCount = 0L
var deltaDirCount = 0L
var deltaSize = 0L
var lastModified = Date(0)
fun isEmpty() = (
deltaFileCount == 0L &&
deltaDirCount == 0L &&
deltaSize == 0L &&
lastModified.time == 0L
)
}
@@ -0,0 +1,173 @@
package net.syncthing.java.bep.index
import net.syncthing.java.bep.BlockExchangeProtos
import net.syncthing.java.core.beans.BlockInfo
import net.syncthing.java.core.beans.FileBlocks
import net.syncthing.java.core.beans.FileInfo
import net.syncthing.java.core.interfaces.IndexTransaction
import org.bouncycastle.util.encoders.Hex
import org.slf4j.LoggerFactory
import java.io.IOException
import java.util.*
object IndexElementProcessor {
private val logger = LoggerFactory.getLogger(IndexElementProcessor::class.java)
fun pushRecords(
transaction: IndexTransaction,
folder: String,
updates: List<BlockExchangeProtos.FileInfo>,
oldRecords: Map<String, FileInfo>,
folderStatsUpdateCollector: FolderStatsUpdateCollector,
enableDetailedException: Boolean
): List<FileInfo> {
// this always keeps the last version per path
val filesToProcess = updates
.sortedBy { it.sequence }
.reversed()
.distinctBy { it.name /* this is the whole path */ }
.reversed()
val preparedUpdates = filesToProcess.mapNotNull {
try {
prepareUpdate(folder, it)
} catch (ex: Exception) {
if (enableDetailedException) {
throw IOException("error processing index update: ${it.name}", ex)
} else {
throw ex
}
}
}
val updatesToApply = preparedUpdates.filter { shouldUpdateRecord(oldRecords[it.first.path], it.first) }
transaction.updateFileInfoAndBlocks(
fileInfos = updatesToApply.map { it.first },
fileBlocks = updatesToApply.mapNotNull { it.second }
)
for ((newRecord) in updatesToApply) {
updateFolderStatsCollector(oldRecords[newRecord.path], newRecord, folderStatsUpdateCollector)
}
return updatesToApply.map { it.first }
}
fun pushRecord(
transaction: IndexTransaction,
folder: String,
bepFileInfo: BlockExchangeProtos.FileInfo,
folderStatsUpdateCollector: FolderStatsUpdateCollector,
oldRecord: FileInfo?
): FileInfo? {
val update = prepareUpdate(folder, bepFileInfo)
return if (update != null) {
addRecord(
transaction = transaction,
newRecord = update.first,
fileBlocks = update.second,
folderStatsUpdateCollector = folderStatsUpdateCollector,
oldRecord = oldRecord
)
} else {
null
}
}
private fun prepareUpdate(
folder: String,
bepFileInfo: BlockExchangeProtos.FileInfo
): Pair<FileInfo, FileBlocks?>? {
val builder = FileInfo.Builder()
.setFolder(folder)
.setPath(bepFileInfo.name)
.setLastModified(Date(bepFileInfo.modifiedS * 1000 + bepFileInfo.modifiedNs / 1000000))
.setVersionList((if (bepFileInfo.hasVersion()) bepFileInfo.version.countersList else null ?: emptyList()).map { record -> FileInfo.Version(record.id, record.value) })
.setDeleted(bepFileInfo.deleted)
var fileBlocks: FileBlocks? = null
when (bepFileInfo.type) {
BlockExchangeProtos.FileInfoType.FILE -> {
fileBlocks = FileBlocks(folder, builder.getPath()!!, ((bepFileInfo.blocksList ?: emptyList())).map { record ->
BlockInfo(record.offset, record.size, Hex.toHexString(record.hash.toByteArray()))
})
builder
.setTypeFile()
.setHash(fileBlocks.hash)
.setSize(bepFileInfo.size)
}
BlockExchangeProtos.FileInfoType.DIRECTORY -> builder.setTypeDir()
else -> {
logger.warn("unsupported file type = {}, discarding file info", bepFileInfo.type)
return null
}
}
return builder.build() to fileBlocks
}
private fun shouldUpdateRecord(
oldRecord: FileInfo?,
newRecord: FileInfo
) = oldRecord == null || newRecord.lastModified >= oldRecord.lastModified
private fun addRecord(
transaction: IndexTransaction,
newRecord: FileInfo,
oldRecord: FileInfo?,
fileBlocks: FileBlocks?,
folderStatsUpdateCollector: FolderStatsUpdateCollector
): FileInfo? {
return if (shouldUpdateRecord(oldRecord, newRecord)) {
logger.trace("discarding record = {}, modified before local record", newRecord)
null
} else {
logger.trace("loaded new record = {}", newRecord)
transaction.updateFileInfo(newRecord, fileBlocks)
updateFolderStatsCollector(oldRecord, newRecord, folderStatsUpdateCollector)
newRecord
}
}
private fun updateFolderStatsCollector(
oldRecord: FileInfo?,
newRecord: FileInfo,
folderStatsUpdateCollector: FolderStatsUpdateCollector
) {
val oldMissing = oldRecord == null || oldRecord.isDeleted
val newMissing = newRecord.isDeleted
val oldSizeMissing = oldMissing || !oldRecord!!.isFile()
val newSizeMissing = newMissing || !newRecord.isFile()
if (!oldSizeMissing) {
folderStatsUpdateCollector.deltaSize -= oldRecord!!.size!!
}
if (!newSizeMissing) {
folderStatsUpdateCollector.deltaSize += newRecord.size!!
}
if (!oldMissing) {
if (oldRecord!!.isFile()) {
folderStatsUpdateCollector.deltaFileCount--
} else if (oldRecord.isDirectory()) {
folderStatsUpdateCollector.deltaDirCount--
}
}
if (!newMissing) {
if (newRecord.isFile()) {
folderStatsUpdateCollector.deltaFileCount++
} else if (newRecord.isDirectory()) {
folderStatsUpdateCollector.deltaDirCount++
}
}
folderStatsUpdateCollector.lastModified = newRecord.lastModified
}
}
@@ -0,0 +1,191 @@
/*
* Copyright (C) 2016 Davide Imbriaco
* Copyright (C) 2018 Jonas Lochmann
*
* This Java file is subject to the terms of the Mozilla Public
* License, v. 2.0. If a copy of the MPL was not distributed with this
* file, You can obtain one at http://mozilla.org/MPL/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 net.syncthing.java.bep.index
import kotlinx.coroutines.channels.BroadcastChannel
import kotlinx.coroutines.channels.consume
import kotlinx.coroutines.withTimeoutOrNull
import net.syncthing.java.bep.BlockExchangeProtos
import net.syncthing.java.bep.connectionactor.ClusterConfigInfo
import net.syncthing.java.bep.connectionactor.ConnectionActorWrapper
import net.syncthing.java.bep.folder.FolderBrowser
import net.syncthing.java.bep.index.browser.IndexBrowser
import net.syncthing.java.core.beans.*
import net.syncthing.java.core.configuration.Configuration
import net.syncthing.java.core.interfaces.IndexRepository
import net.syncthing.java.core.interfaces.IndexTransaction
import net.syncthing.java.core.interfaces.TempRepository
import org.apache.commons.lang3.tuple.Pair
import org.slf4j.LoggerFactory
import java.io.Closeable
import java.io.IOException
data class IndexRecordAcquiredEvent(val folderId: String, val files: List<FileInfo>, val indexInfo: IndexInfo)
class IndexHandler(
configuration: Configuration,
val indexRepository: IndexRepository,
tempRepository: TempRepository,
enableDetailedException: Boolean
) : Closeable {
private val logger = LoggerFactory.getLogger(javaClass)
private val onIndexRecordAcquiredEvents = BroadcastChannel<IndexRecordAcquiredEvent>(capacity = 16)
private val onFullIndexAcquiredEvents = BroadcastChannel<String>(capacity = 16)
private val onFolderStatsUpdatedEvents = BroadcastChannel<FolderStats>(capacity = 16)
private val indexMessageProcessor = IndexMessageQueueProcessor(
indexRepository = indexRepository,
tempRepository = tempRepository,
isRemoteIndexAcquired = ::isRemoteIndexAcquired,
onIndexRecordAcquiredEvents = onIndexRecordAcquiredEvents,
onFullIndexAcquiredEvents = onFullIndexAcquiredEvents,
onFolderStatsUpdatedEvents = onFolderStatsUpdatedEvents,
enableDetailedException = enableDetailedException
)
fun subscribeToOnFullIndexAcquiredEvents() = onFullIndexAcquiredEvents.openSubscription()
fun subscribeToOnIndexRecordAcquiredEvents() = onIndexRecordAcquiredEvents.openSubscription()
fun subscribeFolderStatsUpdatedEvents() = onFolderStatsUpdatedEvents.openSubscription()
fun getNextSequenceNumber() = indexRepository.runInTransaction { it.getSequencer().nextSequence() }
fun clearIndex() {
indexRepository.runInTransaction { it.clearIndex() }
}
private fun isRemoteIndexAcquiredWithoutTransaction(clusterConfigInfo: ClusterConfigInfo, peerDeviceId: DeviceId): Boolean {
return indexRepository.runInTransaction { transaction -> isRemoteIndexAcquired(clusterConfigInfo, peerDeviceId, transaction) }
}
private fun isRemoteIndexAcquired(clusterConfigInfo: ClusterConfigInfo, peerDeviceId: DeviceId, transaction: IndexTransaction): Boolean {
return clusterConfigInfo.sharedFolderIds.find { sharedFolderId ->
// try to find one folder which is not yet ready
val indexSequenceInfo = transaction.findIndexInfoByDeviceAndFolder(peerDeviceId, sharedFolderId)
indexSequenceInfo == null || indexSequenceInfo.localSequence < indexSequenceInfo.maxSequence
} == null
}
// the old implementation kept waiting when index updates were still happening, but waiting 30 seconds should be enough
suspend fun waitForRemoteIndexAcquiredWithTimeout(connectionHandler: ConnectionActorWrapper, timeoutSecs: Long? = null): IndexHandler {
val timeoutMillis = (timeoutSecs ?: DEFAULT_INDEX_TIMEOUT) * 1000
val ok = withTimeoutOrNull(timeoutMillis) {
waitForRemoteIndexAcquiredWithoutTimeout(connectionHandler)
true
} ?: false
if (!ok) {
throw IOException("unable to acquire index from connection $connectionHandler, timeout reached!")
}
return this
}
suspend fun waitForRemoteIndexAcquiredWithoutTimeout(connectionHandler: ConnectionActorWrapper) {
val events = onFullIndexAcquiredEvents.openSubscription()
events.consume {
fun isDone() = isRemoteIndexAcquiredWithoutTransaction(connectionHandler.getClusterConfig(), connectionHandler.deviceId)
if (isDone()) {
return
}
for (event in events) {
if (isDone()) {
return
}
}
}
}
suspend fun handleClusterConfigMessageProcessedEvent(clusterConfig: BlockExchangeProtos.ClusterConfig) {
val updatedIndexInfos = indexRepository.runInTransaction { transaction ->
val updatedIndexInfos = mutableListOf<IndexInfo>()
for (folderRecord in clusterConfig.foldersList) {
val folder = folderRecord.id
logger.debug("acquired folder info from cluster config = {}", folder)
for (deviceRecord in folderRecord.devicesList) {
val deviceId = DeviceId.fromHashData(deviceRecord.id.toByteArray())
if (deviceRecord.hasIndexId() && deviceRecord.hasMaxSequence()) {
val folderIndexInfo = UpdateIndexInfo.updateIndexInfo(transaction, folder, deviceId, deviceRecord.indexId, deviceRecord.maxSequence, null)
logger.debug("acquired folder index info from cluster config = {}", folderIndexInfo)
updatedIndexInfos.add(folderIndexInfo)
}
}
}
updatedIndexInfos
}
updatedIndexInfos.forEach {
onIndexRecordAcquiredEvents.send(
IndexRecordAcquiredEvent(
folderId = it.folderId,
indexInfo = it,
files = emptyList()
)
)
}
}
internal suspend fun handleIndexMessageReceivedEvent(folderId: String, filesList: List<BlockExchangeProtos.FileInfo>, clusterConfigInfo: ClusterConfigInfo, peerDeviceId: DeviceId) {
indexMessageProcessor.handleIndexMessageReceivedEvent(folderId, filesList, clusterConfigInfo, peerDeviceId)
}
fun getFileInfoByPath(folder: String, path: String): FileInfo? {
return indexRepository.runInTransaction { it.findFileInfo(folder, path) }
}
fun getFileInfoAndBlocksByPath(folder: String, path: String): Pair<FileInfo, FileBlocks>? {
return indexRepository.runInTransaction { transaction ->
val fileInfo = transaction.findFileInfo(folder, path)
if (fileInfo == null) {
null
} else {
val fileBlocks = transaction.findFileBlocks(folder, path)
assert(fileInfo.isFile())
checkNotNull(fileBlocks) {"file blocks not found for file info = $fileInfo"}
FileInfo.checkBlocks(fileInfo, fileBlocks)
Pair.of(fileInfo, fileBlocks)
}
}
}
val folderBrowser = FolderBrowser(this, configuration)
val indexBrowser = IndexBrowser(indexRepository, this)
suspend fun sendFolderStatsUpdate(event: FolderStats) {
onFolderStatsUpdatedEvents.send(event)
}
override fun close() {
onIndexRecordAcquiredEvents.close()
onFullIndexAcquiredEvents.close()
indexMessageProcessor.stop()
}
companion object {
private const val DEFAULT_INDEX_TIMEOUT: Long = 30
}
}
@@ -0,0 +1,64 @@
package net.syncthing.java.bep.index
import net.syncthing.java.bep.BlockExchangeProtos
import net.syncthing.java.core.beans.DeviceId
import net.syncthing.java.core.beans.FileInfo
import net.syncthing.java.core.beans.FolderStats
import net.syncthing.java.core.beans.IndexInfo
import net.syncthing.java.core.interfaces.IndexTransaction
import org.slf4j.LoggerFactory
object IndexMessageProcessor {
private val logger = LoggerFactory.getLogger(IndexMessageProcessor::class.java)
fun doHandleIndexMessageReceivedEvent(
message: BlockExchangeProtos.IndexUpdate,
peerDeviceId: DeviceId,
transaction: IndexTransaction,
enableDetailedException: Boolean
): Result {
val folderId = message.folder
logger.debug("processing {} index records for folder {}", message.filesList.size, folderId)
val oldRecords = transaction.findFileInfo(folderId, message.filesList.map { it.name })
val folderStatsUpdateCollector = FolderStatsUpdateCollector(message.folder)
val newRecords = IndexElementProcessor.pushRecords(
transaction = transaction,
oldRecords = oldRecords,
folder = folderId,
folderStatsUpdateCollector = folderStatsUpdateCollector,
updates = message.filesList,
enableDetailedException = enableDetailedException
)
var sequence: Long = -1
for (newRecord in message.filesList) {
sequence = Math.max(newRecord.sequence, sequence)
}
handleFolderStatsUpdate(transaction, folderStatsUpdateCollector)
val newIndexInfo = UpdateIndexInfo.updateIndexInfo(transaction, folderId, peerDeviceId, null, null, sequence)
return Result(newIndexInfo, newRecords.toList(), transaction.findFolderStats(folderId) ?: FolderStats.createDummy(folderId))
}
fun handleFolderStatsUpdate(transaction: IndexTransaction, folderStatsUpdateCollector: FolderStatsUpdateCollector) {
if (folderStatsUpdateCollector.isEmpty()) {
return
}
transaction.updateOrInsertFolderStats(
folder = folderStatsUpdateCollector.folderId,
deltaSize = folderStatsUpdateCollector.deltaSize,
deltaFileCount = folderStatsUpdateCollector.deltaFileCount,
deltaDirCount = folderStatsUpdateCollector.deltaDirCount,
lastUpdate = folderStatsUpdateCollector.lastModified
)
}
data class Result(val newIndexInfo: IndexInfo, val updatedFiles: List<FileInfo>, val newFolderStats: FolderStats)
}
@@ -0,0 +1,148 @@
/*
* Copyright (C) 2016 Davide Imbriaco
* Copyright (C) 2018 Jonas Lochmann
*
* This Java file is subject to the terms of the Mozilla Public
* License, v. 2.0. If a copy of the MPL was not distributed with this
* file, You can obtain one at http://mozilla.org/MPL/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 net.syncthing.java.bep.index
import kotlinx.coroutines.Dispatchers
import kotlinx.coroutines.GlobalScope
import kotlinx.coroutines.Job
import kotlinx.coroutines.channels.BroadcastChannel
import kotlinx.coroutines.channels.Channel
import kotlinx.coroutines.channels.consumeEach
import kotlinx.coroutines.launch
import kotlinx.coroutines.sync.Mutex
import kotlinx.coroutines.sync.withLock
import net.syncthing.java.bep.BlockExchangeProtos
import net.syncthing.java.bep.connectionactor.ClusterConfigInfo
import net.syncthing.java.core.beans.DeviceId
import net.syncthing.java.core.beans.FolderStats
import net.syncthing.java.core.interfaces.IndexRepository
import net.syncthing.java.core.interfaces.IndexTransaction
import net.syncthing.java.core.interfaces.TempRepository
import org.slf4j.LoggerFactory
class IndexMessageQueueProcessor (
private val indexRepository: IndexRepository,
private val tempRepository: TempRepository,
private val onIndexRecordAcquiredEvents: BroadcastChannel<IndexRecordAcquiredEvent>,
private val onFullIndexAcquiredEvents: BroadcastChannel<String>,
private val onFolderStatsUpdatedEvents: BroadcastChannel<FolderStats>,
private val isRemoteIndexAcquired: (ClusterConfigInfo, DeviceId, IndexTransaction) -> Boolean,
private val enableDetailedException: Boolean
) {
private data class IndexUpdateAction(val update: BlockExchangeProtos.IndexUpdate, val clusterConfigInfo: ClusterConfigInfo, val peerDeviceId: DeviceId)
private data class StoredIndexUpdateAction(val updateId: String, val clusterConfigInfo: ClusterConfigInfo, val peerDeviceId: DeviceId)
companion object {
private val logger = LoggerFactory.getLogger(IndexMessageQueueProcessor::class.java)
private const val BATCH_SIZE = 128
}
private val job = Job()
private val indexUpdateIncomingLock = Mutex()
private val indexUpdateProcessStoredQueue = Channel<StoredIndexUpdateAction>(capacity = Channel.UNLIMITED)
private val indexUpdateProcessingQueue = Channel<IndexUpdateAction>(capacity = Channel.RENDEZVOUS)
suspend fun handleIndexMessageReceivedEvent(folderId: String, filesList: List<BlockExchangeProtos.FileInfo>, clusterConfigInfo: ClusterConfigInfo, peerDeviceId: DeviceId) {
filesList.chunked(BATCH_SIZE).forEach { chunck ->
handleIndexMessageReceivedEventWithoutChuncking(folderId, chunck, clusterConfigInfo, peerDeviceId)
}
}
suspend fun handleIndexMessageReceivedEventWithoutChuncking(folderId: String, filesList: List<BlockExchangeProtos.FileInfo>, clusterConfigInfo: ClusterConfigInfo, peerDeviceId: DeviceId) {
indexUpdateIncomingLock.withLock {
logger.info("received index message event, preparing")
val data = BlockExchangeProtos.IndexUpdate.newBuilder()
.addAllFiles(filesList)
.setFolder(folderId)
.build()
if (indexUpdateProcessingQueue.offer(IndexUpdateAction(data, clusterConfigInfo, peerDeviceId))) {
// message is beeing processed now
} else {
val key = tempRepository.pushTempData(data.toByteArray())
logger.debug("received index message event, stored to temp record {}, queuing for processing", key)
indexUpdateProcessStoredQueue.send(StoredIndexUpdateAction(key, clusterConfigInfo, peerDeviceId))
}
}
}
init {
GlobalScope.launch(Dispatchers.IO + job) {
indexUpdateProcessingQueue.consumeEach {
doHandleIndexMessageReceivedEvent(it)
}
}
GlobalScope.launch(Dispatchers.IO + job) {
indexUpdateProcessStoredQueue.consumeEach { action ->
logger.debug("processing index message event from temp record {}", action.updateId)
val data = tempRepository.popTempData(action.updateId)
val message = BlockExchangeProtos.IndexUpdate.parseFrom(data)
indexUpdateProcessingQueue.send(IndexUpdateAction(
message,
action.clusterConfigInfo,
action.peerDeviceId
))
}
}
}
private suspend fun doHandleIndexMessageReceivedEvent(action: IndexUpdateAction) {
val (message, clusterConfigInfo, peerDeviceId) = action
logger.info("processing index message with {} records", message.filesCount)
val (indexResult, wasIndexAcquired) = indexRepository.runInTransaction { indexTransaction ->
val wasIndexAcquiredBefore = isRemoteIndexAcquired(clusterConfigInfo, peerDeviceId, indexTransaction)
val startTime = System.currentTimeMillis()
val indexResult = IndexMessageProcessor.doHandleIndexMessageReceivedEvent(
message = message,
peerDeviceId = peerDeviceId,
transaction = indexTransaction,
enableDetailedException = enableDetailedException
)
val endTime = System.currentTimeMillis()
logger.info("processed {} index records, acquired {} in ${endTime - startTime} ms", message.filesCount, indexResult.updatedFiles.size)
logger.debug("index info = {}", indexResult.newIndexInfo)
indexResult to ((!wasIndexAcquiredBefore) && isRemoteIndexAcquired(clusterConfigInfo, peerDeviceId, indexTransaction))
}
if (indexResult.updatedFiles.isNotEmpty()) {
onIndexRecordAcquiredEvents.send(IndexRecordAcquiredEvent(message.folder, indexResult.updatedFiles, indexResult.newIndexInfo))
}
onFolderStatsUpdatedEvents.send(indexResult.newFolderStats)
if (wasIndexAcquired) {
logger.debug("index acquired")
onFullIndexAcquiredEvents.send(message.folder)
}
}
fun stop() {
logger.info("stopping index record processor")
job.cancel()
}
}
@@ -0,0 +1,50 @@
package net.syncthing.java.bep.index
import net.syncthing.java.core.beans.DeviceId
import net.syncthing.java.core.beans.IndexInfo
import net.syncthing.java.core.interfaces.IndexTransaction
object UpdateIndexInfo {
fun updateIndexInfo(
transaction: IndexTransaction,
folder: String,
deviceId: DeviceId,
indexId: Long?,
maxSequence: Long?,
localSequence: Long?
): IndexInfo {
val oldIndexSequenceInfo = transaction.findIndexInfoByDeviceAndFolder(deviceId, folder)
var newIndexSequenceInfo = oldIndexSequenceInfo ?: kotlin.run {
assert(indexId != null) {
"index sequence info not found, and supplied null index id (folder = $folder, device = $deviceId)"
}
IndexInfo(
folderId = folder,
deviceId = deviceId.deviceId,
indexId = indexId!!,
localSequence = 0,
maxSequence = -1
)
}
if (indexId != null && indexId != newIndexSequenceInfo.indexId) {
newIndexSequenceInfo = newIndexSequenceInfo.copy(indexId = indexId)
}
if (maxSequence != null && maxSequence > newIndexSequenceInfo.maxSequence) {
newIndexSequenceInfo = newIndexSequenceInfo.copy(maxSequence = maxSequence)
}
if (localSequence != null && localSequence > newIndexSequenceInfo.localSequence) {
newIndexSequenceInfo = newIndexSequenceInfo.copy(localSequence = localSequence)
}
if (oldIndexSequenceInfo != newIndexSequenceInfo) {
transaction.updateIndexInfo(newIndexSequenceInfo)
}
return newIndexSequenceInfo
}
}
@@ -0,0 +1,25 @@
package net.syncthing.java.bep.index.browser
import net.syncthing.java.core.beans.FileInfo
import net.syncthing.java.core.utils.PathUtils
sealed class DirectoryListing {
abstract val folder: String
abstract val path: String
}
data class DirectoryContentListing(
val directoryInfo: FileInfo,
val parentEntry: FileInfo?,
val entries: List<FileInfo>
): DirectoryListing() {
override val folder = directoryInfo.folder
override val path = directoryInfo.path
}
data class DirectoryNotFoundListing(
override val folder: String,
override val path: String
): DirectoryListing() {
val theoreticalParentPath = if (PathUtils.isRoot(path)) null else PathUtils.getParentPath(path)
}
@@ -0,0 +1,167 @@
/*
* Copyright (C) 2016 Davide Imbriaco
* Copyright (C) 2018 Jonas Lochmann
*
* This Java file is subject to the terms of the Mozilla Public
* License, v. 2.0. If a copy of the MPL was not distributed with this
* file, You can obtain one at http://mozilla.org/MPL/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 net.syncthing.java.bep.index.browser
import kotlinx.coroutines.Dispatchers
import kotlinx.coroutines.GlobalScope
import kotlinx.coroutines.channels.consume
import kotlinx.coroutines.channels.produce
import kotlinx.coroutines.withContext
import net.syncthing.java.bep.index.IndexHandler
import net.syncthing.java.core.beans.FileInfo
import net.syncthing.java.core.interfaces.IndexRepository
import net.syncthing.java.core.interfaces.IndexTransaction
import net.syncthing.java.core.utils.PathUtils
import java.util.*
class IndexBrowser internal constructor(
private val indexRepository: IndexRepository,
private val indexHandler: IndexHandler
) {
companion object {
val sortAlphabeticallyDirectoriesFirst: Comparator<FileInfo> =
compareBy<FileInfo>({!isParent(it) }, {!it.isDirectory()})
.thenBy { it.fileName.toLowerCase() }
val sortByLastModification: Comparator<FileInfo> =
compareBy<FileInfo>({!isParent(it) }, {it.lastModified})
.thenBy { it.fileName.toLowerCase() }
private fun isParent(fileInfo: FileInfo) = PathUtils.isParent(fileInfo.path)
fun getPathFileName(path: String) = PathUtils.getFileName(path)
const val ROOT_PATH = PathUtils.ROOT_PATH
}
fun getDirectoryListing(folder: String, path: String): DirectoryListing = indexRepository.runInTransaction { indexTransaction ->
val entries = indexTransaction.findNotDeletedFilesByFolderAndParent(folder, path)
val parentPath = if (PathUtils.isRoot(path)) null else PathUtils.getParentPath(path)
val parentEntry = if (PathUtils.isRoot(path)) null else getFileInfoByPathAllowNull(folder, PathUtils.getParentPath(path), indexTransaction)
val directoryInfo = getFileInfoByPathAllowNull(folder, path, indexTransaction)
if ((parentPath != null && parentEntry == null) || directoryInfo == null || directoryInfo.type != FileInfo.FileType.DIRECTORY) {
DirectoryNotFoundListing(folder, path)
} else {
DirectoryContentListing(
entries = entries,
parentEntry = parentEntry,
directoryInfo = directoryInfo
)
}
}
fun streamDirectoryListing(folder: String, path: String) = GlobalScope.produce {
indexHandler.subscribeToOnIndexRecordAcquiredEvents().consume {
val directoryName = PathUtils.getFileName(path)
val parentPath = if (PathUtils.isRoot(path)) null else PathUtils.getParentPath(path)
val parentDirectoryName = if (parentPath != null) PathUtils.getFileName(parentPath) else null
val parentParentPath = if (parentPath == null || PathUtils.isRoot(parentPath)) null else PathUtils.getParentPath(parentPath)
// get the initial state
var (entries, parentEntry, directoryInfo) = withContext (Dispatchers.IO) {
indexRepository.runInTransaction { indexTransaction ->
val entries = indexTransaction.findNotDeletedFilesByFolderAndParent(folder, path)
val parentEntry = if (PathUtils.isRoot(path)) null else getFileInfoByPathAllowNull(folder, PathUtils.getParentPath(path), indexTransaction)
val directoryInfo = getFileInfoByPathAllowNull(folder, path, indexTransaction)
Triple(entries, parentEntry, directoryInfo)
}
}
var previousStatus: DirectoryListing? = null
suspend fun dispatch() {
// let Kotlin understand that the value does not change during running this
val directoryInfo = directoryInfo
val newStatus = if ((parentPath != null && parentEntry == null) || directoryInfo == null || directoryInfo.type != FileInfo.FileType.DIRECTORY) {
DirectoryNotFoundListing(folder, path)
} else {
DirectoryContentListing(
entries = entries,
parentEntry = parentEntry,
directoryInfo = directoryInfo
)
}
if (newStatus != previousStatus) {
previousStatus = newStatus
send(newStatus)
}
}
dispatch()
// handle updates
for (event in this) {
var hadChanges = false
if (event.folderId == folder) {
event.files.forEach { fileUpdate ->
// entry change
if (fileUpdate.parent == path) {
hadChanges = true
entries = entries.filter { it.fileName != fileUpdate.fileName }
if (!fileUpdate.isDeleted) {
entries += listOf(fileUpdate)
}
}
// handle directory info changes
if (fileUpdate.parent == parentPath && fileUpdate.fileName == directoryName) {
directoryInfo = if (fileUpdate.isDeleted) null else fileUpdate
hadChanges = true
}
// handle parent directory info changes
if (fileUpdate.parent == parentParentPath && fileUpdate.fileName == parentDirectoryName) {
parentEntry = if (fileUpdate.isDeleted) null else fileUpdate
hadChanges = true
}
}
}
if (hadChanges) {
dispatch()
}
}
}
}
fun getFileInfoByAbsolutePath(folder: String, path: String): FileInfo = getFileInfoByAbsolutePathAllowNull(folder, path)
?: error("file not found for path = $path")
fun getFileInfoByAbsolutePathAllowNull(folder: String, path: String): FileInfo? {
return if (PathUtils.isRoot(path)) {
FileInfo(folder = folder, type = FileInfo.FileType.DIRECTORY, path = PathUtils.ROOT_PATH)
} else {
indexRepository.runInTransaction { it.findNotDeletedFileInfo(folder, path) }
}
}
fun getFileInfoByPath(folder: String, path: String, transaction: IndexTransaction) = getFileInfoByPathAllowNull(folder, path, transaction)
?: error("file not found for path = $path")
fun getFileInfoByPathAllowNull(folder: String, path: String, transaction: IndexTransaction): FileInfo? {
return if (PathUtils.isRoot(path)) {
FileInfo(folder = folder, type = FileInfo.FileType.DIRECTORY, path = PathUtils.ROOT_PATH)
} else {
transaction.findNotDeletedFileInfo(folder, path)
}
}
}
@@ -0,0 +1,11 @@
package net.syncthing.java.bep.utils
inline fun <T> Iterable<T>.longMaxBy(selector: (T) -> Long, defaultValue: Long): Long {
var max = defaultValue
this.forEach {
max = Math.max(max, selector(it))
}
return max
}
@@ -13,7 +13,10 @@
*/
package net.syncthing.java.client.cli
import kotlinx.coroutines.channels.consume
import kotlinx.coroutines.runBlocking
import net.syncthing.java.bep.index.browser.DirectoryContentListing
import net.syncthing.java.bep.index.browser.IndexBrowser
import net.syncthing.java.client.SyncthingClient
import net.syncthing.java.core.beans.DeviceId
import net.syncthing.java.core.beans.DeviceInfo
@@ -26,7 +29,6 @@ import org.slf4j.LoggerFactory
import java.io.File
import java.io.FileInputStream
import java.io.IOException
import java.util.concurrent.CountDownLatch
class Main(private val commandLine: CommandLine) {
@@ -117,7 +119,6 @@ class Main(private val commandLine: CommandLine) {
System.out.println("file path = $path")
val folder = path.split(":".toRegex()).dropLastWhile({ it.isEmpty() }).toTypedArray()[0]
path = path.split(":".toRegex()).dropLastWhile({ it.isEmpty() }).toTypedArray()[1]
val latch = CountDownLatch(1)
val blockPusher = syncthingClient.getBlockPusher(folder)
val observer = runBlocking {
@@ -169,33 +170,35 @@ class Main(private val commandLine: CommandLine) {
System.out.println("uploaded dir to network")
}
"L" -> {
waitForIndexUpdate(syncthingClient, configuration)
for (folder in syncthingClient.indexHandler.folderList()) {
syncthingClient.indexHandler.newIndexBrowser(folder).use { indexBrowser ->
System.out.println("list folder = ${indexBrowser.folder}")
for (fileInfo in indexBrowser.listFiles()) {
waitForIndexUpdate(syncthingClient)
for (folder in configuration.folders) {
System.out.println("list folder = ${folder}")
val listing = syncthingClient.indexHandler.indexBrowser.getDirectoryListing(folder.folderId, IndexBrowser.ROOT_PATH)
if (listing is DirectoryContentListing) {
for (fileInfo in listing.entries) {
System.out.println("${fileInfo.type.name.substring(0, 1)}\t${fileInfo.describeSize()}\t${fileInfo.path}")
}
}
}
}
"I" -> {
waitForIndexUpdate(syncthingClient, configuration)
waitForIndexUpdate(syncthingClient)
val folderInfo = StringBuilder()
for (folder in syncthingClient.indexHandler.folderList()) {
for (folder in configuration.folders) {
folderInfo.append("\nfolder info: ")
.append(syncthingClient.indexHandler.getFolderInfo(folder))
.append(folder)
folderInfo.append("\nfolder stats: ")
.append(syncthingClient.indexHandler.newFolderBrowser().getFolderStats(folder).dumpInfo())
.append(syncthingClient.indexHandler.folderBrowser.getFolderStatusSync(folder.folderId).stats.infoDump)
.append("\n")
}
System.out.println("folders:\n$folderInfo\n")
}
"l" -> {
var folderInfo = ""
for (folder in syncthingClient.indexHandler.folderList()) {
folderInfo += "\nfolder info: " + syncthingClient.indexHandler.getFolderInfo(folder)
folderInfo += "\nfolder stats: " + syncthingClient.indexHandler.newFolderBrowser().getFolderStats(folder).dumpInfo() + "\n"
for (folder in configuration.folders) {
folderInfo += "\nfolder info: " + folder
folderInfo += "\nfolder stats: " + syncthingClient.indexHandler.folderBrowser.getFolderStatusSync(folder.folderId).stats.infoDump + "\n"
}
System.out.println("folders:\n$folderInfo\n")
}
@@ -211,11 +214,12 @@ class Main(private val commandLine: CommandLine) {
}
@Throws(InterruptedException::class)
private fun waitForIndexUpdate(client: SyncthingClient, configuration: Configuration) {
val latch = CountDownLatch(configuration.peers.size)
client.indexHandler.registerOnFullIndexAcquiredListenersListener {
latch.countDown()
private fun waitForIndexUpdate(client: SyncthingClient) {
// FIXME: what happens if the index update happened already?
runBlocking {
client.indexHandler.subscribeToOnFullIndexAcquiredEvents().consume {
this.receive() // wait until there is one event
}
}
latch.await()
}
}
@@ -16,9 +16,13 @@ package net.syncthing.java.client
import kotlinx.coroutines.GlobalScope
import kotlinx.coroutines.async
import kotlinx.coroutines.runBlocking
import net.syncthing.java.bep.*
import net.syncthing.java.bep.BlockPuller
import net.syncthing.java.bep.BlockPullerStatus
import net.syncthing.java.bep.BlockPusher
import net.syncthing.java.bep.RequestHandlerRegistry
import net.syncthing.java.bep.connectionactor.ConnectionActorGenerator
import net.syncthing.java.bep.connectionactor.ConnectionActorWrapper
import net.syncthing.java.bep.index.IndexHandler
import net.syncthing.java.core.beans.DeviceId
import net.syncthing.java.core.beans.FileInfo
import net.syncthing.java.core.configuration.Configuration
@@ -32,9 +36,10 @@ import java.util.*
class SyncthingClient(
private val configuration: Configuration,
private val repository: IndexRepository,
private val tempRepository: TempRepository
private val tempRepository: TempRepository,
enableDetailedException: Boolean = false
) : Closeable {
val indexHandler = IndexHandler(configuration, repository, tempRepository)
val indexHandler = IndexHandler(configuration, repository, tempRepository, enableDetailedException)
val discoveryHandler = DiscoveryHandler(configuration)
private val onConnectionChangedListeners = Collections.synchronizedList(mutableListOf<(DeviceId) -> Unit>())
@@ -17,7 +17,7 @@ package net.syncthing.java.core.beans
import com.google.gson.stream.JsonReader
import com.google.gson.stream.JsonWriter
open class FolderInfo(val folderId: String, label: String? = null) {
data class FolderInfo(val folderId: String, val label: String) {
companion object {
private const val FOLDER_ID = "folderId"
private const val LABEL = "label"
@@ -43,11 +43,8 @@ open class FolderInfo(val folderId: String, label: String? = null) {
}
}
val label: String
init {
assert(!folderId.isEmpty())
this.label = if (label != null && !label.isEmpty()) label else folderId
}
override fun toString(): String {
@@ -1,5 +1,6 @@
/*
* Copyright (C) 2016 Davide Imbriaco
* Copyright (C) 2018 Jonas Lochmann
*
* This Java file is subject to the terms of the Mozilla Public
* License, v. 2.0. If a copy of the MPL was not distributed with this
@@ -16,87 +17,22 @@ package net.syncthing.java.core.beans
import org.apache.commons.io.FileUtils
import java.util.*
class FolderStats private constructor(val fileCount: Long, val dirCount: Long, val size: Long, val lastUpdate: Date, folder: String, label: String?) : FolderInfo(folder, label) {
fun getRecordCount(): Long = dirCount + fileCount
fun describeSize(): String = FileUtils.byteCountToDisplaySize(size)
fun dumpInfo(): String {
return ("folder " + label + " (" + folderId + ") file count = " + fileCount
+ " dir count = " + dirCount + " folder size = " + describeSize() + " last update = " + lastUpdate)
data class FolderStats(val fileCount: Long, val dirCount: Long, val size: Long, val lastUpdate: Date, val folderId: String) {
companion object {
fun createDummy(folderId: String) = FolderStats(
fileCount = 0,
dirCount = 0,
size = 0,
lastUpdate = Date(0),
folderId = folderId
)
}
override fun toString(): String {
return "FolderStats{folder=$folderId, fileCount=$fileCount, dirCount=$dirCount, size=$size, lastUpdate=$lastUpdate}"
}
val recordCount: Long = dirCount + fileCount
fun copyBuilder(): Builder = Builder(fileCount, dirCount, size, folderId, label)
class Builder {
private var fileCount: Long = 0
private var dirCount: Long = 0
private var size: Long = 0
private var lastUpdate = Date(0)
private var folder: String? = null
private var label: String? = null
constructor()
constructor(fileCount: Long, dirCount: Long, size: Long, folder: String, label: String) {
this.fileCount = fileCount
this.dirCount = dirCount
this.size = size
this.folder = folder
this.label = label
}
fun getFileCount(): Long = fileCount
fun setFileCount(fileCount: Long): Builder {
this.fileCount = fileCount
return this
}
fun getDirCount(): Long = dirCount
fun setDirCount(dirCount: Long): Builder {
this.dirCount = dirCount
return this
}
fun getSize(): Long = size
fun setSize(size: Long): Builder {
this.size = size
return this
}
fun getLastUpdate(): Date = lastUpdate
fun setLastUpdate(lastUpdate: Date): Builder {
this.lastUpdate = lastUpdate
return this
}
fun getFolder(): String? = folder
fun setFolder(folder: String): Builder {
this.folder = folder
return this
}
fun getLabel(): String? = label
fun setLabel(label: String): Builder {
this.label = label
return this
}
fun build(): FolderStats {
return FolderStats(fileCount, dirCount, size, lastUpdate, folder!!, label)
}
val sizeDescription: String by lazy { FileUtils.byteCountToDisplaySize(size) }
val infoDump: String by lazy {
"folder $folderId file count = $fileCount dir count = $dirCount folder size = $sizeDescription last update = $lastUpdate"
}
}
@@ -1,5 +1,6 @@
/*
* Copyright (C) 2016 Davide Imbriaco
* Copyright (C) 2018 Jonas Lochmann
*
* This Java file is subject to the terms of the Mozilla Public
* License, v. 2.0. If a copy of the MPL was not distributed with this
@@ -14,98 +15,11 @@
package net.syncthing.java.core.beans
class IndexInfo private constructor(folder: String, val deviceId: String, val indexId: Long, val localSequence: Long, val maxSequence: Long) : FolderInfo(folder) {
data class IndexInfo(val folderId: String, val deviceId: String, val indexId: Long, val localSequence: Long, val maxSequence: Long) {
fun getCompleted(): Double = if (maxSequence > 0) localSequence.toDouble() / maxSequence else 0.0
val completed: Double = if (maxSequence > 0) localSequence.toDouble() / maxSequence else 0.0
init {
assert(!deviceId.isEmpty())
assert(deviceId.isNotEmpty())
}
fun copyBuilder(): Builder {
return Builder(folderId, indexId, deviceId, localSequence, maxSequence)
}
override fun toString(): String {
return "FolderIndexInfo{indexId=$indexId, folder=$folderId, deviceId=$deviceId, localSequence=$localSequence, maxSequence=$maxSequence}"
}
class Builder {
private var indexId: Long = 0
private var deviceId: String? = null
private var folder: String? = null
private var localSequence: Long = 0
private var maxSequence: Long = 0
internal constructor()
internal constructor(folder: String, indexId: Long, deviceId: String, localSequence: Long, maxSequence: Long) {
assert(!folder.isEmpty())
assert(!deviceId.isEmpty())
this.folder = folder
this.indexId = indexId
this.deviceId = deviceId
this.localSequence = localSequence
this.maxSequence = maxSequence
}
fun getIndexId(): Long {
return indexId
}
fun getDeviceId(): String? {
return deviceId
}
fun getFolder(): String? {
return folder
}
fun getLocalSequence(): Long {
return localSequence
}
fun getMaxSequence(): Long {
return maxSequence
}
fun setIndexId(indexId: Long): Builder {
this.indexId = indexId
return this
}
fun setDeviceId(deviceId: String): Builder {
this.deviceId = deviceId
return this
}
fun setFolder(folder: String): Builder {
this.folder = folder
return this
}
fun setLocalSequence(localSequence: Long): Builder {
this.localSequence = localSequence
return this
}
fun setMaxSequence(maxSequence: Long): Builder {
this.maxSequence = maxSequence
return this
}
fun build(): IndexInfo {
return IndexInfo(folder!!, deviceId!!, indexId, localSequence, maxSequence)
}
}
companion object {
fun newBuilder(): Builder {
return Builder()
}
}
}
@@ -1,5 +1,6 @@
/*
* Copyright (C) 2016 Davide Imbriaco
* Copyright (C) 2018 Jonas Lochmann
*
* This Java file is subject to the terms of the Mozilla Public
* License, v. 2.0. If a copy of the MPL was not distributed with this
@@ -13,46 +14,8 @@
*/
package net.syncthing.java.core.interfaces
import net.syncthing.java.core.beans.*
import java.io.Closeable
import java.util.*
interface IndexRepository: Closeable {
fun setOnFolderStatsUpdatedListener(listener: ((IndexRepository.FolderStatsUpdatedEvent) -> Unit)?)
fun getSequencer(): Sequencer
fun updateIndexInfo(indexInfo: IndexInfo)
fun findIndexInfoByDeviceAndFolder(deviceId: DeviceId, folder: String): IndexInfo?
fun findFileInfo(folder: String, path: String): FileInfo?
fun findFileInfoLastModified(folder: String, path: String): Date?
fun findNotDeletedFileInfo(folder: String, path: String): FileInfo?
fun findFileBlocks(folder: String, path: String): FileBlocks?
fun updateFileInfo(fileInfo: FileInfo, fileBlocks: FileBlocks?)
fun findNotDeletedFilesByFolderAndParent(folder: String, parentPath: String): List<FileInfo>
fun clearIndex()
fun findFolderStats(folder: String): FolderStats?
fun findAllFolderStats(): List<FolderStats>
fun findFileInfoBySearchTerm(query: String): List<FileInfo>
fun countFileInfoBySearchTerm(query: String): Long
abstract class FolderStatsUpdatedEvent {
abstract fun getFolderStats(): List<FolderStats>
}
fun <T> runInTransaction(action: (IndexTransaction) -> T): T
}
@@ -0,0 +1,57 @@
/*
* Copyright (C) 2016 Davide Imbriaco
* Copyright (C) 2018 Jonas Lochmann
*
* This Java file is subject to the terms of the Mozilla Public
* License, v. 2.0. If a copy of the MPL was not distributed with this
* file, You can obtain one at http://mozilla.org/MPL/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 net.syncthing.java.core.interfaces
import net.syncthing.java.core.beans.*
import java.util.*
interface IndexTransaction {
fun getSequencer(): Sequencer
fun updateIndexInfo(indexInfo: IndexInfo)
fun findAllIndexInfos(): List<IndexInfo>
fun findIndexInfoByDeviceAndFolder(deviceId: DeviceId, folder: String): IndexInfo?
fun findFileInfo(folder: String, path: String): FileInfo?
// path to FileInfo
fun findFileInfo(folder: String, path: List<String>): Map<String, FileInfo>
fun findFileInfoLastModified(folder: String, path: String): Date?
fun findNotDeletedFileInfo(folder: String, path: String): FileInfo?
fun findFileBlocks(folder: String, path: String): FileBlocks?
fun updateFileInfo(fileInfo: FileInfo, fileBlocks: FileBlocks?)
fun updateFileInfoAndBlocks(fileInfos: List<FileInfo>, fileBlocks: List<FileBlocks>)
fun findNotDeletedFilesByFolderAndParent(folder: String, parentPath: String): List<FileInfo>
fun clearIndex()
fun findFolderStats(folder: String): FolderStats?
fun findAllFolderStats(): List<FolderStats>
fun updateOrInsertFolderStats(folder: String, deltaFileCount: Long, deltaDirCount: Long, deltaSize: Long, lastUpdate: Date)
fun findFileInfoBySearchTerm(query: String): List<FileInfo>
fun countFileInfoBySearchTerm(query: String): Long
}
@@ -1,56 +0,0 @@
/*
* Copyright (C) 2016 Davide Imbriaco
*
* This Java file is subject to the terms of the Mozilla Public
* License, v. 2.0. If a copy of the MPL was not distributed with this
* file, You can obtain one at http://mozilla.org/MPL/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 net.syncthing.java.core.utils
import org.slf4j.LoggerFactory
import java.util.concurrent.ExecutorService
import java.util.concurrent.Future
import java.util.concurrent.RejectedExecutionException
import java.util.concurrent.TimeUnit
private val logger = LoggerFactory.getLogger(ExecutorService::class.java)
fun ExecutorService.awaitTerminationSafe() {
try {
awaitTermination(2, TimeUnit.SECONDS)
} catch (ex: InterruptedException) {
logger.warn("", ex)
}
}
fun ExecutorService.submitLogging(runnable: Runnable) = submitLogging { runnable.run() }
/**
* Wrapper method for [[ExecutorService.submit]], which silently swallows exceptions. If an exception is thrown in
* [[runnable]], logs the exception and force crashes
*/
fun <T> ExecutorService.submitLogging(runnable: () -> T): Future<T> {
return submit<T>({
try {
runnable()
} catch (e: Exception) {
logger.error("", e)
System.exit(1)
null
}
})
}
fun ExecutorService.trySubmitLogging(runnable: Runnable) {
try {
submitLogging(runnable)
} catch (ex: RejectedExecutionException) {
logger.warn("could not submit task", ex)
}
}
@@ -14,6 +14,7 @@
*/
package net.syncthing.java.discovery.protocol
import kotlinx.coroutines.Dispatchers
import kotlinx.coroutines.GlobalScope
import kotlinx.coroutines.Job
import kotlinx.coroutines.channels.consumeEach
@@ -32,7 +33,7 @@ internal class LocalDiscoveryHandler(private val configuration: Configuration,
private val job = Job()
fun sendAnnounceMessage() {
GlobalScope.launch {
GlobalScope.launch (Dispatchers.IO) {
LocalDiscoveryUtil.sendAnnounceMessage(
ownDeviceId = configuration.localDeviceId,
instanceId = configuration.instanceId
@@ -1,124 +1,32 @@
package net.syncthing.repository.android
import net.syncthing.java.core.beans.*
import net.syncthing.java.core.interfaces.IndexRepository
import net.syncthing.java.core.interfaces.Sequencer
import net.syncthing.java.core.interfaces.IndexTransaction
import net.syncthing.repository.android.database.RepositoryDatabase
import net.syncthing.repository.android.database.item.*
import java.util.*
import java.util.concurrent.Callable
class SqliteIndexRepository(
private val database: RepositoryDatabase,
private val closeDatabaseOnClose: Boolean,
private val clearTempStorageHook: () -> Unit
): IndexRepository {
private var folderStatsChangeListener: ((IndexRepository.FolderStatsUpdatedEvent) -> Unit)? = null
// FileInfo
override fun findFileInfo(folder: String, path: String) = database.fileInfo().findFileInfo(folder, path)?.native
override fun findFileInfoBySearchTerm(query: String) = database.fileInfo().findFileInfoBySearchTerm(query).map { it.native }
override fun findFileInfoLastModified(folder: String, path: String): Date? = database.fileInfo().findFileInfoLastModified(folder, path)?.lastModified
override fun findNotDeletedFileInfo(folder: String, path: String) = database.fileInfo().findNotDeletedFileInfo(folder, path)?.native
override fun findNotDeletedFilesByFolderAndParent(folder: String, parentPath: String) = database.fileInfo().findNotDeletedFilesByFolderAndParent(folder, parentPath).map { it.native }
override fun countFileInfoBySearchTerm(query: String) = database.fileInfo().countFileInfoBySearchTerm(query)
override fun <T> runInTransaction(action: (IndexTransaction) -> T): T {
return database.runInTransaction (object: Callable<T> {
override fun call(): T {
val transaction = SqliteTransaction(
database = database,
threadId = Thread.currentThread().id,
clearTempStorageHook = clearTempStorageHook
)
override fun updateFileInfo(fileInfo: FileInfo, fileBlocks: FileBlocks?) {
val newFileInfo = fileInfo
val newFileBlocks = fileBlocks
database.runInTransaction {
if (newFileBlocks != null) {
FileInfo.checkBlocks(newFileInfo, newFileBlocks)
database.fileBlocks().mergeBlock(FileBlocksItem.fromNative(newFileBlocks))
}
val oldFileInfo = findFileInfo(newFileInfo.folder, newFileInfo.path)
database.fileInfo().updateFileInfo(FileInfoItem.fromNative(newFileInfo))
//update stats
var deltaFileCount = 0L
var deltaDirCount= 0L
var deltaSize = 0L
val oldMissing = oldFileInfo == null || oldFileInfo.isDeleted
val newMissing = newFileInfo.isDeleted
val oldSizeMissing = oldMissing || !oldFileInfo!!.isFile()
val newSizeMissing = newMissing || !newFileInfo.isFile()
if (!oldSizeMissing) {
deltaSize -= oldFileInfo!!.size!!
}
if (!newSizeMissing) {
deltaSize += newFileInfo.size!!
}
if (!oldMissing) {
if (oldFileInfo!!.isFile()) {
deltaFileCount--
} else if (oldFileInfo.isDirectory()) {
deltaDirCount--
}
}
if (!newMissing) {
if (newFileInfo.isFile()) {
deltaFileCount++
} else if (newFileInfo.isDirectory()) {
deltaDirCount++
}
}
val newFolderStats = kotlin.run {
val updatedRows = database.folderStats().updateFolderStats(
folder = newFileInfo.folder,
deltaDirCount = deltaDirCount,
deltaFileCount = deltaFileCount,
deltaSize = deltaSize,
lastUpdate = newFileInfo.lastModified
)
if (updatedRows == 0L) {
database.folderStats().insertFolderStats(FolderStatsItem(
folder = newFileInfo.folder,
dirCount = deltaDirCount,
fileCount = deltaFileCount,
size = deltaSize,
lastUpdate = newFileInfo.lastModified
))
}
database.folderStats().getFolderStats(newFileInfo.folder)!!
}
folderStatsChangeListener?.invoke(object : IndexRepository.FolderStatsUpdatedEvent() {
override fun getFolderStats(): List<FolderStats> {
return listOf(newFolderStats.native)
return try {
action(transaction)
} finally {
transaction.markFinished()
}
}
})
}
}
// FileBlocks
override fun findFileBlocks(folder: String, path: String) = database.fileBlocks().findFileBlocks(folder, path)?.native
// FolderStats
override fun findAllFolderStats() = database.folderStats().findAllFolderStats().map { it.native }
override fun findFolderStats(folder: String): FolderStats? = database.folderStats().findFolderStats(folder)?.native
// IndexInfo
override fun updateIndexInfo(indexInfo: IndexInfo) {
database.folderIndexInfo().updateIndexInfo(FolderIndexInfoItem.fromNative(indexInfo))
}
override fun findIndexInfoByDeviceAndFolder(deviceId: DeviceId, folder: String): IndexInfo? = database.folderIndexInfo().findIndexInfoByDeviceAndFolder(deviceId, folder)?.native
// managment
override fun clearIndex() {
database.clearAllTables()
clearTempStorageHook()
}
override fun close() {
@@ -126,39 +34,4 @@ class SqliteIndexRepository(
database.close()
}
}
// other
private val sequencer = object: Sequencer {
fun getDatabaseEntry(): IndexSequenceItem {
val entry = database.indexSequence().getItem()
if (entry != null) {
return entry
}
val newEntry = IndexSequenceItem(
indexId = Math.abs(Random().nextLong()) + 1,
currentSequence = Math.abs(Random().nextLong()) + 1
)
database.indexSequence().createItem(newEntry)
return newEntry
}
override fun indexId() = getDatabaseEntry().indexId
override fun currentSequence() = getDatabaseEntry().currentSequence
override fun nextSequence(): Long {
database.indexSequence().incrementSequenceNumber(indexId())
return currentSequence()
}
}
override fun getSequencer() = sequencer
override fun setOnFolderStatsUpdatedListener(listener: ((IndexRepository.FolderStatsUpdatedEvent) -> Unit)?) {
folderStatsChangeListener = listener
}
}
@@ -0,0 +1,169 @@
package net.syncthing.repository.android
import net.syncthing.java.core.beans.*
import net.syncthing.java.core.interfaces.IndexTransaction
import net.syncthing.java.core.interfaces.Sequencer
import net.syncthing.repository.android.database.RepositoryDatabase
import net.syncthing.repository.android.database.item.*
import java.util.*
class SqliteTransaction(
private val database: RepositoryDatabase,
private val threadId: Long,
private val clearTempStorageHook: () -> Unit
): IndexTransaction {
private var finished = false
private fun assertAllowed() {
if (finished) {
throw IllegalStateException("tried to use a transaction which is already done")
}
if (Thread.currentThread().id != threadId) {
throw IllegalStateException("tried to access the transaction from an other Thread")
}
}
fun markFinished() {
finished = true
}
private fun <T> runIfAllowed(block: () -> T): T {
assertAllowed()
return block()
}
// FileInfo
override fun findFileInfo(folder: String, path: String) = runIfAllowed {
database.fileInfo().findFileInfo(folder, path)?.native
}
override fun findFileInfo(folder: String, path: List<String>): Map<String, FileInfo> = runIfAllowed {
database.fileInfo().findFileInfo(folder, path)
.map { it.native }
.associateBy { it.path }
}
override fun findFileInfoBySearchTerm(query: String) = runIfAllowed {
database.fileInfo().findFileInfoBySearchTerm(query).map { it.native }
}
override fun findFileInfoLastModified(folder: String, path: String): Date? = runIfAllowed {
database.fileInfo().findFileInfoLastModified(folder, path)?.lastModified
}
override fun findNotDeletedFileInfo(folder: String, path: String) = runIfAllowed {
database.fileInfo().findNotDeletedFileInfo(folder, path)?.native
}
override fun findNotDeletedFilesByFolderAndParent(folder: String, parentPath: String) = runIfAllowed {
database.fileInfo().findNotDeletedFilesByFolderAndParent(folder, parentPath).map { it.native }
}
override fun countFileInfoBySearchTerm(query: String) = runIfAllowed {
database.fileInfo().countFileInfoBySearchTerm(query)
}
override fun updateFileInfo(fileInfo: FileInfo, fileBlocks: FileBlocks?) = runIfAllowed {
if (fileBlocks != null) {
FileInfo.checkBlocks(fileInfo, fileBlocks)
database.fileBlocks().mergeBlock(FileBlocksItem.fromNative(fileBlocks))
}
database.fileInfo().updateFileInfo(FileInfoItem.fromNative(fileInfo))
}
override fun updateFileInfoAndBlocks(fileInfos: List<FileInfo>, fileBlocks: List<FileBlocks>) = runIfAllowed {
if (fileInfos.isNotEmpty()) {
database.fileInfo().updateFileInfo(fileInfos.map { FileInfoItem.fromNative(it) })
}
if (fileBlocks.isNotEmpty()) {
database.fileBlocks().mergeBlocks(fileBlocks.map { FileBlocksItem.fromNative(it) })
}
}
// FileBlocks
override fun findFileBlocks(folder: String, path: String) = runIfAllowed {
database.fileBlocks().findFileBlocks(folder, path)?.native
}
// FolderStats
override fun findAllFolderStats() = runIfAllowed {
database.folderStats().findAllFolderStats().map { it.native }
}
override fun findFolderStats(folder: String): FolderStats? = runIfAllowed {
database.folderStats().findFolderStats(folder)?.native
}
override fun updateOrInsertFolderStats(
folder: String,
deltaFileCount: Long,
deltaDirCount: Long,
deltaSize: Long,
lastUpdate: Date
) = runIfAllowed {
if (database.folderStats().updateFolderStats(folder, deltaFileCount, deltaDirCount, deltaSize, lastUpdate) == 0L) {
database.folderStats().insertFolderStats(FolderStatsItem(folder, deltaFileCount, deltaDirCount, lastUpdate, deltaSize))
}
}
// IndexInfo
override fun updateIndexInfo(indexInfo: IndexInfo) = runIfAllowed {
database.folderIndexInfo().updateIndexInfo(FolderIndexInfoItem.fromNative(indexInfo))
}
override fun findIndexInfoByDeviceAndFolder(deviceId: DeviceId, folder: String): IndexInfo? = runIfAllowed {
database.folderIndexInfo().findIndexInfoByDeviceAndFolder(deviceId, folder)?.native
}
override fun findAllIndexInfos(): List<IndexInfo> = runIfAllowed {
database.folderIndexInfo().findAllIndexInfo().map { it.native }
}
// managment
override fun clearIndex() {
runIfAllowed {
database.clearAllTables()
clearTempStorageHook()
}
}
// other
private val sequencer = object: Sequencer {
private fun getDatabaseEntry(): IndexSequenceItem {
val entry = database.indexSequence().getItem()
if (entry != null) {
return entry
}
val newEntry = IndexSequenceItem(
indexId = Math.abs(Random().nextLong()) + 1,
currentSequence = Math.abs(Random().nextLong()) + 1
)
database.indexSequence().createItem(newEntry)
return newEntry
}
override fun indexId() = runIfAllowed { getDatabaseEntry().indexId }
override fun currentSequence() = runIfAllowed { getDatabaseEntry().currentSequence }
override fun nextSequence(): Long = runIfAllowed {
database.indexSequence().incrementSequenceNumber(indexId())
currentSequence()
}
}
override fun getSequencer() = sequencer
}
@@ -13,4 +13,7 @@ interface FileBlocksDao {
@Insert(onConflict = OnConflictStrategy.REPLACE)
fun mergeBlock(blocksItem: FileBlocksItem)
@Insert(onConflict = OnConflictStrategy.REPLACE)
fun mergeBlocks(blocksItem: List<FileBlocksItem>)
}
@@ -12,6 +12,9 @@ interface FileInfoDao {
@Query("SELECT * FROM file_info WHERE folder = :folder AND path = :path")
fun findFileInfo(folder: String, path: String): FileInfoItem?
@Query("SELECT * FROM file_info WHERE folder = :folder AND path IN (:path)")
fun findFileInfo(folder: String, path: List<String>): List<FileInfoItem>
@Query("SELECT last_modified FROM file_info WHERE folder = :folder AND path = :path")
fun findFileInfoLastModified(folder: String, path: String): FileInfoLastModified?
@@ -31,4 +34,7 @@ interface FileInfoDao {
@Insert(onConflict = OnConflictStrategy.REPLACE)
fun updateFileInfo(info: FileInfoItem)
@Insert(onConflict = OnConflictStrategy.REPLACE)
fun updateFileInfo(info: List<FileInfoItem>)
}
@@ -13,4 +13,7 @@ interface FolderIndexInfoDao {
@Query("SELECT * FROM folder_index_info WHERE device_id = :deviceId AND folder = :folder")
fun findIndexInfoByDeviceAndFolder(deviceId: DeviceId, folder: String): FolderIndexInfoItem?
@Query("SELECT * FROM folder_index_info")
fun findAllIndexInfo(): List<FolderIndexInfoItem>
}
@@ -31,12 +31,12 @@ data class FolderIndexInfoItem(
@delegate:Transient
val native: IndexInfo by lazy {
IndexInfo.newBuilder()
.setFolder(folder)
.setDeviceId(deviceId)
.setIndexId(indexId)
.setLocalSequence(localSequence)
.setMaxSequence(maxSequence)
.build()
IndexInfo(
folderId = folder,
deviceId = deviceId,
indexId = indexId,
localSequence = localSequence,
maxSequence = maxSequence
)
}
}
@@ -25,12 +25,12 @@ data class FolderStatsItem(
) {
@delegate:Transient
val native: FolderStats by lazy {
FolderStats.Builder()
.setFolder(folder)
.setDirCount(dirCount)
.setFileCount(fileCount)
.setSize(size)
.setLastUpdate(lastUpdate)
.build()
FolderStats(
folderId = folder,
dirCount = dirCount,
fileCount = fileCount,
size = size,
lastUpdate = lastUpdate
)
}
}
@@ -13,36 +13,23 @@
*/
package net.syncthing.java.repository.repo
import com.google.protobuf.ByteString
import com.google.protobuf.InvalidProtocolBufferException
import com.zaxxer.hikari.HikariConfig
import com.zaxxer.hikari.HikariDataSource
import net.syncthing.java.bep.BlockExchangeExtraProtos
import net.syncthing.java.bep.BlockExchangeProtos
import net.syncthing.java.core.beans.*
import net.syncthing.java.core.beans.FileInfo.FileType
import net.syncthing.java.core.beans.FileInfo.Version
import net.syncthing.java.core.interfaces.IndexRepository
import net.syncthing.java.core.interfaces.Sequencer
import net.syncthing.java.core.interfaces.IndexTransaction
import net.syncthing.java.core.interfaces.TempRepository
import org.apache.commons.lang3.tuple.Pair
import org.bouncycastle.util.encoders.Hex
import org.slf4j.LoggerFactory
import java.io.Closeable
import java.io.File
import java.sql.Connection
import java.sql.ResultSet
import java.sql.SQLException
import java.sql.Types
import java.util.*
class SqlRepository(databaseFolder: File) : Closeable, IndexRepository, TempRepository {
private val logger = LoggerFactory.getLogger(javaClass)
private var sequencer: Sequencer = IndexRepoSequencer()
private val dataSource: HikariDataSource
// private final ScheduledExecutorService scheduledExecutorService = Executors.newSingleThreadScheduledExecutor();
private var onFolderStatsUpdatedListener: ((IndexRepository.FolderStatsUpdatedEvent) -> Unit)? = null
@Throws(SQLException::class)
private fun getConnection() = dataSource.connection
@@ -78,10 +65,6 @@ class SqlRepository(databaseFolder: File) : Closeable, IndexRepository, TempRepo
logger.debug("database ready")
}
override fun setOnFolderStatsUpdatedListener(listener: ((IndexRepository.FolderStatsUpdatedEvent) -> Unit)?) {
onFolderStatsUpdatedListener = listener
}
private fun checkDb() {
try {
getConnection().use { connection ->
@@ -95,16 +78,17 @@ class SqlRepository(databaseFolder: File) : Closeable, IndexRepository, TempRepo
}
} catch (ex: SQLException) {
logger.warn("Invalid database, resetting db", ex)
initDb()
getConnection().use {
initDb(it)
}
}
}
@Throws(SQLException::class)
private fun initDb() {
private fun initDb(connection: Connection) {
logger.info("init db")
getConnection().use { connection -> connection.prepareStatement("DROP ALL OBJECTS").use { prepareStatement -> prepareStatement.execute() } }
connection.prepareStatement("DROP ALL OBJECTS").use { prepareStatement -> prepareStatement.execute() }
getConnection().use { connection ->
connection.prepareStatement("CREATE TABLE index_sequence (index_id BIGINT NOT NULL PRIMARY KEY, current_sequence BIGINT NOT NULL)").use { prepareStatement -> prepareStatement.execute() }
connection.prepareStatement("CREATE TABLE folder_index_info (folder VARCHAR NOT NULL,"
+ "device_id VARCHAR NOT NULL,"
@@ -150,7 +134,6 @@ class SqlRepository(databaseFolder: File) : Closeable, IndexRepository, TempRepo
prepareStatement.setInt(1, VERSION)
assert(prepareStatement.executeUpdate() == 1)
}
}
logger.info("database initialized")
}
@@ -164,433 +147,28 @@ class SqlRepository(databaseFolder: File) : Closeable, IndexRepository, TempRepo
}
}
override fun getSequencer(): Sequencer = sequencer
override fun <T> runInTransaction(action: (IndexTransaction) -> T): T {
return getConnection().use { connection ->
val transaction = SqlTransaction(connection, ::initDb)
@Throws(SQLException::class)
private fun readFolderIndexInfo(resultSet: ResultSet): IndexInfo {
return IndexInfo.newBuilder()
.setFolder(resultSet.getString("folder"))
.setDeviceId(resultSet.getString("device_id"))
.setIndexId(resultSet.getLong("index_id"))
.setLocalSequence(resultSet.getLong("local_sequence"))
.setMaxSequence(resultSet.getLong("max_sequence"))
.build()
}
try {
connection.autoCommit = false
connection.transactionIsolation = Connection.TRANSACTION_SERIALIZABLE
@Throws(SQLException::class)
override fun updateIndexInfo(indexInfo: IndexInfo) {
getConnection().use { connection ->
connection.prepareStatement("MERGE INTO folder_index_info"
+ " (folder,device_id,index_id,local_sequence,max_sequence)"
+ " VALUES (?,?,?,?,?)").use { prepareStatement ->
prepareStatement.setString(1, indexInfo.folderId)
prepareStatement.setString(2, indexInfo.deviceId)
prepareStatement.setLong(3, indexInfo.indexId)
prepareStatement.setLong(4, indexInfo.localSequence)
prepareStatement.setLong(5, indexInfo.maxSequence)
prepareStatement.executeUpdate()
action(transaction)
} catch (ex: Exception) {
connection.rollback()
throw ex
} finally {
transaction.close()
connection.commit()
connection.autoCommit = true
}
}
}
override fun findIndexInfoByDeviceAndFolder(deviceId: DeviceId, folder: String): IndexInfo? {
val key = Pair.of(deviceId, folder)
return doFindIndexInfoByDeviceAndFolder(key.left, key.right)
}
@Throws(SQLException::class)
private fun doFindIndexInfoByDeviceAndFolder(deviceId: DeviceId, folder: String): IndexInfo? {
getConnection().use { connection ->
connection.prepareStatement("SELECT * FROM folder_index_info WHERE device_id=? AND folder=?").use { prepareStatement ->
prepareStatement.setString(1, deviceId.deviceId)
prepareStatement.setString(2, folder)
val resultSet = prepareStatement.executeQuery()
return if (resultSet.first()) {
readFolderIndexInfo(resultSet)
} else {
null
}
}
}
}
@Throws(SQLException::class)
override fun findFileInfo(folder: String, path: String): FileInfo? {
getConnection().use { connection ->
connection.prepareStatement("SELECT * FROM file_info WHERE folder=? AND path=?").use { prepareStatement ->
prepareStatement.setString(1, folder)
prepareStatement.setString(2, path)
val resultSet = prepareStatement.executeQuery()
return if (resultSet.first()) {
readFileInfo(resultSet)
} else {
null
}
}
}
}
@Throws(SQLException::class)
override fun findFileInfoLastModified(folder: String, path: String): Date? {
getConnection().use { connection ->
connection.prepareStatement("SELECT last_modified FROM file_info WHERE folder=? AND path=?").use { prepareStatement ->
prepareStatement.setString(1, folder)
prepareStatement.setString(2, path)
val resultSet = prepareStatement.executeQuery()
return if (resultSet.first()) {
Date(resultSet.getLong("last_modified"))
} else {
null
}
}
}
}
@Throws(SQLException::class)
override fun findNotDeletedFileInfo(folder: String, path: String): FileInfo? {
getConnection().use { connection ->
connection.prepareStatement("SELECT * FROM file_info WHERE folder=? AND path=? AND is_deleted=FALSE").use { prepareStatement ->
prepareStatement.setString(1, folder)
prepareStatement.setString(2, path)
val resultSet = prepareStatement.executeQuery()
return if (resultSet.first()) {
readFileInfo(resultSet)
} else {
null
}
}
}
}
@Throws(SQLException::class)
private fun readFileInfo(resultSet: ResultSet): FileInfo {
val folder = resultSet.getString("folder")
val path = resultSet.getString("path")
val fileType = FileType.valueOf(resultSet.getString("file_type"))
val lastModified = Date(resultSet.getLong("last_modified"))
val versionList = listOf(Version(resultSet.getLong("version_id"), resultSet.getLong("version_value")))
val isDeleted = resultSet.getBoolean("is_deleted")
val builder = FileInfo.Builder()
.setFolder(folder)
.setPath(path)
.setLastModified(lastModified)
.setVersionList(versionList)
.setDeleted(isDeleted)
return if (fileType == FileType.DIRECTORY) {
builder.setTypeDir().build()
} else {
builder.setTypeFile().setSize(resultSet.getLong("size")).setHash(resultSet.getString("hash")).build()
}
}
@Throws(SQLException::class, InvalidProtocolBufferException::class)
override fun findFileBlocks(folder: String, path: String): FileBlocks? {
getConnection().use { connection ->
connection.prepareStatement("SELECT * FROM file_blocks WHERE folder=? AND path=?").use { prepareStatement ->
prepareStatement.setString(1, folder)
prepareStatement.setString(2, path)
val resultSet = prepareStatement.executeQuery()
return if (resultSet.first()) {
readFileBlocks(resultSet)
} else {
null
}
}
}
}
@Throws(SQLException::class, InvalidProtocolBufferException::class)
private fun readFileBlocks(resultSet: ResultSet): FileBlocks {
val blocks = BlockExchangeExtraProtos.Blocks.parseFrom(resultSet.getBytes("blocks"))
val blockList = blocks.blocksList.map { record ->
BlockInfo(record!!.offset, record.size, Hex.toHexString(record.hash.toByteArray()))
}
return FileBlocks(resultSet.getString("folder"), resultSet.getString("path"), blockList)
}
@Throws(SQLException::class)
override fun updateFileInfo(fileInfo: FileInfo, fileBlocks: FileBlocks?) {
val version = fileInfo.versionList.last()
//TODO open transsaction, rollback
getConnection().use { connection ->
if (fileBlocks != null) {
FileInfo.checkBlocks(fileInfo, fileBlocks)
connection.prepareStatement("MERGE INTO file_blocks"
+ " (folder,path,hash,size,blocks)"
+ " VALUES (?,?,?,?,?)").use { prepareStatement ->
prepareStatement.setString(1, fileBlocks.folder)
prepareStatement.setString(2, fileBlocks.path)
prepareStatement.setString(3, fileBlocks.hash)
prepareStatement.setLong(4, fileBlocks.size)
prepareStatement.setBytes(5, BlockExchangeExtraProtos.Blocks.newBuilder()
.addAllBlocks(fileBlocks.blocks.map { input ->
BlockExchangeProtos.BlockInfo.newBuilder()
.setOffset(input.offset)
.setSize(input.size)
.setHash(ByteString.copyFrom(Hex.decode(input.hash)))
.build()
}).build().toByteArray())
prepareStatement.executeUpdate()
}
}
val oldFileInfo = findFileInfo(fileInfo.folder, fileInfo.path)
connection.prepareStatement("MERGE INTO file_info"
+ " (folder,path,file_name,parent,size,hash,last_modified,file_type,version_id,version_value,is_deleted)"
+ " VALUES (?,?,?,?,?,?,?,?,?,?,?)").use { prepareStatement ->
prepareStatement.setString(1, fileInfo.folder)
prepareStatement.setString(2, fileInfo.path)
prepareStatement.setString(3, fileInfo.fileName)
prepareStatement.setString(4, fileInfo.parent)
prepareStatement.setLong(7, fileInfo.lastModified.time)
prepareStatement.setString(8, fileInfo.type.name)
prepareStatement.setLong(9, version.id)
prepareStatement.setLong(10, version.value)
prepareStatement.setBoolean(11, fileInfo.isDeleted)
if (fileInfo.isDirectory()) {
prepareStatement.setNull(5, Types.BIGINT)
prepareStatement.setNull(6, Types.VARCHAR)
} else {
prepareStatement.setLong(5, fileInfo.size!!)
prepareStatement.setString(6, fileInfo.hash)
}
prepareStatement.executeUpdate()
}
//update stats
var deltaFileCount: Long = 0
var deltaDirCount: Long = 0
var deltaSize: Long = 0
val oldMissing = oldFileInfo == null || oldFileInfo.isDeleted
val newMissing = fileInfo.isDeleted
val oldSizeMissing = oldMissing || !oldFileInfo!!.isFile()
val newSizeMissing = newMissing || !fileInfo.isFile()
if (!oldSizeMissing) {
deltaSize -= oldFileInfo!!.size!!
}
if (!newSizeMissing) {
deltaSize += fileInfo.size!!
}
if (!oldMissing) {
if (oldFileInfo!!.isFile()) {
deltaFileCount--
} else if (oldFileInfo.isDirectory()) {
deltaDirCount--
}
}
if (!newMissing) {
if (fileInfo.isFile()) {
deltaFileCount++
} else if (fileInfo.isDirectory()) {
deltaDirCount++
}
}
val folderStats = updateFolderStats(connection, fileInfo.folder, deltaFileCount, deltaDirCount, deltaSize, fileInfo.lastModified)
onFolderStatsUpdatedListener?.invoke(object : IndexRepository.FolderStatsUpdatedEvent() {
override fun getFolderStats(): List<FolderStats> {
return listOf(folderStats)
}
})
}
}
@Throws(SQLException::class)
override fun findNotDeletedFilesByFolderAndParent(folder: String, parentPath: String): MutableList<FileInfo> {
getConnection().use { connection ->
connection.prepareStatement("SELECT * FROM file_info WHERE folder=? AND parent=? AND is_deleted=FALSE").use { prepareStatement ->
prepareStatement.setString(1, folder)
prepareStatement.setString(2, parentPath)
val resultSet = prepareStatement.executeQuery()
val list = mutableListOf<FileInfo>()
while (resultSet.next()) {
list.add(readFileInfo(resultSet))
}
return list
}
}
}
@Throws(SQLException::class)
override fun findFileInfoBySearchTerm(query: String): List<FileInfo> {
assert(query.isNotBlank())
// checkArgument(maxResult > 0);
// try (Connection connection = getConnection(); PreparedStatement preparedStatement = connection.prepareStatement("SELECT * FROM file_info WHERE LOWER(file_name) LIKE ? AND is_deleted=FALSE LIMIT ?")) {
getConnection().use { connection ->
connection.prepareStatement("SELECT * FROM file_info WHERE LOWER(file_name) REGEXP ? AND is_deleted=FALSE").use { preparedStatement ->
// try (Connection connection = getConnection(); PreparedStatement prepareStatement = connection.prepareStatement("SELECT * FROM file_info LIMIT 10")) {
// preparedStatement.setString(1, "%" + query.trim().toLowerCase() + "%");
preparedStatement.setString(1, query.trim { it <= ' ' }.toLowerCase())
// preparedStatement.setInt(2, maxResult);
val resultSet = preparedStatement.executeQuery()
val list = mutableListOf<FileInfo>()
while (resultSet.next()) {
list.add(readFileInfo(resultSet))
}
return list
}
}
}
@Throws(SQLException::class)
override fun countFileInfoBySearchTerm(query: String): Long {
assert(query.isNotBlank())
getConnection().use { connection ->
connection.prepareStatement("SELECT COUNT(*) FROM file_info WHERE LOWER(file_name) REGEXP ? AND is_deleted=FALSE").use { preparedStatement ->
// try (Connection connection = getConnection(); PreparedStatement preparedStatement = connection.prepareStatement("SELECT COUNT(*) FROM file_info")) {
preparedStatement.setString(1, query.trim { it <= ' ' }.toLowerCase())
val resultSet = preparedStatement.executeQuery()
assert(resultSet.first())
return resultSet.getLong(1)
}
}
}
// FILE INFO - END
override fun clearIndex() {
initDb()
sequencer = IndexRepoSequencer()
}
// FOLDER STATS - BEGIN
@Throws(SQLException::class)
private fun readFolderStats(resultSet: ResultSet): FolderStats {
return FolderStats.Builder()
.setFolder(resultSet.getString("folder"))
.setDirCount(resultSet.getLong("dir_count"))
.setFileCount(resultSet.getLong("file_count"))
.setSize(resultSet.getLong("size"))
.setLastUpdate(Date(resultSet.getLong("last_update")))
.build()
}
override fun findFolderStats(folder: String): FolderStats? {
return doFindFolderStats(folder)
}
@Throws(SQLException::class)
private fun doFindFolderStats(folder: String): FolderStats? {
getConnection().use { connection ->
connection.prepareStatement("SELECT * FROM folder_stats WHERE folder=?").use { prepareStatement ->
prepareStatement.setString(1, folder)
val resultSet = prepareStatement.executeQuery()
return if (resultSet.first()) {
readFolderStats(resultSet)
} else {
null
}
}
}
}
@Throws(SQLException::class)
override fun findAllFolderStats(): List<FolderStats> {
getConnection().use { connection ->
connection.prepareStatement("SELECT * FROM folder_stats").use { prepareStatement ->
val resultSet = prepareStatement.executeQuery()
val list = mutableListOf<FolderStats>()
while (resultSet.next()) {
val folderStats = readFolderStats(resultSet)
list.add(folderStats)
}
return list
}
}
}
@Throws(SQLException::class)
private fun updateFolderStats(connection: Connection, folder: String, deltaFileCount: Long, deltaDirCount: Long, deltaSize: Long, lastUpdate: Date): FolderStats {
val oldFolderStats = findFolderStats(folder)
val newFolderStats: FolderStats
if (oldFolderStats == null) {
newFolderStats = FolderStats.Builder()
.setDirCount(deltaDirCount)
.setFileCount(deltaFileCount)
.setFolder(folder)
.setLastUpdate(lastUpdate)
.setSize(deltaSize)
.build()
} else {
newFolderStats = oldFolderStats.copyBuilder()
.setDirCount(oldFolderStats.dirCount + deltaDirCount)
.setFileCount(oldFolderStats.fileCount + deltaFileCount)
.setSize(oldFolderStats.size + deltaSize)
.setLastUpdate(if (lastUpdate.after(oldFolderStats.lastUpdate)) lastUpdate else oldFolderStats.lastUpdate)
.build()
}
updateFolderStats(connection, newFolderStats)
return newFolderStats
}
// private void updateFolderStats() {
// logger.info("updateFolderStats BEGIN");
// final Map<String, FolderStats.Builder> map = Maps.newHashMap();
// final Function<String, FolderStats.Builder> func = new Function<String, FolderStats.Builder>() {
// @Override
// public FolderStats.Builder apply(String folder) {
// FolderStats.Builder res = map.get(folder);
// if (res == null) {
// res = FolderStats.newBuilder().setFolder(folder);
// map.put(folder, res);
// }
// return res;
// }
// };
// final List<FolderStats> list;
// try (Connection connection = getConnection()) {
// try (PreparedStatement prepareStatement = connection.prepareStatement("SELECT folder, COUNT(*) AS file_count, SUM(size) AS size, MAX(last_modified) AS last_update FROM file_info WHERE file_type=? AND is_deleted=FALSE GROUP BY folder")) {
// prepareStatement.setString(1, FileType.FILE.name());
// ResultSet resultSet = prepareStatement.executeQuery();
// while (resultSet.next()) {
// FolderStats.Builder builder = func.apply(resultSet.getString("folder"));
// builder.setSize(resultSet.getLong("size"));
// builder.setFileCount(resultSet.getLong("file_count"));
// builder.setLastUpdate(new Date(resultSet.getLong("last_update")));
// }
// }
// try (PreparedStatement prepareStatement = connection.prepareStatement("SELECT folder, COUNT(*) AS dir_count FROM file_info WHERE file_type=? AND is_deleted=FALSE GROUP BY folder")) {
// prepareStatement.setString(1, FileType.DIRECTORY.name());
// ResultSet resultSet = prepareStatement.executeQuery();
// while (resultSet.next()) {
// FolderStats.Builder builder = func.apply(resultSet.getString("folder"));
// builder.setDirCount(resultSet.getLong("dir_count"));
// }
// }
// list = Lists.newArrayList(Iterables.transform(map.values(), new Function<FolderStats.Builder, FolderStats>() {
// @Override
// public FolderStats apply(FolderStats.Builder builder) {
// return builder.build();
// }
// }));
// for (FolderStats folderStats : list) {
// updateFolderStats(connection, folderStats);
// }
// } catch (SQLException ex) {
// throw new RuntimeException(ex);
// }
// logger.info("updateFolderStats END");
// eventBus.post(new FolderStatsUpdatedEvent() {
// @Override
// public List<FolderStats> getFolderStats() {
// return Collections.unmodifiableList(list);
// }
// });
// }
@Throws(SQLException::class)
private fun updateFolderStats(connection: Connection, folderStats: FolderStats) {
assert(folderStats.fileCount >= 0)
assert(folderStats.dirCount >= 0)
assert(folderStats.size >= 0)
connection.prepareStatement("MERGE INTO folder_stats"
+ " (folder,file_count,dir_count,size,last_update)"
+ " VALUES (?,?,?,?,?)").use { prepareStatement ->
prepareStatement.setString(1, folderStats.folderId)
prepareStatement.setLong(2, folderStats.fileCount)
prepareStatement.setLong(3, folderStats.dirCount)
prepareStatement.setLong(4, folderStats.size)
prepareStatement.setLong(5, folderStats.lastUpdate.time)
prepareStatement.executeUpdate()
}
}
override fun close() {
logger.info("closing index repository (sql)")
// scheduledExecutorService.shutdown();
@@ -649,68 +227,6 @@ class SqlRepository(databaseFolder: File) : Closeable, IndexRepository, TempRepo
}
}
//SEQUENCER
private inner class IndexRepoSequencer : Sequencer {
private var indexId: Long? = null
private var currentSequence: Long? = null
@Throws(SQLException::class)
@Synchronized private fun loadFromDb() {
getConnection().use { connection ->
connection.prepareStatement("SELECT index_id,current_sequence FROM index_sequence").use { statement ->
val resultSet = statement.executeQuery()
assert(resultSet.first())
indexId = resultSet.getLong("index_id")
currentSequence = resultSet.getLong("current_sequence")
logger.info("loaded index info from db, index_id = {}, current_sequence = {}", indexId, currentSequence)
}
}
}
@Synchronized override fun indexId(): Long {
if (indexId == null) {
loadFromDb()
}
return indexId!!
}
@Throws(SQLException::class)
@Synchronized override fun nextSequence(): Long {
val nextSequence = currentSequence() + 1
getConnection().use { connection ->
connection.prepareStatement("UPDATE index_sequence SET current_sequence=?").use { statement ->
statement.setLong(1, nextSequence)
assert(statement.executeUpdate() == 1)
logger.debug("update local index sequence to {}", nextSequence)
}
}
currentSequence = nextSequence
return nextSequence
}
@Synchronized override fun currentSequence(): Long {
if (currentSequence == null) {
loadFromDb()
}
return currentSequence!!
}
}
@Throws(SQLException::class)
private fun readDeviceAddress(resultSet: ResultSet): DeviceAddress {
val instanceId = resultSet.getLong("instance_id")
return DeviceAddress.Builder()
.setAddress(resultSet.getString("address_url"))
.setDeviceId(DeviceId(resultSet.getString("device_id")))
.setInstanceId(if (instanceId == 0L) null else instanceId)
.setProducer(DeviceAddress.AddressProducer.valueOf(resultSet.getString("address_producer")))
.setScore(resultSet.getInt("address_score"))
.setLastModified(Date(resultSet.getLong("last_modified")))
.build()
}
companion object {
private const val VERSION = 13
}
@@ -0,0 +1,482 @@
/*
* Copyright (C) 2016 Davide Imbriaco
* Copyright (C) 2018 Jonas Lochmann
*
* This Java file is subject to the terms of the Mozilla Public
* License, v. 2.0. If a copy of the MPL was not distributed with this
* file, You can obtain one at http://mozilla.org/MPL/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 net.syncthing.java.repository.repo
import com.google.protobuf.ByteString
import com.google.protobuf.InvalidProtocolBufferException
import net.syncthing.java.bep.BlockExchangeExtraProtos
import net.syncthing.java.bep.BlockExchangeProtos
import net.syncthing.java.core.beans.*
import net.syncthing.java.core.interfaces.IndexTransaction
import net.syncthing.java.core.interfaces.Sequencer
import org.bouncycastle.util.encoders.Hex
import java.sql.Connection
import java.sql.ResultSet
import java.sql.SQLException
import java.sql.Types
import java.util.*
class SqlTransaction(
private val connection: Connection,
private val initDb: (Connection) -> Unit
): IndexTransaction, Sequencer {
private var closed = false
fun close() {
closed = true
}
private fun <T> runIfAllowed(block: () -> T): T {
if (closed) {
throw IllegalStateException("transaction already done")
}
return block()
}
override fun getSequencer() = this
override fun indexId(): Long = runIfAllowed {
connection.prepareStatement("SELECT index_id FROM index_sequence").use { statement ->
val resultSet = statement.executeQuery()
assert(resultSet.first())
resultSet.getLong("index_id")
}
}
override fun currentSequence(): Long = runIfAllowed {
connection.prepareStatement("SELECT current_sequence FROM index_sequence").use { statement ->
val resultSet = statement.executeQuery()
assert(resultSet.first())
resultSet.getLong("current_sequence")
}
}
override fun nextSequence(): Long = runIfAllowed {
connection.prepareStatement("UPDATE index_sequence SET current_sequence = current_sequence + 1").use { statement ->
assert(statement.executeUpdate() == 1)
}
currentSequence()
}
override fun updateIndexInfo(indexInfo: IndexInfo): Unit = runIfAllowed {
connection.prepareStatement("MERGE INTO folder_index_info"
+ " (folder,device_id,index_id,local_sequence,max_sequence)"
+ " VALUES (?,?,?,?,?)").use { prepareStatement ->
prepareStatement.setString(1, indexInfo.folderId)
prepareStatement.setString(2, indexInfo.deviceId)
prepareStatement.setLong(3, indexInfo.indexId)
prepareStatement.setLong(4, indexInfo.localSequence)
prepareStatement.setLong(5, indexInfo.maxSequence)
prepareStatement.executeUpdate()
}
}
override fun findIndexInfoByDeviceAndFolder(deviceId: DeviceId, folder: String): IndexInfo? = runIfAllowed {
doFindIndexInfoByDeviceAndFolder(deviceId, folder)
}
@Throws(SQLException::class)
private fun doFindIndexInfoByDeviceAndFolder(deviceId: DeviceId, folder: String): IndexInfo? = runIfAllowed {
connection.prepareStatement("SELECT * FROM folder_index_info WHERE device_id=? AND folder=?").use { prepareStatement ->
prepareStatement.setString(1, deviceId.deviceId)
prepareStatement.setString(2, folder)
val resultSet = prepareStatement.executeQuery()
if (resultSet.first()) {
readFolderIndexInfo(resultSet)
} else {
null
}
}
}
override fun findAllIndexInfos(): List<IndexInfo> = runIfAllowed {
connection.prepareStatement("SELECT * FROM folder_index_info").use { prepareStatement ->
val resultSet = prepareStatement.executeQuery()
val list = mutableListOf<IndexInfo>()
while (resultSet.next()) {
list.add(readFolderIndexInfo(resultSet))
}
list
}
}
@Throws(SQLException::class)
override fun findFileInfo(folder: String, path: String): FileInfo? = runIfAllowed {
connection.prepareStatement("SELECT * FROM file_info WHERE folder=? AND path=?").use { prepareStatement ->
prepareStatement.setString(1, folder)
prepareStatement.setString(2, path)
val resultSet = prepareStatement.executeQuery()
if (resultSet.first()) {
readFileInfo(resultSet)
} else {
null
}
}
}
override fun findFileInfo(folder: String, path: List<String>): Map<String, FileInfo> = runIfAllowed {
connection.prepareStatement("SELECT * FROM folder_index_info WHERE folder=? AND PATH IN ?").use { prepareStatement ->
prepareStatement.setString(1, folder)
prepareStatement.setArray(2, connection.createArrayOf("VARCHAR", path.toTypedArray()))
val resultSet = prepareStatement.executeQuery()
val map = mutableMapOf<String, FileInfo>()
while (resultSet.next()) {
val item = readFileInfo(resultSet)
map[item.path] = item
}
map
}
}
@Throws(SQLException::class)
override fun findFileInfoLastModified(folder: String, path: String): Date? = runIfAllowed {
connection.prepareStatement("SELECT last_modified FROM file_info WHERE folder=? AND path=?").use { prepareStatement ->
prepareStatement.setString(1, folder)
prepareStatement.setString(2, path)
val resultSet = prepareStatement.executeQuery()
if (resultSet.first()) {
Date(resultSet.getLong("last_modified"))
} else {
null
}
}
}
@Throws(SQLException::class)
override fun findNotDeletedFileInfo(folder: String, path: String): FileInfo? = runIfAllowed {
connection.prepareStatement("SELECT * FROM file_info WHERE folder=? AND path=? AND is_deleted=FALSE").use { prepareStatement ->
prepareStatement.setString(1, folder)
prepareStatement.setString(2, path)
val resultSet = prepareStatement.executeQuery()
if (resultSet.first()) {
readFileInfo(resultSet)
} else {
null
}
}
}
@Throws(SQLException::class)
private fun readFileInfo(resultSet: ResultSet): FileInfo {
val folder = resultSet.getString("folder")
val path = resultSet.getString("path")
val fileType = FileInfo.FileType.valueOf(resultSet.getString("file_type"))
val lastModified = Date(resultSet.getLong("last_modified"))
val versionList = listOf(FileInfo.Version(resultSet.getLong("version_id"), resultSet.getLong("version_value")))
val isDeleted = resultSet.getBoolean("is_deleted")
val builder = FileInfo.Builder()
.setFolder(folder)
.setPath(path)
.setLastModified(lastModified)
.setVersionList(versionList)
.setDeleted(isDeleted)
return if (fileType == FileInfo.FileType.DIRECTORY) {
builder.setTypeDir().build()
} else {
builder.setTypeFile().setSize(resultSet.getLong("size")).setHash(resultSet.getString("hash")).build()
}
}
@Throws(SQLException::class, InvalidProtocolBufferException::class)
override fun findFileBlocks(folder: String, path: String): FileBlocks? = runIfAllowed {
connection.prepareStatement("SELECT * FROM file_blocks WHERE folder=? AND path=?").use { prepareStatement ->
prepareStatement.setString(1, folder)
prepareStatement.setString(2, path)
val resultSet = prepareStatement.executeQuery()
if (resultSet.first()) {
readFileBlocks(resultSet)
} else {
null
}
}
}
@Throws(SQLException::class, InvalidProtocolBufferException::class)
private fun readFileBlocks(resultSet: ResultSet): FileBlocks {
val blocks = BlockExchangeExtraProtos.Blocks.parseFrom(resultSet.getBytes("blocks"))
val blockList = blocks.blocksList.map { record ->
BlockInfo(record!!.offset, record.size, Hex.toHexString(record.hash.toByteArray()))
}
return FileBlocks(resultSet.getString("folder"), resultSet.getString("path"), blockList)
}
@Throws(SQLException::class)
override fun updateFileInfo(fileInfo: FileInfo, fileBlocks: FileBlocks?): Unit = runIfAllowed {
val version = fileInfo.versionList.last()
if (fileBlocks != null) {
FileInfo.checkBlocks(fileInfo, fileBlocks)
connection.prepareStatement("MERGE INTO file_blocks"
+ " (folder,path,hash,size,blocks)"
+ " VALUES (?,?,?,?,?)").use { prepareStatement ->
prepareStatement.setString(1, fileBlocks.folder)
prepareStatement.setString(2, fileBlocks.path)
prepareStatement.setString(3, fileBlocks.hash)
prepareStatement.setLong(4, fileBlocks.size)
prepareStatement.setBytes(5, BlockExchangeExtraProtos.Blocks.newBuilder()
.addAllBlocks(fileBlocks.blocks.map { input ->
BlockExchangeProtos.BlockInfo.newBuilder()
.setOffset(input.offset)
.setSize(input.size)
.setHash(ByteString.copyFrom(Hex.decode(input.hash)))
.build()
}).build().toByteArray())
prepareStatement.executeUpdate()
}
}
connection.prepareStatement("MERGE INTO file_info"
+ " (folder,path,file_name,parent,size,hash,last_modified,file_type,version_id,version_value,is_deleted)"
+ " VALUES (?,?,?,?,?,?,?,?,?,?,?)").use { prepareStatement ->
prepareStatement.setString(1, fileInfo.folder)
prepareStatement.setString(2, fileInfo.path)
prepareStatement.setString(3, fileInfo.fileName)
prepareStatement.setString(4, fileInfo.parent)
prepareStatement.setLong(7, fileInfo.lastModified.time)
prepareStatement.setString(8, fileInfo.type.name)
prepareStatement.setLong(9, version.id)
prepareStatement.setLong(10, version.value)
prepareStatement.setBoolean(11, fileInfo.isDeleted)
if (fileInfo.isDirectory()) {
prepareStatement.setNull(5, Types.BIGINT)
prepareStatement.setNull(6, Types.VARCHAR)
} else {
prepareStatement.setLong(5, fileInfo.size!!)
prepareStatement.setString(6, fileInfo.hash)
}
prepareStatement.executeUpdate()
}
}
override fun updateFileInfoAndBlocks(fileInfos: List<FileInfo>, fileBlocks: List<FileBlocks>) = runIfAllowed {
connection.prepareStatement("MERGE INTO file_blocks"
+ " (folder,path,hash,size,blocks)"
+ " VALUES (?,?,?,?,?)").use { prepareStatement ->
fileBlocks.forEach { block ->
prepareStatement.setString(1, block.folder)
prepareStatement.setString(2, block.path)
prepareStatement.setString(3, block.hash)
prepareStatement.setLong(4, block.size)
prepareStatement.setBytes(5, BlockExchangeExtraProtos.Blocks.newBuilder()
.addAllBlocks(block.blocks.map { input ->
BlockExchangeProtos.BlockInfo.newBuilder()
.setOffset(input.offset)
.setSize(input.size)
.setHash(ByteString.copyFrom(Hex.decode(input.hash)))
.build()
}).build().toByteArray())
prepareStatement.executeUpdate()
}
}
connection.prepareStatement("MERGE INTO file_info"
+ " (folder,path,file_name,parent,size,hash,last_modified,file_type,version_id,version_value,is_deleted)"
+ " VALUES (?,?,?,?,?,?,?,?,?,?,?)").use { prepareStatement ->
fileInfos.forEach { fileInfo ->
val version = fileInfo.versionList.last()
prepareStatement.setString(1, fileInfo.folder)
prepareStatement.setString(2, fileInfo.path)
prepareStatement.setString(3, fileInfo.fileName)
prepareStatement.setString(4, fileInfo.parent)
prepareStatement.setLong(7, fileInfo.lastModified.time)
prepareStatement.setString(8, fileInfo.type.name)
prepareStatement.setLong(9, version.id)
prepareStatement.setLong(10, version.value)
prepareStatement.setBoolean(11, fileInfo.isDeleted)
if (fileInfo.isDirectory()) {
prepareStatement.setNull(5, Types.BIGINT)
prepareStatement.setNull(6, Types.VARCHAR)
} else {
prepareStatement.setLong(5, fileInfo.size!!)
prepareStatement.setString(6, fileInfo.hash)
}
prepareStatement.executeUpdate()
}
}
}
@Throws(SQLException::class)
override fun findNotDeletedFilesByFolderAndParent(folder: String, parentPath: String): MutableList<FileInfo> = runIfAllowed {
connection.prepareStatement("SELECT * FROM file_info WHERE folder=? AND parent=? AND is_deleted=FALSE").use { prepareStatement ->
prepareStatement.setString(1, folder)
prepareStatement.setString(2, parentPath)
val resultSet = prepareStatement.executeQuery()
val list = mutableListOf<FileInfo>()
while (resultSet.next()) {
list.add(readFileInfo(resultSet))
}
list
}
}
@Throws(SQLException::class)
override fun findFileInfoBySearchTerm(query: String): List<FileInfo> = runIfAllowed {
assert(query.isNotBlank())
// checkArgument(maxResult > 0);
// try (Connection connection = getConnection(); PreparedStatement preparedStatement = connection.prepareStatement("SELECT * FROM file_info WHERE LOWER(file_name) LIKE ? AND is_deleted=FALSE LIMIT ?")) {
connection.prepareStatement("SELECT * FROM file_info WHERE LOWER(file_name) REGEXP ? AND is_deleted=FALSE").use { preparedStatement ->
// try (Connection connection = getConnection(); PreparedStatement prepareStatement = connection.prepareStatement("SELECT * FROM file_info LIMIT 10")) {
// preparedStatement.setString(1, "%" + query.trim().toLowerCase() + "%");
preparedStatement.setString(1, query.trim { it <= ' ' }.toLowerCase())
// preparedStatement.setInt(2, maxResult);
val resultSet = preparedStatement.executeQuery()
val list = mutableListOf<FileInfo>()
while (resultSet.next()) {
list.add(readFileInfo(resultSet))
}
list
}
}
@Throws(SQLException::class)
override fun countFileInfoBySearchTerm(query: String): Long = runIfAllowed {
assert(query.isNotBlank())
connection.prepareStatement("SELECT COUNT(*) FROM file_info WHERE LOWER(file_name) REGEXP ? AND is_deleted=FALSE").use { preparedStatement ->
// try (Connection connection = getConnection(); PreparedStatement preparedStatement = connection.prepareStatement("SELECT COUNT(*) FROM file_info")) {
preparedStatement.setString(1, query.trim { it <= ' ' }.toLowerCase())
val resultSet = preparedStatement.executeQuery()
assert(resultSet.first())
resultSet.getLong(1)
}
}
// FILE INFO - END
override fun clearIndex() = runIfAllowed {
initDb(connection)
}
// FOLDER STATS - BEGIN
@Throws(SQLException::class)
private fun readFolderStats(resultSet: ResultSet) = FolderStats(
folderId = resultSet.getString("folder"),
dirCount = resultSet.getLong("dir_count"),
fileCount = resultSet.getLong("file_count"),
size = resultSet.getLong("size"),
lastUpdate = Date(resultSet.getLong("last_update"))
)
override fun findFolderStats(folder: String): FolderStats? {
return doFindFolderStats(folder)
}
@Throws(SQLException::class)
private fun doFindFolderStats(folder: String): FolderStats? = runIfAllowed {
connection.prepareStatement("SELECT * FROM folder_stats WHERE folder=?").use { prepareStatement ->
prepareStatement.setString(1, folder)
val resultSet = prepareStatement.executeQuery()
if (resultSet.first()) {
readFolderStats(resultSet)
} else {
null
}
}
}
@Throws(SQLException::class)
override fun findAllFolderStats(): List<FolderStats> = runIfAllowed {
connection.prepareStatement("SELECT * FROM folder_stats").use { prepareStatement ->
val resultSet = prepareStatement.executeQuery()
val list = mutableListOf<FolderStats>()
while (resultSet.next()) {
val folderStats = readFolderStats(resultSet)
list.add(folderStats)
}
list
}
}
override fun updateOrInsertFolderStats(folder: String, deltaFileCount: Long, deltaDirCount: Long, deltaSize: Long, lastUpdate: Date) {
updateFolderStats(connection, folder, deltaFileCount, deltaDirCount, deltaSize, lastUpdate)
}
@Throws(SQLException::class)
private fun updateFolderStats(
connection: Connection,
folder: String,
deltaFileCount: Long,
deltaDirCount: Long,
deltaSize: Long,
lastUpdate: Date
): FolderStats = runIfAllowed {
val oldFolderStats = findFolderStats(folder)
val newFolderStats: FolderStats
if (oldFolderStats == null) {
newFolderStats = FolderStats(
dirCount = deltaDirCount,
fileCount = deltaFileCount,
folderId = folder,
lastUpdate = lastUpdate,
size = deltaSize
)
} else {
newFolderStats = oldFolderStats.copy(
dirCount = oldFolderStats.dirCount + deltaDirCount,
fileCount = oldFolderStats.fileCount + deltaFileCount,
size = oldFolderStats.size + deltaSize,
lastUpdate = if (lastUpdate.after(oldFolderStats.lastUpdate)) lastUpdate else oldFolderStats.lastUpdate
)
}
updateFolderStats(connection, newFolderStats)
newFolderStats
}
@Throws(SQLException::class)
private fun updateFolderStats(connection: Connection, folderStats: FolderStats) {
assert(folderStats.fileCount >= 0)
assert(folderStats.dirCount >= 0)
assert(folderStats.size >= 0)
connection.prepareStatement("MERGE INTO folder_stats"
+ " (folder,file_count,dir_count,size,last_update)"
+ " VALUES (?,?,?,?,?)").use { prepareStatement ->
prepareStatement.setString(1, folderStats.folderId)
prepareStatement.setLong(2, folderStats.fileCount)
prepareStatement.setLong(3, folderStats.dirCount)
prepareStatement.setLong(4, folderStats.size)
prepareStatement.setLong(5, folderStats.lastUpdate.time)
prepareStatement.executeUpdate()
}
}
@Throws(SQLException::class)
private fun readFolderIndexInfo(resultSet: ResultSet) = IndexInfo(
folderId = resultSet.getString("folder"),
deviceId = resultSet.getString("device_id"),
indexId = resultSet.getLong("index_id"),
localSequence = resultSet.getLong("local_sequence"),
maxSequence = resultSet.getLong("max_sequence")
)
}