54 lines
851 B
Go
54 lines
851 B
Go
|
package fsm
|
||
|
|
|
||
|
|
import (
|
||
|
|
"context"
|
||
|
|
"log"
|
||
|
|
|
||
|
|
"github.com/lni/dragonboat/v4"
|
||
|
|
)
|
||
|
|
|
||
|
|
type SharedState struct {
|
||
|
|
shardID uint64
|
||
|
|
replicaID uint64
|
||
|
|
nh *dragonboat.NodeHost
|
||
|
|
}
|
||
|
|
|
||
|
|
func (ss *SharedState) GetUserState(Userid int64) (us *UserState, err error) {
|
||
|
|
|
||
|
|
ius, err := ss.nh.SyncRead(context.TODO(), ss.shardID, Userid)
|
||
|
|
if err != nil {
|
||
|
|
log.Println(err)
|
||
|
|
return nil, err
|
||
|
|
}
|
||
|
|
|
||
|
|
if ius != nil {
|
||
|
|
return ius.(*UserState), nil
|
||
|
|
}
|
||
|
|
|
||
|
|
// cmd := &Command{
|
||
|
|
// Op: OP_Update,
|
||
|
|
// Key: Userid,
|
||
|
|
// }
|
||
|
|
|
||
|
|
cs := ss.nh.GetNoOPSession(128)
|
||
|
|
|
||
|
|
cmd := &CmdUpdate{UserId: Userid}
|
||
|
|
cmdBuf, err := FsPasser.PackToBytes(cmd)
|
||
|
|
if err != nil {
|
||
|
|
return nil, err
|
||
|
|
}
|
||
|
|
|
||
|
|
result, err := ss.nh.SyncPropose(context.TODO(), cs, cmdBuf)
|
||
|
|
if err != nil {
|
||
|
|
return nil, err
|
||
|
|
}
|
||
|
|
|
||
|
|
us = &UserState{}
|
||
|
|
err = us.Decode(result.Data)
|
||
|
|
if err != nil {
|
||
|
|
return nil, err
|
||
|
|
}
|
||
|
|
|
||
|
|
return us, err
|
||
|
|
}
|