IPropertyTree *getPkgSetRegistry(const char *process, bool readonly)
{
Owned<IRemoteConnection> globalLock = querySDS().connect("/PackageSets/", myProcessSession(), RTM_LOCK_WRITE|RTM_CREATE_QUERY, SDS_LOCK_TIMEOUT);
if (!globalLock)
throw MakeStringException(PKG_DALI_LOOKUP_ERROR, "Unable to connect to PackageSet information in dali /PackageSets");
IPropertyTree *pkgSets = globalLock->queryRoot();
if (!pkgSets)
throw MakeStringException(PKG_DALI_LOOKUP_ERROR, "Unable to open PackageSet information in dali /PackageSets");
if (!process || !*process)
process = "*";
StringBuffer id;
buildPkgSetId(id, process);
//Only lock the branch for the target we're interested in.
VStringBuffer xpath("/PackageSets/PackageSet[@id='%s']", id.str());
Owned<IRemoteConnection> conn = querySDS().connect(xpath.str(), myProcessSession(), readonly ? RTM_LOCK_READ : RTM_LOCK_WRITE, SDS_LOCK_TIMEOUT);
if (!conn)
{
if (readonly)
return NULL;
Owned<IPropertyTree> pkgSet = createPTree();
pkgSet->setProp("@id", id.str());
pkgSet->setProp("@process", process);
pkgSets->addPropTree("PackageSet", pkgSet.getClear());
globalLock->commit();
conn.setown(querySDS().connect(xpath.str(), myProcessSession(), RTM_LOCK_WRITE, SDS_LOCK_TIMEOUT));
}
return (conn) ? conn->getRoot() : NULL;
}
bool CWsPackageProcessEx::onAddPackage(IEspContext &context, IEspAddPackageRequest &req, IEspAddPackageResponse &resp)
{
resp.updateStatus().setCode(0);
StringAttr target(req.getTarget());
StringAttr name(req.getPackageMap());
if (target.isEmpty())
throw MakeStringExceptionDirect(PKG_MISSING_PARAM, "Target cluster parameter required");
if (name.isEmpty())
throw MakeStringExceptionDirect(PKG_MISSING_PARAM, "PackageMap name parameter required");
DBGLOG("%s adding packagemap %s to target %s", context.queryUserId(), name.str(), target.str());
StringBuffer pmid;
if (!req.getGlobalScope())
pmid.append(target).append("::");
pmid.append(name.get());
bool activate = req.getActivate();
unsigned updateFlags = 0;
if (req.getOverWrite())
updateFlags |= (DALI_UPDATEF_PACKAGEMAP | DALI_UPDATEF_REPLACE_FILE | DALI_UPDATEF_CLONE_FROM | DALI_UPDATEF_SUPERFILES);
if (req.getReplacePackageMap())
updateFlags |= DALI_UPDATEF_PACKAGEMAP;
if (req.getUpdateCloneFrom())
updateFlags |= DALI_UPDATEF_CLONE_FROM;
if (req.getUpdateSuperFiles())
updateFlags |= DALI_UPDATEF_SUPERFILES;
if (req.getAppendCluster())
updateFlags |= DALI_UPDATEF_APPEND_CLUSTER;
StringAttr processName(req.getProcess());
Owned<IUserDescriptor> userdesc;
const char *user = context.queryUserId();
const char *password = context.queryPassword();
if (user && *user && *password && *password)
{
userdesc.setown(createUserDescriptor());
userdesc->set(user, password);
}
StringBuffer srcCluster;
StringBuffer daliip;
StringBuffer prefix;
splitDerivedDfsLocation(req.getDaliIp(), srcCluster, daliip, prefix, req.getSourceProcess(), req.getSourceProcess(), NULL, NULL);
StringBuffer pkgSetId;
buildPkgSetId(pkgSetId, processName.get());
StringArray filesNotFound;
addPackageMapInfo(updateFlags, req.getInfo(), filesNotFound, processName, target, pmid, pkgSetId, daliip, srcCluster, prefix, activate, userdesc, req.getAllowForeignFiles(), req.getPreloadAllPackages());
resp.setFilesNotFound(filesNotFound);
StringBuffer msg;
msg.append("Successfully loaded ").append(name.get());
resp.updateStatus().setDescription(msg.str());
return true;
}
bool deletePkgInfo(const char *name, const char *target, const char *process, bool globalScope)
{
Owned<IRemoteConnection> pkgSetsConn = querySDS().connect("/PackageSets/", myProcessSession(), RTM_LOCK_WRITE, SDS_LOCK_TIMEOUT);
if (!pkgSetsConn)
throw MakeStringException(PKG_NONE_DEFINED, "No package sets defined");
IPropertyTree* packageSets = pkgSetsConn->queryRoot();
StringBuffer pkgSetId;
buildPkgSetId(pkgSetId, process);
VStringBuffer pkgSet_xpath("PackageSet[@id='%s']", pkgSetId.str());
IPropertyTree *pkgSetRegistry = packageSets->queryPropTree(pkgSet_xpath.str());
if (!pkgSetRegistry)
throw MakeStringException(PKG_TARGET_NOT_DEFINED, "No package sets defined for %s", process);
StringBuffer lcTarget(target);
target = lcTarget.toLowerCase().str();
StringBuffer lcName(name);
name = lcName.toLowerCase().str();
Owned<IPropertyTree> mapEntry;
StringBuffer xpath;
if (!globalScope)
{
xpath.appendf("PackageMap[@id='%s::%s'][@querySet='%s']", target, name, target);
mapEntry.setown(pkgSetRegistry->getPropTree(xpath.str()));
}
if (!mapEntry)
{
xpath.clear().appendf("PackageMap[@id='%s'][@querySet='%s']", name, target);
mapEntry.setown(pkgSetRegistry->getPropTree(xpath.str()));
if (!mapEntry)
throw MakeStringException(PKG_DELETE_NOT_FOUND, "Unable to delete %s - information not found", lcName.str());
}
StringAttr pmid(mapEntry->queryProp("@id"));
pkgSetRegistry->removeTree(mapEntry);
xpath.clear().appendf("PackageSet/PackageMap[@id='%s']", pmid.get());
if (!packageSets->hasProp(xpath))
{
Owned<IRemoteConnection> pkgMapsConn = querySDS().connect("/PackageMaps/", myProcessSession(), RTM_LOCK_WRITE, SDS_LOCK_TIMEOUT);
if (!pkgMapsConn)
throw MakeStringException(PKG_DALI_LOOKUP_ERROR, "Unable to retrieve PackageMaps information from dali [/PackageMaps]");
IPropertyTree *pkgMaps = pkgMapsConn->queryRoot();
if (!pkgMaps)
throw MakeStringException(PKG_DALI_LOOKUP_ERROR, "Unable to retrieve PackageMaps information from dali [/PackageMaps]");
IPropertyTree *mapTree = pkgMaps->queryPropTree(xpath.clear().appendf("PackageMap[@id='%s']", pmid.get()).str());
if (mapTree)
pkgMaps->removeTree(mapTree);
}
return true;
}
bool CWsPackageProcessEx::onAddPackage(IEspContext &context, IEspAddPackageRequest &req, IEspAddPackageResponse &resp)
{
resp.updateStatus().setCode(0);
StringAttr target(req.getTarget());
StringAttr name(req.getPackageMap());
if (target.isEmpty())
throw MakeStringExceptionDirect(PKG_MISSING_PARAM, "Target cluster parameter required");
if (name.isEmpty())
throw MakeStringExceptionDirect(PKG_MISSING_PARAM, "PackageMap name parameter required");
StringBuffer pmid;
if (!req.getGlobalScope())
pmid.append(target).append("::");
pmid.append(name.get());
bool activate = req.getActivate();
bool overWrite = req.getOverWrite();
StringAttr processName(req.getProcess());
Owned<IUserDescriptor> userdesc;
const char *user = context.queryUserId();
const char *password = context.queryPassword();
if (user && *user && *password && *password)
{
userdesc.setown(createUserDescriptor());
userdesc->set(user, password);
}
StringBuffer srcCluster;
StringBuffer daliip;
StringBuffer prefix;
splitDerivedDfsLocation(req.getDaliIp(), srcCluster, daliip, prefix, req.getSourceProcess(), req.getSourceProcess(), NULL, NULL);
StringBuffer pkgSetId;
buildPkgSetId(pkgSetId, processName.get());
StringArray filesNotFound;
addPackageMapInfo(req.getInfo(), filesNotFound, processName, target, pmid, pkgSetId, daliip, srcCluster, prefix, activate, overWrite, userdesc, req.getAllowForeignFiles(), req.getPreloadAllPackages());
resp.setFilesNotFound(filesNotFound);
StringBuffer msg;
msg.append("Successfully loaded ").append(name.get());
resp.updateStatus().setDescription(msg.str());
return true;
}
void activatePackageMapInfo(const char *target, const char *name, const char *process, bool globalScope, bool activate)
{
if (!target || !*target)
throw MakeStringExceptionDirect(PKG_TARGET_NOT_DEFINED, "No target defined");
if (!name || !*name)
throw MakeStringExceptionDirect(PKG_MISSING_PARAM, "No pmid specified");
Owned<IRemoteConnection> globalLock = querySDS().connect("PackageSets", myProcessSession(), RTM_LOCK_WRITE|RTM_CREATE_QUERY, SDS_LOCK_TIMEOUT);
if (!globalLock)
throw MakeStringException(PKG_DALI_LOOKUP_ERROR, "Unable to retrieve PackageSets information from dali /PackageSets");
StringBuffer lcTarget(target);
target = lcTarget.toLowerCase().str();
StringBuffer lcName(name);
name = lcName.toLowerCase().str();
IPropertyTree *root = globalLock->queryRoot();
if (!root)
throw MakeStringException(PKG_ACTIVATE_NOT_FOUND, "Unable to retrieve PackageSet information");
StringBuffer pkgSetId;
buildPkgSetId(pkgSetId, process);
VStringBuffer xpath("PackageSet[@id='%s']", pkgSetId.str());
IPropertyTree *pkgSetTree = root->queryPropTree(xpath);
if (pkgSetTree)
{
IPropertyTree *mapTree = NULL;
if (!globalScope)
{
xpath.clear().appendf("PackageMap[@querySet='%s'][@id='%s::%s']", target, target, name);
mapTree = pkgSetTree->queryPropTree(xpath);
}
if (!mapTree)
{
xpath.clear().appendf("PackageMap[@querySet='%s'][@id='%s']", target, name);
mapTree = pkgSetTree->queryPropTree(xpath);
}
if (!mapTree)
throw MakeStringException(PKG_ACTIVATE_NOT_FOUND, "PackageMap %s not found on target %s", name, target);
makePackageActive(pkgSetTree, mapTree, target, activate);
}
}
void setProcess(const char *name)
{
process.set(name);
buildPkgSetId(pkgSetId, process);
}