2 * Copyright © 2017-2018 AT&T Intellectual Property.
3 * Modifications Copyright © 2019 Bell Canada.
5 * Licensed under the Apache License, Version 2.0 (the "License");
6 * you may not use this file except in compliance with the License.
7 * You may obtain a copy of the License at
9 * http://www.apache.org/licenses/LICENSE-2.0
11 * Unless required by applicable law or agreed to in writing, software
12 * distributed under the License is distributed on an "AS IS" BASIS,
13 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
14 * See the License for the specific language governing permissions and
15 * limitations under the License.
18 package org.onap.ccsdk.cds.blueprintsprocessor.selfservice.api
20 import io.grpc.StatusException
21 import io.grpc.stub.StreamObserver
22 import org.onap.ccsdk.cds.blueprintsprocessor.core.BluePrintCoreConfiguration
23 import org.onap.ccsdk.cds.blueprintsprocessor.selfservice.api.utils.currentTimestamp
24 import org.onap.ccsdk.cds.controllerblueprints.common.api.CommonHeader
25 import org.onap.ccsdk.cds.controllerblueprints.common.api.Status
26 import org.onap.ccsdk.cds.controllerblueprints.core.interfaces.BluePrintCatalogService
27 import org.onap.ccsdk.cds.controllerblueprints.management.api.BluePrintManagementInput
28 import org.onap.ccsdk.cds.controllerblueprints.management.api.BluePrintManagementOutput
29 import org.onap.ccsdk.cds.controllerblueprints.management.api.BluePrintManagementServiceGrpc
30 import org.slf4j.LoggerFactory
31 import org.springframework.security.access.prepost.PreAuthorize
32 import org.springframework.stereotype.Service
36 open class BluePrintManagementGRPCHandler(private val bluePrintCoreConfiguration: BluePrintCoreConfiguration,
37 private val bluePrintCatalogService: BluePrintCatalogService)
38 : BluePrintManagementServiceGrpc.BluePrintManagementServiceImplBase() {
40 private val log = LoggerFactory.getLogger(BluePrintManagementGRPCHandler::class.java)
42 @PreAuthorize("hasRole('USER')")
43 override fun uploadBlueprint(request: BluePrintManagementInput, responseObserver: StreamObserver<BluePrintManagementOutput>) {
44 val blueprintName = request.blueprintName
45 val blueprintVersion = request.blueprintVersion
46 val blueprint = "blueprint $blueprintName:$blueprintVersion"
48 log.info("request(${request.commonHeader.requestId}): Received upload $blueprint")
50 val blueprintArchivedFilePath = "${bluePrintCoreConfiguration.archivePath}/$blueprintName/$blueprintVersion/$blueprintName.zip"
52 val blueprintArchivedFile = File(blueprintArchivedFilePath)
54 saveToDisk(request, blueprintArchivedFile)
55 val blueprintId = bluePrintCatalogService.saveToDatabase(blueprintArchivedFile)
57 File("${bluePrintCoreConfiguration.archivePath}/$blueprintName").deleteRecursively()
59 responseObserver.onNext(successStatus("Successfully uploaded $blueprint with id($blueprintId)", request.commonHeader))
60 responseObserver.onCompleted()
61 } catch (e: Exception) {
62 failStatus("request(${request.commonHeader.requestId}): Failed to upload $blueprint at path $blueprintArchivedFilePath", e)
66 @PreAuthorize("hasRole('USER')")
67 override fun removeBlueprint(request: BluePrintManagementInput, responseObserver: StreamObserver<BluePrintManagementOutput>) {
68 val blueprintName = request.blueprintName
69 val blueprintVersion = request.blueprintVersion
70 val blueprint = "blueprint $blueprintName:$blueprintVersion"
72 log.info("request(${request.commonHeader.requestId}): Received delete $blueprint")
75 bluePrintCatalogService.deleteFromDatabase(blueprintName, blueprintVersion)
76 responseObserver.onNext(successStatus("Successfully deleted $blueprint", request.commonHeader))
77 responseObserver.onCompleted()
78 } catch (e: Exception) {
79 failStatus("request(${request.commonHeader.requestId}): Failed to delete $blueprint", e)
83 private fun saveToDisk(request: BluePrintManagementInput, blueprintDir: File) {
84 log.info("request(${request.commonHeader.requestId}): Writing CBA File under :${blueprintDir.absolutePath}")
85 if (blueprintDir.exists()) {
86 log.info("request(${request.commonHeader.requestId}): Re-creating blueprint directory(${blueprintDir.absolutePath})")
87 //FileUtils.deleteDirectory(blueprintDir.parentFile)
88 blueprintDir.parentFile.deleteRecursively()
90 blueprintDir.parentFile.mkdirs()
91 //FileUtils.forceMkdir(blueprintDir.parentFile)
92 blueprintDir.writeBytes(request.fileChunk.chunk.toByteArray()).apply {
93 log.info("request(${request.commonHeader.requestId}): CBA file(${blueprintDir.absolutePath} written successfully")
98 private fun successStatus(message: String, header: CommonHeader): BluePrintManagementOutput =
99 BluePrintManagementOutput.newBuilder()
100 .setCommonHeader(header)
101 .setStatus(Status.newBuilder()
102 .setTimestamp(currentTimestamp())
108 private fun failStatus(message: String, e: Exception): StatusException {
109 log.error(message, e)
110 return io.grpc.Status.INTERNAL
111 .withDescription(message)