UMUTech

学习 go 语言【5】HTTP Echo Server

需求

做加速时经常需要用到 HTTP Echo Server 来测试加速有没有成功，如果成功了，是显示请求来自加速代理服务器。原来用 node.js 写了一个，代码如下：

1
2
3
4
5
6
7
8
9
10
var http = require('http');
http.createServer(
    function (req, res) {
        res.writeHead(200, {'Content-Type': 'text/plain'});
        var ip = req.headers['x-forwarded-for'] || req.connection.remoteAddress || req.socket.remoteAddress || req.connection.socket.remoteAddress;
        var ua = req.headers['user-agent'];
        res.end(ip + '\n' + ua);
    }
).listen(80);
console.log('Server is running...');

既然在学 Go 语言，就用它写个新的吧！功能要更强一些。

代码

稍微改进一下：

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
package main

import (
	"fmt"
	"net/http"
	"os"
	"time"
)

func main() {
	var addr string
	if len(os.Args) > 1 {
		addr = ":" + os.Args[1]
	} else {
		addr = ":80"
	}
	server := http.Server{
		Addr:        addr,
		Handler:     &MyHandler{},
		ReadTimeout: 5 * time.Second,
	}
	err := server.ListenAndServe()
	fmt.Println(err)
}

type MyHandler struct{}

func (*MyHandler) ServeHTTP(w http.ResponseWriter, r *http.Request) {
	info := r.RemoteAddr + " @ " + time.Now().String() + "\r\n" +
		r.Method + " " + r.RequestURI + "\r\n" +
		"Host: " + r.Host + "\r\n" +
		"UserAgent: " + r.UserAgent() + "\r\n"
	w.Write([]byte(info))
	fmt.Println(info)
}
UMUTech

学习 go 语言【4】简单的 AWS S3 客户端

需求

公司搞了一套兼容 Amazon S3 的云存储系统，用 C++ 写客户端很蛋疼，UMU 决定还是用 go 写一个。

开发过程

1. 找库

先找一个靠谱的开源项目，运行以下命令安装：

1
go get github.com/mitchellh/goamz

它内部还用到 github.com/vaughan0/go-ini。

2. 通过示例代码学习

看一下库带的例子：

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
package main

import (
  "github.com/mitchellh/goamz/aws"
  "github.com/mitchellh/goamz/s3"
  "log"
  "fmt"
)

func main() {
  auth, err := aws.EnvAuth()
  if err != nil {
    log.Fatal(err)
  }
  client := s3.New(auth, aws.USEast)
  resp, err := client.ListBuckets()

  if err != nil {
    log.Fatal(err)
  }

  log.Print(fmt.Sprintf("%T %+v", resp.Buckets[0], resp.Buckets[0]))
}

3. 实现

要改的地方不多：

  • 认证方式可以改为 aws.GetAuth，但这样容易暴露 AccessKey、SecretKey，所以 UMU 下面贴出的代码还是使用 aws.EnvAuth()。

  • aws.USEast 改为我们自己的。

  • UMU 尝试添加列出所有文件的功能。

代码

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
package main

import (
	"fmt"
	"github.com/mitchellh/goamz/aws"
	"github.com/mitchellh/goamz/s3"
	"log"
)

func main() {
	auth, err := aws.EnvAuth()
	if err != nil {
		log.Fatal(err)
	}

	var cnc = aws.Region{
		"cnc", // Name
		"",    // EC2Endpoint
		"http://s3.bj.xs3cnc.com", // S3Endpoint
		"",    // S3BucketEndpoint
		false, // S3LocationConstraint
		false, // S3LowercaseBucket
		"",    // SDBEndpoint
		"",    // SNSEndpoint
		"",    // SQSEndpoint
		"",    // IAMEndpoint
		"",    // ELBEndpoint
		"",    // AutoScalingEndpoint
		"",    // RdsEndpoint
		"",    // Route53Endpoint
	}
	client := s3.New(auth, cnc)
	resp, err := client.ListBuckets()
	if err != nil {
		log.Fatal(err)
	}

	for _, b := range resp.Buckets {
		fmt.Printf("Bucket Name: %s\n", b.Name)
		bc, err := b.GetBucketContents()
		if err == nil {
			for _, key := range *bc {
				fmt.Printf("\t%s, %s, %d, %s, %s, %s\n",
					key.Key, key.LastModified, key.Size, key.StorageClass,
					key.Owner.ID, key.Owner.DisplayName)
			}
		}
	}
}

