Compare commits
3 Commits
| Author | SHA1 | Date | |
|---|---|---|---|
| f336a2932f | |||
| 0b3e2bf914 | |||
| 6d9009daff |
+2
-2
@@ -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 +1 @@
|
||||
- new connection handling
|
||||
- option for users to get detailed crash reports
|
||||
- bugfixes
|
||||
- new (faster) index handling
|
||||
|
||||
@@ -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>
|
||||
|
||||
@@ -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>
|
||||
|
||||
@@ -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
|
||||
}
|
||||
}
|
||||
+44
-35
@@ -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)
|
||||
|
||||
+1
-1
@@ -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
|
||||
|
||||
+4
@@ -42,4 +42,8 @@ object ConnectionActorUtil {
|
||||
|
||||
deferred.await()
|
||||
}
|
||||
|
||||
suspend fun disconnect(actor: SendChannel<ConnectionAction>) {
|
||||
actor.send(CloseConnectionAction)
|
||||
}
|
||||
}
|
||||
|
||||
+7
-1
@@ -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)
|
||||
|
||||
)
|
||||
}
|
||||
}
|
||||
+17
@@ -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)
|
||||
}
|
||||
+148
@@ -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
|
||||
}
|
||||
}
|
||||
+25
@@ -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()
|
||||
}
|
||||
}
|
||||
|
||||
}
|
||||
|
||||
+2
-39
@@ -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)
|
||||
}
|
||||
}
|
||||
+2
-1
@@ -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
|
||||
|
||||
+15
-142
@@ -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
|
||||
}
|
||||
}
|
||||
|
||||
+169
@@ -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
|
||||
}
|
||||
+3
@@ -13,4 +13,7 @@ interface FileBlocksDao {
|
||||
|
||||
@Insert(onConflict = OnConflictStrategy.REPLACE)
|
||||
fun mergeBlock(blocksItem: FileBlocksItem)
|
||||
|
||||
@Insert(onConflict = OnConflictStrategy.REPLACE)
|
||||
fun mergeBlocks(blocksItem: List<FileBlocksItem>)
|
||||
}
|
||||
|
||||
+6
@@ -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>)
|
||||
}
|
||||
|
||||
+3
@@ -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>
|
||||
}
|
||||
|
||||
+7
-7
@@ -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
|
||||
)
|
||||
}
|
||||
}
|
||||
|
||||
+7
-7
@@ -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
|
||||
)
|
||||
}
|
||||
}
|
||||
|
||||
+22
-506
@@ -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
|
||||
}
|
||||
|
||||
+482
@@ -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")
|
||||
)
|
||||
}
|
||||
Reference in New Issue
Block a user