json.go 1.3 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869
  1. package parser
  2. import (
  3. "encoding/json"
  4. "reflect"
  5. "github.com/langgenius/dify-plugin-daemon/internal/types/validators"
  6. )
  7. func UnmarshalJson[T any](text string) (T, error) {
  8. return UnmarshalJsonBytes[T]([]byte(text))
  9. }
  10. func UnmarshalJsonBytes[T any](data []byte) (T, error) {
  11. var result T
  12. err := json.Unmarshal(data, &result)
  13. if err != nil {
  14. return result, err
  15. }
  16. // skip validate if T is a map
  17. typ := reflect.TypeOf(result)
  18. if typ.Kind() == reflect.Map {
  19. return result, nil
  20. }
  21. if err := validators.GlobalEntitiesValidator.Struct(&result); err != nil {
  22. return result, err
  23. }
  24. return result, err
  25. }
  26. func UnmarshalJsonBytes2Slice[T any](data []byte) ([]T, error) {
  27. var result []T
  28. err := json.Unmarshal(data, &result)
  29. if err != nil {
  30. return nil, err
  31. }
  32. for _, item := range result {
  33. if err := validators.GlobalEntitiesValidator.Struct(&item); err != nil {
  34. return nil, err
  35. }
  36. }
  37. return result, err
  38. }
  39. func MarshalJson[T any](data T) string {
  40. b, _ := json.Marshal(data)
  41. return string(b)
  42. }
  43. func MarshalJsonBytes[T any](data T) []byte {
  44. b, _ := json.Marshal(data)
  45. return b
  46. }
  47. func UnmarshalJsonBytes2Map(data []byte) (map[string]any, error) {
  48. result := map[string]any{}
  49. err := json.Unmarshal(data, &result)
  50. return result, err
  51. }
  52. func UnmarshalJson2Map(json string) (map[string]any, error) {
  53. return UnmarshalJsonBytes2Map([]byte(json))
  54. }