package starmap import ( "encoding/gob" "os" "time" "b612.me/notify" ) func init() { notify.Register(kvMsg{}) notify.Register(starMapErr{}) } type starMapErr struct { Err string } func (s starMapErr) Error() string { return s.Err } func newStarMapErr(err error) error { if err == nil { return nil } return starMapErr{Err: err.Error()} } type kvMsg struct { Key interface{} Value interface{} Err error } type RemoteKv struct { server notify.Server client notify.Client kvmap StarMapKV addr string network string readTimeout time.Duration timeout time.Duration } func NewServer(network, addr string) (*RemoteKv, error) { var err error kv := RemoteKv{ server: notify.NewServer(), kvmap: NewStarMap(), addr: addr, network: network, } err = kv.server.Listen(network, addr) if err == nil { kv.bind() } return &kv, err } func NewClient(network, addr string, dialTimeout time.Duration) (*RemoteKv, error) { var err error kv := RemoteKv{ client: notify.NewClient(), kvmap: NewStarMap(), addr: addr, network: network, timeout: dialTimeout, readTimeout: time.Second * 5, } err = kv.client.ConnectTimeout(network, addr, dialTimeout) return &kv, err } func (kv *RemoteKv) Register(data interface{}) { gob.Register(data) } func (kv *RemoteKv) RegisterAll(data []interface{}) { for _, v := range data { gob.Register(v) } } func (kv *RemoteKv) bind() { //for server kv.server.SetDefaultLink(kv.dispatch) //for client } func (kv *RemoteKv) reconnect() error { if kv.server != nil { return nil } if kv.client != nil { if kv.client.Status().Alive { return nil } return kv.client.ConnectTimeout(kv.network, kv.addr, kv.timeout) } return os.ErrInvalid }