mirror of
https://github.com/v2fly/v2ray-core.git
synced 2024-12-23 02:26:49 -05:00
86 lines
2.1 KiB
Go
86 lines
2.1 KiB
Go
package instman
|
|
|
|
import (
|
|
"context"
|
|
|
|
core "github.com/v2fly/v2ray-core/v5"
|
|
"github.com/v2fly/v2ray-core/v5/common"
|
|
"github.com/v2fly/v2ray-core/v5/features/extension"
|
|
)
|
|
|
|
//go:generate go run github.com/v2fly/v2ray-core/v5/common/errors/errorgen
|
|
|
|
type InstanceMgr struct {
|
|
config *Config // nolint: structcheck
|
|
instances map[string]*core.Instance
|
|
}
|
|
|
|
func (i InstanceMgr) Type() interface{} {
|
|
return extension.InstanceManagementType()
|
|
}
|
|
|
|
func (i InstanceMgr) Start() error {
|
|
return nil
|
|
}
|
|
|
|
func (i InstanceMgr) Close() error {
|
|
return nil
|
|
}
|
|
|
|
func (i InstanceMgr) ListInstance(ctx context.Context) ([]string, error) {
|
|
var instanceNames []string
|
|
for k := range i.instances {
|
|
instanceNames = append(instanceNames, k)
|
|
}
|
|
return instanceNames, nil
|
|
}
|
|
|
|
func (i InstanceMgr) AddInstance(ctx context.Context, name string, config []byte, configType string) error {
|
|
coreConfig, err := core.LoadConfig(configType, config)
|
|
if err != nil {
|
|
return newError("unable to load config").Base(err)
|
|
}
|
|
instance, err := core.New(coreConfig)
|
|
if err != nil {
|
|
return newError("unable to create instance").Base(err)
|
|
}
|
|
i.instances[name] = instance
|
|
return nil
|
|
}
|
|
|
|
func (i InstanceMgr) StartInstance(ctx context.Context, name string) error {
|
|
err := i.instances[name].Start()
|
|
if err != nil {
|
|
return newError("failed to start instance").Base(err)
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (i InstanceMgr) StopInstance(ctx context.Context, name string) error {
|
|
err := i.instances[name].Close()
|
|
if err != nil {
|
|
return newError("failed to stop instance").Base(err)
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (i InstanceMgr) UntrackInstance(ctx context.Context, name string) error {
|
|
delete(i.instances, name)
|
|
return nil
|
|
}
|
|
|
|
func NewInstanceMgr(ctx context.Context, config *Config) (extension.InstanceManagement, error) {
|
|
return InstanceMgr{instances: map[string]*core.Instance{}}, nil
|
|
}
|
|
|
|
func init() {
|
|
common.Must(common.RegisterConfig((*Config)(nil), func(ctx context.Context, config interface{}) (interface{}, error) {
|
|
var f extension.InstanceManagement
|
|
var err error
|
|
if f, err = NewInstanceMgr(ctx, config.(*Config)); err != nil {
|
|
return nil, err
|
|
}
|
|
return f, nil
|
|
}))
|
|
}
|