avoid concurrent write to addressPrefixes #2

Merged
roylee17 merged 1 commit from fix-concurrent-write-to-addressPrefix into master 2018-05-16 00:53:27 +02:00

View file

@ -2,6 +2,7 @@ package address
import ( import (
"errors" "errors"
"github.com/lbryio/lbryschema.go/address/base58" "github.com/lbryio/lbryschema.go/address/base58"
) )
@ -20,16 +21,13 @@ const lbrycrdMain = "lbrycrd_main"
const lbrycrdTestnet = "lbrycrd_testnet" const lbrycrdTestnet = "lbrycrd_testnet"
const lbrycrdRegtest = "lbrycrd_regtest" const lbrycrdRegtest = "lbrycrd_regtest"
var addressPrefixes = map[string][2]byte{} var addressPrefixes = map[string][2]byte{
lbrycrdMain: [2]byte{lbrycrdMainPubkeyPrefix, lbrycrdMainScriptPrefix},
func SetPrefixes() { lbrycrdTestnet: [2]byte{lbrycrdTestnetPubkeyPrefix, lbrycrdTestnetScriptPrefix},
addressPrefixes[lbrycrdMain] = [2]byte{lbrycrdMainPubkeyPrefix, lbrycrdMainScriptPrefix} lbrycrdRegtest: [2]byte{lbrycrdRegtestPubkeyPrefix, lbrycrdRegtestScriptPrefix},
addressPrefixes[lbrycrdTestnet] = [2]byte{lbrycrdTestnetPubkeyPrefix, lbrycrdTestnetScriptPrefix}
addressPrefixes[lbrycrdRegtest] = [2]byte{lbrycrdRegtestPubkeyPrefix, lbrycrdRegtestScriptPrefix}
} }
func PrefixIsValid(address [addressLength]byte, blockchainName string) bool { func PrefixIsValid(address [addressLength]byte, blockchainName string) bool {
SetPrefixes()
prefix := address[0] prefix := address[0]
for _, addrPrefix := range addressPrefixes[blockchainName] { for _, addrPrefix := range addressPrefixes[blockchainName] {
if addrPrefix == prefix { if addrPrefix == prefix {