Feat: update

zijiren233 · Aug 30, 2022 · c072db4 · c072db4
1 parent 30f7dd0
commit c072db4
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 5 deletions.
diff --git a/README.md b/README.md
@@ -4,7 +4,6 @@
 package main
 
 import (
- "bytes"
  "fmt"
  "image/color"
  "io"
@@ -25,7 +24,7 @@ func main() {
  return
  }
  defer f.Close()
- io.Copy(f, bytes.NewReader(pic.GeneratePicture()))
+ io.Copy(f, pic.GeneratePicture())
 }
 ```
 

diff --git a/text2pictrue.go b/text2pictrue.go
@@ -105,11 +105,11 @@ func NewPictureWithBackGround(png *image.RGBA, dpi float64, padding int, fontSiz
  return &p
 }
 
-func (p *picture) GeneratePicture() []byte {
+func (p *picture) GeneratePicture() *bytes.Buffer {
  return saveImage(p.rgba)
 }
 
-func saveImage(rgba *image.RGBA) []byte {
+func saveImage(rgba *image.RGBA) *bytes.Buffer {
  b := bytes.NewBuffer(nil)
  bf := bufio.NewWriter(b)
  if err := png.Encode(bf, rgba); err != nil {
@@ -118,5 +118,5 @@ func saveImage(rgba *image.RGBA) []byte {
  if err := bf.Flush(); err != nil {
  return nil
  }
- return b.Bytes()
+ return b
 }