47 lines
1.0 KiB
Go
47 lines
1.0 KiB
Go
package ai
|
|
|
|
import (
|
|
"context"
|
|
|
|
"gitlab.guxuan.icu/jinshan_community/internal/config"
|
|
|
|
"github.com/gin-gonic/gin"
|
|
"gitlab.guxuan.icu/jinshan_community/internal/mods/ai/api"
|
|
"gitlab.guxuan.icu/jinshan_community/internal/mods/ai/schema"
|
|
"gorm.io/gorm"
|
|
)
|
|
|
|
type Ai struct {
|
|
DB *gorm.DB
|
|
AiRequestAPI *api.AiRequest
|
|
}
|
|
|
|
func (a *Ai) AutoMigrate(ctx context.Context) error {
|
|
return a.DB.AutoMigrate(new(schema.AiRequest))
|
|
}
|
|
|
|
func (a *Ai) Init(ctx context.Context) error {
|
|
if config.C.Storage.DB.AutoMigrate {
|
|
if err := a.AutoMigrate(ctx); err != nil {
|
|
return err
|
|
}
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (a *Ai) RegisterV1Routers(ctx context.Context, v1 *gin.RouterGroup) error {
|
|
aiRequest := v1.Group("ai-requests")
|
|
{
|
|
aiRequest.GET("", a.AiRequestAPI.Query)
|
|
aiRequest.GET(":id", a.AiRequestAPI.Get)
|
|
aiRequest.POST("", a.AiRequestAPI.Create)
|
|
aiRequest.PUT(":id", a.AiRequestAPI.Update)
|
|
aiRequest.DELETE(":id", a.AiRequestAPI.Delete)
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (a *Ai) Release(ctx context.Context) error {
|
|
return nil
|
|
}
|