mirror of
https://github.com/EchoVault/SugarDB.git
synced 2025-10-14 12:13:54 +08:00
Scrapped plugin design in favour of simple command interfaces.
Setup docker build process for running server. Deleted test files.
This commit is contained in:
51
server/mock.go
Normal file
51
server/mock.go
Normal file
@@ -0,0 +1,51 @@
|
||||
package main
|
||||
|
||||
import (
|
||||
"bytes"
|
||||
"io"
|
||||
"sync"
|
||||
)
|
||||
|
||||
type CustomWriter struct {
|
||||
Buf bytes.Buffer
|
||||
}
|
||||
|
||||
func (cw *CustomWriter) Write(p []byte) (int, error) {
|
||||
count := 0
|
||||
|
||||
for _, b := range p {
|
||||
cw.Buf.WriteByte(b)
|
||||
count += 1
|
||||
}
|
||||
|
||||
if count != len(p) {
|
||||
return count, io.ErrShortWrite
|
||||
}
|
||||
|
||||
return count, nil
|
||||
}
|
||||
|
||||
type MockData struct {
|
||||
Mu sync.Mutex
|
||||
Data map[string]interface{}
|
||||
}
|
||||
|
||||
type MockServer struct {
|
||||
Data MockData
|
||||
}
|
||||
|
||||
func (server *MockServer) Lock() {
|
||||
server.Data.Mu.Lock()
|
||||
}
|
||||
|
||||
func (server *MockServer) Unlock() {
|
||||
server.Data.Mu.Unlock()
|
||||
}
|
||||
|
||||
func (server *MockServer) GetData(key string) interface{} {
|
||||
return server.Data.Data[key]
|
||||
}
|
||||
|
||||
func (server *MockServer) SetData(key string, value interface{}) {
|
||||
server.Data.Data[key] = value
|
||||
}
|
Reference in New Issue
Block a user