gerrit.onap Code Review - sdc.git/blob

   1 package org.openecomp.core.tools.store;
   2
   3 import com.datastax.driver.mapping.annotations.Accessor;
   4 import com.datastax.driver.mapping.annotations.Query;
   5 import org.openecomp.core.nosqldb.factory.NoSqlDbFactory;
   6
   7 import java.util.HashSet;
   8 import java.util.Set;
   9
  10 public class NotificationHandler {
  11
  12   public void registerNotificationForUserOnEntity(String user, String entityId) {
  13
  14     Set<String> userSet = new HashSet<>();
  15     userSet.add(user);
  16     NoSqlDbFactory.getInstance().createInterface().getMappingManager()
  17         .createAccessor(NotificationAccessor.class)
  18         .updateNotificationSubscription(userSet, entityId);
  19   }
  20
  21   @Accessor
  22   interface NotificationAccessor {
  23
  24     @Query("UPDATE dox.notification_subscribers SET subscribers = subscribers + ? where " +
  25         "entity_id = ?")
  26     void updateNotificationSubscription(Set<String> users, String entityId);
  27   }
  28 }