1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
|
package client
import (
"bufio"
"fmt"
"net"
"os"
"strings"
"time"
"github.com/kyren223/eko/internal/utils/log"
)
func Run() {
log.SetLevel(log.LevelDebug)
log.Info("Client started, waiting for user input...")
for {
fmt.Print("> ")
input, _ := bufio.NewReader(os.Stdin).ReadString('\n')
input = strings.TrimSpace(input)
if input == ":q" || input == "exit" || input == "quit" {
break
}
log.Debug("Input: %v", input)
err := processRequest(input)
if err != nil {
log.Error("%v", err)
}
}
}
func processRequest(request string) error {
conn, err := net.Dial("tcp", ":7223")
if err != nil {
return fmt.Errorf("Unable to establish connection with server: %v", err)
}
defer conn.Close()
conn.SetDeadline(time.Now().Add(time.Second))
log.Info("Established connection to server: %v", conn.RemoteAddr().String())
_, err = conn.Write([]byte(request))
if err != nil {
return fmt.Errorf("Unable to send request: %v", err)
}
buffer := make([]byte, 1024)
n, err := conn.Read(buffer)
if err != nil {
return fmt.Errorf("Unable to receive response: %v", err)
}
log.Info("Response from server: %v", string(buffer[:n]))
return nil
}
|
