package org.apache.athrift.service;
import java.util.ArrayList;
import java.util.HashMap;
import org.apache.athrift.compiler.Function;
import org.apache.athrift.compiler.Service;
public class ThriftService {
private String name;
private ThriftService extendService;
private HashMap<String, ThriftServiceItem> theFuncNameHMap = new HashMap<String, ThriftServiceItem>();
public ThriftService(Service theServiceParm,
ThriftServicesParser theThriftServicesParserParm) throws Exception
{
name = theServiceParm.getName();
String tmpExtendService = theServiceParm.getExtendService();
if (tmpExtendService != null)
{
ThriftService tmpExtThriftService = (ThriftService)theThriftServicesParserParm.getTheServiceHMap()
.get(tmpExtendService);
if (tmpExtThriftService == null)
{
tmpExtThriftService =
(ThriftService)ServiceCatalog.getInstance()
.getServiceHMap_WithFilePrefix().get(tmpExtendService);
}
if (tmpExtThriftService == null)
{
throw new Exception("û�ҵ���Ӧ��extends����");
}
else
{
extendService = tmpExtThriftService;
}
}
ArrayList<Function> tmpList = theServiceParm.getFunctionList();
for (int i = 0; i < tmpList.size(); i++)
{
Function tmpOneFunction = tmpList.get(i);
ThriftServiceItem tmpNewItem = new ThriftServiceItem(tmpOneFunction,
theThriftServicesParserParm);
theFuncNameHMap.put(tmpNewItem.getName(), tmpNewItem);
}
}
public String getName() {
return name;
}
public ThriftServiceItem getFuncItem(String funcNameParm)
{
ThriftServiceItem retItem = null;
retItem = theFuncNameHMap.get(funcNameParm);
if (retItem == null)
{
if (extendService != null)
{
retItem = extendService.getFuncItem(funcNameParm);
}
}
return retItem;
}
}