1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
|
/*
* Copyright (C) 2018 Bell Canada.
*
* Licensed 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.onap.ccsdk.sli.adaptors.netbox.impl;
import com.google.common.annotations.VisibleForTesting;
import com.google.common.collect.Lists;
import com.google.gson.Gson;
import com.google.gson.GsonBuilder;
import com.google.gson.JsonSerializer;
import java.io.IOException;
import java.io.InputStreamReader;
import java.io.Reader;
import java.sql.SQLException;
import java.util.ArrayList;
import java.util.concurrent.CompletionException;
import org.apache.http.HttpResponse;
import org.onap.ccsdk.sli.adaptors.netbox.api.IpamException;
import org.onap.ccsdk.sli.adaptors.netbox.api.NetboxClient;
import org.onap.ccsdk.sli.adaptors.netbox.model.IPAddress;
import org.onap.ccsdk.sli.adaptors.netbox.model.Prefix;
import org.onap.ccsdk.sli.adaptors.netbox.model.Status;
import org.onap.ccsdk.sli.core.dblib.DbLibService;
import org.onap.ccsdk.sli.core.sli.SvcLogicJavaPlugin;
public class NetboxClientImpl implements NetboxClient, SvcLogicJavaPlugin {
private static final String NEXT_AVAILABLE_IP_IN_PREFIX_PATH = "/api/ipam/prefixes/%s/available-ips/";
private static final String IP_ADDRESS_PATH = "/api/ipam/ip-addresses/%s/";
private static final String EMPTY_STRING = "";
private static final String ID_MISSING_MSG = "Id must be set";
private static final String ASSIGN_IP_SQL_STATEMENT =
"INSERT INTO IPAM_IP_ASSIGNEMENT (service_instance_id, vf_module_id, prefix_id, ip_address_id, ip_adress, ip_status) \n"
+ "VALUES (?, ?, ?, ?, ?, ?)";
private static final String UNASSIGN_IP_SQL_STATEMENT =
"DELETE FROM IPAM_IP_ASSIGNEMENT WHERE service_instance_id = ? AND vf_module_id = ? AND ip_address_id = ?";
private final NetboxHttpClient client;
private final Gson gson;
private final DbLibService dbLibService;
public NetboxClientImpl(final NetboxHttpClient client, final DbLibService dbLibService) {
this.client = client;
this.dbLibService = dbLibService;
final JsonSerializer<Status> vlanStatusDeserializer = (val, type, context) -> val.toJson();
gson = new GsonBuilder()
.registerTypeAdapter(Status.class, vlanStatusDeserializer)
.create();
}
@Override
public IPAddress assign(final Prefix prefix, final String serviceInstanceId, final String vfModuleId)
throws IpamException, SQLException {
checkArgument(prefix.getId() != null);
try {
IPAddress ipAddress = client
.post(String.format(NEXT_AVAILABLE_IP_IN_PREFIX_PATH, prefix.getId()), EMPTY_STRING)
.thenApply(this::getIpAddress)
.toCompletableFuture()
.join();
ArrayList<String> args = Lists.newArrayList(serviceInstanceId,
vfModuleId,
String.valueOf(prefix.getId()),
String.valueOf(ipAddress.getId()),
ipAddress.getAddress(),
ipAddress.getStatus().getLabel());
dbLibService.writeData(ASSIGN_IP_SQL_STATEMENT, args, null);
return ipAddress;
} catch (CompletionException e) {
// Unwrap the CompletionException and wrap in IpamException
throw new IpamException("Fail to assign IP for Prefix(id= " + prefix.getId() + "). " + e.getMessage(),
e.getCause());
}
}
@Override
public void unassign(final IPAddress ipAddress, final String serviceInstanceId, final String vfModuleId)
throws IpamException, SQLException {
checkArgument(ipAddress.getId() != null);
try {
client.delete(String.format(IP_ADDRESS_PATH, ipAddress.getId()))
.thenAccept(this::checkResult)
.toCompletableFuture()
.join();
ArrayList<String> args = Lists.newArrayList(
serviceInstanceId,
vfModuleId,
String.valueOf(ipAddress.getId()));
dbLibService.writeData(UNASSIGN_IP_SQL_STATEMENT, args, null);
} catch (CompletionException e) {
// Unwrap the CompletionException and wrap in IpamException
throw new IpamException(
"Fail to unassign IP for IPAddress(id= " + ipAddress.getId() + "). " + e.getMessage(),
e.getCause());
}
}
@VisibleForTesting
IPAddress getIpAddress(final HttpResponse response) {
if (response.getStatusLine().getStatusCode() != 201) {
throw new IllegalStateException(NetboxHttpClient.getBodyAsString(response));
}
try (final Reader reader = new InputStreamReader(response.getEntity().getContent())) {
return gson.fromJson(reader, IPAddress.class);
} catch (final IOException e) {
throw new IllegalStateException(e);
}
}
private static void checkArgument(final boolean argument) throws IpamException {
if (!argument) {
throw new IpamException(ID_MISSING_MSG);
}
}
private void checkResult(final HttpResponse response) {
if (response.getStatusLine().getStatusCode() - 200 >= 100) {
throw new IllegalStateException(
"Netbox request failed with status: " + NetboxHttpClient.getBodyAsString(response));
}
}
}
|
