66 lines
2.2 KiB
Go
66 lines
2.2 KiB
Go
package envmonitor
|
|
|
|
import (
|
|
"context"
|
|
"ifms/internal/config"
|
|
|
|
"ifms/internal/mods/envmonitor/api"
|
|
"ifms/internal/mods/envmonitor/schema"
|
|
|
|
"github.com/gin-gonic/gin"
|
|
"gorm.io/gorm"
|
|
)
|
|
|
|
type EnvMonitor struct {
|
|
DB *gorm.DB
|
|
EnvMonitorDeviceTypeInfoAPI *api.EnvMonitorDeviceTypeInfo
|
|
EnvMonitorDeviceAPI *api.EnvMonitorDevice
|
|
EnvMonitorDeviceFishPondAPI *api.EnvMonitorDeviceFishPond
|
|
}
|
|
|
|
func (a *EnvMonitor) AutoMigrate(ctx context.Context) error {
|
|
return a.DB.AutoMigrate(new(schema.EnvMonitorDeviceTypeInfo), new(schema.EnvMonitorDevice), new(schema.EnvMonitorDeviceFishPond))
|
|
}
|
|
|
|
func (a *EnvMonitor) Init(ctx context.Context) error {
|
|
if config.C.Storage.DB.AutoMigrate {
|
|
if err := a.AutoMigrate(ctx); err != nil {
|
|
return err
|
|
}
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (a *EnvMonitor) RegisterV1Routers(ctx context.Context, v1 *gin.RouterGroup) error {
|
|
v1 = v1.Group("envmonitor")
|
|
envMonitorDeviceTypeInfo := v1.Group("env-monitor-device-type-infos")
|
|
{
|
|
envMonitorDeviceTypeInfo.GET("", a.EnvMonitorDeviceTypeInfoAPI.Query)
|
|
envMonitorDeviceTypeInfo.GET(":id", a.EnvMonitorDeviceTypeInfoAPI.Get)
|
|
envMonitorDeviceTypeInfo.POST("", a.EnvMonitorDeviceTypeInfoAPI.Create)
|
|
envMonitorDeviceTypeInfo.PUT(":id", a.EnvMonitorDeviceTypeInfoAPI.Update)
|
|
envMonitorDeviceTypeInfo.DELETE(":id", a.EnvMonitorDeviceTypeInfoAPI.Delete)
|
|
}
|
|
envMonitorDevice := v1.Group("env-monitor-devices")
|
|
{
|
|
envMonitorDevice.GET("", a.EnvMonitorDeviceAPI.Query)
|
|
envMonitorDevice.GET(":id", a.EnvMonitorDeviceAPI.Get)
|
|
envMonitorDevice.POST("", a.EnvMonitorDeviceAPI.Create)
|
|
envMonitorDevice.PUT(":id", a.EnvMonitorDeviceAPI.Update)
|
|
envMonitorDevice.DELETE(":id", a.EnvMonitorDeviceAPI.Delete)
|
|
}
|
|
envMonitorDeviceFishPond := v1.Group("env-monitor-device-fish-ponds")
|
|
{
|
|
envMonitorDeviceFishPond.GET("", a.EnvMonitorDeviceFishPondAPI.Query)
|
|
envMonitorDeviceFishPond.GET(":id", a.EnvMonitorDeviceFishPondAPI.Get)
|
|
envMonitorDeviceFishPond.POST("", a.EnvMonitorDeviceFishPondAPI.Create)
|
|
envMonitorDeviceFishPond.PUT(":id", a.EnvMonitorDeviceFishPondAPI.Update)
|
|
envMonitorDeviceFishPond.DELETE(":id", a.EnvMonitorDeviceFishPondAPI.Delete)
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (a *EnvMonitor) Release(ctx context.Context) error {
|
|
return nil
|
|
}
|