From baab8294939718c4fe30acd8038448b9bcf67608 Mon Sep 17 00:00:00 2001
From: weibin <xiaobinqt@163.com>
Date: Sat, 25 Dec 2021 17:13:13 +0800
Subject: [PATCH] fix save avatar

fixed save avatar `image not loaded  try to open it externally to fix format problem`
if response content length is 0, an error is reported.
---
 user.go | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/user.go b/user.go
index a87b5d8..61a8e27 100644
--- a/user.go
+++ b/user.go
@@ -75,6 +75,11 @@ func (u *User) SaveAvatar(filename string) error {
 		}
 	}
 	defer resp.Body.Close()
+	// 写文件前判断下 content length 是否是 0，不然保存的头像会出现
+	// image not loaded  try to open it externally to fix format problem 问题
+	if resp.ContentLength == 0 {
+		return fmt.Errorf("get avatar response content length is 0")
+	}
 	file, err := os.Create(filename)
 	if err != nil {
 		return err