nstream.adapter.nats.NatsIngestingPatch Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of nstream-adapter-nats Show documentation
Templates for consuming from and producing to NATS 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.nats;

import io.nats.client.Message;
import nstream.adapter.common.AdapterUtils;
import nstream.adapter.common.ingress.AssemblyException;
import nstream.adapter.common.ingress.ValueAssembler;
import nstream.adapter.common.schedule.DeferrableException;
import swim.structure.Value;

public class NatsIngestingPatch extends NatsIngestingAgent {

  protected ValueAssembler valueAssembler;

  public NatsIngestingPatch() {
  }

  @Override
  protected void ingest(Message unstructured) throws DeferrableException {
    final Value v;
    try {
      v = assembleResponse(unstructured);
    } catch (Exception e) {
      error(e.getMessage());
      nak(unstructured);
      throw new DeferrableException(nodeUri() + ": Failed to assemble message with value "
          + new String(unstructured.getData()), e);
    }
    try {
      ingestStructure(v);
    } catch (DeferrableException | RuntimeException e) {
      error("Failed to ingest structure: " + e.getMessage());
      throw e;
    }
    ack(unstructured);
  }

  protected void ack(Message unstructured) {
    try {
      unstructured.ack();
    } catch (RuntimeException e) {
      error("Failed to ack message: " + e.getMessage());
    }
  }

  protected void nak(Message unstructured) {
    try {
      unstructured.nak();
    } catch (RuntimeException e) {
      error("Failed to nak message: " + e.getMessage());
    }
  }

  protected Value assembleResponse(Message unstructured) throws AssemblyException {
    return NatsAdapterUtils.assembleConsumerMessage(unstructured.getData(), this.valueAssembler,
        this.ingressSettings.contentTypeOverride());
  }

  protected void ingestStructure(Value messageStructure) throws DeferrableException {
    AdapterUtils.ingressDslRelay(this.ingressSettings.relaySchema(),
        agentContext(), messageStructure);
  }

}