nstream.adapter.druid.DruidEgressConnectionProvision Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of nstream-adapter-druid Show documentation

Templates for consuming from and producing to Apache Druid with Swim

The newest version!

// Copyright 2015-2024 Nstream, inc.
//
// Licensed under the Redis Source Available License 2.0 (RSALv2) Agreement;
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
//     https://redis.com/legal/rsalv2-agreement/
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.

package nstream.adapter.druid;

import java.nio.file.Files;
import java.nio.file.Paths;
import java.util.Properties;
import nstream.adapter.common.provision.AbstractProvision;
import org.apache.hc.client5.http.classic.methods.HttpPost;
import org.apache.hc.core5.http.io.entity.StringEntity;
import swim.util.Log;

public final class DruidEgressConnectionProvision extends AbstractProvision {

  public DruidEgressConnectionProvision() {
  }

  @Override
  protected void loadValue(Log log, Properties properties) {
    try {
      final String ingestionUrl = properties.getProperty("druidEgressUrl");
      final String specFilePath = properties.getProperty("druidEgressSpecFilePath");
      final String jsonPayload = new String(Files.readAllBytes(Paths.get(specFilePath)));
      this.value = new HttpPost(ingestionUrl);
      this.value.setEntity(new StringEntity(jsonPayload));
      this.value.setHeader("Content-type", "application/json");
    } catch (Exception e) {
      this.value = null;
      log.error("Did not load DruidEgressConnectionProvision due to issue loading HttpPost from properties");
      throw new RuntimeException("Failed to load DruidEgressConnectionProvision", e);
    }
  }

  @Override
  protected void unloadValue(Log log) {
    this.value = null;
  }

}