Dinesh Belwalkar | e1e85ad | 2019-07-31 23:06:47 +0000 | [diff] [blame] | 1 | // Copyright 2018 Open Networking Foundation |
| 2 | // |
| 3 | // Licensed under the Apache License, Version 2.0 (the "License"); |
| 4 | // you may not use this file except in compliance with the License. |
| 5 | // You may obtain a copy of the License at |
| 6 | // |
| 7 | // http://www.apache.org/licenses/LICENSE-2.0 |
| 8 | // |
| 9 | // Unless required by applicable law or agreed to in writing, software |
| 10 | // distributed under the License is distributed on an "AS IS" BASIS, |
| 11 | // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
| 12 | // See the License for the specific language governing permissions and |
| 13 | // limitations under the License. |
| 14 | |
| 15 | package main |
| 16 | |
| 17 | import ( |
Dinesh Belwalkar | b5db83f | 2019-10-24 17:27:58 +0000 | [diff] [blame] | 18 | "bufio" |
Dinesh Belwalkar | e1e85ad | 2019-07-31 23:06:47 +0000 | [diff] [blame] | 19 | "bytes" |
Dinesh Belwalkar | b5db83f | 2019-10-24 17:27:58 +0000 | [diff] [blame] | 20 | "crypto/tls" |
| 21 | "fmt" |
| 22 | "github.com/Shopify/sarama" |
Dinesh Belwalkar | 72ecafb | 2019-12-12 00:08:56 +0000 | [diff] [blame] | 23 | "github.com/opencord/device-management/demo_test/proto" |
Dinesh Belwalkar | e63f7f9 | 2019-11-22 23:11:16 +0000 | [diff] [blame] | 24 | log "github.com/sirupsen/logrus" |
Dinesh Belwalkar | b5db83f | 2019-10-24 17:27:58 +0000 | [diff] [blame] | 25 | "golang.org/x/net/context" |
| 26 | "google.golang.org/grpc" |
| 27 | "google.golang.org/grpc/status" |
| 28 | "net" |
| 29 | "net/http" |
| 30 | "os" |
| 31 | "os/exec" |
| 32 | "os/signal" |
| 33 | "strconv" |
Dinesh Belwalkar | e1e85ad | 2019-07-31 23:06:47 +0000 | [diff] [blame] | 34 | "strings" |
mc | cd7e950 | 2019-12-16 22:04:13 +0000 | [diff] [blame^] | 35 | "sort" |
Dinesh Belwalkar | e1e85ad | 2019-07-31 23:06:47 +0000 | [diff] [blame] | 36 | ) |
| 37 | |
Dinesh Belwalkar | b5db83f | 2019-10-24 17:27:58 +0000 | [diff] [blame] | 38 | var REDFISH_ROOT = "/redfish/v1" |
| 39 | var CONTENT_TYPE = "application/json" |
nickhuang | 6b31f8f | 2019-09-26 02:02:14 +0000 | [diff] [blame] | 40 | var EVENTS_MAP = map[string]string{ |
Dinesh Belwalkar | b5db83f | 2019-10-24 17:27:58 +0000 | [diff] [blame] | 41 | "add": "ResourceAdded", |
| 42 | "rm": "ResourceRemoved", |
| 43 | "alert": "Alert", |
| 44 | "update": "Update"} |
nickhuang | 6b31f8f | 2019-09-26 02:02:14 +0000 | [diff] [blame] | 45 | |
Dinesh Belwalkar | b5db83f | 2019-10-24 17:27:58 +0000 | [diff] [blame] | 46 | var default_address string = "localhost:31085" |
Dinesh Belwalkar | b5db83f | 2019-10-24 17:27:58 +0000 | [diff] [blame] | 47 | var importerTopic = "importer" |
Dinesh Belwalkar | e1e85ad | 2019-07-31 23:06:47 +0000 | [diff] [blame] | 48 | var DataConsumer sarama.Consumer |
| 49 | |
Dinesh Belwalkar | b5db83f | 2019-10-24 17:27:58 +0000 | [diff] [blame] | 50 | var cc importer.DeviceManagementClient |
| 51 | var ctx context.Context |
| 52 | var conn *grpc.ClientConn |
nickhuang | 6b31f8f | 2019-09-26 02:02:14 +0000 | [diff] [blame] | 53 | |
dileepbk | 86ef010 | 2019-11-13 00:08:33 +0000 | [diff] [blame] | 54 | func GetCurrentDevices() (error, []string) { |
Dinesh Belwalkar | 72ecafb | 2019-12-12 00:08:56 +0000 | [diff] [blame] | 55 | fmt.Println("Testing GetCurrentDevices") |
dileepbk | 86ef010 | 2019-11-13 00:08:33 +0000 | [diff] [blame] | 56 | empty := new(importer.Empty) |
mc | 20a4b5f | 2019-10-16 20:28:24 +0000 | [diff] [blame] | 57 | var ret_msg *importer.DeviceListByIp |
dileepbk | 86ef010 | 2019-11-13 00:08:33 +0000 | [diff] [blame] | 58 | ret_msg, err := cc.GetCurrentDevices(ctx, empty) |
| 59 | if err != nil { |
| 60 | return err, nil |
| 61 | } else { |
| 62 | return err, ret_msg.Ip |
| 63 | } |
| 64 | } |
| 65 | |
Dinesh Belwalkar | e1e85ad | 2019-07-31 23:06:47 +0000 | [diff] [blame] | 66 | func init() { |
Dinesh Belwalkar | b5db83f | 2019-10-24 17:27:58 +0000 | [diff] [blame] | 67 | Formatter := new(log.TextFormatter) |
| 68 | Formatter.TimestampFormat = "02-01-2006 15:04:05" |
| 69 | Formatter.FullTimestamp = true |
| 70 | log.SetFormatter(Formatter) |
Dinesh Belwalkar | e1e85ad | 2019-07-31 23:06:47 +0000 | [diff] [blame] | 71 | } |
| 72 | |
| 73 | func topicListener(topic *string, master sarama.Consumer) { |
| 74 | log.Info("Starting topicListener for ", *topic) |
| 75 | consumer, err := master.ConsumePartition(*topic, 0, sarama.OffsetOldest) |
| 76 | if err != nil { |
Dinesh Belwalkar | 72ecafb | 2019-12-12 00:08:56 +0000 | [diff] [blame] | 77 | log.Errorf("topicListener panic, topic=[%s]: %s", *topic, err.Error()) |
Dinesh Belwalkar | e1e85ad | 2019-07-31 23:06:47 +0000 | [diff] [blame] | 78 | os.Exit(1) |
| 79 | } |
| 80 | signals := make(chan os.Signal, 1) |
| 81 | signal.Notify(signals, os.Interrupt) |
| 82 | doneCh := make(chan struct{}) |
| 83 | go func() { |
| 84 | for { |
| 85 | select { |
| 86 | case err := <-consumer.Errors(): |
Dinesh Belwalkar | 72ecafb | 2019-12-12 00:08:56 +0000 | [diff] [blame] | 87 | log.Errorf("Consumer error: %s", err.Err) |
Dinesh Belwalkar | e1e85ad | 2019-07-31 23:06:47 +0000 | [diff] [blame] | 88 | case msg := <-consumer.Messages(): |
Dinesh Belwalkar | 72ecafb | 2019-12-12 00:08:56 +0000 | [diff] [blame] | 89 | log.Infof("Got message on topic=[%s]: %s", *topic, string(msg.Value)) |
Dinesh Belwalkar | e1e85ad | 2019-07-31 23:06:47 +0000 | [diff] [blame] | 90 | case <-signals: |
| 91 | log.Warn("Interrupt is detected") |
Dinesh Belwalkar | b5db83f | 2019-10-24 17:27:58 +0000 | [diff] [blame] | 92 | os.Exit(1) |
Dinesh Belwalkar | e1e85ad | 2019-07-31 23:06:47 +0000 | [diff] [blame] | 93 | } |
| 94 | } |
| 95 | }() |
| 96 | <-doneCh |
| 97 | } |
| 98 | |
| 99 | func kafkainit() { |
Dinesh Belwalkar | b5db83f | 2019-10-24 17:27:58 +0000 | [diff] [blame] | 100 | cmd := exec.Command("/bin/sh", "kafka_ip.sh") |
Dinesh Belwalkar | e1e85ad | 2019-07-31 23:06:47 +0000 | [diff] [blame] | 101 | var kafkaIP string |
| 102 | var out bytes.Buffer |
| 103 | cmd.Stdout = &out |
| 104 | err := cmd.Run() |
| 105 | if err != nil { |
| 106 | log.Info(err) |
| 107 | os.Exit(1) |
| 108 | } |
| 109 | |
| 110 | kafkaIP = out.String() |
| 111 | kafkaIP = strings.TrimSuffix(kafkaIP, "\n") |
Dinesh Belwalkar | b5db83f | 2019-10-24 17:27:58 +0000 | [diff] [blame] | 112 | kafkaIP = kafkaIP + ":9092" |
| 113 | fmt.Println("IP address of kafka-cord-0:", kafkaIP) |
Dinesh Belwalkar | e1e85ad | 2019-07-31 23:06:47 +0000 | [diff] [blame] | 114 | config := sarama.NewConfig() |
| 115 | config.Consumer.Return.Errors = true |
| 116 | master, err := sarama.NewConsumer([]string{kafkaIP}, config) |
| 117 | if err != nil { |
| 118 | panic(err) |
| 119 | } |
| 120 | DataConsumer = master |
| 121 | |
| 122 | go topicListener(&importerTopic, master) |
| 123 | } |
| 124 | func main() { |
nickhuang | 6b31f8f | 2019-09-26 02:02:14 +0000 | [diff] [blame] | 125 | http.DefaultTransport.(*http.Transport).TLSClientConfig = &tls.Config{InsecureSkipVerify: true} |
| 126 | fmt.Println("Launching server...") |
Dinesh Belwalkar | e1e85ad | 2019-07-31 23:06:47 +0000 | [diff] [blame] | 127 | log.Info("kafkaInit starting") |
| 128 | kafkainit() |
Dinesh Belwalkar | e1e85ad | 2019-07-31 23:06:47 +0000 | [diff] [blame] | 129 | |
nickhuang | 6b31f8f | 2019-09-26 02:02:14 +0000 | [diff] [blame] | 130 | ln, err := net.Listen("tcp", ":9999") |
Dinesh Belwalkar | b5db83f | 2019-10-24 17:27:58 +0000 | [diff] [blame] | 131 | if err != nil { |
| 132 | fmt.Println("could not listen") |
Dinesh Belwalkar | 72ecafb | 2019-12-12 00:08:56 +0000 | [diff] [blame] | 133 | log.Fatalf("did not listen: %v", err) |
Dinesh Belwalkar | b5db83f | 2019-10-24 17:27:58 +0000 | [diff] [blame] | 134 | } |
nickhuang | 6b31f8f | 2019-09-26 02:02:14 +0000 | [diff] [blame] | 135 | defer ln.Close() |
Dinesh Belwalkar | e1e85ad | 2019-07-31 23:06:47 +0000 | [diff] [blame] | 136 | |
mc | 20a4b5f | 2019-10-16 20:28:24 +0000 | [diff] [blame] | 137 | conn, err = grpc.Dial(default_address, grpc.WithInsecure()) |
| 138 | if err != nil { |
| 139 | fmt.Println("could not connect") |
Dinesh Belwalkar | 72ecafb | 2019-12-12 00:08:56 +0000 | [diff] [blame] | 140 | log.Fatalf("did not connect: %v", err) |
mc | 20a4b5f | 2019-10-16 20:28:24 +0000 | [diff] [blame] | 141 | } |
| 142 | defer conn.Close() |
nickhuang | 6b31f8f | 2019-09-26 02:02:14 +0000 | [diff] [blame] | 143 | |
mc | 20a4b5f | 2019-10-16 20:28:24 +0000 | [diff] [blame] | 144 | cc = importer.NewDeviceManagementClient(conn) |
| 145 | ctx = context.Background() |
nickhuang | 6b31f8f | 2019-09-26 02:02:14 +0000 | [diff] [blame] | 146 | |
mc | 20a4b5f | 2019-10-16 20:28:24 +0000 | [diff] [blame] | 147 | loop := true |
nickhuang | 6b31f8f | 2019-09-26 02:02:14 +0000 | [diff] [blame] | 148 | |
mc | 20a4b5f | 2019-10-16 20:28:24 +0000 | [diff] [blame] | 149 | for loop == true { |
mc | cd7e950 | 2019-12-16 22:04:13 +0000 | [diff] [blame^] | 150 | connS, err := ln.Accept() |
| 151 | if err != nil { |
| 152 | fmt.Println("Accept error") |
| 153 | log.Fatal("Accept error: %v", err) |
| 154 | } |
mc | 20a4b5f | 2019-10-16 20:28:24 +0000 | [diff] [blame] | 155 | cmdstr, _ := bufio.NewReader(connS).ReadString('\n') |
| 156 | cmdstr = strings.TrimSuffix(cmdstr, "\n") |
| 157 | s := strings.Split(cmdstr, " ") |
| 158 | newmessage := "" |
| 159 | cmd := string(s[0]) |
nickhuang | 6b31f8f | 2019-09-26 02:02:14 +0000 | [diff] [blame] | 160 | |
mc | 20a4b5f | 2019-10-16 20:28:24 +0000 | [diff] [blame] | 161 | switch cmd { |
nickhuang | 6b31f8f | 2019-09-26 02:02:14 +0000 | [diff] [blame] | 162 | |
Dinesh Belwalkar | 72ecafb | 2019-12-12 00:08:56 +0000 | [diff] [blame] | 163 | case "attach": |
mc | 20a4b5f | 2019-10-16 20:28:24 +0000 | [diff] [blame] | 164 | if len(s) < 2 { |
mc | cd7e950 | 2019-12-16 22:04:13 +0000 | [diff] [blame^] | 165 | newmessage = newmessage + "invalid command length" + cmdstr + "\n" |
mc | 20a4b5f | 2019-10-16 20:28:24 +0000 | [diff] [blame] | 166 | break |
| 167 | } |
| 168 | var devicelist importer.DeviceList |
| 169 | var ipattached []string |
| 170 | for _, devinfo := range s[1:] { |
| 171 | info := strings.Split(devinfo, ":") |
| 172 | if len(info) != 3 { |
| 173 | newmessage = newmessage + "invalid command " + devinfo + "\n" |
| 174 | continue |
nickhuang | 6b31f8f | 2019-09-26 02:02:14 +0000 | [diff] [blame] | 175 | } |
mc | 20a4b5f | 2019-10-16 20:28:24 +0000 | [diff] [blame] | 176 | deviceinfo := new(importer.DeviceInfo) |
| 177 | deviceinfo.IpAddress = info[0] + ":" + info[1] |
| 178 | freq, err := strconv.ParseUint(info[2], 10, 32) |
Dinesh Belwalkar | 72ecafb | 2019-12-12 00:08:56 +0000 | [diff] [blame] | 179 | if err != nil { |
mc | 20a4b5f | 2019-10-16 20:28:24 +0000 | [diff] [blame] | 180 | newmessage = newmessage + "invalid command " + devinfo + "\n" |
| 181 | continue |
nickhuang | 6b31f8f | 2019-09-26 02:02:14 +0000 | [diff] [blame] | 182 | } |
mc | 20a4b5f | 2019-10-16 20:28:24 +0000 | [diff] [blame] | 183 | deviceinfo.Frequency = uint32(freq) |
| 184 | devicelist.Device = append(devicelist.Device, deviceinfo) |
| 185 | ipattached = append(ipattached, deviceinfo.IpAddress) |
| 186 | } |
mc | cd7e950 | 2019-12-16 22:04:13 +0000 | [diff] [blame^] | 187 | if len(devicelist.Device) == 0 { |
| 188 | break |
| 189 | } |
mc | 20a4b5f | 2019-10-16 20:28:24 +0000 | [diff] [blame] | 190 | _, err := cc.SendDeviceList(ctx, &devicelist) |
| 191 | if err != nil { |
| 192 | errStatus, _ := status.FromError(err) |
mc | cd7e950 | 2019-12-16 22:04:13 +0000 | [diff] [blame^] | 193 | newmessage = newmessage + errStatus.Message() |
mc | 20a4b5f | 2019-10-16 20:28:24 +0000 | [diff] [blame] | 194 | fmt.Printf("attach error - status code %v message %v", errStatus.Code(), errStatus.Message()) |
| 195 | } else { |
mc | cd7e950 | 2019-12-16 22:04:13 +0000 | [diff] [blame^] | 196 | sort.Strings(ipattached) |
mc | 20a4b5f | 2019-10-16 20:28:24 +0000 | [diff] [blame] | 197 | ips := strings.Join(ipattached, " ") |
| 198 | newmessage = newmessage + ips + " attached\n" |
| 199 | } |
Dinesh Belwalkar | 72ecafb | 2019-12-12 00:08:56 +0000 | [diff] [blame] | 200 | case "delete": |
mc | 20a4b5f | 2019-10-16 20:28:24 +0000 | [diff] [blame] | 201 | if len(s) < 2 { |
| 202 | newmessage = newmessage + "invalid command " + cmdstr + "\n" |
| 203 | break |
| 204 | } |
| 205 | var devicelist importer.DeviceListByIp |
| 206 | for _, ip := range s[1:] { |
mc | cd7e950 | 2019-12-16 22:04:13 +0000 | [diff] [blame^] | 207 | addr := strings.Split(ip, ":") |
| 208 | if len(addr) != 2 { |
| 209 | newmessage = newmessage + "invalid address " + ip + "\n" |
| 210 | continue |
| 211 | } |
mc | 20a4b5f | 2019-10-16 20:28:24 +0000 | [diff] [blame] | 212 | devicelist.Ip = append(devicelist.Ip, ip) |
| 213 | } |
mc | cd7e950 | 2019-12-16 22:04:13 +0000 | [diff] [blame^] | 214 | if len(devicelist.Ip) == 0 { |
| 215 | break |
| 216 | } |
mc | 20a4b5f | 2019-10-16 20:28:24 +0000 | [diff] [blame] | 217 | _, err := cc.DeleteDeviceList(ctx, &devicelist) |
| 218 | if err != nil { |
| 219 | errStatus, _ := status.FromError(err) |
mc | cd7e950 | 2019-12-16 22:04:13 +0000 | [diff] [blame^] | 220 | newmessage = newmessage + errStatus.Message() |
mc | 20a4b5f | 2019-10-16 20:28:24 +0000 | [diff] [blame] | 221 | fmt.Printf("delete error - status code %v message %v", errStatus.Code(), errStatus.Message()) |
| 222 | } else { |
mc | cd7e950 | 2019-12-16 22:04:13 +0000 | [diff] [blame^] | 223 | sort.Strings(devicelist.Ip) |
mc | 20a4b5f | 2019-10-16 20:28:24 +0000 | [diff] [blame] | 224 | ips := strings.Join(devicelist.Ip, " ") |
| 225 | newmessage = newmessage + ips + " deleted\n" |
| 226 | } |
Dinesh Belwalkar | 72ecafb | 2019-12-12 00:08:56 +0000 | [diff] [blame] | 227 | case "period": |
mc | 20a4b5f | 2019-10-16 20:28:24 +0000 | [diff] [blame] | 228 | if len(s) != 2 { |
| 229 | newmessage = newmessage + "invalid command " + cmdstr + "\n" |
| 230 | break |
| 231 | } |
| 232 | args := strings.Split(s[1], ":") |
| 233 | if len(args) != 3 { |
| 234 | newmessage = newmessage + "invalid command " + s[1] + "\n" |
| 235 | break |
| 236 | } |
| 237 | ip := args[0] + ":" + args[1] |
| 238 | pv := args[2] |
| 239 | u, err := strconv.ParseUint(pv, 10, 64) |
| 240 | if err != nil { |
| 241 | fmt.Print("ParseUint error!!\n") |
| 242 | } else { |
| 243 | freqinfo := new(importer.FreqInfo) |
| 244 | freqinfo.Frequency = uint32(u) |
| 245 | freqinfo.IpAddress = ip |
| 246 | _, err := cc.SetFrequency(ctx, freqinfo) |
nickhuang | 6b31f8f | 2019-09-26 02:02:14 +0000 | [diff] [blame] | 247 | |
mc | 20a4b5f | 2019-10-16 20:28:24 +0000 | [diff] [blame] | 248 | if err != nil { |
| 249 | errStatus, _ := status.FromError(err) |
| 250 | newmessage = newmessage + errStatus.Message() |
| 251 | fmt.Printf("period error - status code %v message %v", errStatus.Code(), errStatus.Message()) |
Dinesh Belwalkar | b5db83f | 2019-10-24 17:27:58 +0000 | [diff] [blame] | 252 | } else { |
mc | 20a4b5f | 2019-10-16 20:28:24 +0000 | [diff] [blame] | 253 | newmessage = newmessage + "data collection interval configured to " + pv + " seconds\n" |
nickhuang | 6b31f8f | 2019-09-26 02:02:14 +0000 | [diff] [blame] | 254 | } |
mc | 20a4b5f | 2019-10-16 20:28:24 +0000 | [diff] [blame] | 255 | } |
| 256 | case "sub", "unsub": |
| 257 | if len(s) != 2 { |
| 258 | newmessage = newmessage + "invalid command " + cmdstr + "\n" |
| 259 | break |
| 260 | } |
| 261 | args := strings.Split(s[1], ":") |
| 262 | if len(args) < 3 { |
| 263 | newmessage = newmessage + "invalid command " + s[1] + "\n" |
| 264 | break |
| 265 | } |
| 266 | giveneventlist := new(importer.GivenEventList) |
| 267 | giveneventlist.EventIpAddress = args[0] + ":" + args[1] |
| 268 | for _, event := range args[2:] { |
| 269 | if value, ok := EVENTS_MAP[event]; ok { |
| 270 | giveneventlist.Events = append(giveneventlist.Events, value) |
nickhuang | 6b31f8f | 2019-09-26 02:02:14 +0000 | [diff] [blame] | 271 | } |
mc | 20a4b5f | 2019-10-16 20:28:24 +0000 | [diff] [blame] | 272 | } |
mc | cd7e950 | 2019-12-16 22:04:13 +0000 | [diff] [blame^] | 273 | if len(giveneventlist.Events) == 0 { |
| 274 | newmessage = newmessage + "No valid event was given\n" |
| 275 | } |
mc | 20a4b5f | 2019-10-16 20:28:24 +0000 | [diff] [blame] | 276 | var err error |
| 277 | if cmd == "sub" { |
mc | cd7e950 | 2019-12-16 22:04:13 +0000 | [diff] [blame^] | 278 | _, err = cc.SubscribeGivenEvents(ctx, giveneventlist) |
mc | 20a4b5f | 2019-10-16 20:28:24 +0000 | [diff] [blame] | 279 | } else { |
mc | cd7e950 | 2019-12-16 22:04:13 +0000 | [diff] [blame^] | 280 | _, err = cc.UnsubscribeGivenEvents(ctx, giveneventlist) |
mc | 20a4b5f | 2019-10-16 20:28:24 +0000 | [diff] [blame] | 281 | } |
| 282 | if err != nil { |
| 283 | errStatus, _ := status.FromError(err) |
| 284 | newmessage = newmessage + errStatus.Message() |
| 285 | fmt.Printf("Un/subscribe error - status code %v message %v", errStatus.Code(), errStatus.Message()) |
| 286 | } else { |
Dinesh Belwalkar | 72ecafb | 2019-12-12 00:08:56 +0000 | [diff] [blame] | 287 | newmessage = newmessage + cmd + " successful\n" |
mc | 20a4b5f | 2019-10-16 20:28:24 +0000 | [diff] [blame] | 288 | } |
| 289 | case "showeventlist": |
| 290 | if len(s) != 2 { |
mc | cd7e950 | 2019-12-16 22:04:13 +0000 | [diff] [blame^] | 291 | newmessage = newmessage + "invalid command " + cmdstr + "\n" |
mc | 20a4b5f | 2019-10-16 20:28:24 +0000 | [diff] [blame] | 292 | break |
| 293 | } |
| 294 | currentdeviceinfo := new(importer.Device) |
| 295 | currentdeviceinfo.IpAddress = s[1] |
| 296 | ret_msg, err := cc.GetEventList(ctx, currentdeviceinfo) |
| 297 | if err != nil { |
| 298 | errStatus, _ := status.FromError(err) |
| 299 | newmessage = errStatus.Message() |
| 300 | fmt.Printf("showeventlist error - status code %v message %v", errStatus.Code(), errStatus.Message()) |
| 301 | } else { |
| 302 | fmt.Print("showeventlist ", ret_msg.Events) |
mc | cd7e950 | 2019-12-16 22:04:13 +0000 | [diff] [blame^] | 303 | sort.Strings(ret_msg.Events[:]) |
| 304 | newmessage = strings.Join(ret_msg.Events[:], " ") |
| 305 | newmessage = newmessage + "\n" |
mc | 20a4b5f | 2019-10-16 20:28:24 +0000 | [diff] [blame] | 306 | } |
| 307 | case "showdeviceeventlist": |
| 308 | if len(s) != 2 { |
| 309 | newmessage = newmessage + "invalid command " + s[1] + "\n" |
| 310 | break |
| 311 | } |
| 312 | currentdeviceinfo := new(importer.Device) |
| 313 | currentdeviceinfo.IpAddress = s[1] |
| 314 | ret_msg, err := cc.GetCurrentEventList(ctx, currentdeviceinfo) |
| 315 | if err != nil { |
| 316 | errStatus, _ := status.FromError(err) |
| 317 | fmt.Printf("showdeviceeventlist error - status code %v message %v", errStatus.Code(), errStatus.Message()) |
| 318 | newmessage = newmessage + errStatus.Message() |
| 319 | } else { |
| 320 | fmt.Print("showdeviceeventlist ", ret_msg.Events) |
mc | cd7e950 | 2019-12-16 22:04:13 +0000 | [diff] [blame^] | 321 | sort.Strings(ret_msg.Events[:]) |
| 322 | newmessage = strings.Join(ret_msg.Events[:], " ") |
| 323 | newmessage = newmessage + "\n" |
mc | 20a4b5f | 2019-10-16 20:28:24 +0000 | [diff] [blame] | 324 | } |
| 325 | case "cleardeviceeventlist": |
| 326 | if len(s) != 2 { |
| 327 | newmessage = newmessage + "invalid command " + s[1] + "\n" |
| 328 | break |
| 329 | } |
| 330 | currentdeviceinfo := new(importer.Device) |
| 331 | currentdeviceinfo.IpAddress = s[1] |
| 332 | _, err := cc.ClearCurrentEventList(ctx, currentdeviceinfo) |
| 333 | if err != nil { |
| 334 | errStatus, _ := status.FromError(err) |
| 335 | newmessage = newmessage + errStatus.Message() |
| 336 | fmt.Printf("cleardeviceeventlist error - status code %v message %v", errStatus.Code(), errStatus.Message()) |
| 337 | } else { |
| 338 | newmessage = newmessage + currentdeviceinfo.IpAddress + " events cleared\n" |
| 339 | } |
| 340 | case "QUIT": |
| 341 | loop = false |
| 342 | newmessage = "QUIT" |
Dinesh Belwalkar | b5db83f | 2019-10-24 17:27:58 +0000 | [diff] [blame] | 343 | |
Dinesh Belwalkar | 72ecafb | 2019-12-12 00:08:56 +0000 | [diff] [blame] | 344 | case "showdevices": |
| 345 | cmd_size := len(s) |
| 346 | fmt.Print("cmd is :", cmd, cmd_size) |
| 347 | if cmd_size > 2 || cmd_size < 0 { |
| 348 | fmt.Print("error event !!") |
| 349 | newmessage = "error event !!" |
| 350 | } else { |
| 351 | err, currentlist := GetCurrentDevices() |
dileepbk | 86ef010 | 2019-11-13 00:08:33 +0000 | [diff] [blame] | 352 | |
Dinesh Belwalkar | 72ecafb | 2019-12-12 00:08:56 +0000 | [diff] [blame] | 353 | if err != nil { |
| 354 | errStatus, _ := status.FromError(err) |
| 355 | fmt.Println(errStatus.Message()) |
| 356 | fmt.Println(errStatus.Code()) |
| 357 | newmessage = errStatus.Message() |
| 358 | fmt.Print("showdevices error!!") |
| 359 | } else { |
| 360 | fmt.Print("showdevices ", currentlist) |
mc | cd7e950 | 2019-12-16 22:04:13 +0000 | [diff] [blame^] | 361 | sort.Strings(currentlist[:]) |
| 362 | newmessage = strings.Join(currentlist[:], " ") |
| 363 | newmessage = newmessage + "\n" |
dileepbk | 86ef010 | 2019-11-13 00:08:33 +0000 | [diff] [blame] | 364 | } |
Dinesh Belwalkar | 72ecafb | 2019-12-12 00:08:56 +0000 | [diff] [blame] | 365 | } |
mc | 20a4b5f | 2019-10-16 20:28:24 +0000 | [diff] [blame] | 366 | default: |
| 367 | newmessage = newmessage + "invalid command " + cmdstr + "\n" |
Dinesh Belwalkar | b5db83f | 2019-10-24 17:27:58 +0000 | [diff] [blame] | 368 | } |
mc | cd7e950 | 2019-12-16 22:04:13 +0000 | [diff] [blame^] | 369 | // send string back to client |
| 370 | connS.Write([]byte(newmessage + ";")) |
nickhuang | 6b31f8f | 2019-09-26 02:02:14 +0000 | [diff] [blame] | 371 | } |
Dinesh Belwalkar | e1e85ad | 2019-07-31 23:06:47 +0000 | [diff] [blame] | 372 | } |