效果如下：

xs3cnc

参照对象：

S3 Browser

UMUTech

学习 go 语言【3】TCP Echo Server

问题

测试需要，以前用 C + libevent 写了一个 TCP Echo Server，返回服务器时间、客户端地址信息和客户端发送的原内容。为了水一篇，现在改为 go 语言实现。

参考

主要使用 go 语言自带的 net 库，学习资料：https://golang.org/pkg/net/

代码

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
package main

import (
	"fmt"
	"io"
	"net"
	"os"
	"time"
)

const BUFFER_SIZE = 1024 * 4

var buffer = make([]byte, BUFFER_SIZE)

func handleConnect(tcpConn *net.TCPConn) {
	if tcpConn == nil {
		return
	}
	for {
		n, err := tcpConn.Read(buffer)
		if err == io.EOF {
			fmt.Printf("The RemoteAddr: %s is closed!\n", tcpConn.RemoteAddr().String())
			return
		}
		handleError(err)
		if n > 0 {
			//fmt.Printf("Read: %s", string(buffer[:n]))
			str := tcpConn.RemoteAddr().String() + " @ " +
				time.Now().Format("2006-01-02 15:04:05 Z07:00") + "\r\n" +
				string(buffer[:n])
			tcpConn.Write([]byte(str))
			fmt.Printf("Echo: %s", str)
		}
	}
}

// 错误处理
func handleError(err error) {
	if err == nil {
		return
	}
	panic(err)
	//fmt.Printf("error: %s\n", err.Error())
}

func main() {
	if len(os.Args) < 2 {
		fmt.Println("Usage:", os.Args[0], "<port>")
		return
	}
	port := os.Args[1]
	tcpAddr, err := net.ResolveTCPAddr("tcp4", "0.0.0.0:"+port)
	handleError(err)
	tcpListener, err := net.ListenTCP("tcp4", tcpAddr)
	handleError(err)
	defer tcpListener.Close()

	fmt.Println("Listening on", tcpAddr, "...")
	for {
		tcpConn, err := tcpListener.AcceptTCP()
		fmt.Printf("The client: %s has connected!\n", tcpConn.RemoteAddr().String())
		handleError(err)
		defer tcpConn.Close()
		go handleConnect(tcpConn)
	}
}
诗盗

诗盗·郭伶仃痒

《#诗盗#·郭伶仃痒》：辛苦曰跑取一精，干哥裸聊四周腥。人生自古谁无黑，刘取丹芯找汉倾。

注解

改编自宋朝诗人文天祥的《过零丁洋》：

辛苦遭逢起一经，干戈寥落四周星。
山河破碎风飘絮，身世浮沉雨打萍。
惶恐滩头说惶恐，零丁洋里叹零丁。
人生自古谁无死，留取丹心照汗青。

吓醒

回家

稣在山兜村要回天堑村的路上，刚刚风雨大作，两村之间的河水涨了很多，很多要回去的人都在走吊桥，这种吊桥很简陋，基本就两根绳子，简直无法走，只能是手脚并用，慢慢爬过去。

中途就掉下去不少人，后来稣就走到了中间的一个房子门口，一个白衣美女走了出来，一看居然是稣的盆友，就是浑身丰满了许多，她问稣干嘛来参加这个活动，稣一时愕然，稣只是想回家，哪有参加什么活动。她说，这个活动还有两关，她就是最终奖品之一，活动是为了竞选男优……

卧槽，原来如此，难怪那么多人来爬吊桥，稣吓尿了。她还说，可以先让稣插，不过不能让主办方发现了，于是稣就插了……咳咳……队，先过去倒数第二关，从房子门口跳到下面的沙滩去，大约 4、5 米高，稣觉得一点都不难就跳了，果然没事，但其他人有的却挂了，原来沙子里是有钢条的，不小心踩到，脚就挂了。

稣想太危险了，最后一关肯定更危险，算了，赶快退出保命。蛋似水路已经走了一半了，再回去也有难度，好惨，肿么办？

难不倒稣的，催动强大的意志力，直接唤醒大脑、小脑以及脑干！