openbilibili/app/interface/main/app-view/dao/dm/dao.go

package dm

import (
	"context"
	"go-common/app/interface/main/app-view/conf"
	dm "go-common/app/interface/main/dm2/model"
	dmrpc "go-common/app/interface/main/dm2/rpc/client"
	"go-common/library/net/metadata"

	"github.com/pkg/errors"
)

type Dao struct {
	dmRPC *dmrpc.Service
}

func New(c *conf.Config) (d *Dao) {
	return &Dao{
		dmRPC: dmrpc.New(c.DMRPC),
	}
}

func (d *Dao) SubjectInfos(c context.Context, typ int32, plat int8, oids ...int64) (res map[int64]*dm.SubjectInfo, err error) {
	ip := metadata.String(c, metadata.RemoteIP)
	arg := &dm.ArgOids{Type: typ, Plat: plat, Oids: oids, RealIP: ip}
	if res, err = d.dmRPC.SubjectInfos(c, arg); err != nil {
		err = errors.Wrapf(err, "%v", arg)
	}
	return
}