fixed EOF error, added INCR, DECR command

Author: SojebSikder

.github/workflows/release.yml

@@ -0,0 +1,27 @@
+name: Release
+
+on:
+  push:
+    tags:
+      - "v*"
+
+permissions:
+  contents: write
+
+jobs:
+  goreleaser:
+    runs-on: ubuntu-latest
+    steps:
+      - uses: actions/checkout@v4
+        with:
+          fetch-depth: 0
+      - uses: actions/setup-go@v5
+        with:
+          go-version-file: "go.mod"
+          cache: true
+      - name: Run GoReleaser
+        uses: goreleaser/goreleaser-action@v6
+        with:
+          args: release --clean
+        env:
+          GITHUB_TOKEN: ${{ secrets.GITHUB_TOKEN }}

.goreleaser.yml

@@ -0,0 +1,29 @@
+version: 2
+
+before:
+  hooks:
+    - go mod tidy
+
+builds:
+  - env:
+      - CGO_ENABLED=0
+    mod_timestamp: '{{ .CommitTimestamp }}'
+    flags:
+      - -trimpath
+    ldflags:
+      - '-s -w -X main.version={{.Version}} -X main.commit={{.Commit}}'
+    goos:
+      - windows
+      - linux
+      - darwin
+    goarch:
+      - amd64
+      - arm64
+    binary: '{{ .ProjectName }}_{{ .Os }}_{{ .Arch }}'
+
+checksum:
+  name_template: '{{ .ProjectName }}_SHA256SUMS'
+  algorithm: sha256
+
+changelog:
+  disable: true

README.md

@@ -35,4 +35,6 @@ set name sojeb
 - HLEN
 - FLUSHALL
 - DEL
-- INFO
+- INFO
+- INCR
+- DECR

build.sh

