Mgo(Go)의 모델로서의 인터페이스: 할당 가능성 오류 해결

다양한 유형의 노드로 구성된 워크플로를 처리할 때 다음과 같은 문제가 발생할 수 있습니다. Golang 인터페이스를 활용하여 다양한 동작을 캡슐화하고 싶은 유혹이 있습니다. 그러나 이러한 워크플로를 MongoDB에 저장하는 것은 독특한 과제를 안겨줍니다.

할당 가능성 문제

찾기 작업 중에 발생한 오류는 디코딩된 문서와 노드 인터페이스. Mgo에는 각 내장 노드의 특정 유형을 결정하는 데 필요한 유형 정보가 부족합니다. 이로 인해 할당 불일치가 발생합니다.

대체 접근 방식: 유형 정보 포함

이 장애물을 극복하려면 포함된 노드 내에 유형 정보를 캡슐화하는 것이 좋습니다. Node 인터페이스와 특정 유형을 나타내는 Type 필드로 구성된 새로운 구조체인 NodeWithType을 정의합니다.

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

사용자 정의 BSON 비마샬링

디코딩된 문서와 Go 구조체 사이의 간격을 메우려면 NodeWithType에 대한 SetBSON 함수를 구현하세요. 이 함수는 Type 필드를 디코딩하고, 해당 유형의 인스턴스를 생성하고, 이에 역마샬링을 위임합니다.

func (nt *NodeWithType) SetBSON(r bson.Raw) error {
   // Decode Type field
   var typeName string
   err := bson.Unmarshal(r["Type"], &typeName)
   if err != nil {
      return err
   }

   // Create an instance of the correct type based on Type field
   var node Node
   switch typeName {
   case "EmailNode":
      node = &EmailNode{}
   case "TwitterNode":
      node = &TwitterNode{}
   }

   // Unmarshal the remaining fields into the created instance
   if err = node.SetBSON(r[typeName]); err != nil {
      return err
   }

   // Assign the created instance to the Node field
   nt.Node = node

   return nil
}

Workflow 구조체에서 Node 조각을 NodeWithType 조각으로 바꿉니다.

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

이 접근 방식을 통해 Mgo는 이제 다양한 유형의 내장 노드가 있는 워크플로를 성공적으로 찾고 역마샬링할 수 있습니다.