From 320da09f364c03def78ba9539ed72fcfd3591091 Mon Sep 17 00:00:00 2001
From: Xyfacai <xyfacai@gmail.com>
Date: Mon, 8 Apr 2024 23:51:51 +0800
Subject: [PATCH] =?UTF-8?q?fix:=20=E4=BF=AE=E5=A4=8D=E6=B8=A0=E9=81=93?=
 =?UTF-8?q?=E4=B8=80=E6=AC=A1=E6=80=A7=E6=B7=BB=E5=8A=A0=E5=BE=88=E5=A4=9A?=
 =?UTF-8?q?model=E5=A4=B1=E8=B4=A5?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

修复渠道一次性添加很多model并且group多
提示失败 too many SQL variables
---
 model/ability.go | 12 +++++++++++-
 1 file changed, 11 insertions(+), 1 deletion(-)

diff --git a/model/ability.go b/model/ability.go
index 01fea9e..7fd52bc 100644
--- a/model/ability.go
+++ b/model/ability.go
@@ -3,6 +3,7 @@ package model
 import (
 	"errors"
 	"fmt"
+	"github.com/samber/lo"
 	"gorm.io/gorm"
 	"one-api/common"
 	"strings"
@@ -134,7 +135,16 @@ func (channel *Channel) AddAbilities() error {
 			abilities = append(abilities, ability)
 		}
 	}
-	return DB.Create(&abilities).Error
+	if len(abilities) == 0 {
+		return nil
+	}
+	for _, chunk := range lo.Chunk(abilities, 50) {
+		err := DB.Create(&chunk).Error
+		if err != nil {
+			return err
+		}
+	}
+	return nil
 }
 
 func (channel *Channel) DeleteAbilities() error {