utils.go

package phantomjs

import (
	"fmt"
	"net"
	"regexp"
	"strconv"
	"strings"
)

var (
	route_param_reg *regexp.Regexp
	ip_addr_reg     *regexp.Regexp
)

func init() {
	route_param_reg = regexp.MustCompile(":[0-9a-zA-Z_]+")
	ip_addr_reg = regexp.MustCompile("^(.*):([0-9]+)$")
}

func RouteTemplate(route string, args map[string]interface{}) (string, error) {
	argKeys := route_param_reg.FindAllString(route, -1)
	if args == nil {
		args = map[string]interface{}{}
	}

	for _, argKey := range argKeys {
		key := strings.TrimLeft(argKey, ":")
		val, ok := args[key]
		if !ok {
			return "", fmt.Errorf("RouteTemplate: need param %s", key)
		}
		route = strings.Replace(route, argKey, fmt.Sprintf("%v", val), -1)
	}
	return route, nil
}

func FindFreePort() int {
	nl, err := net.Listen("tcp", ":")
	if err != nil {
		panic(err)
	}
	defer nl.Close()

	addrParts := ip_addr_reg.FindStringSubmatch(nl.Addr().String())
	if len(addrParts) != 3 {
		panic("invalid addr")
	}
	port, _ := strconv.Atoi(addrParts[2])

	return port
}

func IsConnectAble(addr string) bool {
	conn, err := net.Dial("tcp", addr)
	if err != nil {
		return false
	}
	defer conn.Close()

	return true
}