traetox / goaio Goto Github PK

package compared
     
const (
    goroutineNum = 50                                                                                                                                                                                                                                                           
    rwNum        = 1000
    fileSize     = 4 * 100 * 1024 * 1024
    sloat        = fileSize / (4 * 1024)
)    
     
// only use in linux
const O_DIRECT = 0x100000

package compared

import (
	"fmt"
	"math/rand"
	"os"
	"sync"
	"testing"
	"time"
)

func dorw(wg *sync.WaitGroup, file *os.File) {
	buf := make([]byte, 4096)
	for i := 0; i < rwNum; i++ {
		randNum := rand.Int31()
		op := randNum % int32(2)
		off := int64((randNum % int32(sloat)) * 4096)

                start := time.Now()
		if op == 0 {

			n, _ := file.ReadAt(buf, off)
			if n != 4096 {
				panic("n != 4096")
			}
		} else {

			n, _ := file.WriteAt(buf, off)
			if n != 4096 {
				panic("n != 4096")
			}
		}
               end := time.Now()

               fmt.Printf("op:%d %d\n", op, end.UnixNano() - start.UnixNano())
	}

	wg.Done()
}

func TestRW(t *testing.T) {
	// 环境准备
	file, err := os.OpenFile("1.txt", os.O_CREATE|os.O_TRUNC|os.O_WRONLY, 0644)
	if err != nil {
		t.Fatalf("open 1.txt failed, err: %v", err)
	}
	// 创建指定大小的文件
	file.Seek(fileSize, 0)
	file.Write([]byte("aaa"))
	file.Close()

	file, err = os.OpenFile("1.txt", os.O_RDWR | O_DIRECT, 0644)
	if err != nil {
		t.Fatalf("open 1.txt failed, err: %v", err)
	}

	start := time.Now()
	wg := &sync.WaitGroup{}
	wg.Add(goroutineNum)
	for i := 0; i < goroutineNum; i++ {
		go dorw(wg, file)
	}
	wg.Wait()
	end := time.Now()

	fmt.Printf("rw delay %d ms\n", (end.UnixNano()-start.UnixNano())/(1000*1000))

	file.Close()
}

package compared

import (
	"fmt"
	"go-aio"
	"math/rand"
	"os"
	"sync"
	"testing"
	"time"
)

func do(wg *sync.WaitGroup, file *os.File, gaio *aio.AIO) {
	buf := make([]byte, 4096)
	for i := 0; i < rwNum; i++ {
		randNum := rand.Int31()
		op := randNum % int32(2)
		off := int64((randNum % int32(sloat)) * 4096)

                start := time.Now()
		if op == 0 {
			n, _ := gaio.DoRequest(aio.ReadOP, file, buf, off)
			if n != 4096 {
				panic("n != 4096")
			}
		} else {
			n, _ := gaio.DoRequest(aio.WriteOP, file, buf, off)
			if n != 4096 {
				panic("n != 4096")
			}

		}
               end := time.Now()

               fmt.Printf("op:%d %d\n", op, end.UnixNano() - start.UnixNano())
	}

	wg.Done()
}

func TestAIO(t *testing.T) {
	// 环境准备
	file, err := os.OpenFile("1.txt", os.O_CREATE|os.O_TRUNC|os.O_WRONLY, 0644)
	if err != nil {
		t.Fatalf("open 1.txt failed, err: %v", err)
	}
	// 创建指定大小的文件
	file.Seek(fileSize, 0)
	file.Write([]byte("aaa"))
	file.Close()

	file, err = os.OpenFile("1.txt", os.O_RDWR | O_DIRECT, 0644)
	if err != nil {
		t.Fatalf("open 1.txt failed, err: %v", err)
	}

	gaio := aio.Setup(8, 4096)

	start := time.Now()
	wg := &sync.WaitGroup{}
	wg.Add(goroutineNum)
	for i := 0; i < goroutineNum; i++ {
		go do(wg, file, gaio)
	}
	wg.Wait()
	end := time.Now()

	fmt.Printf("aio rw delay %d ms\n", (end.UnixNano()-start.UnixNano())/(1000*1000))

	file.Close()
}