#include "types.hh"
#include "util.hh"
#include "affinity.hh"

#if HAVE_SCHED_H
#include <sched.h>
#endif

namespace nix {


#if HAVE_SCHED_SETAFFINITY
static bool didSaveAffinity = false;
static cpu_set_t savedAffinity;
#endif


void setAffinityTo(int cpu)
{
#if HAVE_SCHED_SETAFFINITY
    if (sched_getaffinity(0, sizeof(cpu_set_t), &savedAffinity) == -1) return;
    didSaveAffinity = true;
    printMsg(lvlDebug, format("locking this thread to CPU %1%") % cpu);
    cpu_set_t newAffinity;
    CPU_ZERO(&newAffinity);
    CPU_SET(cpu, &newAffinity);
    if (sched_setaffinity(0, sizeof(cpu_set_t), &newAffinity) == -1)
        printMsg(lvlError, format("failed to lock thread to CPU %1%") % cpu);
#endif
}


int lockToCurrentCPU()
{
#if HAVE_SCHED_SETAFFINITY
    int cpu = sched_getcpu();
    if (cpu != -1) setAffinityTo(cpu);
    return cpu;
#else
    return -1;
#endif
}


void restoreAffinity()
{
#if HAVE_SCHED_SETAFFINITY
    if (!didSaveAffinity) return;
    if (sched_setaffinity(0, sizeof(cpu_set_t), &savedAffinity) == -1)
        printMsg(lvlError, "failed to restore affinity %1%");
#endif
}


}
<a href='/guix/refs/?id=dd80d13bea8239dc77518204cfaa8aee8c8ac097'>refs</a><a class='active' href='/guix/log/tests/cpan.scm'>log</a><a href='/guix/tree/tests/cpan.scm?id=dd80d13bea8239dc77518204cfaa8aee8c8ac097'>tree</a><a href='/guix/commit/tests/cpan.scm?id=dd80d13bea8239dc77518204cfaa8aee8c8ac097'>commit</a><a href='/guix/diff/tests/cpan.scm?id=dd80d13bea8239dc77518204cfaa8aee8c8ac097'>diff</a></td><td class='form'><form class='right' method='get' action='/guix/log/tests/cpan.scm'>
<input type='hidden' name='id' value='dd80d13bea8239dc77518204cfaa8aee8c8ac097'/><select name='qt'>
<option value='grep'>log msg</option>
<option value='author'>author</option>
<option value='committer'>committer</option>
<option value='range'>range</option>
</select>
<input class='txt' type='search' size='10' name='q' value=''/>
<input type='submit' value='search'/>
</form>
</td></tr></table>
<div class='path'>path: <a href='/guix/log/?id=dd80d13bea8239dc77518204cfaa8aee8c8ac097'>root</a>/<a href='/guix/log/tests?id=dd80d13bea8239dc77518204cfaa8aee8c8ac097'>tests</a>/<a href='/guix/log/tests/cpan.scm?id=dd80d13bea8239dc77518204cfaa8aee8c8ac097'>cpan.scm</a></div><div class='content'><table class='list nowrap'><tr class='nohover'><th class='left'>Age</th><th class='left'>Commit message (<a href='/guix/log/tests/cpan.scm?id=dd80d13bea8239dc77518204cfaa8aee8c8ac097&amp;showmsg=1'>Expand</a>)</th><th class='left'>Author</th></tr>
<tr><td><span title='2020-01-15 18:40:43 +0100'>2020-01-15</span></td><td><a href='/guix/commit/tests/cpan.scm?id=4aea90b1876179aab8d603a42533a6bdf97ccd3c'>import: cpan: Rewrite tests to use an HTTP server instead of mocking.</a><span class='msg-avail'>...</span></td><td>Ludovic Courtès</td></tr>
<tr><td><span title='2020-01-15 18:40:43 +0100'>2020-01-15</span></td><td><a href='/guix/commit/tests/cpan.scm?id=69f132554c6bd23df4610a21e636bde5f0578174'>import: cpan: Rewrite to use 'define-json-mapping'.</a><span class='msg-avail'>...</span></td><td>Ludovic Courtès</td></tr>
<tr><td><span title='2019-08-22 14:24:11 -0400'>2019-08-22</span></td><td><a href='/guix/commit/tests/cpan.scm?id=d020821c0bd2206a5f3d4db155f2a9a3de7dc670'>Revert "import: cpan: Adapt for the change to guile-json version 3."</a><span class='msg-avail'>...</span></td><td>Mark H Weaver</td></tr>
<tr><td><span title='2019-08-21 18:29:30 +0100'>2019-08-21</span></td><td><a href='/guix/commit/tests/cpan.scm?id=01ce7af25add55514f737af48ea6c127bedfde67'>import: cpan: Adapt for the change to guile-json version 3.</a><span class='msg-avail'>...</span></td><td>Christopher Baines</td></tr>
<tr><td><span title='2018-09-04 17:25:11 +0200'>2018-09-04</span></td><td><a href='/guix/commit/tests/cpan.scm?id=ca719424455465fca4b872c371daf2a46de88b33'>Switch to Guile-Gcrypt.</a><span class='msg-avail'>...</span></td><td>Ludovic Courtès</td></tr>
<tr><td><span title='2018-08-24 18:04:07 +0200'>2018-08-24</span></td><td><a href='/guix/commit/tests/cpan.scm?id=dccd29d17dc6ca6a4f2820ce5f5d4d81f4768e4a'>import: cpan: Adjust test to new URL.</a><span class='msg-avail'>...</span></td><td>Ludovic Courtès</td></tr>
<tr><td><span title='2017-10-30 17:20:42 +0100'>2017-10-30</span></td><td><a href='/guix/commit/tests/cpan.scm?id=b5c7574b2f6fe7752b02eaaa8a745a35b3beb2a1'>Revert "import: cpan: Use HTTPS for home pages."</a><span class='msg-avail'>...</span></td><td>Tobias Geerinckx-Rice</td></tr>
<tr><td><span title='2017-10-29 17:32:35 +0100'>2017-10-29</span></td><td><a href='/guix/commit/tests/cpan.scm?id=29f7bf59d5d4d4b848eaedc6766bb4e02cae20d3'>import: cpan: Use HTTPS for home pages.</a><span class='msg-avail'>...</span></td><td>Tobias Geerinckx-Rice</td></tr>
<tr><td><span title='2017-10-28 18:20:59 -0500'>2017-10-28</span></td><td><a href='/guix/commit/tests/cpan.scm?id=e4bc1727302b0e1e255ea5cf4e2ccf33cafe7296'>import: cpan: Add trailing "/" on home-page.</a><span class='msg-avail'>...</span></td><td>Eric Bavier</td></tr>
<tr><td><span title='2017-10-28 18:20:58 -0500'>2017-10-28</span></td><td><a href='/guix/commit/tests/cpan.scm?id=23055424f2628470a864106dbc4d4acb003399ee'>import: cpan: Propagate imported dependencies.</a><span class='msg-avail'>...</span></td><td>Eric Bavier</td></tr>
<tr><td><span title='2017-09-14 00:10:13 +0200'>2017-09-14</span></td><td><a href='/guix/commit/tests/cpan.scm?id=79e5e00f4f0b5a6b2cbf13f60bcd1fd7029f8e08'>import: cpan: Adjust expected license in tests.</a><span class='msg-avail'>...</span></td><td>Ludovic Courtès</td></tr>
<tr><td><span title='2017-06-07 15:06:34 +0200'>2017-06-07</span></td><td><a href='/guix/commit/tests/cpan.scm?id=4679dd6967c21e21c740cd88e17191b8e2aac5ee'>import: cpan: Update CPAN importer to use MetaCPAN v1 API.</a><span class='msg-avail'>...</span></td><td>James Richardson</td></tr>