aboutsummaryrefslogtreecommitdiffstats
path: root/packages/base/src/files/install/elk/bin/elk.sh
blob: 3429905c4b1910511dc7bfaba7c54dec7300d19f (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
#!/bin/bash

ELK_MAP_SUCCESS_RE="\"acknowledged\": *true"

function usage() {
	echo -n "syntax: $(basename $0) "
	echo -n "[--debug] "
	echo -n "([--audit] |"	
}

function log() {
	echo "$(date +"%Y-%m-%d_%H-%M-%S") $1" >> ${POLICY_HOME}/logs/elk.log
	echo "$1"
}

function delete_index() {
	if [[ $DEBUG == y ]]; then
		echo "-- ${FUNCNAME[0]} $@ --"
		set -x
	fi
	
	COMMAND="curl --silent -X DELETE http://localhost:9200/policy"
	OUTPUT="$(${COMMAND} 2>&1)"
	RC=$?
	
	log "${RC}: ${COMMAND}"
	log "${OUTPUT}"
	
	if [[ ${RC} != 0 ]] || [[ ! ${OUTPUT} =~ ${ELK_MAP_SUCCESS_RE} ]]; then
		log "WARNING: curl: delete /policy: ${RC}"
		return 1
	fi
	
	log "OK: curl: delete /policy: ${OUTPUT}"
	return 0
}

function create_index() {
	if [[ $DEBUG == y ]]; then
		echo "-- ${FUNCNAME[0]} $@ --"
		set -x
	fi
	
	COMMAND="curl --silent -X PUT http://localhost:9200/policy"
	OUTPUT="$(${COMMAND} 2>&1)"
	RC=$?
	
	log "${RC}: ${COMMAND}"
	log "${OUTPUT}"
	
	if [[ ${RC} != 0 ]] || [[ ! ${OUTPUT} =~ ${ELK_MAP_SUCCESS_RE} ]]; then
		log "ERROR: curl: put /policy: ${RC}"
		return 1
	fi
		
	log "OK: curl: put /policy."
	return 0
}

function check_elk_status() {
	if [[ $DEBUG == y ]]; then
		echo "-- ${FUNCNAME[0]} $@ --"
		set -x
	fi
		
	${POLICY_HOME}/etc/init.d/elk status
	if [[ $? != 0 ]]; then
		log "ERROR: elk is down.   Aborting .."
		exit 1
	fi
}

function check_elk_policy_index() {
	if [[ $DEBUG == y ]]; then
		echo "-- ${FUNCNAME[0]} $@ --"
		set -x
	fi
	
	COMMAND="curl --silent -X GET http://localhost:9200/policy"
	OUTPUT="$(${COMMAND} 2>&1)"
	RC=$?
	
	log "${RC}: ${COMMAND}"
	
	if [[ ${RC} != 0 ]] || [[ ! ${OUTPUT} =~ policy ]]; then
		log "ERROR: curl: get /policy: ${RC}"
		return 1
	fi
		
	log "OK: curl: get /policy."
	return 0		
}

#The Script will update the policy data on querying from database as a bulk to Elastic Database
function audit() {
	if [[ $DEBUG == y ]]; then
		echo "-- ${FUNCNAME[0]} $@ --"
		set -x
	fi
	
	check_elk_status
	
	if ! check_elk_policy_index; then
		echo "policy index does not exist. So, Policy Index is created."
		if ! create_index; then
			echo "abort: policy index creation failed."
			exit 1
		fi
	fi
	
	$JAVA_HOME/bin/java -cp $POLICY_HOME/servers/pap/webapps/pap/WEB-INF/lib/ONAP-PAP-REST-*.jar:$POLICY_HOME/servers/pap/webapps/pap/WEB-INF/lib/*: -DPROPERTY_FILE=$POLICY_HOME/servers/pap/bin/policyelk.properties org.onap.policy.pap.xacml.rest.elk.client.ElasticSearchPolicyUpdate
}

#########################################################################
##
## script execution body
##
#########################################################################

DEBUG=n
OPERATION=none

until [[ -z "$1" ]]; do
	case $1 in
		-d|--debug|debug) 	DEBUG=y
							set -x
							;;
		-a|--audit|audit) 	OPERATION=audit
							;;																	
		*)	usage
			exit 1
			;;
	esac
	shift
done

# operation validation
case $OPERATION in
	audit)	;;
	*)		echo "invalid operation (${OPERATION}).";
			usage
			exit 1
			;;
esac

if [[ -z ${POLICY_HOME} ]]; then
	echo "error: POLICY_HOME is unset."
	exit 1
fi

log "**** $OPERATION ****"

if pidof -o %PPID -x $(basename $0) > /dev/null 2>&1; then
	echo "WARNING: an $(basename $0) process is already running.  Exiting."
	exit 1
fi

case $OPERATION in
	audit)	
		audit
		;;		
	*)	echo "invalid operation (${OPERATION}).";
		usage
		exit 1
		;;
esac