mikesafonov / spring-boot-starter-smpp Goto Github PK

private DeliveryReport toReport(DeliverSm deliverSm) throws DeliveryReceiptException {
        byte[] shortMessage = deliverSm.getShortMessage();
        String sms = new String(shortMessage);
        DeliveryReceipt deliveryReceipt = DeliveryReceipt.parseShortMessage(sms, DateTimeZone.UTC, false);
        return DeliveryReport.of(deliveryReceipt, clientId);
    }

public class ResponseSmppSessionHandler extends DefaultSmppSessionHandler {

    private final String clientId;
    private final List<DeliveryReportConsumer> deliveryReportConsumers;
    private final List<ReceivedMessageConsumer> receivedMessageConsumers;

    public ResponseSmppSessionHandler(String clientId, List<DeliveryReportConsumer> deliveryReportConsumers, List<ReceivedMessageConsumer> receivedMessageConsumers) {
        this.clientId = requireNonNull(clientId);
        this.deliveryReportConsumers = requireNonNull(deliveryReportConsumers);
        this.receivedMessageConsumers = requireNonNull(receivedMessageConsumers);
    }

    @Override
    public PduResponse firePduRequestReceived(PduRequest pduRequest) {
        log.debug(pduRequest.toString());
        if (isDelivery(pduRequest)) {
            processReport(pduRequest);
        }

        return pduRequest.createResponse();
    }

    private boolean isDelivery(PduRequest pduRequest) {
        return pduRequest.isRequest() && pduRequest.getClass() == DeliverSm.class;
    }

    private void processReport(PduRequest pduRequest) {
        DeliverSm dlr = (DeliverSm) pduRequest;
        try {
            if(dlr.getEsmClass() == 0x04){
                DeliveryReport report = toReport(dlr);
                for (DeliveryReportConsumer deliveryReportConsumer : deliveryReportConsumers) {
                    deliveryReportConsumer.accept(report);
                }
            }
            else{
                ReceivedMessage receivedMessage = toReceivedMessage(dlr);
                for (ReceivedMessageConsumer consumer: receivedMessageConsumers){
                    consumer.accept(receivedMessage);
                }
            }

        } catch (DeliveryReceiptException e) {
            log.error(e.getMessage(), e);
        }
    }

    private ReceivedMessage toReceivedMessage(DeliverSm deliverSm){
        ReceivedMessage receivedMessage = new ReceivedMessage();
        receivedMessage.setText(new String(deliverSm.getShortMessage()));
        receivedMessage.setSource(deliverSm.getSourceAddress().getAddress());
        receivedMessage.setMsisdn(deliverSm.getDestAddress().getAddress());
        receivedMessage.setMessageId(Integer.toString(deliverSm.getSequenceNumber()));
        return receivedMessage;
    }

    private DeliveryReport toReport(DeliverSm deliverSm) throws DeliveryReceiptException {
        byte[] shortMessage = deliverSm.getShortMessage();
        String sms = new String(shortMessage);
        DeliveryReceipt deliveryReceipt = DeliveryReceipt.parseShortMessage(sms, DateTimeZone.UTC, false);
        return DeliveryReport.of(deliveryReceipt, clientId);
    }
}