com.jdroid.javaweb.firebase.fcm.PushServiceImpl Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of jdroid-java-webapp-firebase-fcm Show documentation
Firebase FCM support
The newest version!
package com.jdroid.javaweb.firebase.fcm;

import com.google.firebase.messaging.Message;
import com.jdroid.java.concurrent.ExecutorUtils;
import com.jdroid.java.date.DateUtils;
import com.jdroid.javaweb.application.Application;
import com.jdroid.javaweb.config.CoreConfigParameter;

import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;

import java.util.Map.Entry;

@Service
public class PushServiceImpl implements PushService {
	
	@Autowired
	private DeviceRepository deviceRepository;

	@Autowired(required = false)
	private PushServiceListener pushServiceListener;

	private FcmSenderListener fcmSenderListener = new SenderListener();

	@Override
	public void addDevice(Device device, Boolean updateLastActiveTimestamp) {
		device.setId(generateId(device.getDeviceType(), device.getInstanceId()));
		
		Device deviceToUpdate = deviceRepository.get(device.getId());
		if (deviceToUpdate != null) {
			if (isDeviceUpdateRequired(deviceToUpdate, device)) {
				deviceToUpdate.setLastActiveTimestamp(updateLastActiveTimestamp ? DateUtils.INSTANCE.nowMillis() : device.getLastActiveTimestamp());
				deviceToUpdate.setRegistrationToken(device.getRegistrationToken());
				deviceToUpdate.setDeviceGroupId(device.getDeviceGroupId());
				deviceToUpdate.setAppVersionCode(device.getAppVersionCode());
				deviceToUpdate.setDeviceOsVersion(device.getDeviceOsVersion());
				deviceToUpdate.setAcceptLanguage(device.getAcceptLanguage());
				deviceToUpdate.setExtras(device.getExtras());
				deviceRepository.update(deviceToUpdate);
				if (pushServiceListener != null) {
					pushServiceListener.onUpdateDevice(deviceToUpdate.getInstanceId(), deviceToUpdate.getDeviceType());
				}
				
			}
		} else {
			Long now = DateUtils.INSTANCE.nowMillis();
			device.setCreationTimestamp(now);
			device.setLastActiveTimestamp(now);
			deviceRepository.add(device);
			if (pushServiceListener != null) {
				pushServiceListener.onAddDevice(device.getInstanceId(), device.getDeviceType());
			}
		}
	}
	
	protected Boolean isDeviceUpdateRequired(Device oldDevice, Device newDevice) {
		newDevice.setLastActiveTimestamp(oldDevice.getLastActiveTimestamp());
		newDevice.setId(oldDevice.getId());
		return !oldDevice.equals(newDevice) || DateUtils.INSTANCE.nowMillis() - oldDevice.getLastActiveTimestamp() > Application.get().getRemoteConfigLoader().getLong(CoreConfigParameter.DEVICE_UPDATE_REQUIRED_DURATION);
	}
	
	private String generateId(DeviceType deviceType, String instanceId) {
		return deviceType.getUserAgent() + "-" + instanceId;
	}

	@Override
	public void removeDevice(String instanceId, DeviceType deviceType) {
		Device deviceToRemove = deviceRepository.get(generateId(deviceType, instanceId));
		if (deviceToRemove != null) {
			deviceRepository.remove(deviceToRemove);
			if (pushServiceListener != null) {
				pushServiceListener.onRemoveDevice(deviceToRemove.getInstanceId(), deviceToRemove.getDeviceType());
			}
		}
	}

	@Override
	public void send(Message message) {
		ExecutorUtils.INSTANCE.execute(new Runnable() {
			@Override
			public void run() {
				FcmSender.get().send(message, fcmSenderListener);
			}
		});
	}

	private class SenderListener implements FcmSenderListener {
		@Override
		public void onSuccessfulSend(MessageSendingResponse messageSendingResponse) {

			// TODO By Google recommendation, we should remove all the tokens that weren't recently updated
			for (String each : messageSendingResponse.getRegistrationTokensToRemove()) {
				Device deviceToRemove = deviceRepository.findByRegistrationToken(each, messageSendingResponse.getDeviceType());
				if (deviceToRemove != null) {
					deviceRepository.remove(deviceToRemove);
					if (pushServiceListener != null) {
						pushServiceListener.onRemoveDevice(deviceToRemove.getInstanceId(), deviceToRemove.getDeviceType());
					}
				}
			}

			for (Entry entry : messageSendingResponse.getRegistrationTokensToReplace().entrySet()) {
				Device deviceToUpdate = deviceRepository.findByRegistrationToken(entry.getKey(), messageSendingResponse.getDeviceType());
				if (deviceToUpdate != null) {
					deviceToUpdate.setRegistrationToken(entry.getValue());
					deviceRepository.update(deviceToUpdate);
					if (pushServiceListener != null) {
						pushServiceListener.onUpdateDevice(deviceToUpdate.getInstanceId(), deviceToUpdate.getDeviceType());
					}
				}
			}
		}

		@Override
		public void onErrorSend(String errorCode) {

		}
	}
}