58 lines
1003 B
Go
58 lines
1003 B
Go
|
// +build dev
|
||
|
|
|
||
|
|
package wtdb
|
||
|
|
|
||
|
|
import "sync"
|
||
|
|
|
||
|
|
type MockDB struct {
|
||
|
|
mu sync.Mutex
|
||
|
|
sessions map[SessionID]*SessionInfo
|
||
|
|
}
|
||
|
|
|
||
|
|
func NewMockDB() *MockDB {
|
||
|
|
return &MockDB{
|
||
|
|
sessions: make(map[SessionID]*SessionInfo),
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func (db *MockDB) InsertStateUpdate(update *SessionStateUpdate) (uint16, error) {
|
||
|
|
db.mu.Lock()
|
||
|
|
defer db.mu.Unlock()
|
||
|
|
|
||
|
|
info, ok := db.sessions[update.ID]
|
||
|
|
if !ok {
|
||
|
|
return 0, ErrSessionNotFound
|
||
|
|
}
|
||
|
|
|
||
|
|
err := info.AcceptUpdateSequence(update.SeqNum, update.LastApplied)
|
||
|
|
if err != nil {
|
||
|
|
return info.LastApplied, err
|
||
|
|
}
|
||
|
|
|
||
|
|
return info.LastApplied, nil
|
||
|
|
}
|
||
|
|
|
||
|
|
func (db *MockDB) GetSessionInfo(id *SessionID) (*SessionInfo, error) {
|
||
|
|
db.mu.Lock()
|
||
|
|
defer db.mu.Unlock()
|
||
|
|
|
||
|
|
if info, ok := db.sessions[*id]; ok {
|
||
|
|
return info, nil
|
||
|
|
}
|
||
|
|
|
||
|
|
return nil, ErrSessionNotFound
|
||
|
|
}
|
||
|
|
|
||
|
|
func (db *MockDB) InsertSessionInfo(info *SessionInfo) error {
|
||
|
|
db.mu.Lock()
|
||
|
|
defer db.mu.Unlock()
|
||
|
|
|
||
|
|
if _, ok := db.sessions[info.ID]; ok {
|
||
|
|
return ErrSessionAlreadyExists
|
||
|
|
}
|
||
|
|
|
||
|
|
db.sessions[info.ID] = info
|
||
|
|
|
||
|
|
return nil
|
||
|
|
}
|