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