mirror of
https://github.com/EchoVault/SugarDB.git
synced 2025-10-06 00:16:53 +08:00
80 lines
2.5 KiB
Go
80 lines
2.5 KiB
Go
// Copyright 2024 Kelvin Clement Mwinuka
|
|
//
|
|
// Licensed under the Apache License, Version 2.0 (the "License");
|
|
// you may not use this file except in compliance with the License.
|
|
// You may obtain a copy of the License at
|
|
//
|
|
// http://www.apache.org/licenses/LICENSE-2.0
|
|
//
|
|
// Unless required by applicable law or agreed to in writing, software
|
|
// distributed under the License is distributed on an "AS IS" BASIS,
|
|
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
// See the License for the specific language governing permissions and
|
|
// limitations under the License.
|
|
|
|
package connection
|
|
|
|
import (
|
|
"errors"
|
|
"fmt"
|
|
|
|
"github.com/echovault/echovault/internal"
|
|
"github.com/echovault/echovault/internal/constants"
|
|
)
|
|
|
|
func handlePing(params internal.HandlerFuncParams) ([]byte, error) {
|
|
switch len(params.Command) {
|
|
default:
|
|
return nil, errors.New(constants.WrongArgsResponse)
|
|
case 1:
|
|
return []byte("+PONG\r\n"), nil
|
|
case 2:
|
|
return []byte(fmt.Sprintf("$%d\r\n%s\r\n", len(params.Command[1]), params.Command[1])), nil
|
|
}
|
|
}
|
|
|
|
func handleEcho(params internal.HandlerFuncParams) ([]byte, error) {
|
|
if len(params.Command) != 2 {
|
|
return nil, errors.New(constants.WrongArgsResponse)
|
|
}
|
|
return []byte(fmt.Sprintf("$%d\r\n%s\r\n", len(params.Command[1]), params.Command[1])), nil
|
|
}
|
|
|
|
func Commands() []internal.Command {
|
|
return []internal.Command{
|
|
{
|
|
Command: "ping",
|
|
Module: constants.ConnectionModule,
|
|
Categories: []string{constants.ConnectionCategory, constants.FastCategory},
|
|
Description: `(PING [message])
|
|
Ping the echovault server. If a message is provided, the message will be echoed back to the client.
|
|
Otherwise, the server will return "PONG".`,
|
|
Sync: false,
|
|
KeyExtractionFunc: func(cmd []string) (internal.KeyExtractionFuncResult, error) {
|
|
return internal.KeyExtractionFuncResult{
|
|
Channels: make([]string, 0),
|
|
ReadKeys: make([]string, 0),
|
|
WriteKeys: make([]string, 0),
|
|
}, nil
|
|
},
|
|
HandlerFunc: handlePing,
|
|
},
|
|
{
|
|
Command: "echo",
|
|
Module: constants.ConnectionModule,
|
|
Categories: []string{constants.ConnectionCategory, constants.FastCategory},
|
|
Description: `(ECHO message)
|
|
Echo the message back to the client.`,
|
|
Sync: false,
|
|
KeyExtractionFunc: func(cmd []string) (internal.KeyExtractionFuncResult, error) {
|
|
return internal.KeyExtractionFuncResult{
|
|
Channels: make([]string, 0),
|
|
ReadKeys: make([]string, 0),
|
|
WriteKeys: make([]string, 0),
|
|
}, nil
|
|
},
|
|
HandlerFunc: handleEcho,
|
|
},
|
|
}
|
|
}
|