memcache read from file

gistfile1.go

package main

import (
	"bufio"
	"bytes"
	"encoding/base64"
	"flag"
	"fmt"
	"io"
	"os"
	"strings"

	"github.com/bradfitz/gomemcache/memcache"
)

func main() {
	flag.Parse()
	path := flag.Arg(0)
	mem := memcache.New("localhost:11211")

	var (
		file   *os.File
		part   []byte
		prefix bool
		err    error

		ih    string
		value []byte
	)

	if file, err = os.Open(path); err != nil {
		fmt.Println(err)
		return
	}
	reader := bufio.NewReader(file)
	buffer := new(bytes.Buffer)

	for {
		if part, prefix, err = reader.ReadLine(); err != nil {
			break
		}
		buffer.Write(part)
		if !prefix {
			s := strings.SplitN(buffer.String(), "|", 2)
			ih = strings.ToLower(s[0])
			_, err := fmt.Sscanf(ih, "%x", &ih)
			if err != nil {
				fmt.Println("Could not read hex-encoded string", err)
			}
			ih = base64.StdEncoding.EncodeToString([]byte(ih))
			value = []byte(s[1])
			buffer.Reset()
		}
		if err = mem.Set(&memcache.Item{Key: ih, Value: value}); err != nil {
			fmt.Printf("Error writing key '%q': %v\n", ih, err)
			return
		} else {
			fmt.Println("wrote:", ih)
		}
	}
	if err == io.EOF {
		err = nil
	}
	return
}