bool SgeQueueInterface::startJob(Structure *s) { SSHConnection *ssh = m_opt->ssh()->getFreeConnection(); if (ssh == NULL) { m_opt->warning(tr("Cannot connect to ssh server")); return false; } QWriteLocker wlocker (s->lock()); QString command = "cd \"" + s->getRempath() + "\" && " + m_qsub + " job.sh"; QString stdout_str; QString stderr_str; int ec; if (!ssh->execute(command, stdout_str, stderr_str, ec) || ec != 0) { m_opt->warning(tr("Error executing %1: %2") .arg(command).arg(stderr_str)); m_opt->ssh()->unlockConnection(ssh); return false; } m_opt->ssh()->unlockConnection(ssh); // Assuming stdout_str value is // your job <jobID> ('job.sh') has been submitted unsigned int jobID = stdout_str.split(QRegExp("\\s+"))[2].toUInt(); s->setJobID(jobID); s->startOptTimer(); return true; }
bool SyncTransferThread::connectToRemoteHost() { QMap<QString, QString> host; BaseStorage *storage = BaseStorage::instance(); this->remoteHost = host = storage->getHost(sess_name); q_debug()<<host; // RemoteHostConnectThread *conn // = new RemoteHostConnectThread(host["user_name"], host["password"], host["host_name"], // host["port"].toShort(), host["pubkey"]); // conn->run(); // ssh2_sess = (LIBSSH2_SESSION*)conn->get_ssh2_sess(); // delete conn; conn = NULL; SSHConnection *conn = new SSHConnection(); conn->setHostInfo(host); conn->connect(); ssh2_sess = conn->sess; Q_ASSERT(ssh2_sess != NULL); ssh2_sftp = libssh2_sftp_init(ssh2_sess); if (ssh2_sftp != NULL) { Q_ASSERT(ssh2_sftp != NULL); q_debug()<<"connect ok"; return true; } return false; }
void run() { SSHConnection* conn = manager->getFreeConnection(); conn->copyDirectoryToServer(from, to); conn->removeRemoteDirectory(to); manager->unlockConnection(conn); }
void SSHManagerCLITest::lockAllAndExecute() { QList<SSHConnection*> list; for (int i = 0; i < NUM_CONN; i++) { list.append(manager->getFreeConnection()); } QString command = "expr 2 + 4"; QString stdout_str, stderr_str; SSHConnection* conn; int ec; for (int i = 0; i < NUM_CONN; i++) { conn = list.at(i); QVERIFY(conn->execute(command, stdout_str, stderr_str, ec)); QCOMPARE(ec, 0); QCOMPARE(stdout_str, QString("6\n")); QVERIFY(stderr_str.isEmpty()); } for (int i = 0; i < NUM_CONN; i++) { manager->unlockConnection(list.at(i)); } }
bool SgeQueueInterface::stopJob(Structure *s) { SSHConnection *ssh = m_opt->ssh()->getFreeConnection(); if (ssh == NULL) { m_opt->warning(tr("Cannot connect to ssh server")); return false; } // lock structure QWriteLocker locker (s->lock()); // jobid has not been set, cannot delete! if (s->getJobID() == 0) { if (m_cleanRemoteOnStop) { this->cleanRemoteDirectory(s, ssh); } m_opt->ssh()->unlockConnection(ssh); return true; } const QString command = m_qdel + " " + QString::number(s->getJobID()); // Execute QString stdout_str; QString stderr_str; int ec; bool ret = true; if (!ssh->execute(command, stdout_str, stderr_str, ec) || ec != 0) { // Most likely job is already gone from queue. ret = false; } s->setJobID(0); s->stopOptTimer(); m_opt->ssh()->unlockConnection(ssh); return ret; }
QStringList SgeQueueInterface::getQueueList() const { // recast queue mutex as mutable for safe access: QReadWriteLock &queueMutex = const_cast<QReadWriteLock&> (m_queueMutex); queueMutex.lockForRead(); // Limit queries to once per second if (m_queueTimeStamp.isValid() && // QDateTime::msecsTo is not implemented until Qt 4.7 #if QT_VERSION >= 0x040700 m_queueTimeStamp.msecsTo(QDateTime::currentDateTime()) <= 1000*m_interval #else // Check if day is the same. If not, refresh. Otherwise check // msecsTo current time (m_queueTimeStamp.date() == QDate::currentDate() && m_queueTimeStamp.time().msecsTo(QTime::currentTime()) <= 1000*m_interval) #endif // QT_VERSION >= 4.7 ) { // If the cache is valid, return it QStringList ret (m_queueData); queueMutex.unlock(); return ret; } // Otherwise, store a copy of the current timestamp and switch // queuemutex to writelock QDateTime oldTimeStamp (m_queueTimeStamp); queueMutex.unlock(); // Relock mutex QWriteLocker queueLocker (&queueMutex); // Non-fatal assert: Check current timestamp against the // oldTimeStamp from earlier. If they don't match, another thread // has already updated the queueData, so tail-recurse this // function and try again. if (m_queueTimeStamp.time().msecsTo(oldTimeStamp.time()) != 0) { queueLocker.unlock(); return this->getQueueList(); } // Queue will be updated -- cast queue cache and timestamp as // mutable QStringList &queueData = const_cast<QStringList&> (m_queueData); QDateTime &queueTimeStamp = const_cast<QDateTime&> (m_queueTimeStamp); // Get SSH connection SSHConnection *ssh = m_opt->ssh()->getFreeConnection(); if (ssh == NULL) { m_opt->warning(tr("Cannot connect to ssh server")); queueTimeStamp = QDateTime::currentDateTime(); queueData.clear(); queueData << "CommError"; QStringList ret (m_queueData); return ret; } QString command = m_qstat + " | grep " + m_opt->username; // Execute QString stdout_str; QString stderr_str; int ec; // Valid exit codes for grep: (0) matches found, execution successful // (1) no matches found, execution successful // (2) execution unsuccessful if (!ssh->execute(command, stdout_str, stderr_str, ec) || (ec != 0 && ec != 1 ) ) { m_opt->ssh()->unlockConnection(ssh); m_opt->warning(tr("Error executing %1: (%2) %3\n\t" "Using cached queue data.") .arg(command) .arg(QString::number(ec)) .arg(stderr_str)); queueTimeStamp = QDateTime::currentDateTime(); QStringList ret (m_queueData); return ret; } m_opt->ssh()->unlockConnection(ssh); queueData = stdout_str.split("\n", QString::SkipEmptyParts); QStringList ret (m_queueData); queueTimeStamp = QDateTime::currentDateTime(); return ret; }