Add ImportDatabase

pendo-io · Mar 13, 2021 · d8960c2 · d8960c2
1 parent 3e5fc63
commit d8960c2
Showing 1 changed file with 11 additions and 1 deletion.
diff --git a/mysql/resource_database.go b/mysql/resource_database.go
@@ -22,7 +22,7 @@ func resourceDatabase() *schema.Resource {
 		Read:   ReadDatabase,
 		Delete: DeleteDatabase,
 		Importer: &schema.ResourceImporter{
-			State: schema.ImportStatePassthrough,
+			State: ImportDatabase,
 		},
 		Schema: map[string]*schema.Schema{
 			"name": {
@@ -194,3 +194,13 @@ func extractIdentAfter(sql string, keyword string) string {
 
 	return ""
 }
+
+func ImportDatabase(d *schema.ResourceData, meta interface{}) ([]*schema.ResourceData, error) {
+	err := ReadDatabase(d, meta)
+
+	if err != nil {
+		return nil, err
+	}
+
+	return []*schema.ResourceData{d}, nil
+}