Matteo Scandolo | 8df63df | 2019-09-12 10:34:32 -0700 | [diff] [blame] | 1 | /* |
| 2 | * Portions copyright 2019-present Open Networking Foundation |
| 3 | * Original copyright 2019-present Ciena Corporation |
| 4 | * |
| 5 | * Licensed under the Apache License, Version 2.0 (the "License"); |
| 6 | * you may not use this file except in compliance with the License. |
| 7 | * You may obtain a copy of the License at |
| 8 | * |
| 9 | * http://www.apache.org/licenses/LICENSE-2.0 |
| 10 | * |
| 11 | * Unless required by applicable law or agreed to in writing, software |
| 12 | * distributed under the License is distributed on an "AS IS" BASIS, |
| 13 | * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
| 14 | * See the License for the specific language governing permissions and |
| 15 | * limitations under the License. |
| 16 | */ |
| 17 | |
| 18 | package commands |
| 19 | |
| 20 | import ( |
| 21 | "context" |
Matteo Scandolo | 10f965c | 2019-09-24 10:40:46 -0700 | [diff] [blame] | 22 | "fmt" |
Anand S Katti | 0954135 | 2020-01-29 15:54:01 +0530 | [diff] [blame] | 23 | "os" |
| 24 | "strconv" |
| 25 | "strings" |
| 26 | |
Matteo Scandolo | 8df63df | 2019-09-12 10:34:32 -0700 | [diff] [blame] | 27 | "github.com/jessevdk/go-flags" |
Anand S Katti | 0954135 | 2020-01-29 15:54:01 +0530 | [diff] [blame] | 28 | "github.com/olekukonko/tablewriter" |
Matteo Scandolo | 8df63df | 2019-09-12 10:34:32 -0700 | [diff] [blame] | 29 | pb "github.com/opencord/bbsim/api/bbsim" |
| 30 | "github.com/opencord/bbsim/internal/bbsimctl/config" |
| 31 | "github.com/opencord/cordctl/pkg/format" |
| 32 | log "github.com/sirupsen/logrus" |
Matteo Scandolo | 8df63df | 2019-09-12 10:34:32 -0700 | [diff] [blame] | 33 | ) |
| 34 | |
| 35 | const ( |
Matteo Scandolo | 8a57481 | 2021-05-20 15:18:53 -0700 | [diff] [blame] | 36 | DEFAULT_ONU_DEVICE_HEADER_FORMAT = "table{{ .PonPortID }}\t{{ .ID }}\t{{ .SerialNumber }}\t{{ .OperState }}\t{{ .InternalState }}\t{{ .ImageSoftwareExpectedSections }}\t{{ .ImageSoftwareReceivedSections }}\t{{ .ActiveImageEntityId }}\t{{ .CommittedImageEntityId }}" |
| 37 | DEFAULT_ONU_DEVICE_HEADER_FORMAT_WITH_SERVICES = "table{{ .PonPortID }}\t{{ .ID }}\t{{ .SerialNumber }}\t{{ .OperState }}\t{{ .InternalState }}\t{{ .ImageSoftwareExpectedSections }}\t{{ .ImageSoftwareReceivedSections }}\t{{ .ActiveImageEntityId }}\t{{ .CommittedImageEntityId }}\t{{ .Unis }}" |
Matteo Scandolo | 8df63df | 2019-09-12 10:34:32 -0700 | [diff] [blame] | 38 | ) |
| 39 | |
Matteo Scandolo | 10f965c | 2019-09-24 10:40:46 -0700 | [diff] [blame] | 40 | type OnuSnString string |
Arjun E K | 57a7fcb | 2020-01-30 06:44:45 +0000 | [diff] [blame] | 41 | type IgmpSubAction string |
Onur Kalinagac | 9f9faca | 2021-01-21 14:04:34 +0000 | [diff] [blame] | 42 | type GroupAddress string |
Arjun E K | 57a7fcb | 2020-01-30 06:44:45 +0000 | [diff] [blame] | 43 | |
| 44 | const IgmpJoinKey string = "join" |
| 45 | const IgmpLeaveKey string = "leave" |
Arjun E K | dd443f0 | 2020-02-07 15:24:01 +0000 | [diff] [blame] | 46 | const IgmpJoinKeyV3 string = "joinv3" |
Arjun E K | 57a7fcb | 2020-01-30 06:44:45 +0000 | [diff] [blame] | 47 | |
Matteo Scandolo | 618a658 | 2020-09-09 12:21:29 -0700 | [diff] [blame] | 48 | var IgmpAllowedActions = []string{IgmpJoinKey, IgmpLeaveKey, IgmpJoinKeyV3} |
| 49 | |
Matteo Scandolo | 4a03626 | 2020-08-17 15:56:13 -0700 | [diff] [blame] | 50 | type ONUList struct { |
| 51 | Verbose bool `short:"v" long:"verbose" description:"Print all the informations we have about ONUs"` |
| 52 | } |
Matteo Scandolo | d2ca2c7 | 2019-10-04 16:50:22 -0700 | [diff] [blame] | 53 | |
| 54 | type ONUGet struct { |
Matteo Scandolo | 4a03626 | 2020-08-17 15:56:13 -0700 | [diff] [blame] | 55 | Verbose bool `short:"v" long:"verbose" description:"Print all the informations we have about ONUs"` |
| 56 | Args struct { |
| 57 | OnuSn OnuSnString |
| 58 | } `positional-args:"yes" required:"yes"` |
| 59 | } |
| 60 | |
Matteo Scandolo | ef4e8f8 | 2021-05-17 11:20:49 -0700 | [diff] [blame] | 61 | type ONUUnis struct { |
Matteo Scandolo | 8a57481 | 2021-05-20 15:18:53 -0700 | [diff] [blame] | 62 | Verbose bool `short:"v" long:"verbose" description:"Print all the informations we have about UNIs"` |
| 63 | Args struct { |
Matteo Scandolo | ef4e8f8 | 2021-05-17 11:20:49 -0700 | [diff] [blame] | 64 | OnuSn OnuSnString |
| 65 | } `positional-args:"yes" required:"yes"` |
| 66 | } |
| 67 | |
Matteo Scandolo | 10f965c | 2019-09-24 10:40:46 -0700 | [diff] [blame] | 68 | type ONUShutDown struct { |
| 69 | Args struct { |
| 70 | OnuSn OnuSnString |
| 71 | } `positional-args:"yes" required:"yes"` |
Matteo Scandolo | 8df63df | 2019-09-12 10:34:32 -0700 | [diff] [blame] | 72 | } |
| 73 | |
Matteo Scandolo | 10f965c | 2019-09-24 10:40:46 -0700 | [diff] [blame] | 74 | type ONUPowerOn struct { |
| 75 | Args struct { |
| 76 | OnuSn OnuSnString |
| 77 | } `positional-args:"yes" required:"yes"` |
| 78 | } |
| 79 | |
Matteo Scandolo | e383d5d | 2019-10-25 14:47:27 -0700 | [diff] [blame] | 80 | type ONUEapolRestart struct { |
| 81 | Args struct { |
| 82 | OnuSn OnuSnString |
| 83 | } `positional-args:"yes" required:"yes"` |
| 84 | } |
| 85 | |
| 86 | type ONUDhcpRestart struct { |
| 87 | Args struct { |
| 88 | OnuSn OnuSnString |
| 89 | } `positional-args:"yes" required:"yes"` |
| 90 | } |
| 91 | |
Arjun E K | 57a7fcb | 2020-01-30 06:44:45 +0000 | [diff] [blame] | 92 | type ONUIgmp struct { |
| 93 | Args struct { |
Onur Kalinagac | 9f9faca | 2021-01-21 14:04:34 +0000 | [diff] [blame] | 94 | OnuSn OnuSnString |
| 95 | SubAction IgmpSubAction |
| 96 | GroupAddress GroupAddress |
Arjun E K | 57a7fcb | 2020-01-30 06:44:45 +0000 | [diff] [blame] | 97 | } `positional-args:"yes" required:"yes"` |
| 98 | } |
| 99 | |
Pragya Arya | 8bdb453 | 2020-03-02 17:08:09 +0530 | [diff] [blame] | 100 | type ONUTrafficSchedulers struct { |
| 101 | Args struct { |
| 102 | OnuSn OnuSnString |
| 103 | } `positional-args:"yes" required:"yes"` |
| 104 | } |
| 105 | |
| 106 | type ONUFlows struct { |
| 107 | Args struct { |
| 108 | OnuSn OnuSnString |
| 109 | } `positional-args:"yes" required:"yes"` |
| 110 | } |
| 111 | |
Nitin Subramanian | b0a333a | 2021-07-08 15:01:41 -0700 | [diff] [blame] | 112 | type ONUServices struct { |
| 113 | Args struct { |
| 114 | OnuSn OnuSnString |
| 115 | } `positional-args:"yes"` |
| 116 | } |
| 117 | |
Matteo Scandolo | 10f965c | 2019-09-24 10:40:46 -0700 | [diff] [blame] | 118 | type ONUOptions struct { |
Anand S Katti | 0954135 | 2020-01-29 15:54:01 +0530 | [diff] [blame] | 119 | List ONUList `command:"list"` |
| 120 | Get ONUGet `command:"get"` |
| 121 | ShutDown ONUShutDown `command:"shutdown"` |
| 122 | PowerOn ONUPowerOn `command:"poweron"` |
| 123 | RestartEapol ONUEapolRestart `command:"auth_restart"` |
| 124 | RestartDchp ONUDhcpRestart `command:"dhcp_restart"` |
| 125 | Igmp ONUIgmp `command:"igmp"` |
| 126 | TrafficSchedulers ONUTrafficSchedulers `command:"traffic_schedulers"` |
Anand S Katti | 86552f9 | 2020-03-03 21:56:32 +0530 | [diff] [blame] | 127 | Alarms AlarmOptions `command:"alarms"` |
Pragya Arya | 8bdb453 | 2020-03-02 17:08:09 +0530 | [diff] [blame] | 128 | Flows ONUFlows `command:"flows"` |
Nitin Subramanian | b0a333a | 2021-07-08 15:01:41 -0700 | [diff] [blame] | 129 | Services ONUServices `command:"services"` |
Matteo Scandolo | 10f965c | 2019-09-24 10:40:46 -0700 | [diff] [blame] | 130 | } |
| 131 | |
| 132 | func RegisterONUCommands(parser *flags.Parser) { |
Shrey Baid | 688b424 | 2020-07-10 20:40:10 +0530 | [diff] [blame] | 133 | _, _ = parser.AddCommand("onu", "ONU Commands", "Commands to query and manipulate ONU devices", &ONUOptions{}) |
Matteo Scandolo | 10f965c | 2019-09-24 10:40:46 -0700 | [diff] [blame] | 134 | } |
| 135 | |
Matteo Scandolo | 10f965c | 2019-09-24 10:40:46 -0700 | [diff] [blame] | 136 | func getONUs() *pb.ONUs { |
| 137 | |
| 138 | client, conn := connect() |
Matteo Scandolo | 8df63df | 2019-09-12 10:34:32 -0700 | [diff] [blame] | 139 | defer conn.Close() |
Matteo Scandolo | 8df63df | 2019-09-12 10:34:32 -0700 | [diff] [blame] | 140 | |
| 141 | // Contact the server and print out its response. |
Matteo Scandolo | 8df63df | 2019-09-12 10:34:32 -0700 | [diff] [blame] | 142 | ctx, cancel := context.WithTimeout(context.Background(), config.GlobalConfig.Grpc.Timeout) |
| 143 | defer cancel() |
Matteo Scandolo | 10f965c | 2019-09-24 10:40:46 -0700 | [diff] [blame] | 144 | |
| 145 | onus, err := client.GetONUs(ctx, &pb.Empty{}) |
Matteo Scandolo | 8df63df | 2019-09-12 10:34:32 -0700 | [diff] [blame] | 146 | if err != nil { |
Matteo Scandolo | 2bf742a | 2019-10-01 11:33:34 -0700 | [diff] [blame] | 147 | log.Fatalf("could not get OLT: %v", err) |
Matteo Scandolo | 8df63df | 2019-09-12 10:34:32 -0700 | [diff] [blame] | 148 | return nil |
| 149 | } |
| 150 | return onus |
| 151 | } |
| 152 | |
Matteo Scandolo | 10f965c | 2019-09-24 10:40:46 -0700 | [diff] [blame] | 153 | func (options *ONUList) Execute(args []string) error { |
Matteo Scandolo | 8df63df | 2019-09-12 10:34:32 -0700 | [diff] [blame] | 154 | onus := getONUs() |
| 155 | |
| 156 | // print out |
Matteo Scandolo | 4a03626 | 2020-08-17 15:56:13 -0700 | [diff] [blame] | 157 | var tableFormat format.Format |
| 158 | if options.Verbose { |
| 159 | tableFormat = format.Format(DEFAULT_ONU_DEVICE_HEADER_FORMAT_WITH_SERVICES) |
| 160 | } else { |
| 161 | tableFormat = format.Format(DEFAULT_ONU_DEVICE_HEADER_FORMAT) |
| 162 | } |
Matteo Scandolo | 8df63df | 2019-09-12 10:34:32 -0700 | [diff] [blame] | 163 | if err := tableFormat.Execute(os.Stdout, true, onus.Items); err != nil { |
| 164 | log.Fatalf("Error while formatting ONUs table: %s", err) |
| 165 | } |
| 166 | |
| 167 | return nil |
| 168 | } |
Matteo Scandolo | 10f965c | 2019-09-24 10:40:46 -0700 | [diff] [blame] | 169 | |
Matteo Scandolo | d2ca2c7 | 2019-10-04 16:50:22 -0700 | [diff] [blame] | 170 | func (options *ONUGet) Execute(args []string) error { |
| 171 | client, conn := connect() |
| 172 | defer conn.Close() |
| 173 | |
| 174 | ctx, cancel := context.WithTimeout(context.Background(), config.GlobalConfig.Grpc.Timeout) |
| 175 | defer cancel() |
| 176 | req := pb.ONURequest{ |
| 177 | SerialNumber: string(options.Args.OnuSn), |
| 178 | } |
| 179 | res, err := client.GetONU(ctx, &req) |
| 180 | |
| 181 | if err != nil { |
Matteo Scandolo | 4a03626 | 2020-08-17 15:56:13 -0700 | [diff] [blame] | 182 | log.Fatalf("Cannot not get ONU %s: %v", options.Args.OnuSn, err) |
Matteo Scandolo | d2ca2c7 | 2019-10-04 16:50:22 -0700 | [diff] [blame] | 183 | return err |
| 184 | } |
| 185 | |
Matteo Scandolo | 4a03626 | 2020-08-17 15:56:13 -0700 | [diff] [blame] | 186 | var tableFormat format.Format |
| 187 | if options.Verbose { |
| 188 | tableFormat = format.Format(DEFAULT_ONU_DEVICE_HEADER_FORMAT_WITH_SERVICES) |
| 189 | } else { |
| 190 | tableFormat = format.Format(DEFAULT_ONU_DEVICE_HEADER_FORMAT) |
| 191 | } |
Matteo Scandolo | d2ca2c7 | 2019-10-04 16:50:22 -0700 | [diff] [blame] | 192 | if err := tableFormat.Execute(os.Stdout, true, []*pb.ONU{res}); err != nil { |
| 193 | log.Fatalf("Error while formatting ONUs table: %s", err) |
| 194 | } |
| 195 | |
| 196 | return nil |
| 197 | } |
| 198 | |
Matteo Scandolo | 10f965c | 2019-09-24 10:40:46 -0700 | [diff] [blame] | 199 | func (options *ONUShutDown) Execute(args []string) error { |
| 200 | |
| 201 | client, conn := connect() |
| 202 | defer conn.Close() |
| 203 | |
| 204 | ctx, cancel := context.WithTimeout(context.Background(), config.GlobalConfig.Grpc.Timeout) |
| 205 | defer cancel() |
| 206 | req := pb.ONURequest{ |
| 207 | SerialNumber: string(options.Args.OnuSn), |
| 208 | } |
| 209 | res, err := client.ShutdownONU(ctx, &req) |
| 210 | |
| 211 | if err != nil { |
Matteo Scandolo | e383d5d | 2019-10-25 14:47:27 -0700 | [diff] [blame] | 212 | log.Fatalf("Cannot shutdown ONU %s: %v", options.Args.OnuSn, err) |
Matteo Scandolo | 10f965c | 2019-09-24 10:40:46 -0700 | [diff] [blame] | 213 | return err |
| 214 | } |
| 215 | |
| 216 | fmt.Println(fmt.Sprintf("[Status: %d] %s", res.StatusCode, res.Message)) |
| 217 | |
| 218 | return nil |
| 219 | } |
| 220 | |
| 221 | func (options *ONUPowerOn) Execute(args []string) error { |
| 222 | client, conn := connect() |
| 223 | defer conn.Close() |
| 224 | |
| 225 | ctx, cancel := context.WithTimeout(context.Background(), config.GlobalConfig.Grpc.Timeout) |
| 226 | defer cancel() |
| 227 | req := pb.ONURequest{ |
| 228 | SerialNumber: string(options.Args.OnuSn), |
| 229 | } |
| 230 | res, err := client.PoweronONU(ctx, &req) |
| 231 | |
| 232 | if err != nil { |
Matteo Scandolo | e383d5d | 2019-10-25 14:47:27 -0700 | [diff] [blame] | 233 | log.Fatalf("Cannot power on ONU %s: %v", options.Args.OnuSn, err) |
| 234 | return err |
| 235 | } |
| 236 | |
| 237 | fmt.Println(fmt.Sprintf("[Status: %d] %s", res.StatusCode, res.Message)) |
| 238 | |
| 239 | return nil |
| 240 | } |
| 241 | |
| 242 | func (options *ONUEapolRestart) Execute(args []string) error { |
| 243 | client, conn := connect() |
| 244 | defer conn.Close() |
| 245 | |
| 246 | ctx, cancel := context.WithTimeout(context.Background(), config.GlobalConfig.Grpc.Timeout) |
| 247 | defer cancel() |
| 248 | req := pb.ONURequest{ |
| 249 | SerialNumber: string(options.Args.OnuSn), |
| 250 | } |
| 251 | res, err := client.RestartEapol(ctx, &req) |
| 252 | |
| 253 | if err != nil { |
| 254 | log.Fatalf("Cannot restart EAPOL for ONU %s: %v", options.Args.OnuSn, err) |
| 255 | return err |
| 256 | } |
| 257 | |
| 258 | fmt.Println(fmt.Sprintf("[Status: %d] %s", res.StatusCode, res.Message)) |
| 259 | |
| 260 | return nil |
| 261 | } |
| 262 | |
| 263 | func (options *ONUDhcpRestart) Execute(args []string) error { |
| 264 | client, conn := connect() |
| 265 | defer conn.Close() |
| 266 | |
| 267 | ctx, cancel := context.WithTimeout(context.Background(), config.GlobalConfig.Grpc.Timeout) |
| 268 | defer cancel() |
| 269 | req := pb.ONURequest{ |
| 270 | SerialNumber: string(options.Args.OnuSn), |
| 271 | } |
| 272 | res, err := client.RestartDhcp(ctx, &req) |
| 273 | |
| 274 | if err != nil { |
| 275 | log.Fatalf("Cannot restart DHCP for ONU %s: %v", options.Args.OnuSn, err) |
Matteo Scandolo | 10f965c | 2019-09-24 10:40:46 -0700 | [diff] [blame] | 276 | return err |
| 277 | } |
| 278 | |
| 279 | fmt.Println(fmt.Sprintf("[Status: %d] %s", res.StatusCode, res.Message)) |
| 280 | |
| 281 | return nil |
| 282 | } |
| 283 | |
Arjun E K | 57a7fcb | 2020-01-30 06:44:45 +0000 | [diff] [blame] | 284 | func (options *ONUIgmp) Execute(args []string) error { |
| 285 | client, conn := connect() |
| 286 | defer conn.Close() |
| 287 | |
| 288 | ctx, cancel := context.WithTimeout(context.Background(), config.GlobalConfig.Grpc.Timeout) |
| 289 | defer cancel() |
| 290 | |
| 291 | req := pb.ONURequest{ |
| 292 | SerialNumber: string(options.Args.OnuSn), |
| 293 | } |
| 294 | |
| 295 | var subActionVal pb.SubActionTypes |
| 296 | if string(options.Args.SubAction) == IgmpJoinKey { |
| 297 | subActionVal = pb.SubActionTypes_JOIN |
| 298 | } else if string(options.Args.SubAction) == IgmpLeaveKey { |
| 299 | subActionVal = pb.SubActionTypes_LEAVE |
Anand S Katti | 0954135 | 2020-01-29 15:54:01 +0530 | [diff] [blame] | 300 | } else if string(options.Args.SubAction) == IgmpJoinKeyV3 { |
| 301 | subActionVal = pb.SubActionTypes_JOINV3 |
| 302 | } |
Arjun E K | 57a7fcb | 2020-01-30 06:44:45 +0000 | [diff] [blame] | 303 | |
| 304 | igmpReq := pb.IgmpRequest{ |
| 305 | OnuReq: &req, |
| 306 | SubActionVal: subActionVal, |
Onur Kalinagac | 9f9faca | 2021-01-21 14:04:34 +0000 | [diff] [blame] | 307 | GroupAddress: string(options.Args.GroupAddress), |
Arjun E K | 57a7fcb | 2020-01-30 06:44:45 +0000 | [diff] [blame] | 308 | } |
| 309 | res, err := client.GetONU(ctx, igmpReq.OnuReq) |
| 310 | if err != nil { |
| 311 | log.WithFields(log.Fields{ |
| 312 | "SerialNumber": options.Args.OnuSn, |
| 313 | }).Errorf("Cannot not get details on ONU error: %v", err) |
| 314 | } |
| 315 | log.WithFields(log.Fields{ |
| 316 | "SerialNumber": igmpReq.OnuReq.SerialNumber, |
Onur Kalinagac | 9f9faca | 2021-01-21 14:04:34 +0000 | [diff] [blame] | 317 | }).Debugf("ONU has identified : %s", res) |
Arjun E K | 57a7fcb | 2020-01-30 06:44:45 +0000 | [diff] [blame] | 318 | |
| 319 | igmpRes, igmpErr := client.ChangeIgmpState(ctx, &igmpReq) |
| 320 | if igmpErr != nil { |
| 321 | log.WithFields(log.Fields{ |
| 322 | "SubAction": options.Args.SubAction, |
| 323 | }).Errorf("Could not process Action: error: %v", igmpErr) |
| 324 | } else { |
| 325 | log.WithFields(log.Fields{ |
| 326 | "SubAction": options.Args.SubAction, |
| 327 | }).Debugf("igmp state has been changed with response: %s", |
| 328 | igmpRes.Message) |
| 329 | } |
| 330 | |
| 331 | return nil |
| 332 | } |
| 333 | |
Pragya Arya | 8bdb453 | 2020-03-02 17:08:09 +0530 | [diff] [blame] | 334 | func (options *ONUFlows) Execute(args []string) error { |
| 335 | client, conn := connect() |
| 336 | defer conn.Close() |
| 337 | |
| 338 | ctx, cancel := context.WithTimeout(context.Background(), config.GlobalConfig.Grpc.Timeout) |
| 339 | defer cancel() |
| 340 | req := pb.ONURequest{ |
| 341 | SerialNumber: string(options.Args.OnuSn), |
| 342 | } |
| 343 | res, err := client.GetFlows(ctx, &req) |
| 344 | if err != nil { |
| 345 | log.Errorf("Cannot get flows for ONU %s: %v", options.Args.OnuSn, err) |
| 346 | return err |
| 347 | } |
| 348 | |
| 349 | if res.Flows == nil { |
| 350 | fmt.Println(fmt.Sprintf("ONU %s has no flows", options.Args.OnuSn)) |
| 351 | return nil |
| 352 | } |
| 353 | |
| 354 | flowHeader := []string{ |
| 355 | "access_intf_id", |
| 356 | "onu_id", |
| 357 | "uni_id", |
| 358 | "flow_id", |
| 359 | "flow_type", |
| 360 | "eth_type", |
| 361 | "alloc_id", |
| 362 | "network_intf_id", |
| 363 | "gemport_id", |
| 364 | "classifier", |
| 365 | "action", |
| 366 | "priority", |
| 367 | "cookie", |
| 368 | "port_no", |
| 369 | } |
| 370 | |
| 371 | tableFlow := tablewriter.NewWriter(os.Stdout) |
| 372 | tableFlow.SetRowLine(true) |
| 373 | fmt.Fprintf(os.Stdout, "ONU Flows:\n") |
| 374 | tableFlow.SetHeader(flowHeader) |
| 375 | |
| 376 | for _, flow := range res.Flows { |
| 377 | flowInfo := []string{} |
| 378 | flowInfo = append(flowInfo, |
| 379 | strconv.Itoa(int(flow.AccessIntfId)), |
| 380 | strconv.Itoa(int(flow.OnuId)), |
| 381 | strconv.Itoa(int(flow.UniId)), |
| 382 | strconv.Itoa(int(flow.FlowId)), |
| 383 | flow.FlowType, |
| 384 | fmt.Sprintf("%x", flow.Classifier.EthType), |
| 385 | strconv.Itoa(int(flow.AllocId)), |
| 386 | strconv.Itoa(int(flow.NetworkIntfId)), |
| 387 | strconv.Itoa(int(flow.GemportId)), |
| 388 | flow.Classifier.String(), |
| 389 | flow.Action.String(), |
| 390 | strconv.Itoa(int(flow.Priority)), |
| 391 | strconv.Itoa(int(flow.Cookie)), |
| 392 | strconv.Itoa(int(flow.PortNo)), |
| 393 | ) |
| 394 | tableFlow.Append(flowInfo) |
| 395 | } |
| 396 | tableFlow.Render() |
| 397 | tableFlow.SetNewLine("") |
| 398 | return nil |
| 399 | } |
| 400 | |
Matteo Scandolo | 10f965c | 2019-09-24 10:40:46 -0700 | [diff] [blame] | 401 | func (onuSn *OnuSnString) Complete(match string) []flags.Completion { |
| 402 | client, conn := connect() |
| 403 | defer conn.Close() |
| 404 | |
| 405 | ctx, cancel := context.WithTimeout(context.Background(), config.GlobalConfig.Grpc.Timeout) |
| 406 | defer cancel() |
| 407 | |
| 408 | onus, err := client.GetONUs(ctx, &pb.Empty{}) |
| 409 | if err != nil { |
Matteo Scandolo | 86e8ce6 | 2019-10-11 12:03:10 -0700 | [diff] [blame] | 410 | log.Fatalf("could not get ONUs: %v", err) |
Matteo Scandolo | 10f965c | 2019-09-24 10:40:46 -0700 | [diff] [blame] | 411 | return nil |
| 412 | } |
| 413 | |
| 414 | list := make([]flags.Completion, 0) |
| 415 | for _, k := range onus.Items { |
| 416 | if strings.HasPrefix(k.SerialNumber, match) { |
| 417 | list = append(list, flags.Completion{Item: k.SerialNumber}) |
| 418 | } |
| 419 | } |
| 420 | |
| 421 | return list |
| 422 | } |
Anand S Katti | 0954135 | 2020-01-29 15:54:01 +0530 | [diff] [blame] | 423 | |
Matteo Scandolo | 618a658 | 2020-09-09 12:21:29 -0700 | [diff] [blame] | 424 | func (onuSn *IgmpSubAction) Complete(match string) []flags.Completion { |
| 425 | list := make([]flags.Completion, 0) |
| 426 | for _, k := range IgmpAllowedActions { |
| 427 | if strings.HasPrefix(k, match) { |
| 428 | list = append(list, flags.Completion{Item: k}) |
| 429 | } |
| 430 | } |
| 431 | return list |
| 432 | } |
| 433 | |
Anand S Katti | 0954135 | 2020-01-29 15:54:01 +0530 | [diff] [blame] | 434 | func (options *ONUTrafficSchedulers) Execute(args []string) error { |
| 435 | client, conn := connect() |
| 436 | defer conn.Close() |
| 437 | |
| 438 | ctx, cancel := context.WithTimeout(context.Background(), config.GlobalConfig.Grpc.Timeout) |
| 439 | defer cancel() |
| 440 | req := pb.ONURequest{ |
| 441 | SerialNumber: string(options.Args.OnuSn), |
| 442 | } |
| 443 | res, err := client.GetOnuTrafficSchedulers(ctx, &req) |
| 444 | if err != nil { |
| 445 | log.Fatalf("Cannot get traffic schedulers for ONU %s: %v", options.Args.OnuSn, err) |
| 446 | return err |
| 447 | } |
| 448 | |
| 449 | if res.TraffSchedulers == nil { |
| 450 | log.Fatalf("Cannot get traffic schedulers for ONU: %s (unavailable)", options.Args.OnuSn) |
| 451 | return nil |
| 452 | } |
| 453 | |
| 454 | SchedulerHeader := []string{"Direction", |
| 455 | "AllocId", |
| 456 | "Scheduler.Direction", |
| 457 | "Scheduler.AdditionalBw", |
| 458 | "Scheduler.Priority", |
| 459 | "Scheduler.Weight", |
| 460 | "Scheduler.SchedPolicy", |
| 461 | } |
| 462 | |
| 463 | ShapingInfoHeader := []string{"InferredAdditionBwIndication", |
| 464 | "Cbs", |
| 465 | "Cir", |
| 466 | "Gir", |
| 467 | "Pbs", |
| 468 | "Pir", |
| 469 | } |
| 470 | |
| 471 | SchedulerVals := []string{} |
| 472 | ShapingInfoVals := []string{} |
| 473 | for _, v := range res.TraffSchedulers.TrafficScheds { |
| 474 | SchedulerVals = append(SchedulerVals, |
| 475 | v.GetDirection().String(), |
| 476 | strconv.Itoa(int(v.GetAllocId())), |
| 477 | v.Scheduler.GetDirection().String(), |
| 478 | v.Scheduler.GetAdditionalBw().String(), |
| 479 | strconv.Itoa(int(v.Scheduler.GetPriority())), |
| 480 | strconv.Itoa(int(v.Scheduler.GetWeight())), |
| 481 | v.GetScheduler().GetSchedPolicy().String(), |
| 482 | ) |
| 483 | |
| 484 | ShapingInfoVals = append(ShapingInfoVals, |
| 485 | v.TrafficShapingInfo.GetAddBwInd().String(), |
| 486 | strconv.Itoa(int(v.TrafficShapingInfo.GetCbs())), |
| 487 | strconv.Itoa(int(v.TrafficShapingInfo.GetCir())), |
| 488 | strconv.Itoa(int(v.TrafficShapingInfo.GetGir())), |
| 489 | strconv.Itoa(int(v.TrafficShapingInfo.GetPbs())), |
| 490 | strconv.Itoa(int(v.TrafficShapingInfo.GetPir())), |
| 491 | ) |
| 492 | } |
| 493 | |
| 494 | fmt.Fprintf(os.Stdout, "OnuId: %d \n", int(res.TraffSchedulers.OnuId)) |
| 495 | fmt.Fprintf(os.Stdout, "IntfId: %d \n", int(res.TraffSchedulers.IntfId)) |
| 496 | fmt.Fprintf(os.Stdout, "UniId: %d \n", int(res.TraffSchedulers.UniId)) |
| 497 | fmt.Fprintf(os.Stdout, "OnuPortNo: %d \n", int(res.TraffSchedulers.PortNo)) |
| 498 | |
| 499 | tableSched := tablewriter.NewWriter(os.Stdout) |
| 500 | tableSched.SetRowLine(true) |
| 501 | fmt.Fprintf(os.Stdout, "Traffic Schedulers Info:\n") |
| 502 | tableSched.SetHeader(SchedulerHeader) |
| 503 | tableSched.Append(SchedulerVals) |
| 504 | tableSched.Render() |
| 505 | tableSched.SetNewLine("") |
| 506 | |
| 507 | tableShap := tablewriter.NewWriter(os.Stdout) |
| 508 | tableShap.SetRowLine(true) |
| 509 | fmt.Fprintf(os.Stdout, "Traffic Shaping Info:\n") |
| 510 | tableShap.SetHeader(ShapingInfoHeader) |
| 511 | tableShap.Append(ShapingInfoVals) |
| 512 | tableShap.Render() |
| 513 | |
| 514 | return nil |
| 515 | } |
Nitin Subramanian | b0a333a | 2021-07-08 15:01:41 -0700 | [diff] [blame] | 516 | |
| 517 | func (options *ONUServices) Execute(args []string) error { |
| 518 | services, err := getServices(string(options.Args.OnuSn), "") |
| 519 | |
| 520 | if err != nil { |
| 521 | log.Errorf("Cannot get services for ONU %s: %v", options.Args.OnuSn, err) |
| 522 | return err |
| 523 | } |
| 524 | |
| 525 | // print out |
| 526 | tableFormat := format.Format(DEFAULT_SERVICE_HEADER_FORMAT) |
| 527 | if err := tableFormat.Execute(os.Stdout, true, services.Items); err != nil { |
| 528 | log.Fatalf("Error while formatting ONUs table: %s", err) |
| 529 | } |
| 530 | |
| 531 | return nil |
| 532 | } |