作者:倒霉熊丶JO_205 | 来源:互联网 | 2023-05-17 19:14
用golang生成6位数的唯一id,5分钟生成八十万唯一的id。
package randid_test
import (
"fmt"
"math/rand"
"testing"
"time"
"io"
"os"
"strconv"
)
func RandomTestBase() {
nums := generateRandomNumber(100000, 999999, 800000)
fmt.Println(len(nums))
var wireteString = ""
var filename = "./output1.txt"
var f *os.File
var err1 error
if checkFileIsExist(filename) {
f, err1 = os.OpenFile(filename, os.O_APPEND, 0666)
fmt.Println("文件存在")
} else {
f, err1 = os.Create(filename)
fmt.Println("文件不存在")
}
checkErr(err1)
wireteString = "insert into king_identifier (recordid) values ";
n, err1 := io.WriteString(f, wireteString)
checkErr(err1)
fmt.Printf("写入 %d 个字节n", n)
for k,v := range nums {
fmt.Println(k)
fmt.Println(v)
wireteString = "("+strconv.Itoa(v)+"),"
n, err1 := io.WriteString(f, wireteString)
checkErr(err1)
fmt.Printf("写入 %d 个字节n", n)
}
io.WriteString(f, ";")
}
func generateRandomNumber(start int, end int, count int) []int {
if end return nil
}
nums := make([]int, 0)
r := rand.New(rand.NewSource(time.Now().UnixNano()))
for len(nums)
num := r.Intn((end - start)) + start
exist := false
for _, v := range nums {
if v == num {
exist = true
break
}
}
if !exist {
nums = append(nums, num)
}
}
return nums
}
func checkErr(err error) {
if err != nil {
panic(err)
}
}
func checkFileIsExist(filename string) bool {
var exist = true
if _, err := os.Stat(filename); os.IsNotExist(err) {
exist = false
}
return exist
}
func TestTransRate(t *testing.T) {
RandomTestBase()
}