92 lines
4.1 KiB
Java
92 lines
4.1 KiB
Java
package org.mercury_im.messenger.xmpp.repository;
|
|
|
|
import org.mercury_im.core.util.ThreadUtils;
|
|
import org.mercury_im.messenger.xmpp.model.ChatModel;
|
|
import org.mercury_im.messenger.xmpp.model.ContactModel;
|
|
import org.mercury_im.messenger.xmpp.model.PeerModel;
|
|
import org.mercury_im.messenger.xmpp.util.ChatAndPossiblyContact;
|
|
|
|
import java.util.ArrayList;
|
|
import java.util.List;
|
|
import java.util.logging.Level;
|
|
import java.util.logging.Logger;
|
|
|
|
import javax.inject.Inject;
|
|
import javax.inject.Named;
|
|
|
|
import io.reactivex.Observable;
|
|
import io.reactivex.Scheduler;
|
|
import io.requery.Persistable;
|
|
import io.requery.reactivex.ReactiveEntityStore;
|
|
import io.requery.reactivex.ReactiveResult;
|
|
|
|
public class ChatRepository extends AbstractRepository<ChatModel> {
|
|
|
|
private final Logger LOGGER = Logger.getLogger(ChatRepository.class.getName());
|
|
|
|
@Inject
|
|
public ChatRepository(ReactiveEntityStore<Persistable> data,
|
|
@Named(value = ThreadUtils.SCHEDULER_IO) Scheduler subscriberScheduler,
|
|
@Named(value = ThreadUtils.SCHEDULER_UI) Scheduler observerScheduler) {
|
|
super(ChatModel.class, data, subscriberScheduler, observerScheduler);
|
|
}
|
|
|
|
public Observable<ReactiveResult<ChatModel>> getChatWith(PeerModel entityModel) {
|
|
return getChatWith(entityModel.getAccount().getId(), entityModel.getAddress());
|
|
}
|
|
|
|
public Observable<ReactiveResult<ChatModel>> getChatWith(long accountId, String address) {
|
|
return data().select(ChatModel.class).join(PeerModel.class).on(ChatModel.PEER_ID.eq(PeerModel.ID))
|
|
.where(PeerModel.ACCOUNT_ID.eq(accountId).and(PeerModel.ADDRESS.eq(address)))
|
|
.get().observableResult()
|
|
.doOnError(error -> LOGGER.log(Level.WARNING, "An error occurred while getting chat", error))
|
|
.subscribeOn(subscriberScheduler())
|
|
.observeOn(observerScheduler());
|
|
}
|
|
|
|
public Observable<ReactiveResult<ChatModel>> getVisibleChats() {
|
|
return data().select(ChatModel.class).where(ChatModel.DISPLAYED.eq(true))
|
|
.get().observableResult()
|
|
.subscribeOn(subscriberScheduler())
|
|
.observeOn(observerScheduler());
|
|
}
|
|
|
|
public Observable<List<ChatAndPossiblyContact>> getChatsAndContacts() {
|
|
return Observable.fromCallable(() -> {
|
|
List<ChatModel> chats = data().select(ChatModel.class)
|
|
.from(ChatModel.class).leftJoin(ContactModel.class).on(ChatModel.PEER_ID.eq(ContactModel.ENTITY_ID))
|
|
.get().toList();
|
|
List<ChatAndPossiblyContact> chatsAndContacts = new ArrayList<>(chats.size());
|
|
for (ChatModel chatModel : chats) {
|
|
ContactModel contactModel = data().select(ContactModel.class).from(ContactModel.class).where(ContactModel.ENTITY_ID.eq(chatModel.getPeer().getId()))
|
|
.get().first();
|
|
chatsAndContacts.add(new ChatAndPossiblyContact(chatModel, contactModel));
|
|
}
|
|
return chatsAndContacts;
|
|
})
|
|
.subscribeOn(subscriberScheduler())
|
|
.observeOn(observerScheduler());
|
|
}
|
|
|
|
public Observable<ChatAndPossiblyContact> getChatAndContact(long accountId, String address) {
|
|
|
|
return Observable.fromCallable(() -> {
|
|
ChatModel chat = data().select(ChatModel.class)
|
|
.from(ChatModel.class).join(PeerModel.class).on(ChatModel.PEER_ID.eq(PeerModel.ID))
|
|
.leftJoin(ContactModel.class).on(PeerModel.ID.eq(ContactModel.ENTITY_ID))
|
|
.where(PeerModel.ACCOUNT_ID.eq(accountId).and(PeerModel.ADDRESS.eq(address)))
|
|
.get().firstOrNull();
|
|
if (chat == null) {
|
|
return null;
|
|
}
|
|
|
|
ContactModel contactModel = data().select(ContactModel.class).from(ContactModel.class)
|
|
.where(ContactModel.ENTITY_ID.eq(chat.getPeer().getId()))
|
|
.get().firstOrNull();
|
|
return new ChatAndPossiblyContact(chat, contactModel);
|
|
})
|
|
.subscribeOn(subscriberScheduler())
|
|
.observeOn(observerScheduler());
|
|
}
|
|
}
|