func RegisterDevice()

in aws/sagemaker.go [77:114]

• 34 lines of code
• 3 McCabe index (conditional complexity)

func RegisterDevice(client SagemakerClient, fleetName *string, deviceName *string, iotThingName *string, targetPlatform *cli.TargetPlatform) {

	getDeviceOutput := GetDevice(client, fleetName, deviceName)

	targetOsKey := "os"
	targetArchKey := "arch"
	targetAccelerator := "accelerator"

	if getDeviceOutput == nil {
		_, err := client.RegisterDevices(context.TODO(), &sagemaker.RegisterDevicesInput{
			DeviceFleetName: fleetName,
			Devices: []types.Device{
				{
					DeviceName:   deviceName,
					IotThingName: iotThingName,
				},
			},
			Tags: []types.Tag{
				{
					Key:   &targetOsKey,
					Value: &targetPlatform.Os,
				},
				{
					Key:   &targetArchKey,
					Value: &targetPlatform.Arch,
				},
				{
					Key:   &targetAccelerator,
					Value: &targetPlatform.Accelerator,
				},
			},
		})

		if err != nil {
			log.Fatalf("Failed to register device %s with fleet %s. Encountered error %s", *fleetName, *deviceName, err)
		}
	}
}