tavern/core/rejson_handler.go

package core

import (
	"context"
	"encoding/json"

	"github.com/go-redis/redis/v8"
	"github.com/nitishm/go-rejson/v4"
	"github.com/nitishm/go-rejson/v4/rjs"
)

// gocommon으로 옮길 거
type RejsonHandler struct {
	*redis.Client
	inner *rejson.Handler
	ctx   context.Context
}

func NewReJSONHandler(ctx context.Context, redisClient *redis.Client) *RejsonHandler {
	inner := rejson.NewReJSONHandler()
	inner.SetGoRedisClientWithContext(ctx, redisClient)
	return &RejsonHandler{
		Client: redisClient,
		inner:  inner,
		ctx:    ctx,
	}
}

func respToArray[T any](resp any) []T {
	resArr := resp.([]any)
	v := make([]T, len(resArr))
	for i, e := range resArr {
		v[i] = e.(T)
	}
	return v
}

func (rh *RejsonHandler) JSONMSet(key string, kv map[string]any) error {
	pl := rh.Pipeline()
	for path, obj := range kv {
		b, err := json.Marshal(obj)
		if err != nil {
			return err
		}
		pl.Do(rh.ctx, []any{"JSON.SET"}, key, path, b)
	}

	cmders, err := pl.Exec(rh.ctx)
	if err != nil {
		return err
	}

	for _, cmder := range cmders {
		if cmder.Err() != nil {
			return cmder.Err()
		}
	}
	return nil
}

func (rh *RejsonHandler) JSONSet(key, path string, obj any, opts ...rjs.SetOption) (res interface{}, err error) {
	b, err := json.Marshal(obj)
	if err != nil {
		return nil, err
	}
	args := []any{
		[]any{"JSON.SET"},
		key,
		path,
		b,
	}
	if len(opts) > 0 {
		args = append(args, opts[0])
	}

	return rh.Do(rh.ctx, args...).Result()
}

func (rh *RejsonHandler) JSONGet(key, path string, opts ...rjs.GetOption) (res interface{}, err error) {
	return rh.inner.JSONGet(key, path, opts...)
}

func (rh *RejsonHandler) JSONGetString(key, path string) ([]string, error) {
	res, err := rh.inner.JSONResp(key, path)
	if err != nil {
		return nil, err
	}
	return respToArray[string](res), nil
}

func (rh *RejsonHandler) JSONGetInt64(key, path string) ([]int64, error) {
	res, err := rh.inner.JSONResp(key, path)
	if err != nil {
		return nil, err
	}
	return respToArray[int64](res), nil
}

func (rh *RejsonHandler) JSONMGet(path string, keys ...string) (res interface{}, err error) {
	return rh.inner.JSONMGet(path, keys...)
}

func (rh *RejsonHandler) JSONMDel(key string, paths []string) error {
	pl := rh.Pipeline()
	for _, path := range paths {
		name, args, err := rjs.CommandBuilder(rjs.ReJSONCommandDEL, key, path)
		if err != nil {
			return err
		}
		args = append([]interface{}{name}, args...)
		pl.Do(rh.ctx, args...)
	}
	_, err := pl.Exec(rh.ctx)
	return err
}

func (rh *RejsonHandler) JSONDel(key, path string) (res interface{}, err error) {
	return rh.inner.JSONDel(key, path)
}

func (rh *RejsonHandler) JSONType(key, path string) (res interface{}, err error) {
	return rh.inner.JSONType(key, path)
}

func (rh *RejsonHandler) JSONNumIncrBy(key, path string, number int) (res interface{}, err error) {
	return rh.inner.JSONNumIncrBy(key, path, number)
}

func (rh *RejsonHandler) JSONNumMultBy(key, path string, number int) (res interface{}, err error) {
	return rh.inner.JSONNumMultBy(key, path, number)
}

func (rh *RejsonHandler) JSONStrAppend(key, path string, jsonstring string) (res interface{}, err error) {
	return rh.inner.JSONStrAppend(key, path, jsonstring)
}

func (rh *RejsonHandler) JSONStrLen(key, path string) (res interface{}, err error) {
	return rh.inner.JSONStrLen(key, path)
}

func (rh *RejsonHandler) JSONArrAppend(key, path string, values ...interface{}) (res interface{}, err error) {
	return rh.inner.JSONArrAppend(key, path, values...)
}

func (rh *RejsonHandler) JSONArrLen(key, path string) (res interface{}, err error) {
	return rh.inner.JSONArrLen(key, path)
}

func (rh *RejsonHandler) JSONArrPop(key, path string, index int) (res interface{}, err error) {
	return rh.inner.JSONArrPop(key, path, index)
}

func (rh *RejsonHandler) JSONArrIndex(key, path string, jsonValue interface{}, optionalRange ...int) (res interface{}, err error) {
	return rh.inner.JSONArrIndex(key, path, jsonValue, optionalRange...)
}

func (rh *RejsonHandler) JSONArrTrim(key, path string, start, end int) (res interface{}, err error) {
	return rh.inner.JSONArrTrim(key, path, start, end)
}

func (rh *RejsonHandler) JSONArrInsert(key, path string, index int, values ...interface{}) (res interface{}, err error) {
	return rh.inner.JSONArrInsert(key, path, index, values...)
}

func (rh *RejsonHandler) JSONObjKeys(key, path string) ([]string, error) {
	name, args, err := rjs.CommandBuilder(rjs.ReJSONCommandOBJKEYS, key, path)
	if err != nil {
		return nil, err
	}

	args = append([]interface{}{name}, args...)
	res, err := rh.Do(rh.ctx, args...).Result()
	if err != nil {
		return nil, err
	}

	resArr := res.([]any)
	resArr = resArr[0].([]any)
	slc := make([]string, len(resArr))

	for i, r := range resArr {
		slc[i] = r.(string)
	}

	return slc, nil
}

func (rh *RejsonHandler) JSONObjLen(key, path string) (res interface{}, err error) {
	return rh.inner.JSONObjLen(key, path)
}

func (rh *RejsonHandler) JSONObjLenInt64(key, path string) (int64, error) {
	res, err := rh.inner.JSONObjLen(key, path)
	if err != nil {
		return 0, err
	}
	return res.([]any)[0].(int64), nil
}

func (rh *RejsonHandler) JSONDebug(subCmd rjs.DebugSubCommand, key, path string) (res interface{}, err error) {
	return rh.inner.JSONDebug(subCmd, key, path)
}

func (rh *RejsonHandler) JSONForget(key, path string) (res interface{}, err error) {
	return rh.inner.JSONForget(key, path)
}

func (rh *RejsonHandler) JSONResp(key, path string) (res interface{}, err error) {
	return rh.inner.JSONResp(key, path)
}