generate table, add error but fail

osbzr · Oct 18, 2019 · e4deaf3 · GoodERPJeff · Oct 18, 2019 · e4deaf3
1 parent 9618629
commit e4deaf3
Show file tree

Hide file tree

Showing 3 changed files with 9 additions and 3 deletions.
diff --git a/models/models.go b/models/models.go
@@ -113,6 +113,7 @@ func init() {
 		new(OAuth2AuthorizationCode),
 		new(OAuth2Grant),
 		new(Task),
+		new(Transfer),
 	)
 
 	gonicNames := []string{"SSL", "UID"}

diff --git a/models/transfer_point.go b/models/transfer_point.go
@@ -26,9 +26,9 @@ func TransferPoint(FromID int64, Why string, ToID int64, Qty int) (err error) {
 		return err
 	}
 
-//	if _, err = sess.Insert(&Transfer{FromID: FromID, ToID: ToID, Why: Why, Qty: Qty}); err != nil {
-//		return err
-//	}
+	if _, err = sess.Insert(&Transfer{FromID: FromID, ToID: ToID, Why: Why, Qty: Qty}); err != nil {
+		return err
+	}
 
 	if _, err = sess.Exec("UPDATE `user` SET point = point + ? WHERE id = ?", Qty, ToID); err != nil {
 		return err

diff --git a/routers/user/profile.go b/routers/user/profile.go
@@ -280,6 +280,11 @@ func TransferP(ctx *context.Context) {
 	var err error
 	Qty, err := strconv.Atoi(ctx.Query("qty"))
 	if err != nil {
+		ctx.Flash.Error("请输入数字")
+		return
+	}
+	if ctx.User.Point < Qty {
+		ctx.Flash.Error("余额不足！")
 		return
 	}
 	err = models.TransferPoint(ctx.User.ID,