Get Volume data-source improvement (#177)

civo/datasource_volume.go

@@ -74,20 +74,24 @@ func dataSourceVolumeRead(_ context.Context, d *schema.ResourceData, m interface
 		log.Printf("[INFO] Getting the volume by id")
 		volume, err := apiClient.FindVolume(id.(string))
 		if err != nil {
-			return diag.Errorf("[ERR] failed to retrive volume: %s", err)
+			return diag.Errorf("[ERR] failed to retrieve volume: %s", err)
 		}
 
 		foundVolume = volume
 	} else if name, ok := d.GetOk("name"); ok {
 		log.Printf("[INFO] Getting the volume by name")
 		volume, err := apiClient.FindVolume(name.(string))
 		if err != nil {
-			return diag.Errorf("[ERR] failed to retrive volume: %s", err)
+			return diag.Errorf("[ERR] failed to retrieve volume: %s", err)
 		}
 
 		foundVolume = volume
 	}
 
+	//Check for nil pointer
+	if foundVolume == nil {
+		return diag.Errorf("[ERR] failed to retrieve volume")
+	}
 	d.SetId(foundVolume.ID)
 	d.Set("name", foundVolume.Name)
 	d.Set("size_gb", foundVolume.SizeGigabytes)

examples/data-sources/civo_volume/data-source.tf

@@ -1,3 +1,7 @@
-data "civo_volume" "mysql" {
-    name = "database-mysql"
+data "civo_volume" "myvolume" {
+  name = "test-volume-name"
+}
+
+output "volume_output" {
+  value = data.civo_volume.myvolume
 }