| 123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916 |
- package controller
- import (
- "encoding/json"
- "fmt"
- "net/http"
- "one-api/common"
- "one-api/constant"
- "one-api/model"
- "strconv"
- "strings"
- "github.com/gin-gonic/gin"
- )
- type OpenAIModel struct {
- ID string `json:"id"`
- Object string `json:"object"`
- Created int64 `json:"created"`
- OwnedBy string `json:"owned_by"`
- Permission []struct {
- ID string `json:"id"`
- Object string `json:"object"`
- Created int64 `json:"created"`
- AllowCreateEngine bool `json:"allow_create_engine"`
- AllowSampling bool `json:"allow_sampling"`
- AllowLogprobs bool `json:"allow_logprobs"`
- AllowSearchIndices bool `json:"allow_search_indices"`
- AllowView bool `json:"allow_view"`
- AllowFineTuning bool `json:"allow_fine_tuning"`
- Organization string `json:"organization"`
- Group string `json:"group"`
- IsBlocking bool `json:"is_blocking"`
- } `json:"permission"`
- Root string `json:"root"`
- Parent string `json:"parent"`
- }
- type OpenAIModelsResponse struct {
- Data []OpenAIModel `json:"data"`
- Success bool `json:"success"`
- }
- func parseStatusFilter(statusParam string) int {
- switch strings.ToLower(statusParam) {
- case "enabled", "1":
- return common.ChannelStatusEnabled
- case "disabled", "0":
- return 0
- default:
- return -1
- }
- }
- func GetAllChannels(c *gin.Context) {
- pageInfo := common.GetPageQuery(c)
- channelData := make([]*model.Channel, 0)
- idSort, _ := strconv.ParseBool(c.Query("id_sort"))
- enableTagMode, _ := strconv.ParseBool(c.Query("tag_mode"))
- statusParam := c.Query("status")
- // statusFilter: -1 all, 1 enabled, 0 disabled (include auto & manual)
- statusFilter := parseStatusFilter(statusParam)
- // type filter
- typeStr := c.Query("type")
- typeFilter := -1
- if typeStr != "" {
- if t, err := strconv.Atoi(typeStr); err == nil {
- typeFilter = t
- }
- }
- var total int64
- if enableTagMode {
- tags, err := model.GetPaginatedTags(pageInfo.GetStartIdx(), pageInfo.GetPageSize())
- if err != nil {
- c.JSON(http.StatusOK, gin.H{"success": false, "message": err.Error()})
- return
- }
- for _, tag := range tags {
- if tag == nil || *tag == "" {
- continue
- }
- tagChannels, err := model.GetChannelsByTag(*tag, idSort)
- if err != nil {
- continue
- }
- filtered := make([]*model.Channel, 0)
- for _, ch := range tagChannels {
- if statusFilter == common.ChannelStatusEnabled && ch.Status != common.ChannelStatusEnabled {
- continue
- }
- if statusFilter == 0 && ch.Status == common.ChannelStatusEnabled {
- continue
- }
- if typeFilter >= 0 && ch.Type != typeFilter {
- continue
- }
- filtered = append(filtered, ch)
- }
- channelData = append(channelData, filtered...)
- }
- total, _ = model.CountAllTags()
- } else {
- baseQuery := model.DB.Model(&model.Channel{})
- if typeFilter >= 0 {
- baseQuery = baseQuery.Where("type = ?", typeFilter)
- }
- if statusFilter == common.ChannelStatusEnabled {
- baseQuery = baseQuery.Where("status = ?", common.ChannelStatusEnabled)
- } else if statusFilter == 0 {
- baseQuery = baseQuery.Where("status != ?", common.ChannelStatusEnabled)
- }
- baseQuery.Count(&total)
- order := "priority desc"
- if idSort {
- order = "id desc"
- }
- err := baseQuery.Order(order).Limit(pageInfo.GetPageSize()).Offset(pageInfo.GetStartIdx()).Omit("key").Find(&channelData).Error
- if err != nil {
- c.JSON(http.StatusOK, gin.H{"success": false, "message": err.Error()})
- return
- }
- }
- countQuery := model.DB.Model(&model.Channel{})
- if statusFilter == common.ChannelStatusEnabled {
- countQuery = countQuery.Where("status = ?", common.ChannelStatusEnabled)
- } else if statusFilter == 0 {
- countQuery = countQuery.Where("status != ?", common.ChannelStatusEnabled)
- }
- var results []struct {
- Type int64
- Count int64
- }
- _ = countQuery.Select("type, count(*) as count").Group("type").Find(&results).Error
- typeCounts := make(map[int64]int64)
- for _, r := range results {
- typeCounts[r.Type] = r.Count
- }
- common.ApiSuccess(c, gin.H{
- "items": channelData,
- "total": total,
- "page": pageInfo.GetPage(),
- "page_size": pageInfo.GetPageSize(),
- "type_counts": typeCounts,
- })
- return
- }
- func FetchUpstreamModels(c *gin.Context) {
- id, err := strconv.Atoi(c.Param("id"))
- if err != nil {
- common.ApiError(c, err)
- return
- }
- channel, err := model.GetChannelById(id, true)
- if err != nil {
- common.ApiError(c, err)
- return
- }
- baseURL := constant.ChannelBaseURLs[channel.Type]
- if channel.GetBaseURL() != "" {
- baseURL = channel.GetBaseURL()
- }
- url := fmt.Sprintf("%s/v1/models", baseURL)
- switch channel.Type {
- case constant.ChannelTypeGemini:
- url = fmt.Sprintf("%s/v1beta/openai/models", baseURL)
- case constant.ChannelTypeAli:
- url = fmt.Sprintf("%s/compatible-mode/v1/models", baseURL)
- }
- body, err := GetResponseBody("GET", url, channel, GetAuthHeader(channel.Key))
- if err != nil {
- common.ApiError(c, err)
- return
- }
- var result OpenAIModelsResponse
- if err = json.Unmarshal(body, &result); err != nil {
- c.JSON(http.StatusOK, gin.H{
- "success": false,
- "message": fmt.Sprintf("解析响应失败: %s", err.Error()),
- })
- return
- }
- var ids []string
- for _, model := range result.Data {
- id := model.ID
- if channel.Type == constant.ChannelTypeGemini {
- id = strings.TrimPrefix(id, "models/")
- }
- ids = append(ids, id)
- }
- c.JSON(http.StatusOK, gin.H{
- "success": true,
- "message": "",
- "data": ids,
- })
- }
- func FixChannelsAbilities(c *gin.Context) {
- success, fails, err := model.FixAbility()
- if err != nil {
- common.ApiError(c, err)
- return
- }
- c.JSON(http.StatusOK, gin.H{
- "success": true,
- "message": "",
- "data": gin.H{
- "success": success,
- "fails": fails,
- },
- })
- }
- func SearchChannels(c *gin.Context) {
- keyword := c.Query("keyword")
- group := c.Query("group")
- modelKeyword := c.Query("model")
- statusParam := c.Query("status")
- statusFilter := parseStatusFilter(statusParam)
- idSort, _ := strconv.ParseBool(c.Query("id_sort"))
- enableTagMode, _ := strconv.ParseBool(c.Query("tag_mode"))
- channelData := make([]*model.Channel, 0)
- if enableTagMode {
- tags, err := model.SearchTags(keyword, group, modelKeyword, idSort)
- if err != nil {
- c.JSON(http.StatusOK, gin.H{
- "success": false,
- "message": err.Error(),
- })
- return
- }
- for _, tag := range tags {
- if tag != nil && *tag != "" {
- tagChannel, err := model.GetChannelsByTag(*tag, idSort)
- if err == nil {
- channelData = append(channelData, tagChannel...)
- }
- }
- }
- } else {
- channels, err := model.SearchChannels(keyword, group, modelKeyword, idSort)
- if err != nil {
- c.JSON(http.StatusOK, gin.H{
- "success": false,
- "message": err.Error(),
- })
- return
- }
- channelData = channels
- }
- if statusFilter == common.ChannelStatusEnabled || statusFilter == 0 {
- filtered := make([]*model.Channel, 0, len(channelData))
- for _, ch := range channelData {
- if statusFilter == common.ChannelStatusEnabled && ch.Status != common.ChannelStatusEnabled {
- continue
- }
- if statusFilter == 0 && ch.Status == common.ChannelStatusEnabled {
- continue
- }
- filtered = append(filtered, ch)
- }
- channelData = filtered
- }
- // calculate type counts for search results
- typeCounts := make(map[int64]int64)
- for _, channel := range channelData {
- typeCounts[int64(channel.Type)]++
- }
- typeParam := c.Query("type")
- typeFilter := -1
- if typeParam != "" {
- if tp, err := strconv.Atoi(typeParam); err == nil {
- typeFilter = tp
- }
- }
- if typeFilter >= 0 {
- filtered := make([]*model.Channel, 0, len(channelData))
- for _, ch := range channelData {
- if ch.Type == typeFilter {
- filtered = append(filtered, ch)
- }
- }
- channelData = filtered
- }
- page, _ := strconv.Atoi(c.DefaultQuery("p", "1"))
- pageSize, _ := strconv.Atoi(c.DefaultQuery("page_size", "20"))
- if page < 1 {
- page = 1
- }
- if pageSize <= 0 {
- pageSize = 20
- }
- total := len(channelData)
- startIdx := (page - 1) * pageSize
- if startIdx > total {
- startIdx = total
- }
- endIdx := startIdx + pageSize
- if endIdx > total {
- endIdx = total
- }
- pagedData := channelData[startIdx:endIdx]
- c.JSON(http.StatusOK, gin.H{
- "success": true,
- "message": "",
- "data": gin.H{
- "items": pagedData,
- "total": total,
- "type_counts": typeCounts,
- },
- })
- return
- }
- func GetChannel(c *gin.Context) {
- id, err := strconv.Atoi(c.Param("id"))
- if err != nil {
- common.ApiError(c, err)
- return
- }
- channel, err := model.GetChannelById(id, false)
- if err != nil {
- common.ApiError(c, err)
- return
- }
- c.JSON(http.StatusOK, gin.H{
- "success": true,
- "message": "",
- "data": channel,
- })
- return
- }
- // validateChannel 通用的渠道校验函数
- func validateChannel(channel *model.Channel, isAdd bool) error {
- // 校验 channel settings
- if err := channel.ValidateSettings(); err != nil {
- return fmt.Errorf("渠道额外设置[channel setting] 格式错误:%s", err.Error())
- }
- // 如果是添加操作,检查 channel 和 key 是否为空
- if isAdd {
- if channel == nil || channel.Key == "" {
- return fmt.Errorf("channel cannot be empty")
- }
- // 检查模型名称长度是否超过 255
- for _, m := range channel.GetModels() {
- if len(m) > 255 {
- return fmt.Errorf("模型名称过长: %s", m)
- }
- }
- }
- // VertexAI 特殊校验
- if channel.Type == constant.ChannelTypeVertexAi {
- if channel.Other == "" {
- return fmt.Errorf("部署地区不能为空")
- }
- regionMap, err := common.StrToMap(channel.Other)
- if err != nil {
- return fmt.Errorf("部署地区必须是标准的Json格式,例如{\"default\": \"us-central1\", \"region2\": \"us-east1\"}")
- }
- if regionMap["default"] == nil {
- return fmt.Errorf("部署地区必须包含default字段")
- }
- }
- return nil
- }
- type AddChannelRequest struct {
- Mode string `json:"mode"`
- MultiKeyMode constant.MultiKeyMode `json:"multi_key_mode"`
- Channel *model.Channel `json:"channel"`
- }
- func getVertexArrayKeys(keys string) ([]string, error) {
- if keys == "" {
- return nil, nil
- }
- var keyArray []interface{}
- err := common.Unmarshal([]byte(keys), &keyArray)
- if err != nil {
- return nil, fmt.Errorf("批量添加 Vertex AI 必须使用标准的JsonArray格式,例如[{key1}, {key2}...],请检查输入: %w", err)
- }
- cleanKeys := make([]string, 0, len(keyArray))
- for _, key := range keyArray {
- var keyStr string
- switch v := key.(type) {
- case string:
- keyStr = strings.TrimSpace(v)
- default:
- bytes, err := json.Marshal(v)
- if err != nil {
- return nil, fmt.Errorf("Vertex AI key JSON 编码失败: %w", err)
- }
- keyStr = string(bytes)
- }
- if keyStr != "" {
- cleanKeys = append(cleanKeys, keyStr)
- }
- }
- if len(cleanKeys) == 0 {
- return nil, fmt.Errorf("批量添加 Vertex AI 的 keys 不能为空")
- }
- return cleanKeys, nil
- }
- func AddChannel(c *gin.Context) {
- addChannelRequest := AddChannelRequest{}
- err := c.ShouldBindJSON(&addChannelRequest)
- if err != nil {
- common.ApiError(c, err)
- return
- }
- // 使用统一的校验函数
- if err := validateChannel(addChannelRequest.Channel, true); err != nil {
- c.JSON(http.StatusOK, gin.H{
- "success": false,
- "message": err.Error(),
- })
- return
- }
- addChannelRequest.Channel.CreatedTime = common.GetTimestamp()
- keys := make([]string, 0)
- switch addChannelRequest.Mode {
- case "multi_to_single":
- addChannelRequest.Channel.ChannelInfo.IsMultiKey = true
- addChannelRequest.Channel.ChannelInfo.MultiKeyMode = addChannelRequest.MultiKeyMode
- if addChannelRequest.Channel.Type == constant.ChannelTypeVertexAi {
- array, err := getVertexArrayKeys(addChannelRequest.Channel.Key)
- if err != nil {
- c.JSON(http.StatusOK, gin.H{
- "success": false,
- "message": err.Error(),
- })
- return
- }
- addChannelRequest.Channel.ChannelInfo.MultiKeySize = len(array)
- addChannelRequest.Channel.Key = strings.Join(array, "\n")
- } else {
- cleanKeys := make([]string, 0)
- for _, key := range strings.Split(addChannelRequest.Channel.Key, "\n") {
- if key == "" {
- continue
- }
- key = strings.TrimSpace(key)
- cleanKeys = append(cleanKeys, key)
- }
- addChannelRequest.Channel.ChannelInfo.MultiKeySize = len(cleanKeys)
- addChannelRequest.Channel.Key = strings.Join(cleanKeys, "\n")
- }
- keys = []string{addChannelRequest.Channel.Key}
- case "batch":
- if addChannelRequest.Channel.Type == constant.ChannelTypeVertexAi {
- // multi json
- keys, err = getVertexArrayKeys(addChannelRequest.Channel.Key)
- if err != nil {
- c.JSON(http.StatusOK, gin.H{
- "success": false,
- "message": err.Error(),
- })
- return
- }
- } else {
- keys = strings.Split(addChannelRequest.Channel.Key, "\n")
- }
- case "single":
- keys = []string{addChannelRequest.Channel.Key}
- default:
- c.JSON(http.StatusOK, gin.H{
- "success": false,
- "message": "不支持的添加模式",
- })
- return
- }
- channels := make([]model.Channel, 0, len(keys))
- for _, key := range keys {
- if key == "" {
- continue
- }
- localChannel := addChannelRequest.Channel
- localChannel.Key = key
- channels = append(channels, *localChannel)
- }
- err = model.BatchInsertChannels(channels)
- if err != nil {
- common.ApiError(c, err)
- return
- }
- c.JSON(http.StatusOK, gin.H{
- "success": true,
- "message": "",
- })
- return
- }
- func DeleteChannel(c *gin.Context) {
- id, _ := strconv.Atoi(c.Param("id"))
- channel := model.Channel{Id: id}
- err := channel.Delete()
- if err != nil {
- common.ApiError(c, err)
- return
- }
- model.InitChannelCache()
- c.JSON(http.StatusOK, gin.H{
- "success": true,
- "message": "",
- })
- return
- }
- func DeleteDisabledChannel(c *gin.Context) {
- rows, err := model.DeleteDisabledChannel()
- if err != nil {
- common.ApiError(c, err)
- return
- }
- model.InitChannelCache()
- c.JSON(http.StatusOK, gin.H{
- "success": true,
- "message": "",
- "data": rows,
- })
- return
- }
- type ChannelTag struct {
- Tag string `json:"tag"`
- NewTag *string `json:"new_tag"`
- Priority *int64 `json:"priority"`
- Weight *uint `json:"weight"`
- ModelMapping *string `json:"model_mapping"`
- Models *string `json:"models"`
- Groups *string `json:"groups"`
- }
- func DisableTagChannels(c *gin.Context) {
- channelTag := ChannelTag{}
- err := c.ShouldBindJSON(&channelTag)
- if err != nil || channelTag.Tag == "" {
- c.JSON(http.StatusOK, gin.H{
- "success": false,
- "message": "参数错误",
- })
- return
- }
- err = model.DisableChannelByTag(channelTag.Tag)
- if err != nil {
- common.ApiError(c, err)
- return
- }
- model.InitChannelCache()
- c.JSON(http.StatusOK, gin.H{
- "success": true,
- "message": "",
- })
- return
- }
- func EnableTagChannels(c *gin.Context) {
- channelTag := ChannelTag{}
- err := c.ShouldBindJSON(&channelTag)
- if err != nil || channelTag.Tag == "" {
- c.JSON(http.StatusOK, gin.H{
- "success": false,
- "message": "参数错误",
- })
- return
- }
- err = model.EnableChannelByTag(channelTag.Tag)
- if err != nil {
- common.ApiError(c, err)
- return
- }
- model.InitChannelCache()
- c.JSON(http.StatusOK, gin.H{
- "success": true,
- "message": "",
- })
- return
- }
- func EditTagChannels(c *gin.Context) {
- channelTag := ChannelTag{}
- err := c.ShouldBindJSON(&channelTag)
- if err != nil {
- c.JSON(http.StatusOK, gin.H{
- "success": false,
- "message": "参数错误",
- })
- return
- }
- if channelTag.Tag == "" {
- c.JSON(http.StatusOK, gin.H{
- "success": false,
- "message": "tag不能为空",
- })
- return
- }
- err = model.EditChannelByTag(channelTag.Tag, channelTag.NewTag, channelTag.ModelMapping, channelTag.Models, channelTag.Groups, channelTag.Priority, channelTag.Weight)
- if err != nil {
- common.ApiError(c, err)
- return
- }
- model.InitChannelCache()
- c.JSON(http.StatusOK, gin.H{
- "success": true,
- "message": "",
- })
- return
- }
- type ChannelBatch struct {
- Ids []int `json:"ids"`
- Tag *string `json:"tag"`
- }
- func DeleteChannelBatch(c *gin.Context) {
- channelBatch := ChannelBatch{}
- err := c.ShouldBindJSON(&channelBatch)
- if err != nil || len(channelBatch.Ids) == 0 {
- c.JSON(http.StatusOK, gin.H{
- "success": false,
- "message": "参数错误",
- })
- return
- }
- err = model.BatchDeleteChannels(channelBatch.Ids)
- if err != nil {
- common.ApiError(c, err)
- return
- }
- model.InitChannelCache()
- c.JSON(http.StatusOK, gin.H{
- "success": true,
- "message": "",
- "data": len(channelBatch.Ids),
- })
- return
- }
- type PatchChannel struct {
- model.Channel
- MultiKeyMode *string `json:"multi_key_mode"`
- }
- func UpdateChannel(c *gin.Context) {
- channel := PatchChannel{}
- err := c.ShouldBindJSON(&channel)
- if err != nil {
- common.ApiError(c, err)
- return
- }
- // 使用统一的校验函数
- if err := validateChannel(&channel.Channel, false); err != nil {
- c.JSON(http.StatusOK, gin.H{
- "success": false,
- "message": err.Error(),
- })
- return
- }
- // Preserve existing ChannelInfo to ensure multi-key channels keep correct state even if the client does not send ChannelInfo in the request.
- originChannel, err := model.GetChannelById(channel.Id, false)
- if err != nil {
- c.JSON(http.StatusOK, gin.H{
- "success": false,
- "message": err.Error(),
- })
- return
- }
- // Always copy the original ChannelInfo so that fields like IsMultiKey and MultiKeySize are retained.
- channel.ChannelInfo = originChannel.ChannelInfo
- // If the request explicitly specifies a new MultiKeyMode, apply it on top of the original info.
- if channel.MultiKeyMode != nil && *channel.MultiKeyMode != "" {
- channel.ChannelInfo.MultiKeyMode = constant.MultiKeyMode(*channel.MultiKeyMode)
- }
- err = channel.Update()
- if err != nil {
- common.ApiError(c, err)
- return
- }
- model.InitChannelCache()
- channel.Key = ""
- c.JSON(http.StatusOK, gin.H{
- "success": true,
- "message": "",
- "data": channel,
- })
- return
- }
- func FetchModels(c *gin.Context) {
- var req struct {
- BaseURL string `json:"base_url"`
- Type int `json:"type"`
- Key string `json:"key"`
- }
- if err := c.ShouldBindJSON(&req); err != nil {
- c.JSON(http.StatusBadRequest, gin.H{
- "success": false,
- "message": "Invalid request",
- })
- return
- }
- baseURL := req.BaseURL
- if baseURL == "" {
- baseURL = constant.ChannelBaseURLs[req.Type]
- }
- client := &http.Client{}
- url := fmt.Sprintf("%s/v1/models", baseURL)
- request, err := http.NewRequest("GET", url, nil)
- if err != nil {
- c.JSON(http.StatusInternalServerError, gin.H{
- "success": false,
- "message": err.Error(),
- })
- return
- }
- // remove line breaks and extra spaces.
- key := strings.TrimSpace(req.Key)
- // If the key contains a line break, only take the first part.
- key = strings.Split(key, "\n")[0]
- request.Header.Set("Authorization", "Bearer "+key)
- response, err := client.Do(request)
- if err != nil {
- c.JSON(http.StatusInternalServerError, gin.H{
- "success": false,
- "message": err.Error(),
- })
- return
- }
- //check status code
- if response.StatusCode != http.StatusOK {
- c.JSON(http.StatusInternalServerError, gin.H{
- "success": false,
- "message": "Failed to fetch models",
- })
- return
- }
- defer response.Body.Close()
- var result struct {
- Data []struct {
- ID string `json:"id"`
- } `json:"data"`
- }
- if err := json.NewDecoder(response.Body).Decode(&result); err != nil {
- c.JSON(http.StatusInternalServerError, gin.H{
- "success": false,
- "message": err.Error(),
- })
- return
- }
- var models []string
- for _, model := range result.Data {
- models = append(models, model.ID)
- }
- c.JSON(http.StatusOK, gin.H{
- "success": true,
- "data": models,
- })
- }
- func BatchSetChannelTag(c *gin.Context) {
- channelBatch := ChannelBatch{}
- err := c.ShouldBindJSON(&channelBatch)
- if err != nil || len(channelBatch.Ids) == 0 {
- c.JSON(http.StatusOK, gin.H{
- "success": false,
- "message": "参数错误",
- })
- return
- }
- err = model.BatchSetChannelTag(channelBatch.Ids, channelBatch.Tag)
- if err != nil {
- common.ApiError(c, err)
- return
- }
- model.InitChannelCache()
- c.JSON(http.StatusOK, gin.H{
- "success": true,
- "message": "",
- "data": len(channelBatch.Ids),
- })
- return
- }
- func GetTagModels(c *gin.Context) {
- tag := c.Query("tag")
- if tag == "" {
- c.JSON(http.StatusBadRequest, gin.H{
- "success": false,
- "message": "tag不能为空",
- })
- return
- }
- channels, err := model.GetChannelsByTag(tag, false) // Assuming false for idSort is fine here
- if err != nil {
- c.JSON(http.StatusInternalServerError, gin.H{
- "success": false,
- "message": err.Error(),
- })
- return
- }
- var longestModels string
- maxLength := 0
- // Find the longest models string among all channels with the given tag
- for _, channel := range channels {
- if channel.Models != "" {
- currentModels := strings.Split(channel.Models, ",")
- if len(currentModels) > maxLength {
- maxLength = len(currentModels)
- longestModels = channel.Models
- }
- }
- }
- c.JSON(http.StatusOK, gin.H{
- "success": true,
- "message": "",
- "data": longestModels,
- })
- return
- }
- // CopyChannel handles cloning an existing channel with its key.
- // POST /api/channel/copy/:id
- // Optional query params:
- //
- // suffix - string appended to the original name (default "_复制")
- // reset_balance - bool, when true will reset balance & used_quota to 0 (default true)
- func CopyChannel(c *gin.Context) {
- id, err := strconv.Atoi(c.Param("id"))
- if err != nil {
- c.JSON(http.StatusOK, gin.H{"success": false, "message": "invalid id"})
- return
- }
- suffix := c.DefaultQuery("suffix", "_复制")
- resetBalance := true
- if rbStr := c.DefaultQuery("reset_balance", "true"); rbStr != "" {
- if v, err := strconv.ParseBool(rbStr); err == nil {
- resetBalance = v
- }
- }
- // fetch original channel with key
- origin, err := model.GetChannelById(id, true)
- if err != nil {
- c.JSON(http.StatusOK, gin.H{"success": false, "message": err.Error()})
- return
- }
- // clone channel
- clone := *origin // shallow copy is sufficient as we will overwrite primitives
- clone.Id = 0 // let DB auto-generate
- clone.CreatedTime = common.GetTimestamp()
- clone.Name = origin.Name + suffix
- clone.TestTime = 0
- clone.ResponseTime = 0
- if resetBalance {
- clone.Balance = 0
- clone.UsedQuota = 0
- }
- // insert
- if err := model.BatchInsertChannels([]model.Channel{clone}); err != nil {
- c.JSON(http.StatusOK, gin.H{"success": false, "message": err.Error()})
- return
- }
- model.InitChannelCache()
- // success
- c.JSON(http.StatusOK, gin.H{"success": true, "message": "", "data": gin.H{"id": clone.Id}})
- }
|