59 lines
1.2 KiB
Go
59 lines
1.2 KiB
Go
package controller
|
|
|
|
import (
|
|
"encoding/json"
|
|
"fmt"
|
|
"net/http"
|
|
"opencatd-open/internal/dto"
|
|
|
|
"github.com/gin-gonic/gin"
|
|
)
|
|
|
|
func (p *Proxy) HandleModels(c *gin.Context) {
|
|
models, err := p.getCache()
|
|
if err != nil {
|
|
dto.Fail(c, http.StatusBadGateway, err.Error())
|
|
return
|
|
}
|
|
type _model struct {
|
|
ID string `json:"id"`
|
|
}
|
|
var ms []_model
|
|
for _, model := range models {
|
|
ms = append(ms, _model{ID: model})
|
|
}
|
|
dto.Success(c, ms)
|
|
}
|
|
|
|
func (p *Proxy) setCache() error {
|
|
apikeys, err := p.apiKeyDao.FindKeys(nil)
|
|
models := make(map[string]bool)
|
|
if err == nil && len(apikeys) > 0 {
|
|
for _, k := range apikeys {
|
|
if len(k.SupportModelsArray) > 0 {
|
|
for _, sm := range k.SupportModelsArray {
|
|
models[sm] = true
|
|
}
|
|
} else {
|
|
var sma []string
|
|
json.Unmarshal([]byte(*k.SupportModels), &sma) // nolint:errCheck
|
|
for _, sm := range sma {
|
|
models[sm] = true
|
|
}
|
|
}
|
|
}
|
|
} else {
|
|
return fmt.Errorf("empty data")
|
|
}
|
|
var support_models []string
|
|
for m, _ := range models {
|
|
support_models = append(support_models, m)
|
|
}
|
|
return p.cache.Set("models", support_models)
|
|
}
|
|
|
|
func (p *Proxy) getCache() ([]string, error) {
|
|
models, err := p.cache.Get("models")
|
|
return models.([]string), err
|
|
}
|