aboutsummaryrefslogtreecommitdiffstats
path: root/src/sch/core.clj
blob: 1852c76239518cc983e09cbc1a7443a0e66b7b8f (plain)
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
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
; ============LICENSE_START=======================================================
; org.onap.dcae
; ================================================================================
; Copyright (c) 2017 AT&T Intellectual Property. All rights reserved.
; ================================================================================
; 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.
; ============LICENSE_END=========================================================
;
; ECOMP is a trademark and service mark of AT&T Intellectual Property.

(ns sch.core
  (:require [clojure.java.io :refer :all]
            [cheshire.core :refer [parse-stream parse-string]]
            [taoensso.timbre :as timbre :refer [info error]]
            [taoensso.timbre.appenders.3rd-party.rolling :refer [rolling-appender]]
            [sch.handle :refer [handle-change-event! download-artifacts! deploy-artifacts!
                                deployed-ok deployed-error deployed-already]]
            [sch.asdc-client :refer [get-service-metadata! create-asdc-conn get-consumer-id]]
            [sch.inventory-client :refer [create-inventory-conn]]
            [sch.parse :refer [get-dcae-artifact-types pick-out-artifact]]
            [sch.util :refer [read-config]]
            [clojure.string :as strlib]
            )
  (:import (org.onap.sdc.impl DistributionClientFactory)
           (org.onap.sdc.api.consumer IConfiguration INotificationCallback
                                            IDistributionStatusMessage)
           (org.onap.sdc.utils DistributionActionResultEnum DistributionStatusEnum)
           (com.google.gson Gson)
           )
  (:gen-class))


(defn process-event-from-local-file!
  [event-file-path]
  (with-open [rdr (reader event-file-path)]
    (parse-stream rdr true))
  )

; Distribution client code
; TODO: Should be moved to separate namespace

(defn send-distribution-status!
  "Convenience function used to send distribution status messages"
  [dist-client distribution-id consumer-id artifact status]
  (let [dist-message (proxy [IDistributionStatusMessage] []
                       (getDistributionID [] distribution-id)
                       (getConsumerID [] consumer-id)
                       (getTimestamp []
                         (. java.lang.System currentTimeMillis))
                       (getArtifactURL [] (:artifactURL artifact))
                       (getStatus [] status))
        resp (.sendDeploymentStatus dist-client dist-message)
        ]
    (if (not= (.getDistributionActionResult resp) (. DistributionActionResultEnum SUCCESS))
      (error (str "Problem sending status: " (:artifactName artifact) ", "
                  (str (.getDistributionMessageResult resp)))))
    ))

(defn deploy-artifacts-ex!
  "Enhanced deploy artifacts function

  After calling deploy-artifacts!, this method takes the results and sends out
  appropriate distribution status messages per artifact processed"
  [inventory-uri service-metadata requests send-dist-status]
  (let [[to-post posted to-delete deleted] (deploy-artifacts! inventory-uri service-metadata
                                                              requests)
        pick-out-artifact (partial pick-out-artifact service-metadata)]

    (dorun (map #(send-dist-status (pick-out-artifact %)
                                   (. DistributionStatusEnum DEPLOY_OK))
                (deployed-ok to-post posted)))
    (dorun (map #(send-dist-status (pick-out-artifact %)
                                   (. DistributionStatusEnum DEPLOY_ERROR))
                (deployed-error to-post posted)))
    (dorun (map #(send-dist-status (pick-out-artifact %)
                                   (. DistributionStatusEnum ALREADY_DEPLOYED))
                (deployed-already requests to-post)))
    ; REVIEW: How about the deleted service types?
    ))


