"io/ioutil" has been deprecated

sue445 · Jul 13, 2024 · af21686 · af21686
1 parent b6fd56a
commit af21686
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 11 deletions.
diff --git a/lib/erd_generator.go b/lib/erd_generator.go
@@ -4,7 +4,6 @@ import (
 	"fmt"
 	"github.com/cockroachdb/errors"
 	"github.com/sue445/plant_erd/db"
-	"io/ioutil"
 	"os"
 	"regexp"
 )
@@ -97,7 +96,7 @@ func (g *ErdGenerator) output(content string) error {
 	}
 
 	// Output to file
-	err := ioutil.WriteFile(g.Filepath, []byte(content), 0644)
+	err := os.WriteFile(g.Filepath, []byte(content), 0644)
 	if err != nil {
 		return errors.WithStack(err)
 	}

diff --git a/lib/erd_generator_test.go b/lib/erd_generator_test.go
@@ -3,7 +3,6 @@ package lib
 import (
 	"bytes"
 	"io"
-	"io/ioutil"
 	"os"
 	"path/filepath"
 	"testing"
@@ -208,13 +207,7 @@ func TestErdGenerator_generateMermaidErd(t *testing.T) {
 }
 
 func TestErdGenerator_output_ToFile(t *testing.T) {
-	dir, err := ioutil.TempDir("", "example")
-
-	if err != nil {
-		panic(err)
-	}
-
-	defer os.RemoveAll(dir)
+	dir := t.TempDir()
 
 	filePath := filepath.Join(dir, "erd.txt")
 	g := &ErdGenerator{
@@ -223,7 +216,7 @@ func TestErdGenerator_output_ToFile(t *testing.T) {
 
 	g.output("aaa")
 
-	data, err := ioutil.ReadFile(filePath)
+	data, err := os.ReadFile(filePath)
 
 	if assert.NoError(t, err) {
 		str := string(data)