List of usage examples for org.hibernate.criterion Restrictions eq
public static SimpleExpression eq(String propertyName, Object value)
From source file:au.edu.uts.eng.remotelabs.schedserver.reports.intf.Reports.java
License:Open Source License
@SuppressWarnings("unchecked") @Override/*from ww w . j a v a2s . c om*/ public QuerySessionAccessResponse querySessionAccess(final QuerySessionAccess querySessionAccess) { /* Request parameters. */ final QuerySessionAccessType request = querySessionAccess.getQuerySessionAccess(); final QueryFilterType filter = request.getQuerySelect(); String debug = "Received " + this.getClass().getSimpleName() + "#querySessionAccess with params: select operator=" + filter.getOperator() + ", type=" + filter.getTypeForQuery().toString() + ", like=" + filter.getQueryLike(); if (request.getStartTime() != null) debug += ", start=" + request.getStartTime().getTime(); if (request.getEndTime() != null) debug += ", end=" + request.getEndTime().getTime(); this.logger.debug(debug); /* Response parameters. */ final QuerySessionAccessResponse response = new QuerySessionAccessResponse(); final QuerySessionAccessResponseType respType = new QuerySessionAccessResponseType(); final PaginationType page = new PaginationType(); page.setNumberOfPages(1); page.setPageLength(0); page.setPageNumber(1); respType.setPagination(page); response.setQuerySessionAccessResponse(respType); final org.hibernate.Session db = DataAccessActivator.getNewSession(); try { RequestorType uid = request.getRequestor(); final User requestor = this.getUserFromUserID(uid, db); if (requestor == null) { this.logger.info("Unable to generate report because the user has not been found. Supplied " + "credentials ID=" + uid.getUserID() + ", namespace=" + uid.getUserNamespace() + ", " + "name=" + uid.getUserName() + '.'); return response; } /* We are only interested in sessions that are complete. */ Criteria query = db.createCriteria(Session.class).add(Restrictions.eq("active", Boolean.FALSE)) .add(Restrictions.isNotNull("removalTime")).addOrder(Order.asc("requestTime")); /* If the user has requested a time period, only add that to query. */ if (request.getStartTime() != null) query.add(Restrictions.ge("requestTime", request.getStartTime().getTime())); if (request.getEndTime() != null) query.add(Restrictions.le("requestTime", request.getEndTime().getTime())); if (request.getPagination() != null) { /* Add the requested pagination. */ final PaginationType pages = request.getPagination(); final int noPages = pages.getNumberOfPages(); final int pageNumber = pages.getPageNumber(); final int pageLength = pages.getPageLength(); if (noPages > 1) query.setMaxResults(pageLength); if (pageNumber > 1) query.setFirstResult((pageNumber - 1) * pageLength); } if (filter.getTypeForQuery() == TypeForQuery.RIG) { /* Only administrators can do rig queries. */ if (!User.ADMIN.equals(requestor.getPersona())) { this.logger.warn("Cannot provide session report for user '" + requestor.qName() + "' because their persona '" + requestor.getPersona() + "' does not allow rig reporting."); return response; } query.add(Restrictions.eq("assignedRigName", filter.getQueryLike())); } else if (filter.getTypeForQuery() == TypeForQuery.RIG_TYPE) { /* Only administrators can do rig type queries. */ if (!User.ADMIN.equals(requestor.getPersona())) { this.logger.warn("Cannot provide session report for user '" + requestor.qName() + "' because their persona '" + requestor.getPersona() + "' does not allow rig type reporting."); return response; } final RigType rigType = new RigTypeDao(db).findByName(filter.getQueryLike()); if (rigType == null) { this.logger.warn("Cannot provide session report because rig type '" + filter.getQueryLike() + "' not found."); return response; } query.createCriteria("rig").add(Restrictions.eq("rigType", rigType)); } else if (filter.getTypeForQuery() == TypeForQuery.REQUEST_CAPABILITIES) { /* Only administrators can do request capabilities queries. */ if (!User.ADMIN.equals(requestor.getPersona())) { this.logger.warn("Cannot provide session report for user '" + requestor.qName() + "' because their persona '" + requestor.getPersona() + "' does not allow request capabilities reporting."); return response; } final RequestCapabilities reqCaps = new RequestCapabilitiesDao(db) .findCapabilites(filter.getQueryLike()); if (reqCaps == null) { this.logger.warn("Cannot provide session report because request capabilities '" + filter.getQueryLike() + "' not found."); return response; } List<Rig> capRigs = new ArrayList<Rig>(); for (MatchingCapabilities match : reqCaps.getMatchingCapabilitieses()) { capRigs.addAll(match.getRigCapabilities().getRigs()); } if (capRigs.size() == 0) { this.logger.warn( "Cannot provide session report because there are no rigs with request capabilities '" + reqCaps.getCapabilities() + "'."); return response; } query.add(Restrictions.in("rig", capRigs)); } else if (filter.getTypeForQuery() == TypeForQuery.USER_CLASS) { final UserClass userClass = new UserClassDao(db).findByName(filter.getQueryLike()); if (userClass == null) { this.logger.warn("Cannot provide session report because user class '" + filter.getQueryLike() + "' was not found."); return response; } if (User.ACADEMIC.equals(requestor.getPersona())) { /* An academic may only generate reports for the classes * they have have reporting permission in. */ boolean hasPerm = false; final Iterator<AcademicPermission> it = requestor.getAcademicPermissions().iterator(); while (it.hasNext()) { final AcademicPermission ap = it.next(); if (ap.getUserClass().getId().equals(userClass.getId()) && ap.isCanGenerateReports()) { hasPerm = true; break; } } if (!hasPerm) { this.logger.info("Unable to generate report for user class " + userClass.getName() + " because the user " + requestor.qName() + " does not own or have permission to report it."); return response; } this.logger.debug("Academic " + requestor.qName() + " has permission to generate report from user class " + userClass.getName() + '.'); } else if (!User.ADMIN.equals(requestor.getPersona())) { this.logger.warn("Cannot provide a user session report for user '" + requestor.qName() + "' because their persona '" + requestor.getPersona() + "' does not allow reporting."); return response; } query.createCriteria("resourcePermission").add(Restrictions.eq("userClass", userClass)); } else if (filter.getTypeForQuery() == TypeForQuery.USER) { final User user = new UserDao(db).findByQName(filter.getQueryLike()); if (user == null) { this.logger.warn("Cannot provide session report because user '" + filter.getQueryLike() + "' was not found."); return response; } if (User.ACADEMIC.equals(requestor.getPersona())) { /* The report can only contain sessions originating from the user * classes the academic has reporting permission in. */ List<ResourcePermission> perms = new ArrayList<ResourcePermission>(); final Iterator<AcademicPermission> it = requestor.getAcademicPermissions().iterator(); while (it.hasNext()) { final AcademicPermission ap = it.next(); if (!ap.isCanGenerateReports()) continue; perms.addAll(ap.getUserClass().getResourcePermissions()); } if (perms.size() == 0) { this.logger.info("Unable to generate report for user " + user.qName() + " because the user " + requestor.qName() + " does not own or have permission to report " + "on any of the users permissions."); return response; } query.add(Restrictions.in("resourcePermission", perms)); } else if (!User.ADMIN.equals(requestor.getPersona())) { this.logger.warn("Cannot provide a user session report for user '" + requestor.qName() + "' because their persona '" + requestor.getPersona() + "' does not allow reporting."); return response; } query.add(Restrictions.eq("user", user)); } else { this.logger.error("Cannot provide a session report because the query type '" + filter.getTypeForQuery().toString() + "' was not understood."); return response; } for (final Session s : (List<Session>) query.list()) { final AccessReportType report = new AccessReportType(); /* Set user who was in session. */ final RequestorType sUser = new RequestorType(); final UserNSNameSequence nsSequence = new UserNSNameSequence(); nsSequence.setUserName(s.getUserName()); nsSequence.setUserNamespace(s.getUserNamespace()); sUser.setRequestorNSName(nsSequence); sUser.setUserQName(s.getUserNamespace() + ':' + s.getUserName()); report.setUser(sUser); /* User class. */ if (s.getResourcePermission() != null) report.setUserClass(s.getResourcePermission().getUserClass().getName()); /* Rig details. */ report.setRigName(s.getAssignedRigName()); if (s.getRig() != null) report.setRigType(s.getRig().getRigType().getName()); /* Session start. */ Calendar cal = Calendar.getInstance(); cal.setTime(s.getRequestTime()); report.setQueueStartTime(cal); /* Session timings. */ if (s.getAssignmentTime() != null) { final int queueD = (int) ((s.getAssignmentTime().getTime() - s.getRequestTime().getTime()) / 1000); report.setQueueDuration(queueD); cal = Calendar.getInstance(); cal.setTime(s.getAssignmentTime()); report.setSessionStartTime(cal); final int sessionD = (int) ((s.getRemovalTime().getTime() - s.getAssignmentTime().getTime()) / 1000); report.setSessionDuration(sessionD); } else { final int queueD = (int) ((s.getRemovalTime().getTime() - s.getRequestTime().getTime()) / 1000); report.setQueueDuration(queueD); cal = Calendar.getInstance(); cal.setTime(s.getRemovalTime()); report.setSessionStartTime(cal); report.setSessionDuration(0); } /* Session end. */ cal = Calendar.getInstance(); cal.setTime(s.getRemovalTime()); report.setSessionEndTime(cal); report.setReasonForTermination(s.getRemovalReason()); respType.addAccessReportData(report); } } finally { db.close(); } return response; }
From source file:au.edu.uts.eng.remotelabs.schedserver.reports.intf.Reports.java
License:Open Source License
@SuppressWarnings("unchecked") @Override/* w w w . j a v a2 s . c o m*/ public QuerySessionReportResponse querySessionReport(final QuerySessionReport querySessionReport) { /** Request parameters. **/ final QuerySessionReportType qSRReq = querySessionReport.getQuerySessionReport(); String debug = "Received " + this.getClass().getSimpleName() + "#querySessionReport with params:"; debug += "Requestor: " + qSRReq.getRequestor() + ", QuerySelect: " + qSRReq.getQuerySelect().toString(); if (qSRReq.getQueryConstraints() != null) { debug += ", QueryConstraints: " + qSRReq.getQueryConstraints().toString(); //DODGY only first displayed } debug += ", start time: " + qSRReq.getStartTime() + ", end time: " + qSRReq.getEndTime() + ", pagination: " + qSRReq.getPagination(); this.logger.debug(debug); final RequestorType uid = qSRReq.getRequestor(); /** Response parameters. */ final QuerySessionReportResponse resp = new QuerySessionReportResponse(); final QuerySessionReportResponseType respType = new QuerySessionReportResponseType(); final PaginationType page = new PaginationType(); page.setNumberOfPages(1); page.setPageLength(0); page.setPageNumber(1); respType.setPagination(page); respType.setSessionCount(0); respType.setTotalQueueDuration(0); respType.setTotalSessionDuration(0); resp.setQuerySessionReportResponse(respType); final org.hibernate.Session ses = DataAccessActivator.getNewSession(); /* Set up query from request parameters*/ try { /* ---------------------------------------------------------------- * -- Load the requestor. -- * ---------------------------------------------------------------- */ final User user = this.getUserFromUserID(uid, ses); if (user == null) { this.logger.info("Unable to generate report because the user has not been found. Supplied " + "credentials ID=" + uid.getUserID() + ", namespace=" + uid.getUserNamespace() + ", " + "name=" + uid.getUserName() + '.'); return resp; } final String persona = user.getPersona(); final Criteria cri = ses.createCriteria(Session.class); // Order by request time cri.addOrder(Order.asc("requestTime")); /* Add restriction - removal time cannot be null - these are in session records */ cri.add(Restrictions.isNotNull("removalTime")); /* First Query Filter - this contains grouping for the query */ final QueryFilterType query0 = qSRReq.getQuerySelect(); /* ---------------------------------------------------------------- * -- Get the Query type and process accordingly -- * -- 1. Rig -- * -- 2. Rig Type -- * -- 3. User Class -- * -- 4. User -- * -- 5. Capabilities (not yet implemented) -- * ---------------------------------------------------------------- */ if (query0.getTypeForQuery() == TypeForQuery.RIG) { /* ---------------------------------------------------------------- * -- 1. Rig Information only available to ADMIN, to be mediated -- * -- by interface. No criteria on Requestor -- * ---------------------------------------------------------------- */ /* ---------------------------------------------------------------- * -- 1a. Get Sessions that match the rig -- * ---------------------------------------------------------------- */ cri.add(Restrictions.eq("assignedRigName", query0.getQueryLike())); // Select time period if (qSRReq.getStartTime() != null) { cri.add(Restrictions.ge("requestTime", qSRReq.getStartTime().getTime())); } if (qSRReq.getEndTime() != null) { cri.add(Restrictions.le("requestTime", qSRReq.getEndTime().getTime())); } Long idCount = new Long(-1); final SortedMap<User, SessionStatistics> recordMap = new TreeMap<User, SessionStatistics>( new UserComparator()); /* Contains list of users without user_id in session. Indexed with namespace:name */ final Map<String, SessionStatistics> userMap = new HashMap<String, SessionStatistics>(); final List<Session> list = cri.list(); for (final Session o : list) { /* ---------------------------------------------------------------- * -- 1b. Iterate through sessions and create user records -- * ---------------------------------------------------------------- */ /* Create check for user_id not being null - deleted user */ if (o.getUser() != null) { final User u = o.getUser(); if (recordMap.containsKey(u)) { recordMap.get(u).addRecord(o); } else { final SessionStatistics userRec = new SessionStatistics(); userRec.addRecord(o); recordMap.put(u, userRec); } } else { final String nameIndex = o.getUserNamespace() + QNAME_DELIM + o.getUserName(); /* Does 'namespace:name' indexed list contain user */ if (userMap.containsKey(nameIndex)) { userMap.get(nameIndex).addRecord(o); } else { final User u = new User(); u.setName(o.getUserName()); u.setNamespace(o.getUserNamespace()); u.setId(idCount); idCount--; final SessionStatistics userRec = new SessionStatistics(); userRec.addRecord(o); recordMap.put(u, userRec); userMap.put(nameIndex, userRec); } } } /* ---------------------------------------------------------------- * -- 1c. Get pagination requirements so correct records are -- * -- returned. -- * ---------------------------------------------------------------- */ int pageNumber = 1; int pageLength = recordMap.size(); int recordCount = 0; int totalSessionCount = 0; int totalSessionDuration = 0; int totalQueueDuration = 0; if (qSRReq.getPagination() != null) { final PaginationType pages = qSRReq.getPagination(); pageNumber = pages.getPageNumber(); pageLength = pages.getPageLength(); respType.setPagination(page); } /* ---------------------------------------------------------------- * -- 1d. Iterate through user records and calculate report data -- * ---------------------------------------------------------------- */ for (final Entry<User, SessionStatistics> e : recordMap.entrySet()) { recordCount++; totalSessionCount += e.getValue().getSessionCount(); totalQueueDuration += e.getValue().getTotalQueueDuration(); totalSessionDuration += e.getValue().getTotalSessionDuration(); if ((recordCount > (pageNumber - 1) * pageLength) && (recordCount <= pageNumber * pageLength)) { final SessionReportType reportType = new SessionReportType(); //Get user from session object final RequestorType user0 = new RequestorType(); final UserNSNameSequence nsSequence = new UserNSNameSequence(); nsSequence.setUserName(e.getKey().getName()); nsSequence.setUserNamespace(e.getKey().getNamespace()); user0.setRequestorNSName(nsSequence); reportType.setUser(user0); reportType.setRigName(query0.getQueryLike()); reportType.setAveQueueDuration(e.getValue().getAverageQueueDuration()); reportType.setMedQueueDuration(e.getValue().getMedianQueueDuration()); reportType.setMinQueueDuration(e.getValue().getMinimumQueueDuration()); reportType.setMaxQueueDuration(e.getValue().getMaximumQueueDuration()); reportType.setTotalQueueDuration(e.getValue().getTotalQueueDuration()); reportType.setAveSessionDuration(e.getValue().getAverageSessionDuration()); reportType.setMedSessionDuration(e.getValue().getMedianSessionDuration()); reportType.setMinSessionDuration(e.getValue().getMinimumSessionDuration()); reportType.setMaxSessionDuration(e.getValue().getMaximumSessionDuration()); reportType.setTotalSessionDuration(e.getValue().getTotalSessionDuration()); reportType.setSessionCount(e.getValue().getSessionCount()); respType.addSessionReport(reportType); } } respType.setSessionCount(totalSessionCount); respType.setTotalQueueDuration(totalQueueDuration); respType.setTotalSessionDuration(totalSessionDuration); resp.setQuerySessionReportResponse(respType); } else if (query0.getTypeForQuery() == TypeForQuery.RIG_TYPE) { /* ---------------------------------------------------------------- * -- 2. Rig Type Information only available to ADMIN, to be -- * -- mediated by interface. No criteria on Requestor -- * ---------------------------------------------------------------- */ final RigTypeDao rTypeDAO = new RigTypeDao(ses); final RigType rType = rTypeDAO.findByName(query0.getQueryLike()); if (rType == null) { this.logger.warn("No valid rig type found - " + query0.getTypeForQuery().toString()); return resp; } /* ---------------------------------------------------------------- * -- 2a. Get Sessions that match the rig type -- * ---------------------------------------------------------------- */ // Select Sessions where rig value is of the correct Rig Type cri.createCriteria("rig").add(Restrictions.eq("rigType", rType)); // Select time period if (qSRReq.getStartTime() != null) { cri.add(Restrictions.ge("requestTime", qSRReq.getStartTime().getTime())); } if (qSRReq.getEndTime() != null) { cri.add(Restrictions.le("requestTime", qSRReq.getEndTime().getTime())); } Long idCount = new Long(-1); final SortedMap<User, SessionStatistics> recordMap = new TreeMap<User, SessionStatistics>( new UserComparator()); /* Contains list of users without user_id in session. Indexed with namespace:name */ final Map<String, SessionStatistics> userMap = new HashMap<String, SessionStatistics>(); final List<Session> list = cri.list(); for (final Session o : list) { /* ---------------------------------------------------------------- * -- 2b. Iterate through sessions and create user records -- * ---------------------------------------------------------------- */ /* Create check for user_id not being null - deleted user */ if (o.getUser() != null) { final User u = o.getUser(); if (recordMap.containsKey(u)) { recordMap.get(u).addRecord(o); } else { final SessionStatistics userRec = new SessionStatistics(); userRec.addRecord(o); recordMap.put(u, userRec); } } else { final String nameIndex = o.getUserNamespace() + QNAME_DELIM + o.getUserName(); /* Does 'namespace:name' indexed list contain user */ if (userMap.containsKey(nameIndex)) { userMap.get(nameIndex).addRecord(o); } else { final User u = new User(); u.setName(o.getUserName()); u.setNamespace(o.getUserNamespace()); u.setId(idCount); idCount--; final SessionStatistics userRec = new SessionStatistics(); userRec.addRecord(o); recordMap.put(u, userRec); userMap.put(nameIndex, userRec); } } } /* ---------------------------------------------------------------- * -- 2c. Get pagination requirements so correct records are -- * -- returned. -- * ---------------------------------------------------------------- */ int pageNumber = 1; int pageLength = recordMap.size(); int recordCount = 0; int totalSessionCount = 0; int totalSessionDuration = 0; int totalQueueDuration = 0; if (qSRReq.getPagination() != null) { final PaginationType pages = qSRReq.getPagination(); pageNumber = pages.getPageNumber(); pageLength = pages.getPageLength(); } /* ---------------------------------------------------------------- * -- 2d. Iterate through user records and calculate report data -- * ---------------------------------------------------------------- */ for (final Entry<User, SessionStatistics> e : recordMap.entrySet()) { recordCount++; totalSessionCount += e.getValue().getSessionCount(); totalQueueDuration += e.getValue().getTotalQueueDuration(); totalSessionDuration += e.getValue().getTotalSessionDuration(); if ((recordCount > (pageNumber - 1) * pageLength) && (recordCount <= pageNumber * pageLength)) { final SessionReportType reportType = new SessionReportType(); //Get user from session object final RequestorType user0 = new RequestorType(); final UserNSNameSequence nsSequence = new UserNSNameSequence(); nsSequence.setUserName(e.getKey().getName()); nsSequence.setUserNamespace(e.getKey().getNamespace()); user0.setRequestorNSName(nsSequence); reportType.setUser(user0); reportType.setRigType(rType.getName()); reportType.setAveQueueDuration(e.getValue().getAverageQueueDuration()); reportType.setMedQueueDuration(e.getValue().getMedianQueueDuration()); reportType.setMinQueueDuration(e.getValue().getMinimumQueueDuration()); reportType.setMaxQueueDuration(e.getValue().getMaximumQueueDuration()); reportType.setTotalQueueDuration(e.getValue().getTotalQueueDuration()); reportType.setAveSessionDuration(e.getValue().getAverageSessionDuration()); reportType.setMedSessionDuration(e.getValue().getMedianSessionDuration()); reportType.setMinSessionDuration(e.getValue().getMinimumSessionDuration()); reportType.setMaxSessionDuration(e.getValue().getMaximumSessionDuration()); reportType.setTotalSessionDuration(e.getValue().getTotalSessionDuration()); reportType.setSessionCount(e.getValue().getSessionCount()); respType.addSessionReport(reportType); } } respType.setSessionCount(totalSessionCount); respType.setTotalQueueDuration(totalQueueDuration); respType.setTotalSessionDuration(totalSessionDuration); resp.setQuerySessionReportResponse(respType); } else if (query0.getTypeForQuery() == TypeForQuery.USER_CLASS) { /* ---------------------------------------------------------------- * -- 3. User Class Information only available to ADMIN and -- * -- ACADEMIC users with permissions to generate reports -- * -- for this class. -- * ---------------------------------------------------------------- */ final UserClassDao uClassDAO = new UserClassDao(ses); final UserClass uClass = uClassDAO.findByName(query0.getQueryLike()); if (uClass == null) { this.logger.warn("No valid user class found - " + query0.getTypeForQuery().toString()); return resp; } /* ---------------------------------------------------------------- * Check that the requestor has permissions to request the report. * If persona = USER, no reports (USERs will not get here) * If persona = ADMIN, any report * If persona = ACADEMIC, only for classes they own if they can genrate reports * ---------------------------------------------------------------- */ if (User.ACADEMIC.equals(persona)) { /* An academic may generate reports for their own classes only. */ boolean hasPerm = false; final Iterator<AcademicPermission> apIt = user.getAcademicPermissions().iterator(); while (apIt.hasNext()) { final AcademicPermission ap = apIt.next(); if (ap.getUserClass().getId().equals(uClass.getId()) && ap.isCanGenerateReports()) { hasPerm = true; break; } } if (!hasPerm) { this.logger.info("Unable to generate report for user class " + uClass.getName() + " because the user " + user.getNamespace() + ':' + user.getName() + " does not own or have academic permission to it."); return resp; } this.logger.debug("Academic " + user.getNamespace() + ':' + user.getName() + " has permission to " + "generate report from user class" + uClass.getName() + '.'); } /* ---------------------------------------------------------------- * -- 3a. Get Sessions that match the user class -- * ---------------------------------------------------------------- */ //Select sessions where the resource permission associated is for this user class cri.createCriteria("resourcePermission").add(Restrictions.eq("userClass", uClass)); // Select time period if (qSRReq.getStartTime() != null) { cri.add(Restrictions.ge("requestTime", qSRReq.getStartTime().getTime())); } if (qSRReq.getEndTime() != null) { cri.add(Restrictions.le("requestTime", qSRReq.getEndTime().getTime())); } //Query Filter to be used for multiple selections in later versions of reporting. //QueryFilterType queryFilters[] = qSAReq.getQueryConstraints(); final SortedMap<User, SessionStatistics> recordMap = new TreeMap<User, SessionStatistics>( new UserComparator()); final List<Session> list = cri.list(); for (final Session o : list) { /* ---------------------------------------------------------------- * -- 3b. Iterate through sessions and create user records -- * ---------------------------------------------------------------- */ /* Create check for user_id not being null - deleted user */ if (o.getUser() != null) { final User u = o.getUser(); if (recordMap.containsKey(u)) { recordMap.get(u).addRecord(o); } else { final SessionStatistics userRec = new SessionStatistics(); userRec.addRecord(o); recordMap.put(u, userRec); } } } /* ---------------------------------------------------------------- * -- 3c. Get pagination requirements so correct records are -- * -- returned. -- * ---------------------------------------------------------------- */ int pageNumber = 1; int pageLength = recordMap.size(); int recordCount = 0; int totalSessionCount = 0; int totalSessionDuration = 0; int totalQueueDuration = 0; if (qSRReq.getPagination() != null) { final PaginationType pages = qSRReq.getPagination(); pageNumber = pages.getPageNumber(); pageLength = pages.getPageLength(); } /* ---------------------------------------------------------------- * -- 3d. Iterate through user records and calculate report data -- * ---------------------------------------------------------------- */ for (final Entry<User, SessionStatistics> e : recordMap.entrySet()) { recordCount++; totalSessionCount += e.getValue().getSessionCount(); totalQueueDuration += e.getValue().getTotalQueueDuration(); totalSessionDuration += e.getValue().getTotalSessionDuration(); if ((recordCount > (pageNumber - 1) * pageLength) && (recordCount <= pageNumber * pageLength)) { final SessionReportType reportType = new SessionReportType(); //Get user from session object final RequestorType user0 = new RequestorType(); final UserNSNameSequence nsSequence = new UserNSNameSequence(); nsSequence.setUserName(e.getKey().getName()); nsSequence.setUserNamespace(e.getKey().getNamespace()); user0.setRequestorNSName(nsSequence); reportType.setUser(user0); reportType.setUserClass(uClass.getName()); reportType.setAveQueueDuration(e.getValue().getAverageQueueDuration()); reportType.setMedQueueDuration(e.getValue().getMedianQueueDuration()); reportType.setMinQueueDuration(e.getValue().getMinimumQueueDuration()); reportType.setMaxQueueDuration(e.getValue().getMaximumQueueDuration()); reportType.setTotalQueueDuration(e.getValue().getTotalQueueDuration()); reportType.setAveSessionDuration(e.getValue().getAverageSessionDuration()); reportType.setMedSessionDuration(e.getValue().getMedianSessionDuration()); reportType.setMinSessionDuration(e.getValue().getMinimumSessionDuration()); reportType.setMaxSessionDuration(e.getValue().getMaximumSessionDuration()); reportType.setTotalSessionDuration(e.getValue().getTotalSessionDuration()); reportType.setSessionCount(e.getValue().getSessionCount()); respType.addSessionReport(reportType); } } respType.setSessionCount(totalSessionCount); respType.setTotalQueueDuration(totalQueueDuration); respType.setTotalSessionDuration(totalSessionDuration); resp.setQuerySessionReportResponse(respType); } else if (query0.getTypeForQuery() == TypeForQuery.USER) { /* ---------------------------------------------------------------- * -- 4. User Information only available to ADMIN and -- * -- ACADEMIC users with permissions to generate reports -- * -- for this classes to which this user belongs. -- * -- NOTE: User nam expected as: namespace:username -- * ---------------------------------------------------------------- */ final UserDao userDAO = new UserDao(ses); final String idParts[] = query0.getQueryLike().split(Reports.QNAME_DELIM, 2); final String ns = idParts[0]; final String name = (idParts.length > 1) ? idParts[1] : idParts[0]; final User user0 = userDAO.findByName(ns, name); if (user0 == null) { this.logger.warn("No valid user found - " + query0.getTypeForQuery().toString()); return resp; } /* ---------------------------------------------------------------- * Check that the requestor has permissions to request the report. * If persona = USER, no reports (USERs will not get here) * If persona = ADMIN, any report * If persona = ACADEMIC, only for users belonging to * classes they can generate reports for * ---------------------------------------------------------------- */ if (User.ACADEMIC.equals(persona)) { /* An academic may generate reports for their own classes only. */ boolean hasPerm = false; final Iterator<AcademicPermission> apIt = user.getAcademicPermissions().iterator(); while (apIt.hasNext()) { final AcademicPermission ap = apIt.next(); final Iterator<UserAssociation> uaIt = user0.getUserAssociations().iterator(); while (uaIt.hasNext()) { final UserAssociation ua = uaIt.next(); if (ap.getUserClass().getId().equals(ua.getUserClass().getId()) && ap.isCanGenerateReports()) { hasPerm = true; break; } } if (hasPerm) { break; } } if (!hasPerm) { this.logger.info("Unable to generate report for user class " + user0.getName() + " because the user " + user.getNamespace() + ':' + user.getName() + " does not own or have academic permission to it."); return resp; } this.logger.debug("Academic " + user.getNamespace() + ':' + user.getName() + " has permission to " + "generate report from user class" + user0.getName() + '.'); } /* ---------------------------------------------------------------- * -- 4a. Get Sessions that match the user -- * ---------------------------------------------------------------- */ //Select sessions where the resource permission associated is for this user class cri.add(Restrictions.eq("user", user0)); // Select time period if (qSRReq.getStartTime() != null) { cri.add(Restrictions.ge("requestTime", qSRReq.getStartTime().getTime())); } if (qSRReq.getEndTime() != null) { cri.add(Restrictions.le("requestTime", qSRReq.getEndTime().getTime())); } //Query Filter to be used for multiple selections in later versions of reporting. //QueryFilterType queryFilters[] = qSAReq.getQueryConstraints(); final SortedMap<User, SessionStatistics> recordMap = new TreeMap<User, SessionStatistics>( new UserComparator()); final List<Session> list = cri.list(); for (final Session o : list) { /* ---------------------------------------------------------------- * -- 4b. Iterate through sessions and create user records -- * ---------------------------------------------------------------- */ final User u = o.getUser(); if (recordMap.containsKey(u)) { recordMap.get(u).addRecord(o); } else { final SessionStatistics userRec = new SessionStatistics(); userRec.addRecord(o); recordMap.put(u, userRec); } } /* ---------------------------------------------------------------- * -- 4c. Get pagination requirements so correct records are -- * -- returned. -- * ---------------------------------------------------------------- */ int pageNumber = 1; int pageLength = recordMap.size(); int recordCount = 0; int totalSessionCount = 0; int totalSessionDuration = 0; int totalQueueDuration = 0; if (qSRReq.getPagination() != null) { final PaginationType pages = qSRReq.getPagination(); pageNumber = pages.getPageNumber(); pageLength = pages.getPageLength(); } /* ---------------------------------------------------------------- * -- 4d. Iterate through user records and calculate report data -- * ---------------------------------------------------------------- */ for (final Entry<User, SessionStatistics> e : recordMap.entrySet()) { recordCount++; totalSessionCount += e.getValue().getSessionCount(); totalQueueDuration += e.getValue().getTotalQueueDuration(); totalSessionDuration += e.getValue().getTotalSessionDuration(); if ((recordCount > (pageNumber - 1) * pageLength) && (recordCount <= pageNumber * pageLength)) { final SessionReportType reportType = new SessionReportType(); //Get user from session object final RequestorType user1 = new RequestorType(); final UserNSNameSequence nsSequence = new UserNSNameSequence(); nsSequence.setUserName(e.getKey().getName()); nsSequence.setUserNamespace(e.getKey().getNamespace()); user1.setRequestorNSName(nsSequence); reportType.setUser(user1); reportType.setUser(user1); reportType.setAveQueueDuration(e.getValue().getAverageQueueDuration()); reportType.setMedQueueDuration(e.getValue().getMedianQueueDuration()); reportType.setMinQueueDuration(e.getValue().getMinimumQueueDuration()); reportType.setMaxQueueDuration(e.getValue().getMaximumQueueDuration()); reportType.setTotalQueueDuration(e.getValue().getTotalQueueDuration()); reportType.setAveSessionDuration(e.getValue().getAverageSessionDuration()); reportType.setMedSessionDuration(e.getValue().getMedianSessionDuration()); reportType.setMinSessionDuration(e.getValue().getMinimumSessionDuration()); reportType.setMaxSessionDuration(e.getValue().getMaximumSessionDuration()); reportType.setTotalSessionDuration(e.getValue().getTotalSessionDuration()); reportType.setSessionCount(e.getValue().getSessionCount()); respType.addSessionReport(reportType); } } respType.setSessionCount(totalSessionCount); respType.setTotalQueueDuration(totalQueueDuration); respType.setTotalSessionDuration(totalSessionDuration); resp.setQuerySessionReportResponse(respType); } } finally { ses.close(); } return resp; }
From source file:au.edu.uts.eng.remotelabs.schedserver.rigmanagement.impl.RigMaintenanceNotifier.java
License:Open Source License
@SuppressWarnings("unchecked") @Override//from w w w. j a v a2 s.c o m public void run() { org.hibernate.Session db = DataAccessActivator.getNewSession(); if (db == null) { this.logger .error("Unable to run rig maintenance notification service because unable to obtain a database " + "session."); return; } Date ps = new Date(); Date pe = new Date(System.currentTimeMillis() + RUN_PERIOD * 1000); try { /* ---------------------------------------------------------------- * -- For maintenance periods that are starting, terminate the -- * -- rig's session and set the rig to maintenance mode. -- * ---------------------------------------------------------------- */ Criteria qu = db.createCriteria(RigOfflineSchedule.class).add(Restrictions.eq("active", Boolean.TRUE)) .add(Restrictions.ge("startTime", ps)).add(Restrictions.lt("startTime", pe)) .createCriteria("rig").add(Restrictions.eq("active", Boolean.TRUE)) .add(Restrictions.eq("online", Boolean.TRUE)); for (RigOfflineSchedule sched : (List<RigOfflineSchedule>) qu.list()) { Rig rig = sched.getRig(); this.logger.info("Going to notify rig " + rig.getName() + " to go into maintenance state at time " + ps + '.'); /* Check the rig isn't in session, if it is terminate the session. */ if (sched.getRig().isInSession()) { Session ses = rig.getSession(); this.logger.info("Need to kick off user " + ses.getUser().qName() + " from rig " + rig.getName() + " because rig is going into maintenance."); ses.setActive(false); ses.setRemovalTime(pe); ses.setRemovalReason("Rig going into maintenance."); if (this.notTest) new RigReleaser().release(ses, db); } rig.setOnline(false); rig.setOfflineReason("Rig going into maintenance."); new RigLogDao(db).addOfflineLog(rig, "Maintenance period starting."); db.beginTransaction(); db.flush(); db.getTransaction().commit(); if (this.notTest) new RigMaintenance().putMaintenance(rig, true, db); } /* ---------------------------------------------------------------- * -- For maintenance periods that are ending, notify the rig to -- * -- end maintenance mode. -- * ---------------------------------------------------------------- */ qu = db.createCriteria(RigOfflineSchedule.class).add(Restrictions.eq("active", Boolean.TRUE)) .add(Restrictions.ge("endTime", ps)).add(Restrictions.lt("endTime", pe)); List<RigOfflineSchedule> endOffline = qu.list(); for (RigOfflineSchedule sched : endOffline) { sched.setActive(false); Rig rig = sched.getRig(); if (rig.isActive()) { this.logger.info("Going to notify rig " + rig.getName() + " to clear maintenance state at time " + ps + '.'); if (this.notTest) new RigMaintenance().clearMaintenance(rig, db); } else { this.logger.warn("Mainteance period for rig " + rig.getName() + " is ending but it is not " + "registered."); } /* Notify the booking engine. */ BookingEngineService service = RigManagementActivator.getBookingService(); if (service != null) service.clearRigOffline(sched, db); } if (endOffline.size() > 0) { db.beginTransaction(); db.flush(); db.getTransaction().commit(); } } catch (HibernateException ex) { this.logger .error("Caught database exception in rig maintenance notification service. Exception reason: " + ex.getMessage() + '.'); if (db.isDirty()) { db.getTransaction().rollback(); } } finally { db.close(); } }
From source file:au.edu.uts.eng.remotelabs.schedserver.rigmanagement.intf.RigManagement.java
License:Open Source License
@SuppressWarnings("unchecked") @Override/*from w w w . ja v a 2 s . c om*/ public GetRigResponse getRig(GetRig request) { String name = request.getGetRig().getName(); this.logger.debug("Received RigManagement#getRig with parameters: name=" + name + '.'); GetRigResponse response = new GetRigResponse(); RigType rigParam = new RigType(); rigParam.setName(name); RigTypeIDType typeID = new RigTypeIDType(); typeID.setName("Unknown."); rigParam.setRigType(typeID); rigParam.setCapabilities("Unknown."); response.setGetRigResponse(rigParam); RigDao dao = new RigDao(); try { Rig rig = dao.findByName(name); if (rig == null) return response; rig.getSessions(); typeID.setName(rig.getRigType().getName()); rigParam.setCapabilities(rig.getRigCapabilities().getCapabilities()); rigParam.setIsRegistered(rig.isActive()); rigParam.setIsOnline(rig.isOnline()); if (rig.isInSession()) { rigParam.setIsInSession(true); rigParam.setSessionUser(rig.getSession().getUser().qName()); } rigParam.setIsAlarmed(!(rig.isActive() && rig.isOnline()) && !(new RigOfflineScheduleDao(dao.getSession()).isOffline(rig))); if (rig.getOfflineReason() != null) rigParam.setOfflineReason(rig.getOfflineReason()); if (rig.getContactUrl() != null) rigParam.setContactURL(rig.getContactUrl()); for (RigLog log : (List<RigLog>) dao.getSession().createCriteria(RigLog.class) .add(Restrictions.eq("rig", rig)).addOrder(Order.desc("timeStamp")).setMaxResults(10).list()) { RigLogType logParam = new RigLogType(); Calendar ts = Calendar.getInstance(); ts.setTime(log.getTimeStamp()); logParam.setTimestamp(ts); logParam.setReason(log.getReason()); if (RigLog.ONLINE.equals(log.getOldState())) logParam.setOldState(RigStateType.ONLINE); else if (RigLog.OFFLINE.equals(log.getOldState())) logParam.setOldState(RigStateType.OFFLINE); else if (RigLog.NOT_REGISTERED.equals(log.getOldState())) logParam.setOldState(RigStateType.NOT_REGISTERED); if (RigLog.ONLINE.equals(log.getNewState())) logParam.setNewState(RigStateType.ONLINE); else if (RigLog.OFFLINE.equals(log.getNewState())) logParam.setNewState(RigStateType.OFFLINE); else if (RigLog.NOT_REGISTERED.equals(log.getNewState())) logParam.setNewState(RigStateType.NOT_REGISTERED); rigParam.addLastLog(logParam); } for (RigOfflineSchedule sched : new RigOfflineScheduleDao(dao.getSession()).getOfflinePeriods(rig)) { OfflinePeriodType offline = new OfflinePeriodType(); offline.setId(sched.getId().intValue()); offline.setReason(sched.getReason()); Calendar start = Calendar.getInstance(); start.setTime(sched.getStartTime()); offline.setStart(start); Calendar end = Calendar.getInstance(); end.setTime(sched.getEndTime()); offline.setEnd(end); rigParam.addOfflinePeriods(offline); } } finally { dao.closeSession(); } return response; }
From source file:au.edu.uts.eng.remotelabs.schedserver.rigmanagement.intf.RigManagement.java
License:Open Source License
@Override public PutRigOfflineResponse putRigOffline(PutRigOffline putRigOffline) { PutRigOfflineType param = putRigOffline.getPutRigOffline(); this.logger.debug("Received RigManagement#putRigOffline with params: requestor ID=" + param.getRequestorID() + "requestor namespace=" + param.getRequestorNameSpace() + ", requestor name" + param.getRequestorName() + ", rig name=" + param.getRig().getName() + ", offline start=" + param.getStart().getTime() + ", offline end=" + param.getEnd().getTime() + ", reason=" + param.getReason() + '.'); PutRigOfflineResponse response = new PutRigOfflineResponse(); OperationResponseType result = new OperationResponseType(); response.setPutRigOfflineResponse(result); RigDao dao = new RigDao(); try {//from ww w .j av a2 s. c o m if (!this.isAuthorised(param, dao.getSession())) { this.logger.warn("Unable to put a rig offline because the user is not authorised to perform this " + "operation (not a ADMIN)."); result.setFailureCode(1); result.setFailureReason("Not authorised."); return response; } Rig rig = dao.findByName(param.getRig().getName()); if (rig == null) { this.logger.warn("Unable to put a rig offline because the rig with name " + param.getRig().getName() + " was not found."); result.setFailureCode(2); result.setFailureReason("Rig not found."); return response; } if (param.getStart().after(param.getEnd())) { this.logger .warn("Unable to put a rig offline because the offline start " + param.getStart().getTime() + " is after the offline end " + param.getEnd().getTime() + "."); result.setFailureCode(3); result.setFailureReason("Start after end."); return response; } Date startDate = param.getStart().getTime(); Date endDate = param.getEnd().getTime(); if ((Integer) dao.getSession().createCriteria(RigOfflineSchedule.class) .add(Restrictions.eq("active", Boolean.TRUE)).add(Restrictions.eq("rig", rig)) .add(Restrictions.disjunction() .add(Restrictions.and(Restrictions.gt("startTime", startDate), Restrictions.lt("endTime", endDate))) .add(Restrictions.and(Restrictions.lt("startTime", startDate), Restrictions.gt("endTime", endDate))) .add(Restrictions.and(Restrictions.lt("startTime", startDate), Restrictions.gt("endTime", endDate))) .add(Restrictions.and(Restrictions.lt("startTime", startDate), Restrictions.gt("endTime", endDate)))) .setProjection(Projections.rowCount()).uniqueResult() > 0) { this.logger.warn("Unable to put a rig offline because there is a concurrent rig offline period."); result.setFailureCode(4); result.setFailureReason("Concurrent offline period."); return response; } result.setSuccessful(true); RigOfflineSchedule offline = new RigOfflineSchedule(); offline.setActive(true); offline.setRig(rig); offline.setStartTime(startDate); offline.setEndTime(endDate); offline.setReason(param.getReason()); new RigOfflineScheduleDao(dao.getSession()).persist(offline); /* Notify the booking engine. */ BookingEngineService service = RigManagementActivator.getBookingService(); if (service != null) service.putRigOffline(offline, dao.getSession()); /* If the period is currently active, clear the rig maintenance state. */ Date now = new Date(); if (rig.isActive() && rig.isOnline() && offline.getStartTime().before(now) && offline.getEndTime().after(now)) { this.logger.info("Setting maintenance state on rig " + rig.getName() + '.'); if (this.notTest) new RigMaintenance().putMaintenance(rig, true, dao.getSession()); rig.setOnline(false); rig.setOfflineReason("In maintenance."); new RigLogDao(dao.getSession()).addOfflineLog(rig, "In maintenance."); dao.flush(); } } finally { dao.closeSession(); } return response; }
From source file:au.edu.uts.eng.remotelabs.schedserver.rigprovider.impl.StatusTimeoutChecker.java
License:Open Source License
@Override @SuppressWarnings("unchecked") public void run() { org.hibernate.Session db = null;//from w ww . j a v a2s.c o m try { db = DataAccessActivator.getNewSession(); if (db == null) { this.logger.error("Unable to obtain a database session for the rig status time out checker. " + "Ensure the database is running and configured database details are correct."); return; } RigLogDao rigLogDao = new RigLogDao(db); /* Get all the rigs that have timed out. */ List<Rig> timedOut = db.createCriteria(Rig.class).add(Restrictions.eq("managed", true)) // Unmanaged rigs need not provide a status update .add(Restrictions.eq("active", true)).add(Restrictions.lt("lastUpdateTimestamp", new Date(System.currentTimeMillis() - this.timeout * 1000))) .list(); for (Rig rig : timedOut) { Calendar cal = Calendar.getInstance(); cal.setTime(rig.getLastUpdateTimestamp()); this.logger.warn("Rig " + rig.getName() + " has timed out with last status update received at " + cal.get(Calendar.DATE) + '/' + (cal.get(Calendar.MONTH) + 1) + '/' + cal.get(Calendar.YEAR) + " - " + cal.get(Calendar.HOUR_OF_DAY) + ':' + cal.get(Calendar.MINUTE) + ':' + cal.get(Calendar.SECOND) + ". Making rig inactive."); if (rig.getSession() != null) { this.logger.warn("Timed out rig " + rig.getName() + " is in session so the session is being " + "terminated."); Session ses = rig.getSession(); ses.setActive(false); ses.setRemovalReason("Rig timed out"); ses.setRemovalTime(new Date()); rig.setInSession(false); rig.setSession(null); } rig.setActive(false); rig.setOnline(false); rig.setOfflineReason("Timed out."); rigLogDao.addUnRegisteredLog(rig, "Timed out."); db.beginTransaction(); db.flush(); db.getTransaction().commit(); /* Fire a notification the rig has gone offline. */ for (RigEventListener list : RigProviderActivator.getRigEventListeners()) { list.eventOccurred(RigStateChangeEvent.OFFLINE, rig, db); } } } catch (HibernateException hex) { this.logger.error("Failed to query database to check rig status. Exception: " + hex.getClass().getName() + ", Message:" + hex.getMessage()); if (db != null && db.getTransaction() != null) { try { db.getTransaction().rollback(); } catch (HibernateException ex) { this.logger.error("Exception rolling back up status timeout transaction (Exception: " + ex.getClass().getName() + "," + " Message: " + ex.getMessage() + ")."); } } } finally { try { if (db != null) db.close(); } catch (HibernateException ex) { this.logger.error("Exception cleaning up database session (Exception: " + ex.getClass().getName() + "," + " Message: " + ex.getMessage() + ")."); } } }
From source file:au.edu.uts.eng.remotelabs.schedserver.rigprovider.intf.RigProviderSOAPImpl.java
License:Open Source License
@Override public AddSessionFilesResponse addSessionFiles(AddSessionFiles sessionFiles) { org.hibernate.Session db = null;//from ww w .j a v a 2 s. c o m try { /* Request parameters. */ SessionFiles files = sessionFiles.getAddSessionFiles(); this.logger.debug("Received " + this.getClass().getSimpleName() + "#addSessionFiles with parameters name=" + files.getName() + ", user=" + files.getUser() + ", number of files=" + files.getFiles().length); /* Response parameters. */ AddSessionFilesResponse response = new AddSessionFilesResponse(); ProviderResponse status = new ProviderResponse(); response.setAddSessionFilesResponse(status); db = DataAccessActivator.getNewSession(); /* Load the session that generated the session file. This is the last * session the user was on the rig. */ Session session = (Session) db.createCriteria(Session.class).createAlias("rig", "rig") .createAlias("user", "user").add(Restrictions.eq("rig.name", files.getName())) .add(Restrictions.eq("user.name", files.getUser())).addOrder(Order.desc("activityLastUpdated")) .setMaxResults(1).uniqueResult(); if (session == null) { this.logger.warn("Unable to store session files for user '" + files.getUser() + "' because their " + "session on rig '" + files.getName() + "' was not found."); status.setSuccessful(false); status.setErrorReason("Session not found"); return response; } for (au.edu.uts.eng.remotelabs.schedserver.rigprovider.intf.types.SessionFile file : files.getFiles()) { SessionFile sf = new SessionFile(); sf.setSession(session); sf.setName(file.getName()); sf.setPath(file.getPath()); sf.setTimestamp(file.getTimestamp().getTime()); sf.setTransferMethod(file.getTransfer().toString()); if (SessionFile.ATTACHMENT_TRANSFER.equals(sf.getTransferMethod())) { /* The file needs to be download. */ DataHandler data = file.getFile(); if (data == null) { this.logger.warn("Cannot save session ile with name '" + sf.getName() + "' because it no " + "attached file even though the transfer method is 'ATTACHMENT'."); continue; } // TODO The actual downloading of a file should be handled elsewhere } else if (SessionFile.FILESYSTEM_TRANSFER.equals(sf.getTransferMethod())) { /* As the file is on a shared directory, it should already * exist. */ // FIXME There should be some form of verification here sf.setTransferred(true); } else if (SessionFile.WEBDAV_TRANSFER.equals(sf.getTransferMethod())) { /* The file will be coming later. */ sf.setTransferred(false); } else { this.logger.warn("Cannot save session file with name '" + sf.getName() + "' because its transfer " + "method '" + sf.getTransferMethod() + "' is unknown."); continue; } /* Store the new session file record. */ db.beginTransaction(); db.persist(sf); db.getTransaction().commit(); } status.setSuccessful(true); return response; } finally { if (db != null) db.close(); } }
From source file:au.edu.uts.eng.remotelabs.schedserver.session.impl.RigShutdownSessonStopper.java
License:Open Source License
@Override public void eventOccurred(RigStateChangeEvent event, Rig rig, org.hibernate.Session db) { try {// www . j a va2s.c o m if (event == RigStateChangeEvent.REMOVED) { /* Rig has shutdown, so will need to terminate its assigned * session, if it has one. */ Criteria qu = db.createCriteria(Session.class); qu.add(Restrictions.eq("rig", rig)).add(Restrictions.eq("active", true)); /* Whilst the above criteria are not enforced by a underlying * schema, concecptually there should only ever the one session * the rig is assigned to. */ Session ses = (Session) qu.uniqueResult(); if (ses == null) { this.logger.debug("A session is not active for rig " + rig.getName() + " so there is no need to " + "terminate a session."); return; } this.logger.info("Session for " + ses.getUserNamespace() + ':' + ses.getUserName() + " on " + "rig " + ses.getAssignedRigName() + " is being terminated because the rig has shutdown."); /* Session exists so terminate it. */ ses.setActive(false); ses.setRemovalTime(new Date()); ses.setRemovalReason("Rig " + rig.getName() + " has shutdown."); db.beginTransaction(); db.flush(); db.getTransaction().commit(); } } catch (HibernateException ex) { db.getTransaction().rollback(); this.logger.error("Failed to commit change to terminate session of rig " + rig.getName() + ", messsage: " + ex.getMessage() + '.'); } }
From source file:au.edu.uts.eng.remotelabs.schedserver.session.impl.SessionExpiryChecker.java
License:Open Source License
@SuppressWarnings("unchecked") @Override/*from w w w . ja v a 2 s . co m*/ public void run() { org.hibernate.Session db = null; try { if ((db = DataAccessActivator.getNewSession()) == null) { this.logger.warn("Unable to obtain a database session, for rig session status checker. Ensure the " + "SchedulingServer-DataAccess bundle is installed and active."); return; } boolean kicked = false; Criteria query = db.createCriteria(Session.class); query.add(Restrictions.eq("active", Boolean.TRUE)).add(Restrictions.isNotNull("assignmentTime")); Date now = new Date(); List<Session> sessions = query.list(); for (Session ses : sessions) { ResourcePermission perm = ses.getResourcePermission(); int remaining = ses.getDuration() + // The session time (perm.getAllowedExtensions() - ses.getExtensions()) * perm.getExtensionDuration() - // Extension time Math.round((System.currentTimeMillis() - ses.getAssignmentTime().getTime()) / 1000); // In session time int extension = perm.getSessionDuration() - ses.getDuration() >= TIME_EXT ? TIME_EXT : perm.getExtensionDuration(); /****************************************************************** * For sessions that have been marked for termination, terminate * * those that have no more remaining time. * ******************************************************************/ if (ses.isInGrace()) { if (remaining <= 0) { ses.setActive(false); ses.setRemovalTime(now); db.beginTransaction(); db.flush(); db.getTransaction().commit(); this.logger.info("Terminating session for " + ses.getUserNamespace() + ':' + ses.getUserName() + " on " + ses.getAssignedRigName() + " because it is expired and the grace period has elapsed."); if (this.notTest) new RigReleaser().release(ses, db); } } /****************************************************************** * For sessions with remaining time less than the grace duration: * * 1) If the session has no remaining extensions, mark it for * * termination. * * 2) Else, if the session rig is queued, mark it for * * termination. * * 3) Else, extend the sessions time. * ******************************************************************/ else if (remaining < ses.getRig().getRigType().getLogoffGraceDuration()) { BookingEngineService service; /* Need to make a decision whether to extend time or set for termination. */ if (ses.getExtensions() <= 0 && ses.getDuration() >= perm.getSessionDuration()) { this.logger.info("Session for " + ses.getUserNamespace() + ':' + ses.getUserName() + " on " + "rig " + ses.getAssignedRigName() + " is expired and cannot be extended. Marking " + "session for expiry and giving a grace period."); ses.setInGrace(true); ses.setRemovalReason("No more session time extensions."); db.beginTransaction(); db.flush(); db.getTransaction().commit(); /* Notification warning. */ if (this.notTest) new RigNotifier().notify("Your session will expire in " + remaining + " seconds. " + "Please finish and exit.", ses, db); } else if ((Integer) db.createCriteria(Bookings.class) .add(Restrictions.eq("active", Boolean.TRUE)) .add(Restrictions.eq("user", ses.getUser())).add(Restrictions.ge("startTime", now)) .add(Restrictions.lt("startTime", new Date(System.currentTimeMillis() + extension * 1000))) .setProjection(Projections.rowCount()).uniqueResult() > 0) { this.logger.info("Session for " + ses.getUserNamespace() + ':' + ses.getUserName() + " on " + "rig" + ses.getAssignedRigName() + " is being terminated because the user has a " + "starting booking. Marking session for expiry and giving a grace period."); ses.setInGrace(true); ses.setRemovalReason("User has starting booking."); db.beginTransaction(); db.flush(); db.getTransaction().commit(); /* Notification warning. */ if (this.notTest) new RigNotifier().notify("Your session will expire in " + remaining + " seconds. " + "Please finish and exit. Please note, you have a reservation that starts after this" + " session so do not leave.", ses, db); } else if (QueueInfo.isQueued(ses.getRig(), db) || ((service = SessionActivator.getBookingService()) != null && !service.extendQueuedSession(ses.getRig(), ses, extension, db))) { this.logger.info( "Session for " + ses.getUserNamespace() + ':' + ses.getUserName() + " on " + "rig " + ses.getAssignedRigName() + " is expired and the rig is queued or booked. " + "Marking session for expiry and giving a grace period."); ses.setInGrace(true); ses.setRemovalReason("Rig is queued or booked."); db.beginTransaction(); db.flush(); db.getTransaction().commit(); /* Notification warning. */ if (this.notTest) new RigNotifier().notify("Your session will end in " + remaining + " seconds. " + "After this you be removed, so please logoff.", ses, db); } else { this.logger.info("Session for " + ses.getUserNamespace() + ':' + ses.getUserName() + " on " + "rig " + ses.getAssignedRigName() + " is expired and is having its session time " + "extended by " + extension + " seconds."); if (perm.getSessionDuration() - ses.getDuration() >= TIME_EXT) { ses.setDuration(ses.getDuration() + extension); } else { ses.setExtensions((short) (ses.getExtensions() - 1)); } db.beginTransaction(); db.flush(); db.getTransaction().commit(); } } /****************************************************************** * For sessions created with a user class that can be kicked off, * * if the rig is queued, the user is kicked off immediately. * ******************************************************************/ /* DODGY The 'kicked' flag is to only allow a single kick per * pass. This is allow time for the rig to be released and take the * queued session. This is a hack at best, but should be addressed * by a released - cleaning up meta state. */ else if (!kicked && QueueInfo.isQueued(ses.getRig(), db) && perm.getUserClass().isKickable()) { kicked = true; /* No grace is being given. */ this.logger .info("A kickable user is using a rig that is queued for, so they are being removed."); ses.setActive(false); ses.setRemovalTime(now); ses.setRemovalReason("Resource was queued and user was kickable."); db.beginTransaction(); db.flush(); db.getTransaction().commit(); if (this.notTest) new RigReleaser().release(ses, db); } /****************************************************************** * Finally, for sessions with time still remaining, check * * session activity timeout - if it is not ignored and is ready * * for use. * ******************************************************************/ else if (ses.isReady() && ses.getResourcePermission().isActivityDetected() && (System.currentTimeMillis() - ses.getActivityLastUpdated().getTime()) / 1000 > perm .getSessionActivityTimeout()) { /* Check activity. */ if (this.notTest) new SessionIdleKicker().kickIfIdle(ses, db); } } } catch (HibernateException hex) { this.logger.error("Failed to query database to expired sessions (Exception: " + hex.getClass().getName() + ", Message:" + hex.getMessage() + ")."); if (db != null && db.getTransaction() != null) { try { db.getTransaction().rollback(); } catch (HibernateException ex) { this.logger.error("Exception rolling back session expiry transaction (Exception: " + ex.getClass().getName() + "," + " Message: " + ex.getMessage() + ")."); } } } catch (Throwable thr) { this.logger.error("Caught unchecked exception in session expirty checker. Exception: " + thr.getClass() + ", message: " + thr.getMessage() + '.'); } finally { try { if (db != null) db.close(); } catch (HibernateException ex) { this.logger.error("Exception cleaning up database session (Exception: " + ex.getClass().getName() + "," + " Message: " + ex.getMessage() + ")."); } } }
From source file:au.gov.naa.digipres.spyd.dao.hibernate.HibernateItemRecordDAO.java
License:Open Source License
@Override public ItemRecord getItemRecord(String id) { Session session = HibernateUtil.getSession(); Criteria crit = session.createCriteria(ItemRecord.class); crit.add(Restrictions.eq("id", id)); List results = crit.list();//from w w w .ja v a2s . c o m if ((results == null) || results.size() == 0) { return null; } ItemRecord item = (ItemRecord) results.get(0); session.flush(); session.clear(); return item; }