Package org.jboss.aerogear.simplepush.server.datastore.model

Examples of org.jboss.aerogear.simplepush.server.datastore.model.UserAgentDTO

98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
    @Override
    public boolean saveChannel(final Channel channel) {
        final JpaOperation<Boolean> saveChannel = new JpaOperation<Boolean>() {
            @Override
            public Boolean perform(final EntityManager em) {
                UserAgentDTO userAgent = em.find(UserAgentDTO.class, channel.getUAID());
                if (userAgent == null) {
                    userAgent = new UserAgentDTO(channel.getUAID());
                }
                userAgent.addChannel(channel.getChannelId(), channel.getVersion(), channel.getEndpointToken());
                em.merge(userAgent);
                return Boolean.TRUE;
            }
        };
        try {
View Full Code Here
152
153
154
155
156
157
158
159
160
161
162
163
164
    public Set<String> getChannelIds(final String uaid) {
        final JpaOperation<Set<String>> getChannelIds = new JpaOperation<Set<String>>() {
            @Override
            public Set<String> perform(final EntityManager em) {
                final Set<String> channels = new HashSet<String>();
                final UserAgentDTO userAgent = em.find(UserAgentDTO.class, uaid);
                if (userAgent != null) {
                    for (ChannelDTO dto : userAgent.getChannels()) {
                        channels.add(dto.getChannelId());
                    }
                }
                return channels;
            }
View Full Code Here
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
    @Override
    public void removeChannels(final String uaid) {
        final JpaOperation<Void> removeChannels = new JpaOperation<Void>() {
            @Override
            public Void perform(final EntityManager em) {
                final UserAgentDTO userAgent = em.find(UserAgentDTO.class, uaid);
                if (userAgent != null) {
                    final Set<ChannelDTO> channels = userAgent.getChannels();
                    for (ChannelDTO channelDTO : channels) {
                        em.remove(channelDTO);
                    }
                    channels.clear();
                    userAgent.setChannels(channels);
                }
                return null;
            }
        };
        jpaExecutor.execute(removeChannels);
View Full Code Here
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
    public String saveUnacknowledged(final String channelId, final long version) throws ChannelNotFoundException {
        final JpaOperation<String> saveAcks = new JpaOperation<String>() {
            @Override
            public String perform(final EntityManager em) {
                final ChannelDTO channel = em.find(ChannelDTO.class, channelId);
                final UserAgentDTO userAgent = channel.getUserAgent();
                final Set<AckDTO> dtos = new HashSet<AckDTO>();
                dtos.add(new AckDTO(userAgent, channel.getChannelId(), version));
                userAgent.setAcks(dtos);
                em.merge(userAgent);
                return userAgent.getUaid();
            }
        };
        return jpaExecutor.execute(saveAcks);
    }
View Full Code Here
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
    @Override
    public Set<Ack> getUnacknowledged(final String uaid) {
        final JpaOperation<Set<Ack>> getUnacks = new JpaOperation<Set<Ack>>() {
            @Override
            public Set<Ack> perform(final EntityManager em) {
                final UserAgentDTO userAgent = em.find(UserAgentDTO.class, uaid);
                if (userAgent == null) {
                    return Collections.emptySet();
                }
                final HashSet<Ack> acks = new HashSet<Ack>();
                for (AckDTO ackDTO : userAgent.getAcks()) {
                    acks.add(new AckImpl(ackDTO.getChannelId(), ackDTO.getVersion()));
                }
                return acks;
            }
        };
View Full Code Here
271
272
273
274
275
276
277
278
279
280
281
282
                    channelIds.add(ack.getChannelId());
                }
                final Query delete = em.createQuery("DELETE from AckDTO c where c.channelId in (:channelIds)");
                delete.setParameter("channelIds", channelIds);
                delete.executeUpdate();
                final UserAgentDTO userAgent = em.find(UserAgentDTO.class, uaid);
                final Set<AckDTO> acks = userAgent.getAcks();
                final Set<Ack> unacked = new HashSet<Ack>(acks.size());
                for (AckDTO ackDto : acks) {
                    unacked.add(new AckImpl(ackDto.getChannelId(), ackDto.getVersion()));
                }
                return unacked;
View Full Code Here
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
    @Test
    public void persistUserAgent() {
        final String channelId = UUID.randomUUID().toString();
        final String uaid = UUIDUtil.newUAID();
        final UserAgentDTO userAgent = persist(uaid, channelId, 1, "/endpoint/" + channelId);
        assertThat(entityManager.contains(userAgent), is(true));

        entityManager.getTransaction().begin();
        final UserAgentDTO ua = entityManager.find(UserAgentDTO.class, userAgent.getUaid());
        assertThat(ua.getUaid(), equalTo(uaid.toString()));
        assertThat(ua.getChannels().size(), is(1));
        assertThat(ua.getChannels().iterator().next().getChannelId(), equalTo(channelId));
        assertThat(ua.getChannels().iterator().next().getVersion(), equalTo(1L));
        entityManager.getTransaction().commit();
    }
View Full Code Here
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
    @Test
    public void removeUserAgent() {
        final String channelId = UUID.randomUUID().toString();
        final String uaid = UUIDUtil.newUAID();
        final UserAgentDTO userAgent = persist(uaid, channelId, 1, "/endpoint/" + channelId);
        assertThat(entityManager.contains(userAgent), is(true));
        entityManager.getTransaction().begin();
        entityManager.remove(userAgent);
        entityManager.getTransaction().commit();
        final UserAgentDTO ua = entityManager.find(UserAgentDTO.class, userAgent.getUaid());
        assertThat(ua, is(nullValue()));
        final ChannelDTO channel = entityManager.find(ChannelDTO.class, channelId);
        assertThat(channel, is(nullValue()));
    }
View Full Code Here
119
120
121
122
123
124
125
126
127
128
129
    @Test
    public void persistUpdate() {
        final String channelId = UUID.randomUUID().toString();
        final String uaid = UUIDUtil.newUAID();
        final UserAgentDTO userAgent = persist(uaid, channelId, 10, "/endpoint/" + channelId);

        entityManager.getTransaction().begin();
        final AckDTO update = new AckDTO(userAgent, channelId, 10);
        entityManager.persist(update);
        entityManager.getTransaction().commit();
View Full Code Here
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
    @Test
    public void persistUpdateToSameChannel() {
        final String channelId = UUID.randomUUID().toString();
        final String uaid = UUIDUtil.newUAID();
        final UserAgentDTO userAgent = persist(uaid, channelId, 10, "/endpoint/" + channelId);

        entityManager.getTransaction().begin();
        final AckDTO update = new AckDTO(userAgent, channelId, 10);
        entityManager.persist(update);
        entityManager.getTransaction().commit();

        entityManager.getTransaction().begin();
        final UserAgentDTO userAgentToUpdate = entityManager.find(UserAgentDTO.class, uaid);
        final AckDTO newUpdate = new AckDTO(userAgent, channelId, 11);
        userAgentToUpdate.setAcks(new HashSet<AckDTO>(Arrays.asList(newUpdate)));
        entityManager.merge(userAgentToUpdate);
        entityManager.getTransaction().commit();

        entityManager.getTransaction().begin();
        final AckDTO updated = entityManager.find(AckDTO.class, channelId);
View Full Code Here
TOP

Related Classes of org.jboss.aerogear.simplepush.server.datastore.model.UserAgentDTO

  • org.jboss.aerogear.simplepush.server.datastore.JpaDataStore
  • org.jboss.aerogear.simplepush.server.datastore.JpaEntitiesTest
    • Copyright © 2018 www.massapicom. All rights reserved.
    All source code are property of their respective owners. Java is a trademark of Sun Microsystems, Inc and owned by ORACLE Inc. Contact coftware#gmail.com.