@@ -5,15 +5,15 @@ platforms=("windows/amd64" "windows/arm64" "linux/amd64" "linux/arm64" "darwin/a
 for platform in "${platforms[@]}"; do
     OS=${platform%/*}
     ARCH=${platform#*/}
-    OUTPUT="smem-${OS}-${ARCH}"
+    OUTPUT="build/tunnel-${OS}-${ARCH}"
 
     # Append .exe for Windows
     if [ "$OS" == "windows" ]; then
         OUTPUT+=".exe"
     fi
 
     echo "Building for $OS $ARCH..."
-    GOOS=$OS GOARCH=$ARCH go build -o $OUTPUT
+    GOOS=$OS GOARCH=$ARCH go build -o "$OUTPUT" .
 done
 
-echo "Build completed!"
+echo "Build completed!"

aof.go renamed to internal/aof.go

@@ -1,4 +1,4 @@
-package main
+package internal
 
 import (
 	"bufio"

handler.go renamed to internal/handler.go

@@ -1,7 +1,8 @@
-package main
+package internal
 
 import (
 	"fmt"
+	"strconv"
 	"sync"
 )
 
@@ -17,6 +18,8 @@ var Handlers = map[string]func([]Value) Value{
 	"FLUSHALL": flushall,
 	"DEL":      del,
 	"INFO":     info,
+	"INCR":     incr,
+	"DECR":     decr,
 }
 
 func ping(args []Value) Value {
@@ -222,3 +225,57 @@ func info(args []Value) Value {
 
 	return Value{typ: "bulk", bulk: infoStr}
 }
+
+// incr command which increments the value of a key in the SET
+func incr(args []Value) Value {
+	if len(args) != 1 {
+		return Value{typ: "error", str: "ERR wrong number of arguments for 'incr' command"}
+	}
+
+	key := args[0].bulk
+
+	SETsMu.Lock()
+	value, ok := SETs[key]
+	SETsMu.Unlock()
+
+	if !ok {
+		SETs[key] = "1"
+		return Value{typ: "integer", num: 1}
+	}
+
+	num, err := strconv.Atoi(value)
+	if err != nil {
+		return Value{typ: "error", str: "ERR value is not an integer"}
+	}
+
+	num++
+	SETs[key] = strconv.Itoa(num)
+	return Value{typ: "integer", num: num}
+}
+
+// decr command which decrements the value of a key in the SET
+func decr(args []Value) Value {
+	if len(args) != 1 {
+		return Value{typ: "error", str: "ERR wrong number of arguments for 'decr' command"}
+	}
+
+	key := args[0].bulk
+
+	SETsMu.Lock()
+	value, ok := SETs[key]
+	SETsMu.Unlock()
+
+	if !ok {
+		SETs[key] = "0"
+		return Value{typ: "integer", num: 0}
+	}
+
+	num, err := strconv.Atoi(value)
+	if err != nil {
+		return Value{typ: "error", str: "ERR value is not an integer"}
+	}
+
+	num--
+	SETs[key] = strconv.Itoa(num)
+	return Value{typ: "integer", num: num}
+}

resp.go renamed to internal/resp.go

@@ -1,4 +1,4 @@
-package main
+package internal
 
 import (
 	"bufio"
@@ -195,7 +195,6 @@ func (v Value) marshalInteger() []byte {
 }
 
 // Writer
-
 type Writer struct {
 	writer io.Writer
 }

internal/server.go

@@ -0,0 +1,95 @@
+package internal
+
+import (
+	"flag"
+	"fmt"
+	"net"
+	"strings"
+)
+
+var port = "6379"
+
+func StartServer() {
+	flag.StringVar(&port, "port", "6379", "port to listen on")
+	flag.Parse()
+
+	fmt.Println("Listening on port :" + port)
+
+	// Create a new server
+	listener, err := net.Listen("tcp", ":"+port)
+	if err != nil {
+		fmt.Println("Error listening:", err)
+		return
+	}
+	defer listener.Close()
+
+	aof, err := NewAof("database.aof")
+	if err != nil {
+		fmt.Println("Error opening AOF:", err)
+		return
+	}
+	defer aof.Close()
+
+	// Load data from AOF
+	aof.Read(func(value Value) {
+		command := strings.ToUpper(value.array[0].bulk)
+		args := value.array[1:]
+
+		handler, ok := Handlers[command]
+		if !ok {
+			fmt.Println("Invalid command from AOF:", command)
+			return
+		}
+
+		handler(args)
+	})
+
+	for {
+		conn, err := listener.Accept()
+		if err != nil {
+			fmt.Println("Error accepting connection:", err)
+			continue
+		}
+
+		go handleConnection(conn, aof)
+	}
+}
+
+func handleConnection(conn net.Conn, aof *Aof) {
+	defer conn.Close()
+
+	resp := NewResp(conn)
+	writer := NewWriter(conn)
+
+	for {
+		value, err := resp.Read()
+		if err != nil {
+			if err.Error() != "EOF" {
+				fmt.Println("Error reading:", err)
+			}
+			return
+		}
+
+		if value.typ != "array" || len(value.array) == 0 {
+			writer.Write(Value{typ: "string", str: "Invalid request"})
+			continue
+		}
+
+		command := strings.ToUpper(value.array[0].bulk)
+		args := value.array[1:]
+
+		handler, ok := Handlers[command]
+		if !ok {
+			writer.Write(Value{typ: "string", str: "Unknown command"})
+			continue
+		}
+
+		if command == "SET" || command == "HSET" || command == "DEL" || command == "FLUSHALL" ||
+			command == "HDEL" || command == "INCR" || command == "DECR" {
+			aof.Write(value)
+		}
+
+		result := handler(args)
+		writer.Write(result)
+	}
+}

main.go

@@ -1,92 +1,7 @@
 package main
 
-import (
-	"flag"
-	"fmt"
-	"net"
-	"strings"
-)
-
-var port = "6379"
+import "go-in-memory-database/internal"
 
 func main() {
-	flag.StringVar(&port, "port", "6379", "port to listen on")
-	flag.Parse()
-
-	fmt.Println("Listening on port :" + port)
-
-	// Create a new server
-	l, err := net.Listen("tcp", ":"+port)
-	if err != nil {
-		fmt.Println(err)
-		return
-	}
-
-	aof, err := NewAof("database.aof")
-	if err != nil {
-		fmt.Println(err)
-		return
-	}
-	defer aof.Close()
-
-	aof.Read(func(value Value) {
-		command := strings.ToUpper(value.array[0].bulk)
-		args := value.array[1:]
-
-		handler, ok := Handlers[command]
-		if !ok {
-			fmt.Println("Invalid command: ", command)
-			return
-		}
-
-		handler(args)
-	})
-
-	// Listen for connections
-	conn, err := l.Accept()
-	if err != nil {
-		fmt.Println(err)
-		return
-	}
-
-	defer conn.Close()
-
-	for {
-		resp := NewResp(conn)
-		value, err := resp.Read()
-		if err != nil {
-			fmt.Println(err)
-			return
-		}
-
-		if value.typ != "array" {
-			fmt.Println("Invalid request, expected array")
-			continue
-		}
-
-		if len(value.array) == 0 {
-			fmt.Println("Invalid request, expected array length > 0")
-			continue
-		}
-
-		command := strings.ToUpper(value.array[0].bulk)
-		args := value.array[1:]
-
-		writer := NewWriter(conn)
-
-		handler, ok := Handlers[command]
-		if !ok {
-			fmt.Println("Invalid command: ", command)
-			writer.Write(Value{typ: "string", str: ""})
-			continue
-		}
-
-		// write to aof file if the command does any changes to the database
-		if command == "SET" || command == "HSET" || command == "DEL" || command == "FLUSHALL" || command == "HDEL" {
-			aof.Write(value)
-		}
-
-		result := handler(args)
-		writer.Write(result)
-	}
+	internal.StartServer()
 }