#include <fstream>
#include <boost/function.hpp>
#include <boost/function_output_iterator.hpp>
+#include <algorithm>
#include "zypp/base/Regex.h"
#include "zypp/base/String.h"
#include "zypp/PoolItem.h"
#include "zypp/PoolQueryUtil.tcc"
#include "zypp/ZYppCallbacks.h"
+#include "zypp/sat/SolvAttr.h"
+#include "zypp/sat/Solvable.h"
+#include "zypp/PathInfo.h"
#undef ZYPP_BASE_LOGGER_LOGGROUP
#define ZYPP_BASE_LOGGER_LOGGROUP "locks"
#include "zypp/Locks.h"
-#include "zypp/PathInfo.h"
-using namespace std;
-using namespace zypp;
-using namespace zypp::str;
+using std::endl;
namespace zypp
{
-namespace locks
-{
Locks& Locks::instance()
{
class Locks::Impl
{
public:
- std::list<PoolQuery> locks;
+ LockList locks;
+ LockList toAdd;
+ LockList toRemove;
+ bool locksDirty;
+
+ bool mergeList(callback::SendReport<SavingLocksReport>& report);
+
+ Impl():locksDirty(false){}
};
Locks::Locks() : _pimpl(new Impl){}
-void Locks::saveLocks( const Pathname& file )
+Locks::const_iterator Locks::begin() const
+{ return _pimpl->locks.begin(); }
+
+Locks::const_iterator Locks::end() const
+{ return _pimpl->locks.end(); }
+
+Locks::LockList::size_type Locks::size() const
+{ return _pimpl->locks.size(); }
+
+bool Locks::empty() const
+{ return _pimpl->locks.empty(); }
+
+struct ApplyLock
{
- writePoolQueriesToFile( file, _pimpl->locks.begin(), _pimpl->locks.end() );
-}
+ void operator()(const PoolQuery& query) const
+ {
+ for ( const PoolItem & item : query.poolItem() )
+ {
+ item.status().setLock(true,ResStatus::USER);
+ DBG << "lock "<< item.name();
+ }
+ }
+};
/**
* iterator that takes lock, lock all solvables from query
void operator()(const PoolQuery& query) const
{
- for_( it,query.begin(),query.end() )
- {
- PoolItem item(*it);
- item.status().setLock(true,ResStatus::USER);
- }
-
+ ApplyLock a;a(query);
*out++ = query;
}
private:
OutputIterator& out;
- };
+};
-void Locks::loadLocks( const Pathname& file )
+void Locks::readAndApply( const Pathname& file )
{
- insert_iterator<std::list<PoolQuery> > ii( _pimpl->locks,
- _pimpl->locks.end() );
- LockingOutputIterator<insert_iterator<std::list<PoolQuery> > > lout(ii);
- readPoolQueriesFromFile( file, boost::make_function_output_iterator(lout) );
+ MIL << "read and apply locks from "<<file << endl;
+ PathInfo pinfo(file);
+ if ( pinfo.isExist() )
+ {
+ std::insert_iterator<LockList> ii( _pimpl->locks, _pimpl->locks.end() );
+ LockingOutputIterator<std::insert_iterator<LockList> > lout(ii);
+ readPoolQueriesFromFile( file, boost::make_function_output_iterator(lout) );
+ }
+ else
+ MIL << "file not exist(or cannot be stat), no lock added." << endl;
+
}
+void Locks::read( const Pathname& file )
+{
+ MIL << "read locks from "<<file << endl;
+ PathInfo pinfo(file);
+ if ( pinfo.isExist() )
+ readPoolQueriesFromFile( file, std::insert_iterator<LockList>(_pimpl->locks, _pimpl->locks.end()) );
+ else
+ MIL << "file not exist(or cannot be stat), no lock added." << endl;
+}
+
+
+void Locks::apply() const
+{
+ DBG << "apply locks" << endl;
+ for_each(begin(), end(), ApplyLock());
+}
+
+
void Locks::addLock( const PoolQuery& query )
{
+ MIL << "add new lock" << endl;
for_( it,query.begin(),query.end() )
{
PoolItem item(*it);
item.status().setLock(true,ResStatus::USER);
}
- _pimpl->locks.push_back( query );
+ LockList::iterator i = find(_pimpl->toRemove.begin(),
+ _pimpl->toRemove.end(), query);
+ if ( i != _pimpl->toRemove.end() )
+ {
+ DBG << "query removed from toRemove" << endl;
+ _pimpl->toRemove.erase(i);
+ }
+ else
+ {
+ DBG << "query added as new" << endl;
+ _pimpl->toAdd.push_back( query );
+ }
+}
+
+void Locks::addLock( const IdString& ident_r )
+{
+ sat::Solvable::SplitIdent id(ident_r);
+ addLock(id.kind(),id.name());
+}
+
+void Locks::addLock( const ResKind& kind_r, const C_Str & name_r )
+{
+ addLock(kind_r,IdString(name_r));
+}
+
+void Locks::addLock( const ResKind& kind_r, const IdString& name_r )
+{
+ PoolQuery q;
+ q.addAttribute( sat::SolvAttr::name,name_r.asString() );
+ q.addKind( kind_r );
+ q.setMatchExact();
+ q.setCaseSensitive(true);
+ DBG << "add lock by identifier" << endl;
+ addLock( q );
+}
+
+void Locks::removeLock( const PoolQuery& query )
+{
+ MIL << "remove lock" << endl;
+ for_( it,query.begin(),query.end() )
+ {
+ PoolItem item(*it);
+ item.status().setLock(false,ResStatus::USER);
+ }
+
+ LockList::iterator i = find(_pimpl->toAdd.begin(),
+ _pimpl->toAdd.end(), query);
+ if ( i != _pimpl->toAdd.end() )
+ {
+ DBG << "query removed from added" << endl;
+ _pimpl->toAdd.erase(i);
+ }
+ else
+ {
+ DBG << "needed remove some old lock" << endl;
+ _pimpl->toRemove.push_back( query );
+ }
+}
+
+void Locks::removeLock( const IdString& ident_r )
+{
+ sat::Solvable::SplitIdent id(ident_r);
+ removeLock(id.kind(),id.name());
}
-bool Locks::existEmptyLocks()
+void Locks::removeLock( const ResKind& kind_r, const C_Str & name_r )
+{
+ removeLock(kind_r,IdString(name_r));
+}
+
+void Locks::removeLock( const ResKind &kind_r, const IdString &name_r )
+{
+ PoolQuery q;
+ q.addAttribute( sat::SolvAttr::name,name_r.asString() );
+ q.addKind( kind_r );
+ q.setMatchExact();
+ q.setCaseSensitive(true);
+ q.requireAll();
+ DBG << "remove lock by selectactable" << endl;
+ removeLock(q);
+}
+
+bool Locks::existEmpty() const
{
for_( it, _pimpl->locks.begin(), _pimpl->locks.end() )
{
}
//handle locks during removing
-class LocksRemovePredicate{
+class LocksCleanPredicate{
private:
bool skip_rest;
size_t searched;
callback::SendReport<CleanEmptyLocksReport> &report;
public:
- LocksRemovePredicate(size_t count, callback::SendReport<CleanEmptyLocksReport> &_report): skip_rest(false),searched(0),all(count), report(_report){}
+ LocksCleanPredicate(size_t count, callback::SendReport<CleanEmptyLocksReport> &_report): skip_rest(false),searched(0),all(count), report(_report){}
bool aborted(){ return skip_rest; }
};
-void Locks::removeEmptyLocks()
+void Locks::removeEmpty()
{
+ MIL << "cleaning of locks" << endl;
callback::SendReport<CleanEmptyLocksReport> report;
report->start();
size_t sum = _pimpl->locks.size();
- LocksRemovePredicate p(sum, report);
+ LocksCleanPredicate p(sum, report);
_pimpl->locks.remove_if(p);
if( p.aborted() )
{
+ MIL << "cleaning aborted" << endl;
report->finish(CleanEmptyLocksReport::ABORTED);
}
else
report->finish(CleanEmptyLocksReport::NO_ERROR);
}
+
+ if ( sum != _pimpl->locks.size() ) //some locks has been removed
+ _pimpl->locksDirty = true;
+}
+
+class LocksRemovePredicate
+{
+private:
+ std::set<sat::Solvable>& solvs;
+ const PoolQuery& query;
+ callback::SendReport<SavingLocksReport>& report;
+ bool aborted_;
+
+ //1 for subset of set, 2 only intersect, 0 for not intersect
+ int contains(const PoolQuery& q, std::set<sat::Solvable>& s)
+ {
+ bool intersect = false;
+ for_( it,q.begin(),q.end() )
+ {
+ if ( s.find(*it)!=s.end() )
+ {
+ intersect = true;
+ }
+ else
+ {
+ if (intersect)
+ return 2;
+ }
+ }
+ return intersect ? 1 : 0;
+ }
+
+public:
+ LocksRemovePredicate(std::set<sat::Solvable>& s, const PoolQuery& q,
+ callback::SendReport<SavingLocksReport>& r)
+ : solvs(s), query(q),report(r),aborted_(false) {}
+
+ bool operator()(const PoolQuery& q)
+ {
+ if (aborted())
+ return false;
+ if( q==query )
+ {//identical
+ DBG << "identical queries" << endl;
+ return true;
+ }
+
+ SavingLocksReport::ConflictState cs;
+ switch( contains(q,solvs) )
+ {
+ case 0:
+ return false; //another lock
+ case 1:
+ cs = SavingLocksReport::SAME_RESULTS;
+ break;
+ case 2:
+ cs = SavingLocksReport::INTERSECT;
+ break;
+ default:
+ return true;
+ }
+ MIL << "find conflict: " << cs << endl;
+ switch (report->conflict(q,cs))
+ {
+ case SavingLocksReport::ABORT:
+ aborted_ = true;
+ DBG << "abort merging" << endl;
+ return false;
+ case SavingLocksReport::DELETE:
+ DBG << "force delete" << endl;
+ return true;
+ case SavingLocksReport::IGNORE:
+ DBG << "skip lock" << endl;
+ return false;
+ }
+ WAR << "should not reached, some state is missing" << endl;
+ return false;
+ }
+
+ bool aborted(){ return aborted_; }
+};
+
+bool Locks::Impl::mergeList(callback::SendReport<SavingLocksReport>& report)
+{
+ MIL << "merging list old: " << locks.size()
+ << " to add: " << toAdd.size() << "to remove: " << toRemove.size() << endl;
+ for_(it,toRemove.begin(),toRemove.end())
+ {
+ std::set<sat::Solvable> s(it->begin(),it->end());
+ locks.remove_if(LocksRemovePredicate(s,*it, report));
+ }
+
+ if (!report->progress())
+ return false;
+
+ for_( it, toAdd.begin(), toAdd.end() )
+ {
+ if( std::find( locks.begin(), locks.end(), *it ) == locks.end() )
+ locks.push_back( *it );
+ }
+
+ toAdd.clear();
+ toRemove.clear();
+
+ return true;
+}
+
+void Locks::merge()
+{
+ if( (_pimpl->toAdd.size() | _pimpl->toRemove.size())==0)
+ {
+ return; //nothing to merge
+ }
+
+ callback::SendReport<SavingLocksReport> report;
+ report->start();
+ if (!_pimpl->mergeList(report))
+ {
+ report->finish(SavingLocksReport::ABORTED);
+ return;
+ }
+ DBG << "locks merged" << endl;
+ report->finish(SavingLocksReport::NO_ERROR);
+ _pimpl->locksDirty = true;
+}
+
+void Locks::save( const Pathname& file )
+{
+ if( ((_pimpl->toAdd.size() | _pimpl->toRemove.size())==0)
+ && !_pimpl->locksDirty )
+ {
+ DBG << "nothing changed in locks - no write to file" << endl;
+ return;
+ }
+
+ callback::SendReport<SavingLocksReport> report;
+ report->start();
+
+ if ((_pimpl->toAdd.size() | _pimpl->toRemove.size())!=0)
+ {
+ if (!_pimpl->mergeList(report))
+ {
+ report->finish(SavingLocksReport::ABORTED);
+ return;
+ }
+ }
+
+ DBG << "writed "<< _pimpl->locks.size() << "locks" << endl;
+ writePoolQueriesToFile( file, _pimpl->locks.begin(), _pimpl->locks.end() );
+ report->finish(SavingLocksReport::NO_ERROR);
+}
+
+void Locks::removeDuplicates()
+{
+ size_type sum = size();
+ for_(it,_pimpl->locks.begin(),_pimpl->locks.end())
+ {
+ if ( find(_pimpl->locks.begin(),it,*it) != it )
+ _pimpl->locks.erase(it--); //-- to avoid using break iterator
+ }
+
+ if (sum!=size())
+ _pimpl->locksDirty = true;
}
-} // ns locks
} // ns zypp