Пример #1
0
        public QueryResult SearchVertexes(ContextManager contextMgmt, List <NLUEntity> entities)
        {
            QueryResult responseContent = new QueryResult(false, "Failed", ResponseItemType.Other);

            if (entities == null && entities.Count == 0)
            {
                responseContent = this.msgGenerator.GenerateErrorMessage("无法查找到对应节点,请确定输入的限定条件正确");
                contextMgmt.ExitDialog();
            }
            else
            {
                NLUEntity firstEntity = entities[0];

                List <Vertex> results = new List <Vertex>();

                if (firstEntity.GetEntityType() == "NodeName")
                {
                    Vertex vertex = dsFrame.SearchGraph(firstEntity.GetEntityValue(), contextMgmt.GetSecnarioName(), contextMgmt.GetSavedAttributes());

                    results.Add(vertex);
                }
                else if (firstEntity.GetEntityType() == "Label")
                {
                    results = dsFrame.SearchGraphByLabel(firstEntity.GetEntityValue(), contextMgmt.GetSecnarioName(), contextMgmt.GetSavedAttributes());
                }


                if (results == null || results.Count == 0)
                {
                    responseContent = this.msgGenerator.GenerateErrorMessage("无法查找到对应节点,请确定输入的限定条件正确");
                    contextMgmt.ExitDialog();
                }
                else
                {
                    string headMessage = "";

                    if (entities.Count > 1)
                    {
                        headMessage = firstEntity.GetEntityValue();

                        for (int i = 1; i < entities.Count; i++)
                        {
                            NLUEntity entity = entities[i];

                            if (entity.GetEntityType() == "RelationType")
                            {
                                HashSet <string> relationTypeSet = new HashSet <string>();
                                relationTypeSet.Add(entity.GetEntityValue());

                                List <Vertex> children = new List <Vertex>();

                                foreach (Vertex vertex in results)
                                {
                                    Dictionary <string, List <Vertex> > childrenDict = dsFrame.GetChildren(vertex, relationTypeSet, contextMgmt.GetSavedAttributes(), contextMgmt.GetSecnarioName());
                                    if (childrenDict != null)
                                    {
                                        List <Vertex> currentChildren = childrenDict[entity.GetEntityValue()];
                                        children.AddRange(currentChildren);
                                    }
                                }

                                if (children != null && children.Count > 0)
                                {
                                    headMessage += "的" + entity.GetEntityValue();

                                    results = children;
                                }
                            }
                            else if (entity.GetEntityType() == "Label")
                            {
                                List <Vertex> newResults = new List <Vertex>();
                                foreach (Vertex vertex in results)
                                {
                                    if (vertex.label == entity.GetEntityValue())
                                    {
                                        newResults.Add(vertex);
                                    }
                                }

                                if (newResults.Count > 0)
                                {
                                    headMessage += "的" + entity.GetEntityValue();
                                    results      = newResults;
                                }
                            }
                        }
                    }

                    if (results.Count == 1)
                    {
                        Vertex vertex = results[0];

                        if (vertex.isLeaf())
                        {
                            responseContent = this.msgGenerator.GenerateEndVertexMessage(vertex);
                            contextMgmt.ExitDialog();
                        }
                        else
                        {
                            List <DialogSlot> validSlots = GetValidSlots(contextMgmt);

                            if (validSlots == null || validSlots.Count() == 0)
                            {
                                contextMgmt.StartDialog();
                                responseContent = GetChildren(contextMgmt, vertex, null);
                            }
                            else
                            {
                                contextMgmt.SetSlots(validSlots);
                                contextMgmt.EnterSlotFilling();

                                responseContent = this.msgGenerator.GenerateSlotMessage(validSlots[0]);
                            }
                        }
                    }
                    else
                    {
                        contextMgmt.StartDialog();
                        responseContent = GetMessageForVertexes(contextMgmt, results);
                    }

                    if (!string.IsNullOrWhiteSpace(headMessage) && responseContent.success)
                    {
                        responseContent.responseMessage = headMessage + ":\n" + responseContent.responseMessage;
                    }
                }
            }

            return(responseContent);
        }