:sparkles: saving one-true-pairing based on stable-marriage compromises

backend/db/survey-generator.js

@@ -69,12 +69,53 @@ class DummyProfile {
         this.durationPref = null
         this.locationPref = null
 
+        this.shouldDelete = false
+
         // profile is constructed of 12 answers, 2 for each dimension
         this.profileResponses = this.getRandomAnswers(12, importance)
         this.langPref = this.getRandomAnswers(rand(4), langs)
         this.durationPref = this.getRandomAnswers(1, duration)
         this.locationPref = this.getRandomAnswers(1, location)
+
+        this.matchPref = null
+
+        this.candidateIndex = 0
+        this.otp = null
+        this.forder = []
+    }
+    clearPlaceholderMatches() {
+        this.matchPref = this.matchPref.filter(
+            match => match.shouldDelete == false,
+        )
+        this.forder = this.forder.filter(
+            potentialFiance => potentialFiance.shouldDelete == false,
+        )
+        if (this.otp.shouldDelete) {
+            this.otp = null
+        }
+        // console.log(this.matchPref)
+    }
+    rank(p) {
+        for (let i = 0; i < this.matchPref.length; i++)
+            if (this.matchPref[i] === p) return i
+        return this.matchPref.length + 1
+    }
+    prefers(p) {
+        return this.rank(p) < this.rank(this.otp)
     }
+    nextCandidate() {
+        if (this.candidateIndex >= this.matchPref.length) return null
+        return this.matchPref[this.candidateIndex++]
+    }
+    engageTo(p) {
+        if (p.otp) p.otp.otp = null
+        p.otp = this
+        p.forder.unshift(this)
+        if (this.otp) this.otp.otp = null
+        this.otp = p
+        this.forder.unshift(p)
+    }
+    // My stuff
     getRandomAnswers(count, options) {
         const answers = []
         for (let i = 0; i < count; i++) {
@@ -85,11 +126,25 @@ class DummyProfile {
     }
 }
 
-const generateDummyProfiles = count => {
+function engageEveryone(guys) {
+    var done
+    do {
+        done = true
+        guys.forEach(guy => {
+            if (!guy.otp) {
+                done = false
+                const gal = guy.nextCandidate()
+                if (!gal.otp || gal.prefers(guy)) guy.engageTo(gal)
+            }
+        })
+    } while (!done)
+}
+
+const generateDummyProfiles = (count, startFrom) => {
     const profiles = []
     for (let i = 0; i < count; i++) {
         const dummyProfile = new DummyProfile()
-        dummyProfile.id = i + 1
+        dummyProfile.id = startFrom + i + 1
         profiles.push(dummyProfile)
     }
     profiles.forEach(dummy => {
@@ -105,8 +160,35 @@ const generateDummyProfiles = count => {
     return profiles
 }
 
-const generatedSeekers = generateDummyProfiles(100)
-const generatedProviders = generateDummyProfiles(10)
+const generatedSeekers = generateDummyProfiles(100, 0)
+const generatedProviders = generateDummyProfiles(80, generatedSeekers.length)
+const balanceSeekersAndProviders = (seekers, providers) => {
+    let diff = 0
+    let smallerList = null
+
+    if (seekers.length < providers.length) {
+        diff = providers.length - seekers.length
+        smallerList = seekers
+    } else {
+        diff = seekers.length - providers.length
+        smallerList = providers
+    }
+    let fillerId = seekers.length + providers.length
+    for (let i = 0; i < diff; i++) {
+        const filler = new DummyProfile()
+        filler.id = fillerId + i + 1
+        filler.profileResponses = filler.profileResponses.map(
+            (answer, response_key_id) => {
+                const answerObj = {}
+                // aka: id for the question we asked
+                answerObj[response_key_id] = answer
+                return answerObj
+            },
+        )
+        filler.shouldDelete = true
+        smallerList.push(filler)
+    }
+}
 
 const scoreMatch = (seeker, potentialMatch) => {
     const seekerResponseValues = seeker.profileResponses.map(res =>
@@ -120,24 +202,59 @@ const scoreMatch = (seeker, potentialMatch) => {
     )
 }
 const compareProfile = (seeker, unorderedPotentialMatches) => {
-    const scored = unorderedPotentialMatches.map(potentialMatch => {
-        // add the match to object keyed by score
-        return {
-            profileMatchScore: scoreMatch(seeker, potentialMatch),
-            profile: potentialMatch,
-        }
-    })
+    const scored = unorderedPotentialMatches
+        .map(potentialMatch => {
+            // add the match to object keyed by score
+            return {
+                profileMatchScore: scoreMatch(seeker, potentialMatch),
+                profile: potentialMatch,
+            }
+        })
+        .sort((a, b) => a.profileMatchScore - b.profileMatchScore)
     // return ordered by score
-    return scored.sort((a, b) => a.profileMatchScore - b.profileMatchScore)
+    return scored.map(profileScore => profileScore.profile)
 }
 
+balanceSeekersAndProviders(generatedSeekers, generatedProviders)
+
+// Score
 generatedSeekers.forEach(seeker => {
-    const matchQueue = compareProfile(seeker, generatedProviders).map(
-        provider => ({
-            score: provider.profileMatchScore,
-            profile_id: provider.profile.id,
-        }),
-    )
-    console.log(`\n---|    Results for job_seeker: ${seeker.id}    |---`)
-    console.log(matchQueue)
+    seeker.matchPref = compareProfile(seeker, generatedProviders)
+})
+generatedProviders.forEach(provider => {
+    provider.matchPref = compareProfile(provider, generatedSeekers)
 })
+
+// Everything balanced ready for stable marriage
+if (generatedSeekers.length == generatedProviders.length) {
+    engageEveryone(generatedProviders)
+    generatedSeekers.forEach(seeker => seeker.clearPlaceholderMatches())
+    generatedProviders.forEach(provider => provider.clearPlaceholderMatches())
+
+    console.log('\nuser:', generatedProviders[0].id, '| provider')
+    console.log(
+        'otp',
+        generatedProviders[0].otp ? generatedProviders[0].otp.id : null,
+        'rank',
+        generatedProviders[0].otp
+            ? generatedProviders[0].matchPref
+                  .map(m => m.id)
+                  .indexOf(generatedProviders[0].otp.id) + 1
+            : generatedProviders[0].otp,
+    )
+    console.log(generatedProviders[0].matchPref.map(m => m.id))
+
+    console.log('---')
+    console.log('user:', generatedSeekers[0].id, '| seeker')
+    console.log(
+        'otp',
+        generatedSeekers[0].otp ? generatedSeekers[0].otp.id : null,
+        'rank',
+        generatedSeekers[0].otp
+            ? generatedSeekers[0].matchPref
+                  .map(m => m.id)
+                  .indexOf(generatedSeekers[0].otp.id) + 1
+            : generatedSeekers[0].otp,
+    )
+    console.log(generatedSeekers[0].matchPref.map(m => m.id))
+}