/** * @param host * @param notify * @return */ private boolean remove(SiteHost host, boolean notify) { Logger.info("WorkPool.remove, work site %s", host); boolean success = false; this.lockSingle(); try { mapTime.remove(host); WorkSite site = mapSite.remove(host); if (site != null) { for (Naming naming : site.list()) { SiteSet set = mapNaming.get(naming); if (set != null) { set.remove(host); } if (set == null || set.isEmpty()) { mapNaming.remove(naming); } } success = true; if (notify) { CallPool.getInstance().refreshWorkSite(); } } } catch (Throwable exp) { Logger.error(exp); } finally { this.unlockSingle(); } return success; }
/** * get all worksite * * @return */ public WorkSite[] batch() { WorkSite[] sites = null; super.lockMulti(); try { int size = mapSite.size(); if (size > 0) { sites = new WorkSite[size]; mapSite.values().toArray(sites); } } catch (Throwable exp) { Logger.fatal(exp); } finally { super.unlockMulti(); } return sites; }
/** * refresh work site * * @param host * @return */ public short refresh(SiteHost host) { short code = Response.SERVER_ERROR; this.lockSingle(); try { WorkSite site = mapSite.get(host); if (site != null) { mapTime.put(host, System.currentTimeMillis()); code = Response.ISEE; } else { code = Response.NOTLOGIN; } } catch (Throwable exp) { Logger.error(exp); } finally { this.unlockSingle(); } Logger.debug("WorkPool.refresh, site %s refresh status %d", host, code); return code; }
/** check timeout site */ private void check() { int size = mapTime.size(); if (size == 0) return; ArrayList<SiteHost> dels = new ArrayList<SiteHost>(size); ArrayList<SiteHost> notifys = new ArrayList<SiteHost>(size); super.lockSingle(); try { long nowTime = System.currentTimeMillis(); for (SiteHost host : mapTime.keySet()) { Long value = mapTime.get(host); if (value == null) { dels.add(host); continue; } long time = value.longValue(); if (nowTime - time >= deleteTime) { dels.add(host); } else if (nowTime - time >= refreshTimeout) { notifys.add(host); } } } catch (Throwable exp) { exp.printStackTrace(); } finally { super.unlockSingle(); } // remove timeout site for (SiteHost host : dels) { Logger.error("WorkPool.check, delete timeout site:%s", host); remove(host); } // notify site SiteHost listen = Launcher.getInstance().getLocalHost(); for (SiteHost host : notifys) { Logger.warning("WorkPool.check, notify timeout site:%s", host); this.sendTimeout(host, listen, 2); } }
private boolean add(Site object, boolean notify) { if (object == null || !object.isWork()) { return false; } WorkSite site = (WorkSite) object; SiteHost host = site.getHost(); Logger.info("WorkPool.add, work site %s", host); boolean success = false; this.lockSingle(); try { if (mapSite.containsKey(host)) { return false; } // save site mapSite.put(host, site); mapTime.put(host, System.currentTimeMillis()); for (Naming naming : site.list()) { SiteSet set = mapNaming.get(naming); if (set == null) { set = new SiteSet(); mapNaming.put(naming, set); } set.add(host); } success = true; if (notify) { CallPool.getInstance().refreshWorkSite(); } } catch (Throwable exp) { Logger.error(exp); } finally { this.unlockSingle(); } return success; }
/** * find work site by naming * * @param naming * @return */ public SiteHost[] find(String naming) { Naming s = new Naming(naming); Logger.info("WorkPool.find, find work site by '%s'", s); ArrayList<SiteHost> a = new ArrayList<SiteHost>(); this.lockSingle(); try { SiteSet set = mapNaming.get(s); if (set != null) { a.addAll(set.list()); } } catch (Throwable exp) { Logger.error(exp); } finally { this.unlockSingle(); } if (a.isEmpty()) return null; SiteHost[] hosts = new SiteHost[a.size()]; return a.toArray(hosts); }
/** @return */ public List<SiteHost> gather() { ArrayList<SiteHost> array = new ArrayList<SiteHost>(); array.addAll(mapSite.keySet()); return array; }