コード例 #1
0
 private static ServiceAnnouncement getPrestoAnnouncement(Set<ServiceAnnouncement> announcements) {
   for (ServiceAnnouncement announcement : announcements) {
     if (announcement.getType().equals("presto")) {
       return announcement;
     }
   }
   throw new RuntimeException("Presto announcement not found: " + announcements);
 }
コード例 #2
0
  private static void updateDatasourcesAnnouncement(Announcer announcer, String connectorId) {
    //
    // This code was copied from PrestoServer, and is a hack that should be removed when the data
    // source property is removed
    //

    // get existing announcement
    ServiceAnnouncement announcement = getPrestoAnnouncement(announcer.getServiceAnnouncements());

    // update datasources property
    Map<String, String> properties = new LinkedHashMap<>(announcement.getProperties());
    String property = nullToEmpty(properties.get("datasources"));
    Set<String> datasources =
        new LinkedHashSet<>(
            Splitter.on(',').trimResults().omitEmptyStrings().splitToList(property));
    datasources.add(connectorId);
    properties.put("datasources", Joiner.on(',').join(datasources));

    // update announcement
    announcer.removeServiceAnnouncement(announcement.getId());
    announcer.addServiceAnnouncement(
        serviceAnnouncement(announcement.getType()).addProperties(properties).build());
    announcer.forceAnnounce();
  }