90 Star 491 Fork 149

平凯星辰(北京)科技有限公司/tidb

Create your Gitee Account
Explore and code with more than 12 million developers,Free private repositories !:)
Sign up
Clone or Download
data.go 2.99 KB
Copy Edit Raw Blame History
韩飞 authored 2018-01-22 18:48 . cmd: add importer tool. (#5695)
// Copyright 2016 PingCAP, Inc.
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
// http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// See the License for the specific language governing permissions and
// limitations under the License.
package main
import (
"fmt"
"sync"
"time"
)
var defaultStep int64 = 1
type datum struct {
sync.Mutex
intValue int64
minIntValue int64
maxIntValue int64
timeValue time.Time
step int64
init bool
useRange bool
}
func newDatum() *datum {
return &datum{intValue: -1, step: 1}
}
func (d *datum) setInitInt64Value(step int64, min int64, max int64) {
d.Lock()
defer d.Unlock()
if d.init {
return
}
d.step = step
if min != -1 {
d.minIntValue = min
d.intValue = min
}
if min < max {
d.maxIntValue = max
d.useRange = true
}
d.init = true
}
func (d *datum) uniqInt64() int64 {
d.Lock()
defer d.Unlock()
data := d.intValue
if d.useRange {
if d.intValue+d.step > d.maxIntValue {
return data
}
}
d.intValue += d.step
return data
}
func (d *datum) uniqFloat64() float64 {
data := d.uniqInt64()
return float64(data)
}
func (d *datum) uniqString(n int) string {
d.Lock()
d.intValue++
data := d.intValue
d.Unlock()
var value []byte
for ; ; n-- {
if n == 0 {
break
}
idx := data % int64(len(alphabet))
data = data / int64(len(alphabet))
value = append(value, alphabet[idx])
if data == 0 {
break
}
}
for i, j := 0, len(value)-1; i < j; i, j = i+1, j-1 {
value[i], value[j] = value[j], value[i]
}
return string(value)
}
func (d *datum) uniqTime() string {
d.Lock()
defer d.Unlock()
if d.timeValue.IsZero() {
d.timeValue = time.Now()
} else {
d.timeValue = d.timeValue.Add(time.Duration(d.step) * time.Second)
}
return fmt.Sprintf("%02d:%02d:%02d", d.timeValue.Hour(), d.timeValue.Minute(), d.timeValue.Second())
}
func (d *datum) uniqDate() string {
d.Lock()
defer d.Unlock()
if d.timeValue.IsZero() {
d.timeValue = time.Now()
} else {
d.timeValue = d.timeValue.AddDate(0, 0, int(d.step))
}
return fmt.Sprintf("%04d-%02d-%02d", d.timeValue.Year(), d.timeValue.Month(), d.timeValue.Day())
}
func (d *datum) uniqTimestamp() string {
d.Lock()
defer d.Unlock()
if d.timeValue.IsZero() {
d.timeValue = time.Now()
} else {
d.timeValue = d.timeValue.Add(time.Duration(d.step) * time.Second)
}
return fmt.Sprintf("%04d-%02d-%02d %02d:%02d:%02d",
d.timeValue.Year(), d.timeValue.Month(), d.timeValue.Day(),
d.timeValue.Hour(), d.timeValue.Minute(), d.timeValue.Second())
}
func (d *datum) uniqYear() string {
d.Lock()
defer d.Unlock()
if d.timeValue.IsZero() {
d.timeValue = time.Now()
} else {
d.timeValue = d.timeValue.AddDate(int(d.step), 0, 0)
}
return fmt.Sprintf("%04d", d.timeValue.Year())
}
马建仓 AI 助手
尝试更多
代码解读
代码找茬
代码优化
Go
1
https://gitee.com/pingcap/tidb.git
git@gitee.com:pingcap/tidb.git
pingcap
tidb
tidb
v2.1.0-rc.4

Search