Estou tentando converter uma string retornada de flag.Arg(n)
para uma int
. Qual é a maneira idiomática de fazer isso no Go?
Estou tentando converter uma string retornada de flag.Arg(n)
para uma int
. Qual é a maneira idiomática de fazer isso no Go?
Respostas:
Por exemplo,
package main
import (
"flag"
"fmt"
"os"
"strconv"
)
func main() {
flag.Parse()
s := flag.Arg(0)
// string to int
i, err := strconv.Atoi(s)
if err != nil {
// handle error
fmt.Println(err)
os.Exit(2)
}
fmt.Println(s, i)
}
panic(err)
A maneira mais fácil é usar a strconv.Atoi()
função.
Observe que existem muitas outras maneiras. Por exemplo, fmt.Sscan()
e strconv.ParseInt()
que oferecem maior flexibilidade, você pode especificar a base e o tamanho de bits, por exemplo. Também conforme observado na documentação de strconv.Atoi()
:
Atoi é equivalente a ParseInt (s, 10, 0), convertido para o tipo int.
Aqui está um exemplo usando as funções mencionadas (experimente no Go Playground ):
flag.Parse()
s := flag.Arg(0)
if i, err := strconv.Atoi(s); err == nil {
fmt.Printf("i=%d, type: %T\n", i, i)
}
if i, err := strconv.ParseInt(s, 10, 64); err == nil {
fmt.Printf("i=%d, type: %T\n", i, i)
}
var i int
if _, err := fmt.Sscan(s, &i); err == nil {
fmt.Printf("i=%d, type: %T\n", i, i)
}
Saída (se chamada com argumento "123"
):
i=123, type: int
i=123, type: int64
i=123, type: int
Há também um recurso útil fmt.Sscanf()
que oferece flexibilidade ainda maior, pois, com a sequência de formatação, você pode especificar o formato numérico (como largura, base etc.) junto com caracteres extras adicionais na entrada string
.
Isso é ótimo para analisar seqüências personalizadas mantendo um número. Por exemplo, se sua entrada é fornecida em uma forma em "id:00123"
que você tem um prefixo "id:"
e o número é fixado em 5 dígitos, preenchido com zeros se for menor, isso é muito facilmente analisável assim:
s := "id:00123"
var i int
if _, err := fmt.Sscanf(s, "id:%5d", &i); err == nil {
fmt.Println(i) // Outputs 123
}
ParseInt
especifica?
strconv.ParseInt()
link e você verá imediatamente: ParseInt(s string, base int, bitSize int)
. Portanto, é a base: "ParseInt interpreta uma string s na base fornecida (2 a 36)"
int
for necessário e strconv.ParseInt()
for usado, será necessária a conversão manual do tipo (de int64
para int
).
Aqui estão três maneiras de analisar seqüências de caracteres em números inteiros, do tempo de execução mais rápido ao mais lento:
strconv.ParseInt(...)
o mais rápidostrconv.Atoi(...)
ainda muito rápidofmt.Sscanf(...)
não terrivelmente rápido, mas mais flexívelAqui está uma referência que mostra o tempo de uso e exemplo para cada função:
package main
import "fmt"
import "strconv"
import "testing"
var num = 123456
var numstr = "123456"
func BenchmarkStrconvParseInt(b *testing.B) {
num64 := int64(num)
for i := 0; i < b.N; i++ {
x, err := strconv.ParseInt(numstr, 10, 64)
if x != num64 || err != nil {
b.Error(err)
}
}
}
func BenchmarkAtoi(b *testing.B) {
for i := 0; i < b.N; i++ {
x, err := strconv.Atoi(numstr)
if x != num || err != nil {
b.Error(err)
}
}
}
func BenchmarkFmtSscan(b *testing.B) {
for i := 0; i < b.N; i++ {
var x int
n, err := fmt.Sscanf(numstr, "%d", &x)
if n != 1 || x != num || err != nil {
b.Error(err)
}
}
}
Você pode executá-lo salvando como atoi_test.go
e executando go test -bench=. atoi_test.go
.
goos: darwin
goarch: amd64
BenchmarkStrconvParseInt-8 100000000 17.1 ns/op
BenchmarkAtoi-8 100000000 19.4 ns/op
BenchmarkFmtSscan-8 2000000 693 ns/op
PASS
ok command-line-arguments 5.797s
Tente isto
import ("strconv")
value : = "123"
number,err := strconv.ParseUint(value, 10, 32)
Se você controlar os dados de entrada, poderá usar a mini versão
package main
import (
"testing"
"strconv"
)
func Atoi (s string) int {
var (
n uint64
i int
v byte
)
for ; i < len(s); i++ {
d := s[i]
if '0' <= d && d <= '9' {
v = d - '0'
} else if 'a' <= d && d <= 'z' {
v = d - 'a' + 10
} else if 'A' <= d && d <= 'Z' {
v = d - 'A' + 10
} else {
n = 0; break
}
n *= uint64(10)
n += uint64(v)
}
return int(n)
}
func BenchmarkAtoi(b *testing.B) {
for i := 0; i < b.N; i++ {
in := Atoi("9999")
_ = in
}
}
func BenchmarkStrconvAtoi(b *testing.B) {
for i := 0; i < b.N; i++ {
in, _ := strconv.Atoi("9999")
_ = in
}
}
a opção mais rápida (faça seu cheque, se necessário). Resultado:
Path>go test -bench=. atoi_test.go
goos: windows
goarch: amd64
BenchmarkAtoi-2 100000000 14.6 ns/op
BenchmarkStrconvAtoi-2 30000000 51.2 ns/op
PASS
ok path 3.293s
func main() { ... }
não aceita argumentos e não retorna valor. Use a funçãoos
packageExit
, por exemplo:os.Exit(2).