From 2a48e1f1b92eb9654c1296460abf117f89b8b51e Mon Sep 17 00:00:00 2001
From: ss8651twtw <ss8651twtw@gmail.com>
Date: Sat, 16 Jun 2018 23:11:33 +0800
Subject: [PATCH] Update .travis.yml and coverage script

---
 .travis.yml |  2 +-
 coverage.sh | 23 +++++++++++++++++++++++
 2 files changed, 24 insertions(+), 1 deletion(-)
 create mode 100755 coverage.sh

diff --git a/.travis.yml b/.travis.yml
index 87e421c..c5059e9 100644
--- a/.travis.yml
+++ b/.travis.yml
@@ -16,7 +16,7 @@ install:
  - go get -v ./...
 
 script:
- - go test -v -coverprofile=coverage.txt -covermode=atomic ./...
+ - ./coverage.sh
 
 after_success:
  - bash <(curl -s https://codecov.io/bash)
diff --git a/coverage.sh b/coverage.sh
new file mode 100755
index 0000000..f93a4cb
--- /dev/null
+++ b/coverage.sh
@@ -0,0 +1,23 @@
+#!/bin/sh
+
+set -e
+
+workdir=.cover
+profile=${workdir}/cover.out
+mode=count
+
+generate_cover_data() {
+    rm -rf ${workdir}
+    mkdir ${workdir}
+
+    for pkg in "$@"; do
+        f="${workdir}/$(echo ${pkg} | tr / -).cover"
+        go test -covermode=${mode} -coverprofile=${f} ${pkg}
+    done
+
+    echo "mode: ${mode} > ${profile}"
+    grep -h -v "^mode:" "${workdir}"/*.cover >>"${profile}"
+}
+
+generate_cover_data $(go list ./...)
+mv ${profile} coverage.txt