/* * @Author: LinkLeong link@icewhale.com * @Date: 2022-07-12 09:48:56 * @LastEditors: LinkLeong * @LastEditTime: 2022-09-02 22:10:05 * @FilePath: /CasaOS/service/service.go * @Description: * @Website: https://www.casaos.io * Copyright (c) 2022 by icewhale, All Rights Reserved. */ package service import ( "github.com/IceWhaleTech/CasaOS-Common/external" "github.com/IceWhaleTech/CasaOS-Common/utils/logger" socketio "github.com/googollee/go-socket.io" "github.com/gorilla/websocket" "github.com/patrickmn/go-cache" "go.uber.org/zap" "gorm.io/gorm" ) var Cache *cache.Cache var ( MyService Repository SocketServer *socketio.Server ) var ( WebSocketConns []*websocket.Conn SocketRun bool ) type Repository interface { Casa() CasaService Connections() ConnectionsService Gateway() external.ManagementService Health() HealthService Notify() NotifyServer Rely() RelyService Shares() SharesService System() SystemService Storage() StorageService Storages() StoragesService StoragePath() StoragePathService FsListService() FsListService FsLinkService() FsLinkService FsService() FsService } func NewService(db *gorm.DB, RuntimePath string, socket *socketio.Server) Repository { if socket == nil { logger.Error("socket is nil", zap.Any("error", "socket is nil")) } SocketServer = socket gatewayManagement, err := external.NewManagementService(RuntimePath) if err != nil && len(RuntimePath) > 0 { panic(err) } return &store{ casa: NewCasaService(), connections: NewConnectionsService(db), gateway: gatewayManagement, notify: NewNotifyService(db), rely: NewRelyService(db), system: NewSystemService(), health: NewHealthService(), shares: NewSharesService(db), storage: NewStorageService(), storages: NewStoragesService(), storage_path: NewStoragePathService(), fs_list: NewFsListService(), fs_link: NewFsLinkService(), fs: NewFsService(), } } type store struct { db *gorm.DB casa CasaService notify NotifyServer rely RelyService system SystemService shares SharesService connections ConnectionsService gateway external.ManagementService storage StorageService storages StoragesService storage_path StoragePathService fs_list FsListService fs_link FsLinkService fs FsService health HealthService } func (c *store) FsLinkService() FsLinkService { return c.fs_link } func (c *store) FsService() FsService { return c.fs } func (c *store) FsListService() FsListService { return c.fs_list } func (c *store) StoragePath() StoragePathService { return c.storage_path } func (c *store) Storages() StoragesService { return c.storages } func (c *store) Storage() StorageService { return c.storage } func (c *store) Gateway() external.ManagementService { return c.gateway } func (s *store) Connections() ConnectionsService { return s.connections } func (s *store) Shares() SharesService { return s.shares } func (c *store) Rely() RelyService { return c.rely } func (c *store) System() SystemService { return c.system } func (c *store) Notify() NotifyServer { return c.notify } func (c *store) Casa() CasaService { return c.casa } func (c *store) Health() HealthService { return c.health }