嵌套自定义解组

Question

我需要根据上层字段值（

VerifyingKey

字段）中的变量来实现自定义解组逻辑（到

Type

字段）。我可以举例说明如下情况（源自执行自定义解组后解组剩余的 JSON）

type SingleKey struct {
  Type          string          `json:"type"`
  Others OtherFields   `json:"wrapped_fields"`
}

type OtherFields struct {
  FQDN         string          `json:"fqdn"`
  Address      string          `json:"address"`
  Nonce        int64           `json:"nonce"`
  Challenge    []byte          `json:"challenge"`
  NetworkID    string          `json:"network_id"`
  VerifyingKey []PublicKey     `json:"verifying_key"`
  Signature    []byte          `json:"signature"`
}

// Interface for Public Key
type PublicKey interface {
  // Methods specific to public key functionality
}

// Example concrete implementations (replace with your actual types)
type RsaPublicKey struct {
  PublicKey
  // ... (fields specific to RSA public key)
}

type EcdsaPublicKey struct {
  PublicKey
  // ... (fields specific to ECDSA public key)
}

我尝试了很多替代方案，但是我无法找到解决方案。

Answer 1

为

SingleKey

实现自定义解组器：

type singleKeyMarshal struct {
  Type          string          `json:"type"`
  Others      otherFieldsMarshal `json:"wrapped_fields"`
}

type otherFieldsMarshal struct {
  FQDN         string          `json:"fqdn"`
  Address      string          `json:"address"`
  Nonce        int64           `json:"nonce"`
  Challenge    []byte          `json:"challenge"`
  NetworkID    string          `json:"network_id"`
  VerifyingKey []json.RawMessage     `json:"verifying_key"`
  Signature    []byte          `json:"signature"`
}

func (s *SingleKey) UnmarshalJSON(in []byte) error {

  var sm singleKeyMarshal
  if  err:=json.Unmarshal(in,&sm); err!=nil {
    return err
  }
  // Copy all fields from sm to s
  switch sm.Type {
     // Deal with unmarshaling the public keys
     case "RSA": 
       // This depends on how it is marshaled. If it is rsa.PublicKey:
       key:=[]rsa.PublicKey{}
       if err:=json.Unmarshal(sm.Others.VerifyingKey,&key); err!=nil {
           return err
       }
       s.Others.VerifyingKey=key
       
  }
}