Golang melaksanakan set tugasan

Taskset ialah perintah Linux yang biasa digunakan yang digunakan untuk menentukan pertalian CPU, iaitu, menentukan CPU proses yang dijalankan. Ia boleh mengehadkan proses kepada set CPU tertentu dengan menetapkan topeng. Proses yang dijalankan pada mesin berbilang CPU sering mendapat manfaat daripada menggunakan set tugas untuk mengehadkan pertalian CPU untuk mengoptimumkan prestasi. Dalam artikel ini, kami akan memperkenalkan cara menulis pelaksanaan perintah set tugas mudah menggunakan golang.

Untuk merealisasikan fungsi asas set tugas, kita perlu menggunakan beberapa perpustakaan dan fungsi tertentu dalam golang. Perpustakaan ini termasuk os, bendera dan strconv. Kami akan menggunakan pustaka bendera untuk menghuraikan parameter baris arahan, perpustakaan os untuk mengendalikan fungsi sistem pengendalian, dan perpustakaan strconv untuk menukar jenis data.

Pertama, kita perlu menentukan parameter baris arahan kita, yang termasuk ID proses dan topeng CPU untuk dijalankan. Dalam golang, kita boleh menggunakan pakej bendera untuk mentakrifkan pilihan baris arahan dan menghuraikannya dalam program:

import "flag"

var (
  processID int
  cpuMask   string
)

func init() {
  flag.IntVar(&processID, "p", 0, "pid to modify")
  flag.StringVar(&cpuMask, "c", "", "CPU affintiy mask (in hex)")
}

func main() {
  flag.Parse()
  // ...
}

Di sini, kami mentakrifkan dua pembolehubah processID dan cpuMask. Kami menggunakan flag.IntVar dan flag.StringVar untuk mentakrifkan kedua-dua pembolehubah ini, yang sepadan dengan parameter baris arahan "-p" dan "-c" masing-masing. Maksud parameter ini dijelaskan dalam komen kod.

Kini kita boleh mengendalikan argumen baris arahan dan melaksanakan kod set tugas seperti yang diperlukan. Kami akan menggunakan fungsi sched_setaffinity daripada perpustakaan os untuk menetapkan pertalian CPU. Fungsi ini boleh digunakan sama ada pada satu proses atau kumpulan proses. Dalam contoh kami, kami akan menggunakan ID proses tunggal untuk menetapkan pertalian CPU.

import (
  "fmt"
  "syscall"
)

func setAffinity(pid int, mask string) error {
  if pid <= 0 {
    return fmt.Errorf("invalid PID %d", pid)
  }

  tid, err := syscall.Gettid()
  if err != nil {
    return fmt.Errorf("failed to get TID: %v", err)
  }

  var cpuSet syscall.CPUSet
  _, err = fmt.Sscanf(mask, "%x", &cpuSet)
  if err != nil {
    return fmt.Errorf("failed to scan CPU mask %s: %v", mask, err)
  }

  err = syscall.SchedSetaffinity(tid, &cpuSet)
  if err != nil {
    return fmt.Errorf("failed to set affinity: %v", err)
  }

  return nil
}

Dalam fungsi ini, kami mula-mula menyemak sama ada ID proses adalah nilai yang sah. Kemudian, kami menggunakan fungsi syscall.Gettid untuk mendapatkan ID benang. ID ini akan digunakan untuk menetapkan perkaitan CPU dalam fungsi SchedSetaffinity. Seterusnya, kami menukar topeng CPU yang diluluskan kepada jenis syscall.CPUSet untuk dihantar ke fungsi SchedSetaffinity. Akhir sekali, kami menyemak nilai pulangan fungsi SchedSetaffinity dan mengembalikan sebarang ralat.

Kini, kami telah selesai menulis fungsi setAffinity. Sekarang, kita perlu mengendalikan parameter baris arahan dan memanggil fungsi setAffinity dalam fungsi utama program.

func main() {
  flag.Parse()

  err := setAffinity(processID, cpuMask)
  if err != nil {
    fmt.Printf("Error: %v
", err)
  } else {
    fmt.Println("CPU affinity set successfully")
  }
}

Dalam fungsi ini, kita mula-mula menggunakan fungsi flag.Parse untuk menghuraikan parameter baris arahan. Kemudian, kami memanggil fungsi setAffinity untuk menetapkan pertalian CPU. Jika fungsi mengembalikan ralat, mesej ralat akan dikeluarkan ke konsol, jika tidak, mesej kejayaan akan dikeluarkan.

Kini, kami telah berjaya menulis program set tugasan mudah untuk menetapkan pertalian CPU sesuatu proses. Apabila menggunakan program ini, sila ingat bahawa keistimewaan root diperlukan untuk menjalankannya.

Untuk meringkaskan, kami menulis program set tugasan mudah menggunakan golang yang boleh menetapkan pertalian CPU sesuatu proses. Kami menggunakan pakej bendera untuk melaksanakan penghuraian parameter baris arahan, dan menggunakan pakej os dan syscall untuk melaksanakan tugas sebenar. Ini adalah alat yang sangat berguna untuk mereka yang perlu menjalankan program dalam sistem berbilang CPU.

Atas ialah kandungan terperinci Golang melaksanakan set tugasan. Untuk maklumat lanjut, sila ikut artikel berkaitan lain di laman web China PHP!