102 lines
2.6 KiB
Go
102 lines
2.6 KiB
Go
package share
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"log"
|
|
|
|
"github.com/buraksezer/olric"
|
|
"github.com/buraksezer/olric/client"
|
|
config_olric "github.com/buraksezer/olric/config"
|
|
"github.com/buraksezer/olric/serializer"
|
|
"github.com/yaoapp/gou/session"
|
|
"github.com/yaoapp/kun/exception"
|
|
"github.com/yaoapp/yao/config"
|
|
)
|
|
|
|
var sessServer *olric.Olric
|
|
|
|
// SessionConnect 加载会话信息
|
|
func SessionConnect(conf config.SessionConfig) {
|
|
|
|
var clientConfig = &client.Config{
|
|
Servers: []string{fmt.Sprintf("%s:%d", conf.Host, conf.Port)},
|
|
Serializer: serializer.NewMsgpackSerializer(),
|
|
Client: config_olric.NewClient(),
|
|
}
|
|
|
|
c, err := client.New(clientConfig)
|
|
if err != nil {
|
|
exception.New("会话服务器连接失败 %s", 500, err.Error()).Throw()
|
|
}
|
|
|
|
dm := c.NewDMap("local-session")
|
|
session.MemoryUse(session.ClientDMap{DMap: dm})
|
|
}
|
|
|
|
// SessionServerStop 关闭会话服务器
|
|
func SessionServerStop() {
|
|
if sessServer != nil {
|
|
sessServer.Shutdown(context.Background())
|
|
}
|
|
}
|
|
|
|
// SessionServerStart 启动会话服务器
|
|
func SessionServerStart() {
|
|
c := &config_olric.Config{
|
|
BindAddr: config.Conf.Session.Host,
|
|
BindPort: config.Conf.Session.Port,
|
|
ReadRepair: false,
|
|
ReplicaCount: 1,
|
|
WriteQuorum: 1,
|
|
ReadQuorum: 1,
|
|
MemberCountQuorum: 1,
|
|
Peers: []string{},
|
|
DMaps: &config_olric.DMaps{},
|
|
StorageEngines: config_olric.NewStorageEngine(),
|
|
}
|
|
|
|
m, err := config_olric.NewMemberlistConfig("local")
|
|
if err != nil {
|
|
panic(fmt.Sprintf("unable to create a new memberlist config: %v", err))
|
|
}
|
|
// m.BindAddr = config.Conf.Session.Host
|
|
m.BindPort = config.Conf.Session.Port
|
|
m.AdvertisePort = config.Conf.Session.Port
|
|
c.MemberlistConfig = m
|
|
|
|
// c.MemberlistConfig.BindAddr = config.Conf.Session.Host
|
|
// c.MemberlistConfig.BindPort = 3308
|
|
|
|
// c := config_olric.New("local")
|
|
// c.BindAddr = config.Conf.Session.Host
|
|
// c.BindPort = config.Conf.Session.Port
|
|
|
|
// c.Logger.SetOutput(ioutil.Discard) // 暂时关闭日志
|
|
ctx, cancel := context.WithCancel(context.Background())
|
|
c.Started = func() {
|
|
defer cancel()
|
|
log.Println("[INFO] Olric is ready to accept connections")
|
|
}
|
|
|
|
sessServer, err = olric.New(c)
|
|
|
|
if err != nil {
|
|
log.Fatalf("Failed to create Olric instance: %v", err)
|
|
}
|
|
|
|
go func() {
|
|
err = sessServer.Start() // Call Start at background. It's a blocker call.
|
|
if err != nil {
|
|
log.Fatalf("olric.Start returned an error: %v", err)
|
|
}
|
|
}()
|
|
|
|
<-ctx.Done()
|
|
dm, err := sessServer.NewDMap("local-session")
|
|
if err != nil {
|
|
log.Fatalf("olric.NewDMap returned an error: %v", err)
|
|
}
|
|
|
|
session.MemoryUse(session.ServerDMap{DMap: dm})
|
|
}
|