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
|
#!/bin/bash
if [ ! -e "/root/devices/$(hostname)" ];then
echo "Static slave for $LAVA_MASTER"
exit 0
fi
if [ -z "$LAVA_MASTER_URI" ];then
echo "ERROR: Missing LAVA_MASTER_URI"
exit 11
fi
echo "Dynamic slave for $LAVA_MASTER ($LAVA_MASTER_URI)"
cd /root/lavacli
LAVACLIOPTS="--uri $LAVA_MASTER_URI"
# do a sort of ping for letting master to be up
TIMEOUT=300
while [ $TIMEOUT -ge 1 ];
do
STEP=2
lavacli $LAVACLIOPTS device-types list 2>/dev/null >/dev/null
if [ $? -eq 0 ];then
TIMEOUT=0
else
echo "Wait for master...."
sleep $STEP
fi
TIMEOUT=$(($TIMEOUT-$STEP))
done
# This directory is used for storing device-types already added
mkdir -p /root/.lavadocker/
if [ -e /root/device-types ];then
for i in $(ls /root/device-types/*jinja2)
do
devicetype=$(basename $i |sed 's,.jinja2,,')
echo "Adding custom $devicetype"
lavacli $LAVACLIOPTS device-types list || exit $?
touch /root/.lavadocker/devicetype-$devicetype
done
fi
lavacli $LAVACLIOPTS device-types list > /tmp/device-types.list
if [ $? -ne 0 ];then
exit 1
fi
lavacli $LAVACLIOPTS devices list -a > /tmp/devices.list
if [ $? -ne 0 ];then
exit 1
fi
for worker in $(ls /root/devices/)
do
lavacli $LAVACLIOPTS workers list |grep -q $worker
if [ $? -eq 0 ];then
echo "Remains of $worker, cleaning it"
/usr/local/bin/retire.sh $LAVA_MASTER_URI $worker
#lavacli $LAVACLIOPTS workers update $worker || exit $?
else
echo "Adding worker $worker"
lavacli $LAVACLIOPTS workers add --description "LAVA dispatcher on $(cat /root/phyhostname)" $worker || exit $?
fi
if [ ! -z "$LAVA_DISPATCHER_IP" ];then
echo "Add dispatcher_ip $LAVA_DISPATCHER_IP to $worker"
/usr/local/bin/setdispatcherip.py $LAVA_MASTER_URI $worker $LAVA_DISPATCHER_IP || exit $?
fi
for device in $(ls /root/devices/$worker/)
do
devicename=$(echo $device | sed 's,.jinja2,,')
devicetype=$(grep -h extends /root/devices/$worker/$device| grep -o '[a-zA-Z0-9_-]*.jinja2' | sed 's,.jinja2,,')
if [ -e /root/.lavadocker/devicetype-$devicetype ];then
echo "Skip devicetype $devicetype"
else
echo "Add devicetype $devicetype"
grep -q "$devicetype[[:space:]]" /tmp/device-types.list
if [ $? -eq 0 ];then
echo "Skip devicetype $devicetype"
else
lavacli $LAVACLIOPTS device-types add $devicetype || exit $?
fi
touch /root/.lavadocker/devicetype-$devicetype
fi
echo "Add device $devicename on $worker"
grep -q "$devicename[[:space:]]" /tmp/devices.list
if [ $? -eq 0 ];then
echo "$devicename already present"
#verify if present on another worker
#TODO
lavacli $LAVACLIOPTS devices show $devicename |grep ^worker |grep -q $worker
if [ $? -ne 0 ];then
echo "ERROR: $devicename already present on another worker"
exit 1
fi
DEVICE_HEALTH=$(grep "$devicename[[:space:]]" /tmp/devices.list | sed 's/.*,//')
case "$DEVICE_HEALTH" in
Retired)
echo "DEBUG: Keep $devicename state: $DEVICE_HEALTH"
DEVICE_HEALTH='RETIRED'
;;
Maintenance)
echo "DEBUG: Keep $devicename state: $DEVICE_HEALTH"
DEVICE_HEALTH='MAINTENANCE'
;;
*)
echo "DEBUG: Set $devicename state to UNKNOWN (from $DEVICE_HEALTH)"
DEVICE_HEALTH='UNKNOWN'
;;
esac
lavacli $LAVACLIOPTS devices update --worker $worker --health $DEVICE_HEALTH $devicename || exit $?
# always reset the device dict in case of update of it
lavacli $LAVACLIOPTS devices dict set $devicename /root/devices/$worker/$device || exit $?
else
lavacli $LAVACLIOPTS devices add --type $devicetype --worker $worker $devicename || exit $?
lavacli $LAVACLIOPTS devices dict set $devicename /root/devices/$worker/$device || exit $?
fi
done
done
if [ -e /etc/lava-dispatcher/certificates.d/$(hostname).key ];then
echo "INFO: Enabling encryption"
sed -i 's,.*ENCRYPT=.*,ENCRYPT="--encrypt",' /etc/lava-dispatcher/lava-slave
sed -i "s,.*SLAVE_CERT=.*,SLAVE_CERT=\"--slave-cert /etc/lava-dispatcher/certificates.d/$(hostname).key_secret\"," /etc/lava-dispatcher/lava-slave
(cd /etc/lava-dispatcher/certificates.d; if [ -e master.key ]; then cp master.key $LAVA_MASTER.key; fi)
sed -i "s,.*MASTER_CERT=.*,MASTER_CERT=\"--master-cert /etc/lava-dispatcher/certificates.d/$LAVA_MASTER.key\"," /etc/lava-dispatcher/lava-slave
fi
exit 0
|
