org.apache.camel.component.digitalocean.producer.DigitalOceanSnapshotsProducer Maven / Gradle / Ivy
/*
* Licensed to the Apache Software Foundation (ASF) under one or more
* contributor license agreements. See the NOTICE file distributed with
* this work for additional information regarding copyright ownership.
* The ASF licenses this file to You under the Apache License, Version 2.0
* (the "License"); you may not use this file except in compliance with
* the License. You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* 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 org.apache.camel.component.digitalocean.producer;
import com.myjeeva.digitalocean.exception.DigitalOceanException;
import com.myjeeva.digitalocean.exception.RequestUnsuccessfulException;
import com.myjeeva.digitalocean.pojo.Delete;
import com.myjeeva.digitalocean.pojo.Snapshot;
import com.myjeeva.digitalocean.pojo.Snapshots;
import org.apache.camel.Exchange;
import org.apache.camel.component.digitalocean.DigitalOceanConfiguration;
import org.apache.camel.component.digitalocean.DigitalOceanEndpoint;
import org.apache.camel.component.digitalocean.constants.DigitalOceanHeaders;
import org.apache.camel.component.digitalocean.constants.DigitalOceanSnapshotTypes;
import org.apache.camel.util.ObjectHelper;
/**
* The DigitalOcean producer for Snapshots API.
*/
public class DigitalOceanSnapshotsProducer extends DigitalOceanProducer {
public DigitalOceanSnapshotsProducer(DigitalOceanEndpoint endpoint, DigitalOceanConfiguration configuration) {
super(endpoint, configuration);
}
@Override
public void process(Exchange exchange) throws Exception {
switch (determineOperation(exchange)) {
case list:
getSnapshots(exchange);
break;
case get:
getSnapshot(exchange);
break;
case delete:
deleteSnapshot(exchange);
break;
default:
throw new IllegalArgumentException("Unsupported operation");
}
}
private void getSnapshots(Exchange exchange) throws RequestUnsuccessfulException, DigitalOceanException {
DigitalOceanSnapshotTypes type = exchange.getIn().getHeader(DigitalOceanHeaders.TYPE, DigitalOceanSnapshotTypes.class);
Snapshots snapshots;
if (ObjectHelper.isNotEmpty(type)) {
if (type == DigitalOceanSnapshotTypes.droplet) {
snapshots = getEndpoint().getDigitalOceanClient().getAllDropletSnapshots(configuration.getPage(),
configuration.getPerPage());
} else {
snapshots = getEndpoint().getDigitalOceanClient().getAllVolumeSnapshots(configuration.getPage(),
configuration.getPerPage());
}
} else {
snapshots = getEndpoint().getDigitalOceanClient().getAvailableSnapshots(configuration.getPage(),
configuration.getPerPage());
}
LOG.trace("All Snapshots : page {} / {} per page [{}] ", configuration.getPage(), configuration.getPerPage(),
snapshots.getSnapshots());
exchange.getMessage().setBody(snapshots.getSnapshots());
}
private void getSnapshot(Exchange exchange) throws RequestUnsuccessfulException, DigitalOceanException {
String snapshotId = exchange.getIn().getHeader(DigitalOceanHeaders.ID, String.class);
if (ObjectHelper.isEmpty(snapshotId)) {
throw new IllegalArgumentException(DigitalOceanHeaders.ID + " must be specified");
}
Snapshot snapshot = getEndpoint().getDigitalOceanClient().getSnaphotInfo(snapshotId);
LOG.trace("Snapshot [{}] ", snapshot);
exchange.getMessage().setBody(snapshot);
}
private void deleteSnapshot(Exchange exchange) throws RequestUnsuccessfulException, DigitalOceanException {
String snapshotId = exchange.getIn().getHeader(DigitalOceanHeaders.ID, String.class);
if (ObjectHelper.isEmpty(snapshotId)) {
throw new IllegalArgumentException(DigitalOceanHeaders.ID + " must be specified");
}
Delete delete = getEndpoint().getDigitalOceanClient().deleteSnapshot(snapshotId);
LOG.trace("Delete Snapshot [{}] ", delete);
exchange.getMessage().setBody(delete);
}
}