tianxiaoliang commented on a change in pull request #759:
URL: 
https://github.com/apache/servicecomb-service-center/pull/759#discussion_r533107261



##########
File path: datasource/mongo/dep.go
##########
@@ -21,20 +21,354 @@ import (
        "context"
 
        pb "github.com/go-chassis/cari/discovery"
+       "go.mongodb.org/mongo-driver/bson"
+
+       "errors"
+       "fmt"
+       "strings"
+
+       "github.com/apache/servicecomb-service-center/datasource"
+       "github.com/apache/servicecomb-service-center/datasource/mongo/client"
+       "github.com/apache/servicecomb-service-center/pkg/log"
+       "github.com/apache/servicecomb-service-center/pkg/util"
 )
 
 func (ds *DataSource) SearchProviderDependency(ctx context.Context, request 
*pb.GetDependenciesRequest) (*pb.GetProDependenciesResponse, error) {
-       return &pb.GetProDependenciesResponse{}, nil
+       providerServiceID := request.ServiceId
+       filter := GeneratorServiceFilter(ctx, providerServiceID)
+       provider, err := GetService(ctx, filter)
+       if err != nil {
+               log.Error("GetProviderDependencies failed, provider is 
"+providerServiceID, err)
+               return nil, err
+       }
+       if provider == nil {
+               log.Error(fmt.Sprintf("GetProviderDependencies failed for 
provider %s", providerServiceID), err)
+               return &pb.GetProDependenciesResponse{
+                       Response: pb.CreateResponse(pb.ErrServiceNotExists, 
"Provider does not exist"),
+               }, nil
+       }
+
+       services, err := GetDependencyProviders(ctx, provider.ServiceInfo, 
request)
+       if err != nil {
+               log.Error(fmt.Sprintf("GetProviderDependencies failed, provider 
is %s/%s/%s/%s",
+                       provider.ServiceInfo.Environment, 
provider.ServiceInfo.AppId, provider.ServiceInfo.ServiceName, 
provider.ServiceInfo.Version), err)
+               return &pb.GetProDependenciesResponse{
+                       Response: pb.CreateResponse(pb.ErrInternal, 
err.Error()),
+               }, err
+       }
+
+       return &pb.GetProDependenciesResponse{
+               Response:  pb.CreateResponse(pb.ResponseSuccess, "Get all 
consumers successful."),
+               Consumers: services,
+       }, nil
 }
 
 func (ds *DataSource) SearchConsumerDependency(ctx context.Context, request 
*pb.GetDependenciesRequest) (*pb.GetConDependenciesResponse, error) {
-       return &pb.GetConDependenciesResponse{}, nil
+       consumerID := request.ServiceId
+
+       filter := GeneratorServiceFilter(ctx, consumerID)
+       consumer, err := GetService(ctx, filter)
+       if err != nil {
+               log.Error(fmt.Sprintf("GetConsumerDependencies failed, consumer 
is %s", consumerID), err)
+               return nil, err
+       }
+       if consumer == nil {
+               log.Error(fmt.Sprintf("GetConsumerDependencies failed for 
consumer %s does not exist", consumerID), err)
+               return &pb.GetConDependenciesResponse{
+                       Response: pb.CreateResponse(pb.ErrServiceNotExists, 
"Consumer does not exist"),
+               }, nil
+       }
+
+       services, err := GetDependencyProviders(ctx, consumer.ServiceInfo, 
request)
+       if err != nil {
+               log.Error(fmt.Sprintf("GetConsumerDependencies failed, consumer 
is %s/%s/%s/%s",
+                       consumer.ServiceInfo.Environment, 
consumer.ServiceInfo.AppId, consumer.ServiceInfo.ServiceName, 
consumer.ServiceInfo.Version), err)
+               return &pb.GetConDependenciesResponse{
+                       Response: pb.CreateResponse(pb.ErrInternal, 
err.Error()),
+               }, err
+       }
+
+       return &pb.GetConDependenciesResponse{
+               Response:  pb.CreateResponse(pb.ResponseSuccess, "Get all 
providers successfully."),
+               Providers: services,
+       }, nil
 }
 
 func (ds *DataSource) AddOrUpdateDependencies(ctx context.Context, 
dependencyInfos []*pb.ConsumerDependency, override bool) (*pb.Response, error) {
-       return pb.CreateResponse(pb.ResponseSuccess, ""), nil
+       domainProject := util.ParseDomainProject(ctx)
+       for _, dependencyInfo := range dependencyInfos {
+               consumerFlag := util.StringJoin([]string{
+                       dependencyInfo.Consumer.Environment,
+                       dependencyInfo.Consumer.AppId,
+                       dependencyInfo.Consumer.ServiceName,
+                       dependencyInfo.Consumer.Version}, "/")
+               consumerInfo := 
pb.DependenciesToKeys([]*pb.MicroServiceKey{dependencyInfo.Consumer}, 
domainProject)[0]
+               providersInfo := 
pb.DependenciesToKeys(dependencyInfo.Providers, domainProject)
+
+               rsp := datasource.ParamsChecker(consumerInfo, providersInfo)
+               if rsp != nil {
+                       fmt.Print("put request into dependency queue failed, 
override: ", override, " consumer is",

Review comment:
       绝对不可以用fmt




----------------------------------------------------------------
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.

For queries about this service, please contact Infrastructure at:
[email protected]


Reply via email to