coai/channel/channel.go

package channel

import (
	"chat/utils"
	"math/rand"
	"strings"
)

var defaultMaxRetries = 1

func (c *Channel) GetId() int {
	return c.Id
}

func (c *Channel) GetName() string {
	return c.Name
}

func (c *Channel) GetType() string {
	return c.Type
}

func (c *Channel) GetPriority() int {
	return c.Priority
}

func (c *Channel) GetWeight() int {
	if c.Weight <= 0 {
		return 1
	}
	return c.Weight
}

func (c *Channel) GetModels() []string {
	return c.Models
}

func (c *Channel) GetRetry() int {
	if c.Retry <= 0 {
		return defaultMaxRetries
	}
	return c.Retry
}

func (c *Channel) GetSecret() string {
	return c.Secret
}

func (c *Channel) GetRandomSecret() string {
	arr := strings.Split(c.GetSecret(), "\n")
	idx := rand.Intn(len(arr))
	return arr[idx]
}

func (c *Channel) GetEndpoint() string {
	return c.Endpoint
}

func (c *Channel) GetMapper() string {
	return c.Mapper
}

func (c *Channel) GetReflect() map[string]string {
	if c.Reflect == nil {
		var reflect map[string]string
		arr := strings.Split(c.GetMapper(), "\n")
		for _, item := range arr {
			pair := strings.Split(item, ">")
			if len(pair) == 2 {
				reflect[pair[0]] = pair[1]
			}
		}

		c.Reflect = &reflect
	}

	return *c.Reflect
}

func (c *Channel) GetModelReflect(model string) string {
	ref := c.GetReflect()
	if reflect, ok := ref[model]; ok && len(reflect) > 0 {
		return reflect
	}

	return model
}

func (c *Channel) GetHitModels() []string {
	if c.HitModels == nil {
		var res []string

		models := c.GetModels()
		ref := c.GetReflect()

		for _, model := range models {
			if !utils.Contains(model, res) {
				res = append(res, model)
			}
		}

		for model := range ref {
			if !utils.Contains(model, res) {
				res = append(res, model)
			}
		}

		c.HitModels = &res
	}

	return *c.HitModels
}

func (c *Channel) GetState() bool {
	return c.State
}