(defn create-distribution-client-config
  [config]
  (let [config-asdc (:asdcDistributionClient config)]
    (proxy [IConfiguration] []
      (getAsdcAddress [] (str (:asdcAddress config-asdc)))
      (getMsgBusAddress [] (java.util.ArrayList.
                              (strlib/split (str (:msgBusAddress config-asdc)) #",")))
      (getUser [] (str (:user config-asdc)))
      (getPassword [] (str (:password config-asdc)))
      (getPollingInterval [] (int (:pollingInterval config-asdc)))
      (getPollingTimeout [] (int (:pollingTimeout config-asdc)))
      ; Note: The following didn't work
      ; (. Arrays asList (. ArtifactTypeEnum values))
      ; Also, cannot just use a narrow list of artifact types in order
      ; to handle the deletion scenario.
      (getRelevantArtifactTypes [] (java.util.ArrayList.
                                      (get-dcae-artifact-types)))
      (getConsumerGroup [] (str (:consumerGroup config-asdc)))
      (getConsumerID [] (str (:consumerId config-asdc)))
      (getEnvironmentName [] (str (:environmentName config-asdc)))
      (getKeyStorePath [] (str (:keyStorePath config-asdc)))
      (getKeyStorePassword [] (str (:keyStorePassword config-asdc)))
      (activateServerTLSAuth [] (boolean (:activateServerTLSAuth config-asdc)))
      (isFilterInEmptyResources [] (boolean (:isFilterInEmptyResources config-asdc)))
      (isUseHttpsWithDmaap [] (boolean (:useHttpsWithDmaap config-asdc false)))
      (isConsumeProduceStatusTopic [] (boolean (:isConsumeProduceStatusTopic config-asdc false)))
      )))

(defn run-distribution-client!
  "Entry point to the core production functionality

  Uses the asdc distribution client and to poll for notification events and makes calls
  to handle those events"
  [dist-client-config inventory-uri asdc-conn]
  (let [dist-client (. DistributionClientFactory createDistributionClient)
        dist-client-callback (proxy [INotificationCallback] []
                               (activateCallback [data]
                                 "Callback executed upon notification events

                                 The input parameter is of type `NotificationDataImpl` which fails
                                 to translate via the clojure `bean` call because class is not
                                 public. So mirroring what's done in the distribution client -
                                 use Gson.

                                 Discovered that the notification event and the service metadata
                                 data models are different. Use service metadata because its
                                 richer."
                                 (let [change-event (parse-string (.toJson (Gson.) data) true)
                                       service-id (:serviceUUID change-event)
                                       distribution-id (:distributionID change-event)
                                       service-metadata (get-service-metadata! asdc-conn
                                                                              service-id)
                                       send-dist-status (partial send-distribution-status!
                                                                 dist-client distribution-id
                                                                 (get-consumer-id asdc-conn))
                                       ]

                                   (info (str "Handle change event: " (:serviceName change-event)
                                              ", " distribution-id))

                                   (let [requests (download-artifacts! inventory-uri asdc-conn
                                                                       service-metadata)
                                         artifacts (map #(pick-out-artifact service-metadata %)
                                                        requests)
                                         ]

                                     (dorun (map #(send-dist-status
                                                    % (. DistributionStatusEnum DOWNLOAD_OK))
                                                 artifacts))
                                     (deploy-artifacts-ex! inventory-uri service-metadata
                                                           requests send-dist-status)
                                     )

                                   )))
        ]
    (let [dist-client-init-result (.init dist-client dist-client-config dist-client-callback)]
      (if (= (.getDistributionActionResult dist-client-init-result)
             (. DistributionActionResultEnum SUCCESS))
        (.start dist-client)
        (error dist-client-init-result))
      )))

(defn- setup-logging-rolling
  "Setup logging with the rolling appender"
  [{ {:keys [currentLogFilename rotationFrequency]} :logging }]
  (let [rolling-params (when currentLogFilename { :path currentLogFilename })
        rolling-params (when rotationFrequency
                         (assoc rolling-params :pattern (keyword rotationFrequency)))]
    (timbre/merge-config! { :level :debug :appenders { :rolling (rolling-appender rolling-params) } })
    (info "Setup logging: Rolling appender" (if rolling-params rolling-params "DEFAULT"))
  ))


(defn -main [& args]
  (let [[mode config-path event-path] args
        config (read-config config-path)
        inventory-uri (create-inventory-conn config)
        asdc-conn (create-asdc-conn config)
        ]

    (setup-logging-rolling config)

    (if (= "DEV" (clojure.string/upper-case mode))
      (do
        (info "Development mode")
        (handle-change-event! inventory-uri asdc-conn
                              (process-event-from-local-file! event-path))
        )
      (run-distribution-client!
        (create-distribution-client-config config)
        inventory-uri asdc-conn)
      )

    (info "Done"))
    )