mgo 모델의 인터페이스 유형 이해

MongoDB 및 Go의 맥락에서 인터페이스를 사용한 데이터 모델링은 동적 특성으로 인해 어려움을 겪을 수 있습니다. 인터페이스의. 다음은 발생한 문제에 대한 간략한 설명과 제안된 해결 방법입니다.

인터페이스 유형 문제

MongoDB의 문서 기반 데이터 모델은 유형을 제공하지 않습니다. 포함된 문서에 대한 정보입니다. 인터페이스 유형이 포함된 MongoDB 문서를 Go 구조체로 역마샬링하기 위해 mgo를 사용할 때, mgo는 포함된 각 문서의 특정 유형을 결정할 수 없습니다. 이로 인해 "bson.M 유형의 값은 Node 유형에 할당할 수 없습니다."라는 오류가 발생합니다.

해결책: 인터페이스 유형 래핑

이 제한 사항을 해결하려면 한 가지 접근 방식을 사용하세요. 유형 정보를 제공하는 사용자 정의 구조체에 인터페이스 유형을 래핑하는 것입니다. 이를 통해 mgo는 언마샬링 중에 포함된 문서의 특정 유형을 식별할 수 있습니다.

다음 예를 고려하십시오.

type NodeWithType struct {
   Node Node `bson:"-"`
   Type string
}

type Workflow struct {
   CreatedAt time.Time
   StartedAt time.Time
   CreatedBy string
   Nodes []NodeWithType
}

SetBSON 함수 구현

이 솔루션을 완료하려면 NodeWithType 유형에 대한 SetBSON 함수를 구현해야 합니다. 이 함수는 유형 문자열을 디코딩하고 해당 유형의 인스턴스를 생성한 후 역마샬링합니다.

func (nt *NodeWithType) SetBSON(r bson.Raw) error {
   // Decode the "Type" field and determine the Node type
   var typeStr string
   if err := r.Unmarshal(&typeStr); err != nil {
      return err
   }

   // Create a new instance of the Node type based on the type string
   node, ok := reflect.New(reflect.TypeOf(Node).Elem()).Interface().(Node)
   if !ok {
      return errors.New("invalid Node type")
   }

   // Unmarshal the remaining data into the Node instance
   if err := r.Unmarshal(node); err != nil {
      return err
   }

   // Assign the Node instance to the NodeWithType struct
   nt.Node = node
   return nil
}

결론

이 패턴을 활용하면 다양한 유형의 포함된 문서를 역마샬링하는 기능을 유지하면서 인터페이스를 효과적으로 활용할 수 있습니다. 명시적인 유형 정보를 제공함으로써 mgo는 이러한 문서를 원하는 Go 구조체로 원활하게 디코딩할 수 있습니다.