pipeline.go 24 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729
  1. package core
  2. import (
  3. "bufio"
  4. "fmt"
  5. "io"
  6. "io/ioutil"
  7. "log"
  8. "os"
  9. "path/filepath"
  10. "sort"
  11. "strings"
  12. "time"
  13. "github.com/pkg/errors"
  14. "gopkg.in/src-d/go-git.v4"
  15. "gopkg.in/src-d/go-git.v4/plumbing"
  16. "gopkg.in/src-d/go-git.v4/plumbing/object"
  17. "gopkg.in/src-d/go-git.v4/plumbing/storer"
  18. "gopkg.in/src-d/hercules.v5/internal/pb"
  19. "gopkg.in/src-d/hercules.v5/internal/toposort"
  20. )
  21. // ConfigurationOptionType represents the possible types of a ConfigurationOption's value.
  22. type ConfigurationOptionType int
  23. const (
  24. // BoolConfigurationOption reflects the boolean value type.
  25. BoolConfigurationOption ConfigurationOptionType = iota
  26. // IntConfigurationOption reflects the integer value type.
  27. IntConfigurationOption
  28. // StringConfigurationOption reflects the string value type.
  29. StringConfigurationOption
  30. // FloatConfigurationOption reflects a floating point value type.
  31. FloatConfigurationOption
  32. // StringsConfigurationOption reflects the array of strings value type.
  33. StringsConfigurationOption
  34. )
  35. // String() returns an empty string for the boolean type, "int" for integers and "string" for
  36. // strings. It is used in the command line interface to show the argument's type.
  37. func (opt ConfigurationOptionType) String() string {
  38. switch opt {
  39. case BoolConfigurationOption:
  40. return ""
  41. case IntConfigurationOption:
  42. return "int"
  43. case StringConfigurationOption:
  44. return "string"
  45. case FloatConfigurationOption:
  46. return "float"
  47. case StringsConfigurationOption:
  48. return "string"
  49. }
  50. log.Panicf("Invalid ConfigurationOptionType value %d", opt)
  51. return ""
  52. }
  53. // ConfigurationOption allows for the unified, retrospective way to setup PipelineItem-s.
  54. type ConfigurationOption struct {
  55. // Name identifies the configuration option in facts.
  56. Name string
  57. // Description represents the help text about the configuration option.
  58. Description string
  59. // Flag corresponds to the CLI token with "--" prepended.
  60. Flag string
  61. // Type specifies the kind of the configuration option's value.
  62. Type ConfigurationOptionType
  63. // Default is the initial value of the configuration option.
  64. Default interface{}
  65. }
  66. // FormatDefault converts the default value of ConfigurationOption to string.
  67. // Used in the command line interface to show the argument's default value.
  68. func (opt ConfigurationOption) FormatDefault() string {
  69. if opt.Type == StringsConfigurationOption {
  70. return fmt.Sprintf("\"%s\"", strings.Join(opt.Default.([]string), ","))
  71. }
  72. if opt.Type != StringConfigurationOption {
  73. return fmt.Sprint(opt.Default)
  74. }
  75. return fmt.Sprintf("\"%s\"", opt.Default)
  76. }
  77. // PipelineItem is the interface for all the units in the Git commits analysis pipeline.
  78. type PipelineItem interface {
  79. // Name returns the name of the analysis.
  80. Name() string
  81. // Provides returns the list of keys of reusable calculated entities.
  82. // Other items may depend on them.
  83. Provides() []string
  84. // Requires returns the list of keys of needed entities which must be supplied in Consume().
  85. Requires() []string
  86. // ListConfigurationOptions returns the list of available options which can be consumed by Configure().
  87. ListConfigurationOptions() []ConfigurationOption
  88. // Configure performs the initial setup of the object by applying parameters from facts.
  89. // It allows to create PipelineItems in a universal way.
  90. Configure(facts map[string]interface{})
  91. // Initialize prepares and resets the item. Consume() requires Initialize()
  92. // to be called at least once beforehand.
  93. Initialize(*git.Repository)
  94. // Consume processes the next commit.
  95. // deps contains the required entities which match Depends(). Besides, it always includes
  96. // DependencyCommit and DependencyIndex.
  97. // Returns the calculated entities which match Provides().
  98. Consume(deps map[string]interface{}) (map[string]interface{}, error)
  99. // Fork clones the item the requested number of times. The data links between the clones
  100. // are up to the implementation. Needed to handle Git branches. See also Merge().
  101. // Returns a slice with `n` fresh clones. In other words, it does not include the original item.
  102. Fork(n int) []PipelineItem
  103. // Merge combines several branches together. Each is supposed to have been created with Fork().
  104. // The result is stored in the called item, thus this function returns nothing.
  105. // Merge() must update all the branches, not only self. When several branches merge, some of
  106. // them may continue to live, hence this requirement.
  107. Merge(branches []PipelineItem)
  108. }
  109. // FeaturedPipelineItem enables switching the automatic insertion of pipeline items on or off.
  110. type FeaturedPipelineItem interface {
  111. PipelineItem
  112. // Features returns the list of names which enable this item to be automatically inserted
  113. // in Pipeline.DeployItem().
  114. Features() []string
  115. }
  116. // LeafPipelineItem corresponds to the top level pipeline items which produce the end results.
  117. type LeafPipelineItem interface {
  118. PipelineItem
  119. // Flag returns the cmdline switch to run the analysis. Should be dash-lower-case
  120. // without the leading dashes.
  121. Flag() string
  122. // Description returns the text which explains what the analysis is doing.
  123. // Should start with a capital letter and end with a dot.
  124. Description() string
  125. // Finalize returns the result of the analysis.
  126. Finalize() interface{}
  127. // Serialize encodes the object returned by Finalize() to YAML or Protocol Buffers.
  128. Serialize(result interface{}, binary bool, writer io.Writer) error
  129. }
  130. // ResultMergeablePipelineItem specifies the methods to combine several analysis results together.
  131. type ResultMergeablePipelineItem interface {
  132. LeafPipelineItem
  133. // Deserialize loads the result from Protocol Buffers blob.
  134. Deserialize(pbmessage []byte) (interface{}, error)
  135. // MergeResults joins two results together. Common-s are specified as the global state.
  136. MergeResults(r1, r2 interface{}, c1, c2 *CommonAnalysisResult) interface{}
  137. }
  138. // CommonAnalysisResult holds the information which is always extracted at Pipeline.Run().
  139. type CommonAnalysisResult struct {
  140. // BeginTime is the time of the first commit in the analysed sequence.
  141. BeginTime int64
  142. // EndTime is the time of the last commit in the analysed sequence.
  143. EndTime int64
  144. // CommitsNumber is the number of commits in the analysed sequence.
  145. CommitsNumber int
  146. // RunTime is the duration of Pipeline.Run().
  147. RunTime time.Duration
  148. // RunTimePerItem is the time elapsed by each PipelineItem.
  149. RunTimePerItem map[string]float64
  150. }
  151. // BeginTimeAsTime converts the UNIX timestamp of the beginning to Go time.
  152. func (car *CommonAnalysisResult) BeginTimeAsTime() time.Time {
  153. return time.Unix(car.BeginTime, 0)
  154. }
  155. // EndTimeAsTime converts the UNIX timestamp of the ending to Go time.
  156. func (car *CommonAnalysisResult) EndTimeAsTime() time.Time {
  157. return time.Unix(car.EndTime, 0)
  158. }
  159. // Merge combines the CommonAnalysisResult with an other one.
  160. // We choose the earlier BeginTime, the later EndTime, sum the number of commits and the
  161. // elapsed run times.
  162. func (car *CommonAnalysisResult) Merge(other *CommonAnalysisResult) {
  163. if car.EndTime == 0 || other.BeginTime == 0 {
  164. panic("Merging with an uninitialized CommonAnalysisResult")
  165. }
  166. if other.BeginTime < car.BeginTime {
  167. car.BeginTime = other.BeginTime
  168. }
  169. if other.EndTime > car.EndTime {
  170. car.EndTime = other.EndTime
  171. }
  172. car.CommitsNumber += other.CommitsNumber
  173. car.RunTime += other.RunTime
  174. for key, val := range other.RunTimePerItem {
  175. car.RunTimePerItem[key] += val
  176. }
  177. }
  178. // FillMetadata copies the data to a Protobuf message.
  179. func (car *CommonAnalysisResult) FillMetadata(meta *pb.Metadata) *pb.Metadata {
  180. meta.BeginUnixTime = car.BeginTime
  181. meta.EndUnixTime = car.EndTime
  182. meta.Commits = int32(car.CommitsNumber)
  183. meta.RunTime = car.RunTime.Nanoseconds() / 1e6
  184. meta.RunTimePerItem = car.RunTimePerItem
  185. return meta
  186. }
  187. // Metadata is defined in internal/pb/pb.pb.go - header of the binary file.
  188. type Metadata = pb.Metadata
  189. // MetadataToCommonAnalysisResult copies the data from a Protobuf message.
  190. func MetadataToCommonAnalysisResult(meta *Metadata) *CommonAnalysisResult {
  191. return &CommonAnalysisResult{
  192. BeginTime: meta.BeginUnixTime,
  193. EndTime: meta.EndUnixTime,
  194. CommitsNumber: int(meta.Commits),
  195. RunTime: time.Duration(meta.RunTime * 1e6),
  196. RunTimePerItem: meta.RunTimePerItem,
  197. }
  198. }
  199. // Pipeline is the core Hercules entity which carries several PipelineItems and executes them.
  200. // See the extended example of how a Pipeline works in doc.go
  201. type Pipeline struct {
  202. // OnProgress is the callback which is invoked in Analyse() to output it's
  203. // progress. The first argument is the number of complete steps and the
  204. // second is the total number of steps.
  205. OnProgress func(int, int)
  206. // Repository points to the analysed Git repository struct from go-git.
  207. repository *git.Repository
  208. // Items are the registered building blocks in the pipeline. The order defines the
  209. // execution sequence.
  210. items []PipelineItem
  211. // The collection of parameters to create items.
  212. facts map[string]interface{}
  213. // Feature flags which enable the corresponding items.
  214. features map[string]bool
  215. }
  216. const (
  217. // ConfigPipelineDumpPath is the name of the Pipeline configuration option (Pipeline.Initialize())
  218. // which enables saving the items DAG to the specified file.
  219. ConfigPipelineDumpPath = "Pipeline.DumpPath"
  220. // ConfigPipelineDryRun is the name of the Pipeline configuration option (Pipeline.Initialize())
  221. // which disables Configure() and Initialize() invocation on each PipelineItem during the
  222. // Pipeline initialization.
  223. // Subsequent Run() calls are going to fail. Useful with ConfigPipelineDumpPath=true.
  224. ConfigPipelineDryRun = "Pipeline.DryRun"
  225. // ConfigPipelineCommits is the name of the Pipeline configuration option (Pipeline.Initialize())
  226. // which allows to specify the custom commit sequence. By default, Pipeline.Commits() is used.
  227. ConfigPipelineCommits = "commits"
  228. // DependencyCommit is the name of one of the three items in `deps` supplied to PipelineItem.Consume()
  229. // which always exists. It corresponds to the currently analyzed commit.
  230. DependencyCommit = "commit"
  231. // DependencyIndex is the name of one of the three items in `deps` supplied to PipelineItem.Consume()
  232. // which always exists. It corresponds to the currently analyzed commit's index.
  233. DependencyIndex = "index"
  234. // DependencyIsMerge is the name of one of the three items in `deps` supplied to PipelineItem.Consume()
  235. // which always exists. It indicates whether the analyzed commit is a merge commit.
  236. // Checking the number of parents is not correct - we remove the back edges during the DAG simplification.
  237. DependencyIsMerge = "is_merge"
  238. )
  239. // NewPipeline initializes a new instance of Pipeline struct.
  240. func NewPipeline(repository *git.Repository) *Pipeline {
  241. return &Pipeline{
  242. repository: repository,
  243. items: []PipelineItem{},
  244. facts: map[string]interface{}{},
  245. features: map[string]bool{},
  246. }
  247. }
  248. // GetFact returns the value of the fact with the specified name.
  249. func (pipeline *Pipeline) GetFact(name string) interface{} {
  250. return pipeline.facts[name]
  251. }
  252. // SetFact sets the value of the fact with the specified name.
  253. func (pipeline *Pipeline) SetFact(name string, value interface{}) {
  254. pipeline.facts[name] = value
  255. }
  256. // GetFeature returns the state of the feature with the specified name (enabled/disabled) and
  257. // whether it exists. See also: FeaturedPipelineItem.
  258. func (pipeline *Pipeline) GetFeature(name string) (bool, bool) {
  259. val, exists := pipeline.features[name]
  260. return val, exists
  261. }
  262. // SetFeature sets the value of the feature with the specified name.
  263. // See also: FeaturedPipelineItem.
  264. func (pipeline *Pipeline) SetFeature(name string) {
  265. pipeline.features[name] = true
  266. }
  267. // SetFeaturesFromFlags enables the features which were specified through the command line flags
  268. // which belong to the given PipelineItemRegistry instance.
  269. // See also: AddItem().
  270. func (pipeline *Pipeline) SetFeaturesFromFlags(registry ...*PipelineItemRegistry) {
  271. var ffr *PipelineItemRegistry
  272. if len(registry) == 0 {
  273. ffr = Registry
  274. } else if len(registry) == 1 {
  275. ffr = registry[0]
  276. } else {
  277. panic("Zero or one registry is allowed to be passed.")
  278. }
  279. for _, feature := range ffr.featureFlags.Flags {
  280. pipeline.SetFeature(feature)
  281. }
  282. }
  283. // DeployItem inserts a PipelineItem into the pipeline. It also recursively creates all of it's
  284. // dependencies (PipelineItem.Requires()). Returns the same item as specified in the arguments.
  285. func (pipeline *Pipeline) DeployItem(item PipelineItem) PipelineItem {
  286. fpi, ok := item.(FeaturedPipelineItem)
  287. if ok {
  288. for _, f := range fpi.Features() {
  289. pipeline.SetFeature(f)
  290. }
  291. }
  292. queue := []PipelineItem{}
  293. queue = append(queue, item)
  294. added := map[string]PipelineItem{}
  295. for _, item := range pipeline.items {
  296. added[item.Name()] = item
  297. }
  298. added[item.Name()] = item
  299. pipeline.AddItem(item)
  300. for len(queue) > 0 {
  301. head := queue[0]
  302. queue = queue[1:]
  303. for _, dep := range head.Requires() {
  304. for _, sibling := range Registry.Summon(dep) {
  305. if _, exists := added[sibling.Name()]; !exists {
  306. disabled := false
  307. // If this item supports features, check them against the activated in pipeline.features
  308. if fpi, matches := sibling.(FeaturedPipelineItem); matches {
  309. for _, feature := range fpi.Features() {
  310. if !pipeline.features[feature] {
  311. disabled = true
  312. break
  313. }
  314. }
  315. }
  316. if disabled {
  317. continue
  318. }
  319. added[sibling.Name()] = sibling
  320. queue = append(queue, sibling)
  321. pipeline.AddItem(sibling)
  322. }
  323. }
  324. }
  325. }
  326. return item
  327. }
  328. // AddItem inserts a PipelineItem into the pipeline. It does not check any dependencies.
  329. // See also: DeployItem().
  330. func (pipeline *Pipeline) AddItem(item PipelineItem) PipelineItem {
  331. pipeline.items = append(pipeline.items, item)
  332. return item
  333. }
  334. // RemoveItem deletes a PipelineItem from the pipeline. It leaves all the rest of the items intact.
  335. func (pipeline *Pipeline) RemoveItem(item PipelineItem) {
  336. for i, reg := range pipeline.items {
  337. if reg == item {
  338. pipeline.items = append(pipeline.items[:i], pipeline.items[i+1:]...)
  339. return
  340. }
  341. }
  342. }
  343. // Len returns the number of items in the pipeline.
  344. func (pipeline *Pipeline) Len() int {
  345. return len(pipeline.items)
  346. }
  347. // Commits returns the list of commits from the history similar to `git log` over the HEAD.
  348. // `firstParent` specifies whether to leave only the first parent after each merge
  349. // (`git log --first-parent`) - effectively decreasing the accuracy but increasing performance.
  350. func (pipeline *Pipeline) Commits(firstParent bool) ([]*object.Commit, error) {
  351. var result []*object.Commit
  352. repository := pipeline.repository
  353. head, err := repository.Head()
  354. if err != nil {
  355. if err == plumbing.ErrReferenceNotFound {
  356. refs, errr := repository.References()
  357. if errr != nil {
  358. return nil, errors.Wrap(errr, "unable to list the references")
  359. }
  360. refs.ForEach(func(ref *plumbing.Reference) error {
  361. if strings.HasPrefix(ref.Name().String(), "refs/heads/HEAD/") {
  362. head = ref
  363. return storer.ErrStop
  364. }
  365. return nil
  366. })
  367. }
  368. if head == nil && err != nil {
  369. return nil, errors.Wrap(err, "unable to collect the commit history")
  370. }
  371. }
  372. if firstParent {
  373. commit, err := repository.CommitObject(head.Hash())
  374. if err != nil {
  375. panic(err)
  376. }
  377. // the first parent matches the head
  378. for ; err != io.EOF; commit, err = commit.Parents().Next() {
  379. if err != nil {
  380. panic(err)
  381. }
  382. result = append(result, commit)
  383. }
  384. // reverse the order
  385. for i, j := 0, len(result)-1; i < j; i, j = i+1, j-1 {
  386. result[i], result[j] = result[j], result[i]
  387. }
  388. return result, nil
  389. }
  390. cit, err := repository.Log(&git.LogOptions{From: head.Hash()})
  391. if err != nil {
  392. return nil, errors.Wrap(err, "unable to collect the commit history")
  393. }
  394. defer cit.Close()
  395. cit.ForEach(func(commit *object.Commit) error {
  396. result = append(result, commit)
  397. return nil
  398. })
  399. return result, nil
  400. }
  401. type sortablePipelineItems []PipelineItem
  402. func (items sortablePipelineItems) Len() int {
  403. return len(items)
  404. }
  405. func (items sortablePipelineItems) Less(i, j int) bool {
  406. return items[i].Name() < items[j].Name()
  407. }
  408. func (items sortablePipelineItems) Swap(i, j int) {
  409. items[i], items[j] = items[j], items[i]
  410. }
  411. func (pipeline *Pipeline) resolve(dumpPath string) {
  412. graph := toposort.NewGraph()
  413. sort.Sort(sortablePipelineItems(pipeline.items))
  414. name2item := map[string]PipelineItem{}
  415. ambiguousMap := map[string][]string{}
  416. nameUsages := map[string]int{}
  417. for _, item := range pipeline.items {
  418. nameUsages[item.Name()]++
  419. }
  420. counters := map[string]int{}
  421. for _, item := range pipeline.items {
  422. name := item.Name()
  423. if nameUsages[name] > 1 {
  424. index := counters[item.Name()] + 1
  425. counters[item.Name()] = index
  426. name = fmt.Sprintf("%s_%d", item.Name(), index)
  427. }
  428. graph.AddNode(name)
  429. name2item[name] = item
  430. for _, key := range item.Provides() {
  431. key = "[" + key + "]"
  432. graph.AddNode(key)
  433. if graph.AddEdge(name, key) > 1 {
  434. if ambiguousMap[key] != nil {
  435. fmt.Fprintln(os.Stderr, "Pipeline:")
  436. for _, item2 := range pipeline.items {
  437. if item2 == item {
  438. fmt.Fprint(os.Stderr, "> ")
  439. }
  440. fmt.Fprint(os.Stderr, item2.Name(), " [")
  441. for i, key2 := range item2.Provides() {
  442. fmt.Fprint(os.Stderr, key2)
  443. if i < len(item.Provides())-1 {
  444. fmt.Fprint(os.Stderr, ", ")
  445. }
  446. }
  447. fmt.Fprintln(os.Stderr, "]")
  448. }
  449. panic("Failed to resolve pipeline dependencies: ambiguous graph.")
  450. }
  451. ambiguousMap[key] = graph.FindParents(key)
  452. }
  453. }
  454. }
  455. counters = map[string]int{}
  456. for _, item := range pipeline.items {
  457. name := item.Name()
  458. if nameUsages[name] > 1 {
  459. index := counters[item.Name()] + 1
  460. counters[item.Name()] = index
  461. name = fmt.Sprintf("%s_%d", item.Name(), index)
  462. }
  463. for _, key := range item.Requires() {
  464. key = "[" + key + "]"
  465. if graph.AddEdge(key, name) == 0 {
  466. log.Panicf("Unsatisfied dependency: %s -> %s", key, item.Name())
  467. }
  468. }
  469. }
  470. // Try to break the cycles in some known scenarios.
  471. if len(ambiguousMap) > 0 {
  472. var ambiguous []string
  473. for key := range ambiguousMap {
  474. ambiguous = append(ambiguous, key)
  475. }
  476. sort.Strings(ambiguous)
  477. bfsorder := graph.BreadthSort()
  478. bfsindex := map[string]int{}
  479. for i, s := range bfsorder {
  480. bfsindex[s] = i
  481. }
  482. for len(ambiguous) > 0 {
  483. key := ambiguous[0]
  484. ambiguous = ambiguous[1:]
  485. pair := ambiguousMap[key]
  486. inheritor := pair[1]
  487. if bfsindex[pair[1]] < bfsindex[pair[0]] {
  488. inheritor = pair[0]
  489. }
  490. removed := graph.RemoveEdge(key, inheritor)
  491. cycle := map[string]bool{}
  492. for _, node := range graph.FindCycle(key) {
  493. cycle[node] = true
  494. }
  495. if len(cycle) == 0 {
  496. cycle[inheritor] = true
  497. }
  498. if removed {
  499. graph.AddEdge(key, inheritor)
  500. }
  501. graph.RemoveEdge(inheritor, key)
  502. graph.ReindexNode(inheritor)
  503. // for all nodes key links to except those in cycle, put the link from inheritor
  504. for _, node := range graph.FindChildren(key) {
  505. if _, exists := cycle[node]; !exists {
  506. graph.AddEdge(inheritor, node)
  507. graph.RemoveEdge(key, node)
  508. }
  509. }
  510. graph.ReindexNode(key)
  511. }
  512. }
  513. var graphCopy *toposort.Graph
  514. if dumpPath != "" {
  515. graphCopy = graph.Copy()
  516. }
  517. strplan, ok := graph.Toposort()
  518. if !ok {
  519. panic("Failed to resolve pipeline dependencies: unable to topologically sort the items.")
  520. }
  521. pipeline.items = make([]PipelineItem, 0, len(pipeline.items))
  522. for _, key := range strplan {
  523. if item, ok := name2item[key]; ok {
  524. pipeline.items = append(pipeline.items, item)
  525. }
  526. }
  527. if dumpPath != "" {
  528. // If there is a floating difference, uncomment this:
  529. // fmt.Fprint(os.Stderr, graphCopy.DebugDump())
  530. ioutil.WriteFile(dumpPath, []byte(graphCopy.Serialize(strplan)), 0666)
  531. absPath, _ := filepath.Abs(dumpPath)
  532. log.Printf("Wrote the DAG to %s\n", absPath)
  533. }
  534. }
  535. // Initialize prepares the pipeline for the execution (Run()). This function
  536. // resolves the execution DAG, Configure()-s and Initialize()-s the items in it in the
  537. // topological dependency order. `facts` are passed inside Configure(). They are mutable.
  538. func (pipeline *Pipeline) Initialize(facts map[string]interface{}) {
  539. if facts == nil {
  540. facts = map[string]interface{}{}
  541. }
  542. if _, exists := facts[ConfigPipelineCommits]; !exists {
  543. var err error
  544. facts[ConfigPipelineCommits], err = pipeline.Commits(false)
  545. if err != nil {
  546. log.Panicf("failed to list the commits: %v", err)
  547. }
  548. }
  549. dumpPath, _ := facts[ConfigPipelineDumpPath].(string)
  550. pipeline.resolve(dumpPath)
  551. if dryRun, _ := facts[ConfigPipelineDryRun].(bool); dryRun {
  552. return
  553. }
  554. for _, item := range pipeline.items {
  555. item.Configure(facts)
  556. }
  557. for _, item := range pipeline.items {
  558. item.Initialize(pipeline.repository)
  559. }
  560. }
  561. // Run method executes the pipeline.
  562. //
  563. // `commits` is a slice with the git commits to analyse. Multiple branches are supported.
  564. //
  565. // Returns the mapping from each LeafPipelineItem to the corresponding analysis result.
  566. // There is always a "nil" record with CommonAnalysisResult.
  567. func (pipeline *Pipeline) Run(commits []*object.Commit) (map[LeafPipelineItem]interface{}, error) {
  568. startRunTime := time.Now()
  569. onProgress := pipeline.OnProgress
  570. if onProgress == nil {
  571. onProgress = func(int, int) {}
  572. }
  573. plan := prepareRunPlan(commits)
  574. progressSteps := len(plan) + 2
  575. branches := map[int][]PipelineItem{}
  576. // we will need rootClone if there is more than one root branch
  577. rootClone := cloneItems(pipeline.items, 1)[0]
  578. var newestTime int64
  579. runTimePerItem := map[string]float64{}
  580. commitIndex := 0
  581. for index, step := range plan {
  582. onProgress(index+1, progressSteps)
  583. firstItem := step.Items[0]
  584. switch step.Action {
  585. case runActionCommit:
  586. state := map[string]interface{}{
  587. DependencyCommit: step.Commit,
  588. DependencyIndex: commitIndex,
  589. DependencyIsMerge: (index > 0 &&
  590. plan[index-1].Action == runActionCommit &&
  591. plan[index-1].Commit.Hash == step.Commit.Hash) ||
  592. (index < (len(plan)-1) &&
  593. plan[index+1].Action == runActionCommit &&
  594. plan[index+1].Commit.Hash == step.Commit.Hash),
  595. }
  596. for _, item := range branches[firstItem] {
  597. startTime := time.Now()
  598. update, err := item.Consume(state)
  599. runTimePerItem[item.Name()] += time.Now().Sub(startTime).Seconds()
  600. if err != nil {
  601. log.Printf("%s failed on commit #%d (%d) %s\n",
  602. item.Name(), commitIndex+1, index+1, step.Commit.Hash.String())
  603. return nil, err
  604. }
  605. for _, key := range item.Provides() {
  606. val, ok := update[key]
  607. if !ok {
  608. log.Panicf("%s: Consume() did not return %s", item.Name(), key)
  609. }
  610. state[key] = val
  611. }
  612. }
  613. commitTime := step.Commit.Committer.When.Unix()
  614. if commitTime > newestTime {
  615. newestTime = commitTime
  616. }
  617. commitIndex++
  618. case runActionFork:
  619. for i, clone := range cloneItems(branches[firstItem], len(step.Items)-1) {
  620. branches[step.Items[i+1]] = clone
  621. }
  622. case runActionMerge:
  623. merged := make([][]PipelineItem, len(step.Items))
  624. for i, b := range step.Items {
  625. merged[i] = branches[b]
  626. }
  627. mergeItems(merged)
  628. case runActionEmerge:
  629. if firstItem == rootBranchIndex {
  630. branches[firstItem] = pipeline.items
  631. } else {
  632. branches[firstItem] = cloneItems(rootClone, 1)[0]
  633. }
  634. case runActionDelete:
  635. delete(branches, firstItem)
  636. }
  637. }
  638. onProgress(len(plan)+1, progressSteps)
  639. result := map[LeafPipelineItem]interface{}{}
  640. for index, item := range getMasterBranch(branches) {
  641. if casted, ok := item.(LeafPipelineItem); ok {
  642. result[pipeline.items[index].(LeafPipelineItem)] = casted.Finalize()
  643. }
  644. }
  645. onProgress(progressSteps, progressSteps)
  646. result[nil] = &CommonAnalysisResult{
  647. BeginTime: plan[0].Commit.Committer.When.Unix(),
  648. EndTime: newestTime,
  649. CommitsNumber: len(commits),
  650. RunTime: time.Since(startRunTime),
  651. RunTimePerItem: runTimePerItem,
  652. }
  653. return result, nil
  654. }
  655. // LoadCommitsFromFile reads the file by the specified FS path and generates the sequence of commits
  656. // by interpreting each line as a Git commit hash.
  657. func LoadCommitsFromFile(path string, repository *git.Repository) ([]*object.Commit, error) {
  658. var file io.ReadCloser
  659. if path != "-" {
  660. var err error
  661. file, err = os.Open(path)
  662. if err != nil {
  663. return nil, err
  664. }
  665. defer file.Close()
  666. } else {
  667. file = os.Stdin
  668. }
  669. scanner := bufio.NewScanner(file)
  670. var commits []*object.Commit
  671. for scanner.Scan() {
  672. hash := plumbing.NewHash(scanner.Text())
  673. if len(hash) != 20 {
  674. return nil, errors.New("invalid commit hash " + scanner.Text())
  675. }
  676. commit, err := repository.CommitObject(hash)
  677. if err != nil {
  678. return nil, err
  679. }
  680. commits = append(commits, commit)
  681. }
  682. return commits, nil
  683. }