feats

controllers/file.go

@@ -72,3 +72,52 @@ func (c *BaseController) SaveToExcel(fn string, excel *utils.TinyXLSXEngine) {
 	c.destroyContext(true)
 	c.StopRun()
 }
+
+// UploadBase64Image Upload base64-image to ali-oss
+// @Title Upload image to ali-oss
+// @Description 上传base64图片到阿里云
+// @Param   UpImgStr     form    string  true        "图片控件name"
+// @Success 200 { Url } 图片URL
+// @Failure >300 error message
+func (c *BaseController) UploadBase64Image() {
+	base64Str, err := url.QueryUnescape(c.GetString("base64str"))
+	if err != nil {
+		c.ResponseError(utils.LogError("上传 Base64 图片失败", err))
+	}
+
+	imgURL, err := c.uploadStringToAliOSS(base64Str)
+	if err != nil {
+		c.ResponseError(err)
+	}
+
+	resp := map[string]interface{}{
+		"url": imgURL,
+	}
+
+	c.ResponseJSON(resp)
+}
+
+// uploadStringToAliOSS 上传文件到阿里云
+func (c *BaseController) uploadStringToAliOSS(fStr string) (string, error) {
+	aliConf, ok := c.Configer[AliYunConf]
+	if !ok {
+		return "", errors.New("没有找到阿里云相关配置")
+	}
+
+	endpoint := aliConf.String("oss::Endpoint")
+	accessKeyID := aliConf.String("oss::AccessKeyId")
+	accessKeySecret := aliConf.String("oss::AccessKeySecret")
+	bucket := aliConf.String("oss::Bucket")
+
+	aliCli, err := utils.GetOssClient(endpoint, accessKeyID, accessKeySecret)
+	if err != nil {
+		return "", utils.LogError("配置阿里云客户端失败", err)
+	}
+
+	fileURL, err := utils.UploadStringToBucket(aliCli, bucket, fStr)
+	if err != nil {
+		return "", utils.LogError("上传文件到阿里云失败", err)
+	}
+
+	return fileURL, nil
+}

models/luckdraw/luckdraw.go

@@ -722,7 +722,6 @@ func (m *LuckDrawDao) GetWinning(prizes []model.TaLuckdrawPrize) (*model.TaLuckd
 		}
 	}
 
-	utils.LogError(pList)
 	if len(pList) == 0 {
 		return nil, errors.New("所有奖品已抽完")
 	}

routers/common.go

@@ -287,6 +287,7 @@ func getCommonRoutes(prefix string) beego.LinkNamespace {
 
 		// 文件
 		beego.NSRouter("/file", &controllers.BaseController{}, "post:FileUpload"),
+		beego.NSRouter("/file/base64", &controllers.BaseController{}, "post:UploadBase64Image"),
 
 		// 会员
 		beego.NSRouter("/customer", &customer.CustomerController{}, "get:CustWXList"),

utils/rand_test.go

@@ -13,3 +13,51 @@ func TestGetRand(t *testing.T) {
 		t.Fatalf("TestGetRand error, %s, %s", res1, res2)
 	}
 }
+
+func TestLuckyDraw(t *testing.T) {
+	prizeList := []map[string]interface{}{
+		map[string]interface{}{
+			"prize": "A",
+			"prob":  20,
+		},
+		map[string]interface{}{
+			"prize": "B",
+			"prob":  20,
+		},
+		map[string]interface{}{
+			"prize": "C",
+			"prob":  30,
+		},
+		map[string]interface{}{
+			"prize": "D",
+			"prob":  30,
+		},
+	}
+
+	theA := 0
+	theB := 0
+	theC := 0
+	theD := 0
+
+	for i := 0; i < 10000; i++ {
+		res := utils.LuckyDraw(prizeList).(string)
+
+		switch res {
+		case "A":
+			theA += 1
+		case "B":
+			theB += 1
+		case "C":
+			theC += 1
+		case "D":
+			theD += 1
+		default:
+		}
+
+		// bingoList = append(bingoList, utils.LuckyDraw(prizeList))
+	}
+
+	resStr := `A ==> %d  B ==> %d C ==> %d D ==> %d`
+
+	t.Fatalf(resStr, theA, theB, theC, theD)
+}