Skip to content

Commit 7eec882

Browse files
Frederick-SFrederick-S
committed
Add tcp
1 parent ee024d9 commit 7eec882

4 files changed

Lines changed: 197 additions & 0 deletions

File tree

‎main.go‎

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -3,10 +3,12 @@ package main
33
import (
44
"github.com/read-and-code/cache/cache"
55
"github.com/read-and-code/cache/http"
6+
"github.com/read-and-code/cache/tcp"
67
)
78

89
func main() {
910
cacheProvider := cache.NewCache("inmemory")
1011

12+
go tcp.NewServer(cacheProvider).Listen()
1113
http.NewServer(cacheProvider).Listen()
1214
}

‎tcp/server.go‎

Lines changed: 32 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,32 @@
1+
package tcp
2+
3+
import (
4+
"github.com/read-and-code/cache/cache"
5+
"net"
6+
)
7+
8+
type Server struct {
9+
cache cache.Cache
10+
}
11+
12+
func (server *Server) Listen() {
13+
listener, err := net.Listen("tcp", ":12346")
14+
15+
if err != nil {
16+
panic(err)
17+
}
18+
19+
for {
20+
connection, err := listener.Accept()
21+
22+
if err != nil {
23+
panic(err)
24+
}
25+
26+
go server.process(connection)
27+
}
28+
}
29+
30+
func NewServer(cache cache.Cache) *Server {
31+
return &Server{cache}
32+
}

‎tcp/server_handler.go‎

Lines changed: 122 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,122 @@
1+
package tcp
2+
3+
import (
4+
"bufio"
5+
"io"
6+
"log"
7+
"net"
8+
)
9+
10+
func (server *Server) readKey(reader *bufio.Reader) (string, error) {
11+
keyLength, err := readLength(reader)
12+
13+
if err != nil {
14+
return "", err
15+
}
16+
17+
key := make([]byte, keyLength)
18+
_, err = io.ReadFull(reader, key)
19+
20+
if err != nil {
21+
return "", err
22+
}
23+
24+
return string(key), nil
25+
}
26+
27+
func (server *Server) readKeyAndValue(reader *bufio.Reader) (string, []byte, error) {
28+
keyLength, err := readLength(reader)
29+
30+
if err != nil {
31+
return "", nil, err
32+
}
33+
34+
valueLength, err := readLength(reader)
35+
36+
if err != nil {
37+
return "", nil, err
38+
}
39+
40+
key := make([]byte, keyLength)
41+
_, err = io.ReadFull(reader, key)
42+
43+
if err != nil {
44+
return "", nil, err
45+
}
46+
47+
value := make([]byte, valueLength)
48+
_, err = io.ReadFull(reader, value)
49+
50+
if err != nil {
51+
return "", nil, err
52+
}
53+
54+
return string(key), value, nil
55+
}
56+
57+
func (server *Server) get(connection net.Conn, reader *bufio.Reader) error {
58+
key, err := server.readKey(reader)
59+
60+
if err != nil {
61+
return err
62+
}
63+
64+
value, err := server.cache.Get(key)
65+
66+
return sendResponse(value, err, connection)
67+
}
68+
69+
func (server *Server) set(connection net.Conn, reader *bufio.Reader) error {
70+
key, value, err := server.readKeyAndValue(reader)
71+
72+
if err != nil {
73+
return err
74+
}
75+
76+
return sendResponse(nil, server.cache.Set(key, value), connection)
77+
}
78+
79+
func (server *Server) delete(connection net.Conn, reader *bufio.Reader) error {
80+
key, err := server.readKey(reader)
81+
82+
if err != nil {
83+
return err
84+
}
85+
86+
return sendResponse(nil, server.cache.Delete(key), connection)
87+
}
88+
89+
func (server *Server) process(connection net.Conn) {
90+
defer connection.Close()
91+
92+
reader := bufio.NewReader(connection)
93+
94+
for {
95+
operation, err := reader.ReadByte()
96+
97+
if err != nil {
98+
if err != io.EOF {
99+
log.Println("Close connection due to error:", err)
100+
}
101+
102+
return
103+
}
104+
105+
if operation == 'S' {
106+
err = server.set(connection, reader)
107+
} else if operation == 'G' {
108+
err = server.get(connection, reader)
109+
} else if operation == 'D' {
110+
err = server.delete(connection, reader)
111+
} else {
112+
log.Println("Close connection due to invalid operation:", operation)
113+
}
114+
115+
if err != nil {
116+
log.Println("Close connection due to error:", err)
117+
118+
return
119+
}
120+
}
121+
}
122+

‎tcp/utils.go‎

Lines changed: 41 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,41 @@
1+
package tcp
2+
3+
import (
4+
"bufio"
5+
"fmt"
6+
"net"
7+
"strconv"
8+
"strings"
9+
)
10+
11+
func readLength(reader *bufio.Reader) (int, error) {
12+
text, err := reader.ReadString(' ')
13+
14+
if err != nil {
15+
return 0, err
16+
}
17+
18+
length, err := strconv.Atoi(strings.TrimSpace(text))
19+
20+
if err != nil {
21+
return 0, err
22+
}
23+
24+
return length, nil
25+
}
26+
27+
func sendResponse(value []byte, err error, connection net.Conn) error {
28+
if err != nil {
29+
response := fmt.Sprintf("-%d ", len(err.Error())) + err.Error()
30+
31+
_, err := connection.Write([]byte(response))
32+
33+
return err
34+
}
35+
36+
valueLength := fmt.Sprintf("%d ", len(value))
37+
38+
_, err = connection.Write(append([]byte(valueLength), value...))
39+
40+
return err
41+
}

0 commit comments

Comments
 (0)